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) {}
443 int64 lhs_contracting_dim, int64 rhs_contracting_dim) { in Dot() argument
446 lhs_matcher, rhs_matcher, lhs_contracting_dim, rhs_contracting_dim)); in Dot()
|
| D | algebraic_simplifier.cc | 472 HloInstruction* rhs, int64 rhs_contracting_dim, bool swapped);
1825 const int64 rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat() local
1832 rhs_contracting_dim, /*swapped=*/false)); in OptimizeDotOfConcat()
1837 return OptimizeDotOfConcatHelper(*dot, rhs, rhs_contracting_dim, lhs, in OptimizeDotOfConcat()
1843 HloInstruction* rhs, int64 rhs_contracting_dim, bool swapped) { in OptimizeDotOfConcatHelper() argument
1898 new_dot_dnums.add_lhs_contracting_dimensions(swapped ? rhs_contracting_dim in OptimizeDotOfConcatHelper()
1901 : rhs_contracting_dim); in OptimizeDotOfConcatHelper()
1907 int64 n = rhs->shape().dimensions(1 - rhs_contracting_dim); in OptimizeDotOfConcatHelper()
1911 rhs_slice_shape.set_dimensions(rhs_contracting_dim, sub_k); in OptimizeDotOfConcatHelper()
1915 start_indices[rhs_contracting_dim] = rhs_contracting_dim_offset; in OptimizeDotOfConcatHelper()
[all …]
|
| D | elemental_ir_emitter.cc | 2179 int64 rhs_contracting_dim = dim_numbers.rhs_contracting_dimensions(0); in EmitElementalDot() local
2228 rhs_multi_index.insert(rhs_multi_index.begin() + rhs_contracting_dim, in EmitElementalDot()
|