| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 88 std::vector<int64_t> rhs_batch_dimensions( in ConvertDotDimensionNumbers() local
89 dnums.rhs_batch_dimensions().begin(), dnums.rhs_batch_dimensions().end()); in ConvertDotDimensionNumbers()
95 auto rhs_batch_dims_attr = Convert(rhs_batch_dimensions, builder); in ConvertDotDimensionNumbers()
|
| D | mlir_hlo_to_hlo.cc | 246 auto rhs_batch_dimensions = in Convert_dot_dimension_numbers() local
260 for (const auto& val : rhs_batch_dimensions) { in Convert_dot_dimension_numbers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 107 } else if (!absl::c_linear_search(original_dnums.rhs_batch_dimensions(), in CanonicalizeDot()
120 original_dnums.rhs_batch_dimensions().begin(), in CanonicalizeDot()
121 original_dnums.rhs_batch_dimensions().end()); in CanonicalizeDot()
208 !absl::c_equal(dnums.rhs_batch_dimensions(), canonical_batch_dims)) { in Run()
|
| D | dot_as_convolution_util.cc | 161 dnums.batch_dims.back().rhs = dot_dim_numbs.rhs_batch_dimensions(i); in ParseDotGeneralFromDot()
188 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 | 43 dot.dot_dimension_numbers().rhs_batch_dimensions())) { in CanFoldOperandsIntoDot()
|
| D | shape_inference.cc | 615 absl::Span<const int64> rhs_batch_dimensions = in ValidateDotDimensionNumbers() local
616 AsInt64Slice(dimension_numbers.rhs_batch_dimensions()); in ValidateDotDimensionNumbers()
621 rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
638 !dims_unique(rhs_contracting_dimensions, rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
697 rhs.dimensions(dimension_numbers.rhs_batch_dimensions(i))) { in InferDotOpShape()
724 !absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), i)) { in InferDotOpShape()
|
| D | dynamic_dimension_inference.cc | 493 for (int64 i : dimension_numbers.rhs_batch_dimensions()) { in HandleDot()
522 if (absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), in HandleDot()
|
| D | indexed_array_analysis.cc | 1076 /*batch_dims=*/AsInt64Slice(dim_numbers.rhs_batch_dimensions()))) { in ComputeArrayForDotWithIndexedRhs()
|
| D | algebraic_simplifier.cc | 1777 for (int64 dim : dnums.rhs_batch_dimensions()) { in RemoveDegenerateDimensionFromDot()
2341 AsInt64Slice(dot->dot_dimension_numbers().rhs_batch_dimensions()), in HandleDot()
2392 AsInt64Slice(dot->dot_dimension_numbers().rhs_batch_dimensions()), in HandleDot()
4729 dnums.rhs_batch_dimensions(batch_dim)); in HandleReduce()
4735 dnums.rhs_batch_dimensions(batch_dim)); in HandleReduce()
|
| D | hlo_instructions.cc | 3007 if (!dnums.rhs_batch_dimensions().empty()) { in DotDimensionNumbersToString()
3009 StrJoin(dnums.rhs_batch_dimensions(), ","), "}")); in DotDimensionNumbersToString()
|
| D | hlo_evaluator_typed_visitor.h | 1425 &rhs_index[dnums.rhs_batch_dimensions(i)]});
1437 !absl::c_linear_search(dnums.rhs_batch_dimensions(), i)) {
|
| D | elemental_ir_emitter.cc | 2223 dot_result_index[dim_numbers.rhs_batch_dimensions(i)]); in EmitElementalDot()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 177 dnums.rhs_batch_dimensions.append(0)
579 dim_nums.rhs_batch_dimensions.append(2)
580 dim_nums.rhs_batch_dimensions.append(3)
595 dim_nums.rhs_batch_dimensions.append(0)
610 dim_nums.rhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_gemm_pad_for_tensor_cores.cc | 138 !absl::c_equal(dimension_numbers.rhs_batch_dimensions(), in CheckCanonical()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 499 self.rhs_batch_dimensions = []
524 dot_dims_proto.rhs_batch_dimensions.extend(rhs_batch)
|
| D | xla_client_test.py | 727 dimension_numbers.rhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 235 dimension_numbers.rhs_batch_dimensions(i); in __anonbbc1c2d10202()
275 absl::c_linear_search(dimension_numbers.rhs_batch_dimensions(), in __anonbbc1c2d10202()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 144 …hs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_batch_dimensions\":[]},\"bat…
169 …hs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_batch_dimensions\":[]},\"bat…
194 …hs_contracting_dimensions\":[\"0\"],\"lhs_batch_dimensions\":[],\"rhs_batch_dimensions\":[]},\"bat…
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1101 absl::c_equal(batch_dim_numbers, dim_numbers.rhs_batch_dimensions())); in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 573 repeated int64 rhs_batch_dimensions = 4; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 49 mapping.batch_dims.back().rhs = dnums.rhs_batch_dimensions(i); in HandleDot()
69 if (absl::c_linear_search(dnums.rhs_batch_dimensions(), i) || in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 670 GetI64DenseElementsAttr(hlo_dims.rhs_batch_dimensions()), in EmitGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1014 | 'rhs_batch_dimensions' | repeated int64 | 'rhs' batch dimension numbers |
|