Searched defs:dim_pair (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dgemm_functors.h98 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
99 dim_pair[0].first = 1;
100 dim_pair[0].second = 0;
102 a_matrix.contract(b_matrix, dim_pair);
H A Dmatmul_op.h40 const DimPair& dim_pair) {
41 out.device(d) = in0.contract(in1, dim_pair);
52 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair);
39 MatMul(const Device& d, Out out, In0 in0, In1 in1, const DimPair& dim_pair) argument
H A Dmkl_matmul_op.cc55 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
56 dim_pair[0].first = transpose_a_ ? 0 : 1;
57 dim_pair[0].second = transpose_b_ ? 1 : 0;
60 ctx, a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second),
64 int a_dim_remaining = 1 - dim_pair[0].first;
65 int b_dim_remaining = 1 - dim_pair[0].second;
86 const int m = a.dim_size(1 - dim_pair[0].first);
87 const int k = a.dim_size(dim_pair[0].first);
88 const int n = b.dim_size(1 - dim_pair[
[all...]
H A Dconv_2d.h135 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
136 out.device(d) = in0.contract(in1, dim_pair);
131 operator ()( const Device& d, typename TTypes<T, 2>::Tensor out, typename TTypes<T, 2>::ConstTensor in0, typename TTypes<T, 2>::ConstTensor in1, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) argument
H A Dquantized_matmul_op.cc101 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
102 dim_pair[0].first = transpose_a_ ? 0 : 1;
103 dim_pair[0].second = transpose_b_ ? 1 : 0;
106 a.dim_size(dim_pair[0].first) == b.dim_size(dim_pair[0].second),
115 int a_dim_remaining = 1 - dim_pair[0].first;
116 int b_dim_remaining = 1 - dim_pair[0].second;
130 const size_t k = a.dim_size(dim_pair[0].first);
H A Dmatmul_op.cc77 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
80 if (dim_pair[0].second == 0) {
94 if (dim_pair[0].first == 0) {
107 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
122 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
127 bool was_vector = ExplicitVectorMatrixOptimization<T>(a, b, dim_pair, out);
132 b.matrix<T>(), dim_pair);
260 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
271 const uint64 m = a.dim_size(1 - dim_pair[0].first);
272 const uint64 k = a.dim_size(dim_pair[
75 ExplicitVectorMatrixOptimization( const Tensor& a, const Tensor& b, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, Tensor* out) argument
105 ExplicitVectorMatrixOptimization( const Tensor& a, const Tensor& b, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, Tensor* out) argument
120 launch( OpKernelContext* ctx, const Tensor& a, const Tensor& b, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, std::vector<AlgorithmType>* algorithms, bool use_aututone, Tensor* out) argument
258 launch( OpKernelContext* ctx, const Tensor& a, const Tensor& b, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, std::vector<int64>* algorithms, bool use_autotune, Tensor* out) argument
465 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
513 operator ()( const CPUDevice& d, typename MatMulTypes<T>::out_type out, typename MatMulTypes<T>::in_type in0, typename MatMulTypes<T>::in_type in1, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) argument
526 operator ()( const SYCLDevice& d, typename MatMulTypes<T>::out_type out, typename MatMulTypes<T>::in_type in0, typename MatMulTypes<T>::in_type in1, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) argument
[all...]
H A Dconv_ops.cc80 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
81 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
87 dim_pair);
96 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
97 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
102 filter.shaped<T, 2>({k, filter.dim_size(3)}), dim_pair);
798 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair); \
H A Dsparse_matmul_op.cc1007 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; variable
1008 dim_pair[0].first = transpose_a_ ? 0 : 1;
1009 dim_pair[0].second = transpose_b_ ? 1 : 0;
1012 left->matrix<float>().contract(right->matrix<float>(), dim_pair);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Dmultithreaded_conv.h93 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) {
94 out.device(d) = in0.contract(in1, dim_pair);
130 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
131 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
136 filter, dim_pair);
143 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; local
144 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0);
149 filter, dim_pair);
90 operator ()( const Device& d, EigenMatrix out, ConstEigenMatrix in0, ConstEigenMatrix in1, const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) argument

Completed in 250 milliseconds