Searched defs:dim_pair (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 100 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
114 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gemm_functors.h | 105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
|
| D | conv_ops_fused_impl.h | 267 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
284 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
|
| D | matmul_op.cc | 77 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in ExplicitVectorMatrixOptimization()
107 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in ExplicitVectorMatrixOptimization()
464 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| D | matmul_op.h | 44 const DimPair& dim_pair) { in MatMul()
|
| D | quantized_matmul_op.cc | 101 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| D | mkl_matmul_op.cc | 61 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| D | conv_ops.cc | 89 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
|
| D | sparse_matmul_op.cc | 1015 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3027 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
3042 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
|