Home
last modified time | relevance | path

Searched defs:dim_pair (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_fused_impl.h128 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
145 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
H A Dgemm_functors.h105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator() local
H A Dmatmul_op.h44 const DimPair& dim_pair) { in MatMul()
H A Dquantized_matmul_op.cc116 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
H A Dconv_ops.cc106 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
122 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()() local
H A Dmatmul_op_fused.cc373 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
H A Dsparse_matmul_op.cc1019 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dmultithreaded_conv.h100 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_arith_ops_folder.cc41 for (const auto &dim_pair : llvm::enumerate(dims_attr)) { in VerifyReductionInputAndDims() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_matmul_op.cc57 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute() local
H A Dmkl_matmul_op_fused.cc97 const int dim_pair[] = {1, transpose_b_ ? 1 : 0}; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.cc4308 common_factors, [&](const std::pair<int64_t, int64_t>& dim_pair) { in HandleReshape() argument
4313 common_factors, [&](const std::pair<int64_t, int64_t>& dim_pair) { in HandleReshape() argument
5455 for (auto dim_pair : unmodified_dims) { in HandleReduce() local
5470 for (auto dim_pair : unmodified_dims) { in HandleReduce() local