Searched defs:dim_pair (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_impl.h | 126 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
143 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
|
| D | gemm_functors.h | 105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
|
| D | matmul_op.h | 44 const DimPair& dim_pair) { in MatMul()
|
| D | matmul_op_fused.cc | 186 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| D | quantized_matmul_op.cc | 101 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| D | conv_ops.cc | 100 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
116 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
|
| D | sparse_matmul_op.cc | 1014 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| /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/mkl/ |
| D | mkl_matmul_op.cc | 57 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
|
| D | mkl_matmul_op_fused.cc | 88 const int dim_pair[] = {1, transpose_b_ ? 1 : 0}; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3932 common_factors, [&](const std::pair<int64, int64>& dim_pair) { in HandleReshape() argument
3937 common_factors, [&](const std::pair<int64, int64>& dim_pair) { in HandleReshape() argument
4655 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
4670 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
|