| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 113 } else if (!absl::c_linear_search(original_dnums.rhs_batch_dimensions(), in CanonicalizeDot()
126 original_dnums.rhs_batch_dimensions().begin(), in CanonicalizeDot()
127 original_dnums.rhs_batch_dimensions().end()); in CanonicalizeDot()
222 !absl::c_equal(dnums.rhs_batch_dimensions(), canonical_batch_dims)) { in Run()
|
| D | dot_merger.cc | 62 is_permutation_of_iota(dnums.rhs_batch_dimensions()) && in IsCanonicalDot()
112 !absl::c_equal(dnums_a.rhs_batch_dimensions(), in TryMergeSameOperand()
113 dnums_b.rhs_batch_dimensions()) || in TryMergeSameOperand()
124 << absl::c_equal(dnums_a.rhs_batch_dimensions(), in TryMergeSameOperand()
125 dnums_b.rhs_batch_dimensions()) in TryMergeSameOperand()
|
| D | dot_as_convolution_util.cc | 188 dnums.batch_dims.back().rhs = dot_dim_numbs.rhs_batch_dimensions(i); in ParseDotGeneralFromDot()
215 if (!absl::c_linear_search(dot_dim_numbs.rhs_batch_dimensions(), i) && in ParseDotGeneralFromDot()
|
| D | batch_dot_simplification.cc | 38 batch_dot->dot_dimension_numbers().rhs_batch_dimensions()) || in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | transpose_folding.cc | 264 : dot_dims.rhs_batch_dimensions(); in IsRowColumnTransposeDotOperand()
|
| D | hlo_cse.cc | 174 absl::MakeSpan(dot_dimension_numbers.rhs_batch_dimensions())); in AbslHashValue()
|
| D | shape_inference.cc | 612 absl::Span<const int64_t> rhs_batch_dimensions = in ValidateDotDimensionNumbers() local
613 dimension_numbers.rhs_batch_dimensions(); in ValidateDotDimensionNumbers()
618 rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
635 !dims_unique(rhs_contracting_dimensions, rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
694 rhs.dimensions(dimension_numbers.rhs_batch_dimensions(i))) { in InferDotOpShape()
710 dimension_numbers.rhs_batch_dimensions().size(); in InferDotOpShape()
728 !absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), i)) { in InferDotOpShape()
|
| D | algebraic_simplifier.cc | 1991 for (int64_t dim : dnums.rhs_batch_dimensions()) { in RemoveDegenerateDimensionFromDot()
2047 absl::MakeSpan(dnums.rhs_batch_dimensions()) || in RemoveTransposesFromDotOperands()
2645 rhs, dnums.rhs_batch_dimensions(), in HandleDot()
2690 rhs, dnums.rhs_batch_dimensions(), in HandleDot()
5529 dnums.rhs_batch_dimensions(batch_dim)); in HandleReduce()
5535 dnums.rhs_batch_dimensions(batch_dim)); in HandleReduce()
6073 absl::MakeSpan(dnums.rhs_batch_dimensions()) || in HandleTranspose()
|
| D | dynamic_dimension_inference.cc | 525 for (int64_t i : dimension_numbers.rhs_batch_dimensions()) { in HandleDot()
553 if (absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), i)) { in HandleDot()
|
| D | indexed_array_analysis.cc | 1080 /*batch_dims=*/dim_numbers.rhs_batch_dimensions())) { in ComputeArrayForDotWithIndexedRhs()
|
| D | hlo_evaluator_typed_visitor.h | 1248 !absl::c_linear_search(dnums.rhs_batch_dimensions(), i)) {
1277 rhs_index[dnums.rhs_batch_dimensions(i)] = result_index[idx];
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 215 dnums.rhs_batch_dimensions.append(0)
253 dnums.rhs_batch_dimensions.append(0)
695 dim_nums.rhs_batch_dimensions.append(0)
732 dim_nums.rhs_batch_dimensions.append(2)
733 dim_nums.rhs_batch_dimensions.append(3)
748 dim_nums.rhs_batch_dimensions.append(0)
795 dim_nums.rhs_batch_dimensions.append(0)
808 dim_nums.rhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_broadcast_folding_rewriter.cc | 58 (bcast_operand_index == 1) ? dim_nums->rhs_batch_dimensions() in HandleCustomCall()
|
| D | gpu_layout_assignment.cc | 237 dot_dims.rhs_batch_dimensions().size(), in DotCanSupportShapeWithLayout()
281 dot_dims.rhs_batch_dimensions(); in AddBackendConstraints()
|
| D | cublas_pad_for_gemms.cc | 139 !absl::c_equal(dimension_numbers.rhs_batch_dimensions(), in CheckCanonical()
|
| D | gemm_rewriter.cc | 57 !dot_dims.rhs_batch_dimensions().empty(); in SetName()
|
| D | matmul_utils.cc | 217 : dot_dims.rhs_batch_dimensions(); in CanFoldTransposeOperandIntoDot()
337 dot_dims.rhs_batch_dimensions(), dot_dims.rhs_contracting_dimensions(), in For()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 474 self.rhs_batch_dimensions = []
499 dot_dims_proto.rhs_batch_dimensions.extend(rhs_batch)
|
| D | xla_client.pyi | 128 rhs_batch_dimensions: List[int]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 80 arrayref(dnums.rhs_batch_dimensions()), in ConvertDotDimensionNumbers()
|
| D | mlir_hlo_to_hlo.cc | 358 auto rhs_batch_dimensions = in Convert_dot_dimension_numbers() local
370 for (const auto& val : rhs_batch_dimensions) { in Convert_dot_dimension_numbers()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 266 dimension_numbers.rhs_batch_dimensions(i); in XlaDotShapeFunction()
290 absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), i)) { in XlaDotShapeFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 146 …hs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_batch_dimensions\":[]},\"pre…
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1299 absl::c_equal(batch_dim_numbers, dim_numbers.rhs_batch_dimensions())); in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 621 repeated int64 rhs_batch_dimensions = 4; field
|