Searched refs:rhs_contracting_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 156 int64 rhs_contracting_dim) in HloDotWithContractingDimsMatcher() argument
159 rhs_contracting_dim_(rhs_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 | 395 HloInstruction* rhs, int64 rhs_contracting_dim, bool swapped);
1372 const int64 rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat() local
1379 rhs_contracting_dim, /*swapped=*/false)); in OptimizeDotOfConcat()
1384 return OptimizeDotOfConcatHelper(*dot, rhs, rhs_contracting_dim, lhs, in OptimizeDotOfConcat()
1390 HloInstruction* rhs, int64 rhs_contracting_dim, bool swapped) { in OptimizeDotOfConcatHelper() argument
1445 new_dot_dnums.add_lhs_contracting_dimensions(swapped ? rhs_contracting_dim in OptimizeDotOfConcatHelper()
1448 : rhs_contracting_dim); in OptimizeDotOfConcatHelper()
1454 int64 n = rhs->shape().dimensions(1 - rhs_contracting_dim); in OptimizeDotOfConcatHelper()
1458 rhs_slice_shape.set_dimensions(rhs_contracting_dim, sub_k); in OptimizeDotOfConcatHelper()
1461 start_indices[rhs_contracting_dim] = rhs_contracting_dim_offset; in OptimizeDotOfConcatHelper()
[all …]
|
| D | elemental_ir_emitter.cc | 2120 int64 rhs_contracting_dim = dim_numbers.rhs_contracting_dimensions(0); in EmitElementalDot() local
2169 rhs_multi_index.insert(rhs_multi_index.begin() + rhs_contracting_dim, in EmitElementalDot()
|