Searched refs:lhs_contracting_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 155 ::testing::Matcher<const HloInstruction*> rhs, int64 lhs_contracting_dim, in HloDotWithContractingDimsMatcher() argument
158 lhs_contracting_dim_(lhs_contracting_dim),
407 int64 lhs_contracting_dim, int64 rhs_contracting_dim) { in Dot() argument
410 lhs_matcher, rhs_matcher, lhs_contracting_dim, rhs_contracting_dim)); in Dot()
|
| D | algebraic_simplifier.cc | 394 const HloInstruction& dot, HloInstruction* lhs, int64 lhs_contracting_dim,
1371 const int64 lhs_contracting_dim = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfConcat() local
1378 OptimizeDotOfConcatHelper(*dot, lhs, lhs_contracting_dim, rhs, in OptimizeDotOfConcat()
1385 lhs_contracting_dim, /*swapped=*/true); in OptimizeDotOfConcat()
1389 const HloInstruction& dot, HloInstruction* lhs, int64 lhs_contracting_dim, in OptimizeDotOfConcatHelper() argument
1392 lhs->concatenate_dimension() == lhs_contracting_dim && in OptimizeDotOfConcatHelper()
1446 : lhs_contracting_dim); in OptimizeDotOfConcatHelper()
1447 new_dot_dnums.add_rhs_contracting_dimensions(swapped ? lhs_contracting_dim in OptimizeDotOfConcatHelper()
1456 int64 sub_k = concat_op->shape().dimensions(lhs_contracting_dim); in OptimizeDotOfConcatHelper()
|
| D | elemental_ir_emitter.cc | 2119 int64 lhs_contracting_dim = dim_numbers.lhs_contracting_dimensions(0); in EmitElementalDot() local
2123 hlo->operand(0)->shape().dimensions(lhs_contracting_dim); in EmitElementalDot()
2156 lhs_multi_index.insert(lhs_multi_index.begin() + lhs_contracting_dim, in EmitElementalDot()
|