Home
last modified time | relevance | path

Searched refs:lhs_contracting_dimensions (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc58 const int64 lhs_contracting_dim_number = dnums.lhs_contracting_dimensions(0); in DecomposeBatchDot()
184 if (absl::c_linear_search(original_dnums.lhs_contracting_dimensions(), i)) { in CanonicalizeDot()
204 original_dnums.lhs_contracting_dimensions().begin(), in CanonicalizeDot()
205 original_dnums.lhs_contracting_dimensions().end()); in CanonicalizeDot()
305 dnums.lhs_contracting_dimensions().empty()) { in Run()
Dhlo_matchers.cc224 dim_nums.lhs_contracting_dimensions(0) != lhs_contracting_dim_) { in MatchAndExplain()
227 << absl::StrJoin(dim_nums.lhs_contracting_dimensions(), ",") in MatchAndExplain()
Dbatch_dot_simplification.cc67 new_dim_numbers.lhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
Dtranspose_folding.cc99 0, 1 - new_dim_numbers.lhs_contracting_dimensions(0)); in FoldTransposeIntoDot()
Dshape_inference.cc569 absl::Span<const int64> lhs_contracting_dimensions = in ValidateDotDimensionNumbers() local
570 AsInt64Slice(dimension_numbers.lhs_contracting_dimensions()); in ValidateDotDimensionNumbers()
578 if (!dims_in_range(lhs.rank(), lhs_contracting_dimensions, in ValidateDotDimensionNumbers()
597 if (!dims_unique(lhs_contracting_dimensions, lhs_batch_dimensions) || in ValidateDotDimensionNumbers()
647 dimension_numbers.lhs_contracting_dimensions(i); in InferDotOpShape()
687 if (!absl::c_linear_search(dimension_numbers.lhs_contracting_dimensions(), in InferDotOpShape()
Dalgebraic_simplifier.cc1153 int64 lhs_collapsing_dim = dnums.lhs_contracting_dimensions(0); in HandleDotStrengthReduction()
1371 const int64 lhs_contracting_dim = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfConcat()
1550 int lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfGather()
1663 dot->dot_dimension_numbers().lhs_contracting_dimensions()))); in HandleDot()
1706 dot->dot_dimension_numbers().lhs_contracting_dimensions()))); in HandleDot()
1804 dot->dot_dimension_numbers().lhs_contracting_dimensions(0) == 1 && in HandleDot()
Ddynamic_dimension_inference.cc220 dimension_numbers.lhs_contracting_dimensions(), i)) { in HandleDot()
Dhlo_cost_analysis.cc246 for (auto dim : dnums.lhs_contracting_dimensions()) { in HandleDot()
Dindexed_array_analysis.cc1040 AsInt64Slice(dim_numbers.lhs_contracting_dimensions()), in ComputeArrayForDotWithIndexedLhs()
Dhlo_evaluator_typed_visitor.h1230 const int64 lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in HandleDot()
1317 if (!absl::c_linear_search(dnums.lhs_contracting_dimensions(), i) && in HandleDotSlowPath()
1335 const int64 lhs_dnum = dnums.lhs_contracting_dimensions(i); in HandleDotSlowPath()
Dhlo_instructions.cc2566 StrJoin(dnums.lhs_contracting_dimensions(), ","), in DotDimensionNumbersToString()
Delemental_ir_emitter.cc2119 int64 lhs_contracting_dim = dim_numbers.lhs_contracting_dimensions(0); in EmitElementalDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc416 int64 lhs_reduction_dimension = dim_nums.lhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm()
692 /*m=*/lhs_shape.dimensions(1 - dim_nums.lhs_contracting_dimensions(0)), in GetMatMultDims()
693 /*k=*/lhs_shape.dimensions(dim_nums.lhs_contracting_dimensions(0)), in GetMatMultDims()
696 /*lhs_non_canonical=*/dim_nums.lhs_contracting_dimensions(0) == 0, in GetMatMultDims()
793 dot_info.dim_nums.lhs_contracting_dimensions(0)); in CanEmitTiledLlvmIrGemm()
805 bool lhs_non_canonical = dot_info.dim_nums.lhs_contracting_dimensions(0) == 0; in CanEmitTiledLlvmIrGemm()
975 dot_info.dim_nums.lhs_contracting_dimensions(0) - num_batch_dims); in EmitBatchDotOperation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc78 CHECK_EQ(lhs_shape.dimensions(dim_numbers.lhs_contracting_dimensions(0)), in DotImplementedAsGemm()
Dgemm_thunk.cc388 lhs_data, lhs_shape_, dim_nums.lhs_contracting_dimensions(0) == row_dim); in ExecuteOnStream()
Dir_emitter.cc510 const int64 lhs_reduction_dimension = dnums.lhs_contracting_dimensions(0); in HandleDot()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py149 dnums.lhs_contracting_dimensions.append(2)
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1913 self.lhs_contracting_dimensions = []
1922 dot_dims_proto.lhs_contracting_dimensions.extend(lhs_contract)
Dxla_client_test.py670 dimension_numbers.lhs_contracting_dimensions.append(2)
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto526 repeated int64 lhs_contracting_dimensions = 1; field
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md924 | 'lhs_contracting_dimensions' | repeated int64 | 'lhs' contracting dimension numbers |