/external/tensorflow/tensorflow/core/kernels/ |
H A D | gemm_functors.h | 98 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 D | matmul_op.h | 40 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 D | mkl_matmul_op.cc | 55 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 D | conv_2d.h | 135 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 D | quantized_matmul_op.cc | 101 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 D | matmul_op.cc | 77 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 D | conv_ops.cc | 80 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 D | sparse_matmul_op.cc | 1007 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 D | multithreaded_conv.h | 93 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
|