Searched refs:rhs_batch_dimensions (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 52 rhs_shape.dimensions(dnums.rhs_batch_dimensions(i))); in DecomposeBatchDot()
231 } else if (!absl::c_linear_search(original_dnums.rhs_batch_dimensions(), in CanonicalizeDot()
243 original_dnums.rhs_batch_dimensions().begin(), in CanonicalizeDot()
244 original_dnums.rhs_batch_dimensions().end()); in CanonicalizeDot()
316 !absl::c_equal(dnums.rhs_batch_dimensions(), canonical_batch_dims)) { in Run()
|
| D | dynamic_dimension_inference.cc | 211 dimension_numbers.rhs_batch_dimensions().begin(), in HandleDot()
212 dimension_numbers.rhs_batch_dimensions().end()); in HandleDot()
214 for (int64 i : dimension_numbers.rhs_batch_dimensions()) { in HandleDot()
231 !absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), in HandleDot()
|
| D | shape_inference.cc | 575 absl::Span<const int64> rhs_batch_dimensions = in ValidateDotDimensionNumbers() local
576 AsInt64Slice(dimension_numbers.rhs_batch_dimensions()); in ValidateDotDimensionNumbers()
581 rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
598 !dims_unique(rhs_contracting_dimensions, rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
667 rhs.dimensions(dimension_numbers.rhs_batch_dimensions(i)) || in InferDotOpShape()
670 dimension_numbers.rhs_batch_dimensions(i))) { in InferDotOpShape()
697 !absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), i)) { in InferDotOpShape()
|
| D | algebraic_simplifier.cc | 1167 AsInt64Slice(dnums.rhs_batch_dimensions())); in HandleDotStrengthReduction()
1282 dnums.rhs_batch_dimensions())) { in HandleDotStrengthReduction()
1674 AsInt64Slice(dot->dot_dimension_numbers().rhs_batch_dimensions()), in HandleDot()
1711 AsInt64Slice(dot->dot_dimension_numbers().rhs_batch_dimensions()), in HandleDot()
|
| D | indexed_array_analysis.cc | 1076 /*batch_dims=*/AsInt64Slice(dim_numbers.rhs_batch_dimensions()))) { in ComputeArrayForDotWithIndexedRhs()
|
| D | hlo_instructions.cc | 2569 if (!dnums.rhs_batch_dimensions().empty()) { in DotDimensionNumbersToString()
2571 StrJoin(dnums.rhs_batch_dimensions(), ","), "}")); in DotDimensionNumbersToString()
|
| D | hlo_evaluator_typed_visitor.h | 1312 &rhs_index[dnums.rhs_batch_dimensions(i)]}); in HandleDotSlowPath()
1324 !absl::c_linear_search(dnums.rhs_batch_dimensions(), i)) { in HandleDotSlowPath()
|
| D | elemental_ir_emitter.cc | 2164 dot_result_index[dim_numbers.rhs_batch_dimensions(i)]); in EmitElementalDot()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 152 dnums.rhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1916 self.rhs_batch_dimensions = []
1925 dot_dims_proto.rhs_batch_dimensions.extend(rhs_batch)
|
| D | xla_client_test.py | 673 dimension_numbers.rhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 905 absl::c_equal(batch_dim_numbers, dim_numbers.rhs_batch_dimensions())); in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 532 repeated int64 rhs_batch_dimensions = 4; field
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 927 | 'rhs_batch_dimensions' | repeated int64 | 'rhs' batch dimension numbers |
|