Home
last modified time | relevance | path

Searched refs:lhs_contracting_dimensions_size (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc44 original_dnums.lhs_contracting_dimensions_size(); in CanonicalizeDot()
183 if (dnums.lhs_contracting_dimensions_size() != 1) { in Run()
Dbatch_dot_simplification.cc52 if (dim_numbers.lhs_contracting_dimensions_size() != 1) { in ElideDegenerateBatchDimensionFromBatchDot()
Dtranspose_folding.cc116 CHECK_EQ(new_dim_numbers.lhs_contracting_dimensions_size(), 1); in FoldTransposeIntoDot()
Dhlo_matchers.cc227 if (dim_nums.lhs_contracting_dimensions_size() != 1 || in MatchAndExplain()
Dalgebraic_simplifier.cc1818 if (dnums.lhs_contracting_dimensions_size() != 1 || in OptimizeDotOfConcat()
1961 if (dnums.lhs_contracting_dimensions_size() != 1 || in OptimizeDotOfGather()
2119 if (dnums.lhs_contracting_dimensions_size() != 1) { in OptimizeDotOfReorderContractingDims()
2326 dot->dot_dimension_numbers().lhs_contracting_dimensions_size() == 0) { in HandleDot()
2372 dot->dot_dimension_numbers().lhs_contracting_dimensions_size() == in HandleDot()
2402 dot->dot_dimension_numbers().lhs_contracting_dimensions_size()); in HandleDot()
2437 dot->dot_dimension_numbers().lhs_contracting_dimensions_size()); in HandleDot()
2491 dot->dot_dimension_numbers().lhs_contracting_dimensions_size() == 1 && in HandleDot()
5096 if (lhs->shape().rank() != 1 + dnums.lhs_contracting_dimensions_size()) { in HandleTranspose()
Dhlo_evaluator_typed_visitor.h1417 (lhs_rank - dnums.lhs_contracting_dimensions_size()) +
1443 accumulate_index_sizes.reserve(dnums.lhs_contracting_dimensions_size());
1446 accumulate_index_locations.reserve(dnums.lhs_contracting_dimensions_size());
1447 for (int64 i = 0; i < dnums.lhs_contracting_dimensions_size(); ++i) {
Dshape_inference.cc669 if (dimension_numbers.lhs_contracting_dimensions_size() != in InferDotOpShape()
676 for (int64 i = 0; i < dimension_numbers.lhs_contracting_dimensions_size(); in InferDotOpShape()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc187 if (dimension_numbers.lhs_contracting_dimensions_size() != in __anon3500048a0202()
192 dimension_numbers.lhs_contracting_dimensions_size(), " and ", in __anon3500048a0202()
196 for (int64 i = 0; i < dimension_numbers.lhs_contracting_dimensions_size(); in __anon3500048a0202()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc273 CHECK_EQ(dot_info_.dim_nums.lhs_contracting_dimensions_size(), 1); in EmitLinalgMatmul()
862 CHECK_GE(dim_nums.lhs_contracting_dimensions_size(), 0); in GetMatMultDims()
1095 TF_RET_CHECK(dim_numbers.lhs_contracting_dimensions_size() == 1); in ValidateDotDimensionNumbers()
Dir_emitter.cc843 if (dnums.lhs_contracting_dimensions_size() != 1) { in HandleDot()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc181 int64_t lhs_contracting_dimensions_size = llvm::size( in Verify() local
185 if (lhs_contracting_dimensions_size != rhs_contracting_dimensions_size) { in Verify()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc52 for (int64 i = 0; i < dnums.lhs_contracting_dimensions_size(); ++i) { in HandleDot()