Searched refs:rhs_contracting_dimensions (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 62 const int64 rhs_contracting_dim_number = dnums.rhs_contracting_dimensions(0); in DecomposeBatchDot()
229 if (absl::c_linear_search(original_dnums.rhs_contracting_dimensions(), i)) { in CanonicalizeDot()
246 original_dnums.rhs_contracting_dimensions().begin(), in CanonicalizeDot()
247 original_dnums.rhs_contracting_dimensions().end()); in CanonicalizeDot()
|
| D | hlo_matchers.cc | 233 dim_nums.rhs_contracting_dimensions(0) != rhs_contracting_dim_) { in MatchAndExplain()
236 << absl::StrJoin(dim_nums.rhs_contracting_dimensions(), ",") in MatchAndExplain()
|
| D | batch_dot_simplification.cc | 70 new_dim_numbers.rhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | transpose_folding.cc | 105 0, 1 - new_dim_numbers.rhs_contracting_dimensions(0)); in FoldTransposeIntoDot()
|
| D | shape_inference.cc | 571 absl::Span<const int64> rhs_contracting_dimensions = in ValidateDotDimensionNumbers() local
572 AsInt64Slice(dimension_numbers.rhs_contracting_dimensions()); in ValidateDotDimensionNumbers()
580 !dims_in_range(rhs.rank(), rhs_contracting_dimensions, in ValidateDotDimensionNumbers()
598 !dims_unique(rhs_contracting_dimensions, rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
649 dimension_numbers.rhs_contracting_dimensions(i); in InferDotOpShape()
695 if (!absl::c_linear_search(dimension_numbers.rhs_contracting_dimensions(), in InferDotOpShape()
|
| D | algebraic_simplifier.cc | 1165 int64 rhs_collapsing_dim = dnums.rhs_contracting_dimensions(0); in HandleDotStrengthReduction()
1372 const int64 rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat()
1551 int rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfGather()
1676 dot->dot_dimension_numbers().rhs_contracting_dimensions()))); in HandleDot()
1713 dot->dot_dimension_numbers().rhs_contracting_dimensions()))); in HandleDot()
1805 dot->dot_dimension_numbers().rhs_contracting_dimensions(0) == 0 && in HandleDot()
|
| D | dynamic_dimension_inference.cc | 230 dimension_numbers.rhs_contracting_dimensions(), i) && in HandleDot()
|
| D | indexed_array_analysis.cc | 1075 AsInt64Slice(dim_numbers.rhs_contracting_dimensions()), in ComputeArrayForDotWithIndexedRhs()
|
| D | hlo_evaluator_typed_visitor.h | 1231 const int64 rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in HandleDot()
1323 if (!absl::c_linear_search(dnums.rhs_contracting_dimensions(), i) && in HandleDotSlowPath()
1336 const int64 rhs_dnum = dnums.rhs_contracting_dimensions(i); in HandleDotSlowPath()
|
| D | hlo_instructions.cc | 2574 StrJoin(dnums.rhs_contracting_dimensions(), ","), in DotDimensionNumbersToString()
|
| D | elemental_ir_emitter.cc | 2120 int64 rhs_contracting_dim = dim_numbers.rhs_contracting_dimensions(0); in EmitElementalDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 417 int64 rhs_reduction_dimension = dim_nums.rhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm()
694 /*n=*/rhs_shape.dimensions(1 - dim_nums.rhs_contracting_dimensions(0)), in GetMatMultDims()
698 /*rhs_non_canonical=*/dim_nums.rhs_contracting_dimensions(0) == 1, in GetMatMultDims()
709 if (hlo.dot_dimension_numbers().rhs_contracting_dimensions(0) == 0) { in ProfitableToMakeDotOperandColumnMajor()
806 bool rhs_non_canonical = dot_info.dim_nums.rhs_contracting_dimensions(0) == 1; in CanEmitTiledLlvmIrGemm()
978 dot_info.dim_nums.rhs_contracting_dimensions(0) - num_batch_dims); in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 79 rhs_shape.dimensions(dim_numbers.rhs_contracting_dimensions(0))); in DotImplementedAsGemm()
|
| D | gemm_thunk.cc | 390 rhs_data, rhs_shape_, dim_nums.rhs_contracting_dimensions(0) == col_dim); in ExecuteOnStream()
|
| D | ir_emitter.cc | 511 const int64 rhs_reduction_dimension = dnums.rhs_contracting_dimensions(0); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 150 dnums.rhs_contracting_dimensions.append(1)
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1914 self.rhs_contracting_dimensions = []
1923 dot_dims_proto.rhs_contracting_dimensions.extend(rhs_contract)
|
| D | xla_client_test.py | 671 dimension_numbers.rhs_contracting_dimensions.append(1)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 528 repeated int64 rhs_contracting_dimensions = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 925 | 'rhs_contracting_dimensions' | repeated int64 | 'rhs' contracting dimension numbers |
|