Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc85 std::vector<int64_t> lhs_contracting_dimensions( in ConvertDotDimensionNumbers() local
86 dnums.lhs_contracting_dimensions().begin(), in ConvertDotDimensionNumbers()
87 dnums.lhs_contracting_dimensions().end()); in ConvertDotDimensionNumbers()
96 auto lhs_contracting_dims_attr = Convert(lhs_contracting_dimensions, builder); in ConvertDotDimensionNumbers()
Dmlir_hlo_to_hlo.cc243 auto lhs_contracting_dimensions = in Convert_dot_dimension_numbers() local
244 dot_dimension_numbers_attr.lhs_contracting_dimensions() in Convert_dot_dimension_numbers()
256 for (const auto& val : lhs_contracting_dimensions) { in Convert_dot_dimension_numbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-BatchMatMulV2.mlir20 …numbers = {lhs_batching_dimensions = dense<0> : tensor<1xi64>, lhs_contracting_dimensions = dense<…
33 // CHECK-SAME: lhs_contracting_dimensions = dense<2> : tensor<1xi64>,
45 // CHECK-SAME: lhs_contracting_dimensions = dense<2> : tensor<1xi64>,
56 // CHECK-SAME: lhs_contracting_dimensions = dense<2> : tensor<1xi64>,
67 // CHECK-SAME: lhs_contracting_dimensions = dense<0> : tensor<1xi64>,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlower-general-dot.mlir8 …umbers = {lhs_batching_dimensions = dense<[]> : tensor<0xi64>, lhs_contracting_dimensions = dense<…
23 …umbers = {lhs_batching_dimensions = dense<[]> : tensor<0xi64>, lhs_contracting_dimensions = dense<…
32 …mbers = {lhs_batching_dimensions = dense<[0]> : tensor<1xi64>, lhs_contracting_dimensions = dense<…
Dlhlo_gpu_ops.mlir191 lhs_contracting_dimensions = dense<[1,1]> : tensor<2xi64>,
208 lhs_contracting_dimensions = dense<[1,1]> : tensor<2xi64>,
Dlhlo-legalize-to-affine.mlir174 lhs_contracting_dimensions = dense<1> : tensor<1xi64>,
198 lhs_contracting_dimensions = dense<1> : tensor<1xi64>,
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc58 if (absl::c_linear_search(original_dnums.lhs_contracting_dimensions(), i)) { in CanonicalizeDot()
79 original_dnums.lhs_contracting_dimensions().begin(), in CanonicalizeDot()
80 original_dnums.lhs_contracting_dimensions().end()); in CanonicalizeDot()
197 dnums.lhs_contracting_dimensions().empty()) { in Run()
Ddot_as_convolution_util.cc165 for (int64 i = 0; i < dot_dim_numbs.lhs_contracting_dimensions().size(); in ParseDotGeneralFromDot()
169 dot_dim_numbs.lhs_contracting_dimensions(i); in ParseDotGeneralFromDot()
177 !absl::c_linear_search(dot_dim_numbs.lhs_contracting_dimensions(), i)) { in ParseDotGeneralFromDot()
Dhlo_matchers.cc228 dim_nums.lhs_contracting_dimensions(0) != lhs_contracting_dim_) { in MatchAndExplain()
230 << absl::StrJoin(dim_nums.lhs_contracting_dimensions(), ",") in MatchAndExplain()
Dbatch_dot_simplification.cc85 new_dim_numbers.lhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
Dtranspose_folding.cc127 new_lhs->dimensions(new_dim_numbers.lhs_contracting_dimensions(0))); in FoldTransposeIntoDot()
Dshape_inference.cc609 absl::Span<const int64> lhs_contracting_dimensions = in ValidateDotDimensionNumbers() local
610 AsInt64Slice(dimension_numbers.lhs_contracting_dimensions()); in ValidateDotDimensionNumbers()
618 if (!dims_in_range(lhs.rank(), lhs_contracting_dimensions, in ValidateDotDimensionNumbers()
637 if (!dims_unique(lhs_contracting_dimensions, lhs_batch_dimensions) || in ValidateDotDimensionNumbers()
679 dimension_numbers.lhs_contracting_dimensions(i); in InferDotOpShape()
714 if (!absl::c_linear_search(dimension_numbers.lhs_contracting_dimensions(), in InferDotOpShape()
Dalgebraic_simplifier.cc1770 for (int64 dim : dnums.lhs_contracting_dimensions()) { in RemoveDegenerateDimensionFromDot()
1824 const int64 lhs_contracting_dim = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfConcat()
2004 int lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfGather()
2108 auto lhs_contracting_dims = dnums.lhs_contracting_dimensions(); in OptimizeDotOfReorderContractingDims()
2333 dot->dot_dimension_numbers().lhs_contracting_dimensions()))); in HandleDot()
2383 dot->dot_dimension_numbers().lhs_contracting_dimensions()))); in HandleDot()
2492 dot->dot_dimension_numbers().lhs_contracting_dimensions(0) == 1 && in HandleDot()
5107 dnums.lhs_contracting_dimensions(); in HandleTranspose()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_affine.cc67 if (dot_dimension_numbers.lhs_contracting_dimensions().size() != 1 || in matchAndRewrite()
68 *dot_dimension_numbers.lhs_contracting_dimensions().begin() != 1 || in matchAndRewrite()
Dlower_general_dot.cc150 dot_numbers.lhs_contracting_dimensions(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc294 b_exprs[dot_info_.dim_nums.lhs_contracting_dimensions(0)] = reduce_expr; in EmitLinalgMatmul()
559 int64 lhs_reduction_dimension = dim_nums.lhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm()
868 : lhs_shape.dimensions(1LL - dim_nums.lhs_contracting_dimensions(0)), in GetMatMultDims()
869 /*k=*/lhs_shape.dimensions(dim_nums.lhs_contracting_dimensions(0)), in GetMatMultDims()
875 dim_nums.lhs_contracting_dimensions(0) == 1, in GetMatMultDims()
983 dot_info.dim_nums.lhs_contracting_dimensions(0)); in CanEmitTiledLlvmIrGemm()
995 bool lhs_canonical = dot_info.dim_nums.lhs_contracting_dimensions(0) == 1; in CanEmitTiledLlvmIrGemm()
1171 dot_info.dim_nums.lhs_contracting_dimensions(0) - num_batch_dims); in EmitBatchDotOperation()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py174 dnums.lhs_contracting_dimensions.append(2)
551 dim_nums.lhs_contracting_dimensions.append(2)
565 dim_nums.lhs_contracting_dimensions.append(2)
592 dim_nums.lhs_contracting_dimensions.append(2)
607 dim_nums.lhs_contracting_dimensions.append(1)
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt135 // CHECK-SAME: lhs_contracting_dimensions = dense<1> : tensor<1xi64>
144 …l\":1,\"alpha_imag\":0,\"beta\":0,\"dot_dimension_numbers\":{\"lhs_contracting_dimensions\":[\"1\"…
159 // CHECK-SAME: lhs_contracting_dimensions = dense<1> : tensor<1xi64>
169 …l\":1,\"alpha_imag\":0,\"beta\":1,\"dot_dimension_numbers\":{\"lhs_contracting_dimensions\":[\"1\"…
184 // CHECK-SAME: lhs_contracting_dimensions = dense<1> : tensor<1xi64>
194 …l\":1,\"alpha_imag\":0,\"beta\":1,\"dot_dimension_numbers\":{\"lhs_contracting_dimensions\":[\"1\"…
504 // CHECK-SAME: lhs_contracting_dimensions = dense<2> : tensor<1xi64>,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dsimple.hlo70 lhs_contracting_dimensions: 0
Dimport.hlotxt331 …_numbers = {lhs_batching_dimensions = dense<> : tensor<0xi64>, lhs_contracting_dimensions = dense<…
334 …_numbers = {lhs_batching_dimensions = dense<> : tensor<0xi64>, lhs_contracting_dimensions = dense<…
337 …_numbers = {lhs_batching_dimensions = dense<> : tensor<0xi64>, lhs_contracting_dimensions = dense<…
341 …_numbers = {lhs_batching_dimensions = dense<> : tensor<0xi64>, lhs_contracting_dimensions = dense<…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td26 StructFieldAttr<"lhs_contracting_dimensions", I64ElementsAttr>,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py496 self.lhs_contracting_dimensions = []
521 dot_dims_proto.lhs_contracting_dimensions.extend(lhs_contract)
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc199 dimension_numbers.lhs_contracting_dimensions(i); in __anonbbc1c2d10202()
263 dimension_numbers.lhs_contracting_dimensions(), i) || in __anonbbc1c2d10202()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc244 lhs_buffer, lhs_shape, dim_nums.lhs_contracting_dimensions(0) == row_dim); in RunGemm()
Dir_emission_utils.cc121 CHECK_EQ(lhs_shape.dimensions(dim_numbers.lhs_contracting_dimensions(0)), in IsMatrixMultiplication()

12