| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 60 } else if (absl::c_linear_search(original_dnums.lhs_batch_dimensions(), in CanonicalizeDot()
74 original_dnums.lhs_batch_dimensions().begin(), in CanonicalizeDot()
75 original_dnums.lhs_batch_dimensions().end()); in CanonicalizeDot()
196 if (dnums.lhs_batch_dimensions().empty() && in Run()
201 if (dnums.lhs_batch_dimensions().empty()) { in Run()
207 if (!absl::c_equal(dnums.lhs_batch_dimensions(), canonical_batch_dims) || in Run()
|
| D | batch_dot_simplification.cc | 37 batch_dot->dot_dimension_numbers().lhs_batch_dimensions(), in ElideDegenerateBatchDimensionFromBatchDot()
40 batch_dot->dot_dimension_numbers().lhs_batch_dimensions()))) { in ElideDegenerateBatchDimensionFromBatchDot()
57 for (int64 batch_dim : dim_numbers.lhs_batch_dimensions()) { in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | dot_as_convolution_util.cc | 158 for (int64 i = 0; i < dot_dim_numbs.lhs_batch_dimensions().size(); ++i) { in ParseDotGeneralFromDot()
160 dnums.batch_dims.back().lhs = dot_dim_numbs.lhs_batch_dimensions(i); in ParseDotGeneralFromDot()
176 if (!absl::c_linear_search(dot_dim_numbs.lhs_batch_dimensions(), i) && in ParseDotGeneralFromDot()
|
| D | transpose_folding.cc | 42 if (!absl::c_equal(dot.dot_dimension_numbers().lhs_batch_dimensions(), in CanFoldOperandsIntoDot()
63 dot.dot_dimension_numbers().lhs_batch_dimensions(), in CanFoldOperandsIntoDot()
|
| D | shape_inference.cc | 613 absl::Span<const int64> lhs_batch_dimensions = in ValidateDotDimensionNumbers() local
614 AsInt64Slice(dimension_numbers.lhs_batch_dimensions()); in ValidateDotDimensionNumbers()
619 lhs_batch_dimensions) || in ValidateDotDimensionNumbers()
637 if (!dims_unique(lhs_contracting_dimensions, lhs_batch_dimensions) || in ValidateDotDimensionNumbers()
696 if (lhs.dimensions(dimension_numbers.lhs_batch_dimensions(i)) != in InferDotOpShape()
709 for (int64 lhs_dim : dimension_numbers.lhs_batch_dimensions()) { in InferDotOpShape()
716 !absl::c_linear_search(dimension_numbers.lhs_batch_dimensions(), i)) { in InferDotOpShape()
|
| D | dynamic_dimension_inference.cc | 489 for (int64 i : dimension_numbers.lhs_batch_dimensions()) { in HandleDot()
505 if (absl::c_linear_search(dimension_numbers.lhs_batch_dimensions(), in HandleDot()
|
| D | indexed_array_analysis.cc | 1041 /*batch_dims=*/AsInt64Slice(dim_numbers.lhs_batch_dimensions()))) { in ComputeArrayForDotWithIndexedLhs()
|
| D | algebraic_simplifier.cc | 1764 for (int64 dim : dnums.lhs_batch_dimensions()) { in RemoveDegenerateDimensionFromDot()
2331 AsInt64Slice(dot->dot_dimension_numbers().lhs_batch_dimensions()), in HandleDot()
2381 AsInt64Slice(dot->dot_dimension_numbers().lhs_batch_dimensions()), in HandleDot()
4731 dnums.lhs_batch_dimensions(batch_dim)); in HandleReduce()
4737 dnums.lhs_batch_dimensions(batch_dim)); in HandleReduce()
|
| D | hlo_instructions.cc | 2999 if (!dnums.lhs_batch_dimensions().empty()) { in DotDimensionNumbersToString()
3001 StrJoin(dnums.lhs_batch_dimensions(), ","), "}")); in DotDimensionNumbersToString()
|
| D | hlo_evaluator_typed_visitor.h | 1424 {&lhs_index[dnums.lhs_batch_dimensions(i)],
1431 !absl::c_linear_search(dnums.lhs_batch_dimensions(), i)) {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 90 std::vector<int64_t> lhs_batch_dimensions( in ConvertDotDimensionNumbers() local
91 dnums.lhs_batch_dimensions().begin(), dnums.lhs_batch_dimensions().end()); in ConvertDotDimensionNumbers()
94 auto lhs_batch_dims_attr = Convert(lhs_batch_dimensions, builder); in ConvertDotDimensionNumbers()
|
| D | mlir_hlo_to_hlo.cc | 249 auto lhs_batch_dimensions = in Convert_dot_dimension_numbers() local
264 for (const auto& val : lhs_batch_dimensions) { in Convert_dot_dimension_numbers()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 176 dnums.lhs_batch_dimensions.append(0)
578 dim_nums.lhs_batch_dimensions.append(2)
594 dim_nums.lhs_batch_dimensions.append(0)
609 dim_nums.lhs_batch_dimensions.append(3)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 233 dimension_numbers.lhs_batch_dimensions(i); in __anonbbc1c2d10202()
257 for (int64 lhs_dim : dimension_numbers.lhs_batch_dimensions()) { in __anonbbc1c2d10202()
264 absl::c_linear_search(dimension_numbers.lhs_batch_dimensions(), in __anonbbc1c2d10202()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_gemm_pad_for_tensor_cores.cc | 136 if (!absl::c_equal(dimension_numbers.lhs_batch_dimensions(), in CheckCanonical()
|
| D | gemm_thunk.cc | 197 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in RunGemm()
|
| D | gpu_layout_assignment.cc | 223 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in AddBackendConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 498 self.lhs_batch_dimensions = []
523 dot_dims_proto.lhs_batch_dimensions.extend(lhs_batch)
|
| D | xla_client_test.py | 726 dimension_numbers.lhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 144 …ing_dimensions\":[\"1\"],\"rhs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_…
169 …ing_dimensions\":[\"1\"],\"rhs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_…
194 …ing_dimensions\":[\"1\"],\"rhs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_…
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 462 for (auto d : dnums.lhs_batch_dimensions()) { in Einsum()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1099 absl::c_equal(batch_dim_numbers, dim_numbers.lhs_batch_dimensions())); in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 571 repeated int64 lhs_batch_dimensions = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 48 mapping.batch_dims.back().lhs = dnums.lhs_batch_dimensions(i); in HandleDot()
59 if (absl::c_linear_search(dnums.lhs_batch_dimensions(), i) || in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 669 GetI64DenseElementsAttr(hlo_dims.lhs_batch_dimensions()), in EmitGemm()
|