Searched refs:lhs_batch_dimensions (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 51 CHECK_EQ(lhs_shape.dimensions(dnums.lhs_batch_dimensions(i)), in DecomposeBatchDot()
53 batch_size *= lhs_shape.dimensions(dnums.lhs_batch_dimensions(i)); in DecomposeBatchDot()
186 } else if (absl::c_linear_search(original_dnums.lhs_batch_dimensions(), in CanonicalizeDot()
199 original_dnums.lhs_batch_dimensions().begin(), in CanonicalizeDot()
200 original_dnums.lhs_batch_dimensions().end()); in CanonicalizeDot()
304 if (dnums.lhs_batch_dimensions().empty() && in Run()
309 if (dnums.lhs_batch_dimensions().empty()) { in Run()
315 if (!absl::c_equal(dnums.lhs_batch_dimensions(), canonical_batch_dims) || in Run()
335 if (!dnums.lhs_batch_dimensions().empty()) { in Run()
|
| D | batch_dot_simplification.cc | 39 for (int64 batch_dim : dim_numbers.lhs_batch_dimensions()) { in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | shape_inference.cc | 573 absl::Span<const int64> lhs_batch_dimensions = in ValidateDotDimensionNumbers() local
574 AsInt64Slice(dimension_numbers.lhs_batch_dimensions()); in ValidateDotDimensionNumbers()
579 lhs_batch_dimensions) || in ValidateDotDimensionNumbers()
597 if (!dims_unique(lhs_contracting_dimensions, lhs_batch_dimensions) || in ValidateDotDimensionNumbers()
666 if (lhs.dimensions(dimension_numbers.lhs_batch_dimensions(i)) != in InferDotOpShape()
668 lhs.is_dynamic_dimension(dimension_numbers.lhs_batch_dimensions(i)) != in InferDotOpShape()
682 for (int64 lhs_dim : dimension_numbers.lhs_batch_dimensions()) { in InferDotOpShape()
689 !absl::c_linear_search(dimension_numbers.lhs_batch_dimensions(), i)) { in InferDotOpShape()
|
| D | algebraic_simplifier.cc | 1155 AsInt64Slice(dnums.lhs_batch_dimensions())); in HandleDotStrengthReduction()
1281 !absl::c_equal(dnums.lhs_batch_dimensions(), in HandleDotStrengthReduction()
1305 for (const int64 dim : dnums.lhs_batch_dimensions()) { in HandleDotStrengthReduction()
1661 AsInt64Slice(dot->dot_dimension_numbers().lhs_batch_dimensions()), in HandleDot()
1704 AsInt64Slice(dot->dot_dimension_numbers().lhs_batch_dimensions()), in HandleDot()
|
| D | indexed_array_analysis.cc | 1041 /*batch_dims=*/AsInt64Slice(dim_numbers.lhs_batch_dimensions()))) { in ComputeArrayForDotWithIndexedLhs()
|
| D | hlo_instructions.cc | 2561 if (!dnums.lhs_batch_dimensions().empty()) { in DotDimensionNumbersToString()
2563 StrJoin(dnums.lhs_batch_dimensions(), ","), "}")); in DotDimensionNumbersToString()
|
| D | hlo_evaluator_typed_visitor.h | 1311 {&lhs_index[dnums.lhs_batch_dimensions(i)], in HandleDotSlowPath()
1318 !absl::c_linear_search(dnums.lhs_batch_dimensions(), i)) { in HandleDotSlowPath()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 200 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in AddBackendConstraints()
|
| D | ir_emitter.cc | 514 for (int64 batch_dim : dnums.lhs_batch_dimensions()) { in HandleDot()
541 DCHECK(absl::c_linear_search(dnums.lhs_batch_dimensions(), i)); in HandleDot()
|
| D | gemm_thunk.cc | 335 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 233 for (auto d : dnums.lhs_batch_dimensions()) { in Einsum()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 151 dnums.lhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1915 self.lhs_batch_dimensions = []
1924 dot_dims_proto.lhs_batch_dimensions.extend(lhs_batch)
|
| D | xla_client_test.py | 672 dimension_numbers.lhs_batch_dimensions.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 903 absl::c_equal(batch_dim_numbers, dim_numbers.lhs_batch_dimensions())); in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 530 repeated int64 lhs_batch_dimensions = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 926 | 'lhs_batch_dimensions' | repeated int64 | 'lhs' batch dimension numbers |
|