Home
last modified time | relevance | path

Searched refs:rhs_contracting_dimensions (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc82 std::vector<int64_t> rhs_contracting_dimensions( in ConvertDotDimensionNumbers() local
83 dnums.rhs_contracting_dimensions().begin(), in ConvertDotDimensionNumbers()
84 dnums.rhs_contracting_dimensions().end()); in ConvertDotDimensionNumbers()
97 auto rhs_contracting_dims_attr = Convert(rhs_contracting_dimensions, builder); in ConvertDotDimensionNumbers()
Dmlir_hlo_to_hlo.cc240 auto rhs_contracting_dimensions = in Convert_dot_dimension_numbers() local
241 dot_dimension_numbers_attr.rhs_contracting_dimensions() in Convert_dot_dimension_numbers()
253 for (const auto& val : rhs_contracting_dimensions) { in Convert_dot_dimension_numbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-BatchMatMulV2.mlir20 …or<1xi64>, rhs_batching_dimensions = dense<0> : tensor<1xi64>, rhs_contracting_dimensions = dense<…
35 // CHECK-SAME: rhs_contracting_dimensions = dense<1> : tensor<1xi64>}}
47 // CHECK-SAME: rhs_contracting_dimensions = dense<1> : tensor<1xi64>}}
58 // CHECK-SAME: rhs_contracting_dimensions = dense<1> : tensor<1xi64>}}
69 // CHECK-SAME: rhs_contracting_dimensions = dense<1> : tensor<1xi64>}}
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlower-general-dot.mlir8 …r<1xi64>, rhs_batching_dimensions = dense<[]> : tensor<0xi64>, rhs_contracting_dimensions = dense<…
23 …r<1xi64>, rhs_batching_dimensions = dense<[]> : tensor<0xi64>, rhs_contracting_dimensions = dense<…
32 …<1xi64>, rhs_batching_dimensions = dense<[0]> : tensor<1xi64>, rhs_contracting_dimensions = dense<…
Dlhlo_gpu_ops.mlir192 rhs_contracting_dimensions = dense<[1,1]> : tensor<2xi64>},
209 rhs_contracting_dimensions = dense<[1,1]> : tensor<2xi64>},
Dlhlo-legalize-to-affine.mlir175 rhs_contracting_dimensions = dense<0> : tensor<1xi64>
199 rhs_contracting_dimensions = dense<0> : tensor<1xi64>
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc105 if (absl::c_linear_search(original_dnums.rhs_contracting_dimensions(), i)) { in CanonicalizeDot()
123 original_dnums.rhs_contracting_dimensions().begin(), in CanonicalizeDot()
124 original_dnums.rhs_contracting_dimensions().end()); in CanonicalizeDot()
Dhlo_matchers.cc236 dim_nums.rhs_contracting_dimensions(0) != rhs_contracting_dim_) { in MatchAndExplain()
238 << absl::StrJoin(dim_nums.rhs_contracting_dimensions(), ",") in MatchAndExplain()
Ddot_as_convolution_util.cc171 dot_dim_numbs.rhs_contracting_dimensions(i); in ParseDotGeneralFromDot()
189 !absl::c_linear_search(dot_dim_numbs.rhs_contracting_dimensions(), i)) { in ParseDotGeneralFromDot()
Dbatch_dot_simplification.cc88 new_dim_numbers.rhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
Dtranspose_folding.cc134 new_rhs->dimensions(new_dim_numbers.rhs_contracting_dimensions(0))); in FoldTransposeIntoDot()
Dshape_inference.cc611 absl::Span<const int64> rhs_contracting_dimensions = in ValidateDotDimensionNumbers() local
612 AsInt64Slice(dimension_numbers.rhs_contracting_dimensions()); in ValidateDotDimensionNumbers()
620 !dims_in_range(rhs.rank(), rhs_contracting_dimensions, in ValidateDotDimensionNumbers()
638 !dims_unique(rhs_contracting_dimensions, rhs_batch_dimensions)) { in ValidateDotDimensionNumbers()
681 dimension_numbers.rhs_contracting_dimensions(i); in InferDotOpShape()
722 if (!absl::c_linear_search(dimension_numbers.rhs_contracting_dimensions(), in InferDotOpShape()
Dalgebraic_simplifier.cc1783 for (int64 dim : dnums.rhs_contracting_dimensions()) { in RemoveDegenerateDimensionFromDot()
1825 const int64 rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat()
2005 int rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfGather()
2109 auto rhs_contracting_dims = dnums.rhs_contracting_dimensions(); in OptimizeDotOfReorderContractingDims()
2343 dot->dot_dimension_numbers().rhs_contracting_dimensions()))); in HandleDot()
2394 dot->dot_dimension_numbers().rhs_contracting_dimensions()))); in HandleDot()
2493 dot->dot_dimension_numbers().rhs_contracting_dimensions(0) == 0 && in HandleDot()
5105 dnums.rhs_contracting_dimensions(); in HandleTranspose()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py175 dnums.rhs_contracting_dimensions.append(1)
552 dim_nums.rhs_contracting_dimensions.append(2)
553 dim_nums.rhs_contracting_dimensions.append(3)
566 dim_nums.rhs_contracting_dimensions.append(3)
593 dim_nums.rhs_contracting_dimensions.append(3)
608 dim_nums.rhs_contracting_dimensions.append(2)
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_affine.cc69 dot_dimension_numbers.rhs_contracting_dimensions().size() != 1 || in matchAndRewrite()
70 *dot_dimension_numbers.rhs_contracting_dimensions().begin() != 0) { in matchAndRewrite()
Dlower_general_dot.cc154 dot_numbers.rhs_contracting_dimensions(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt137 // CHECK-SAME: rhs_contracting_dimensions = dense<0> : tensor<1xi64>
144 …_dimension_numbers\":{\"lhs_contracting_dimensions\":[\"1\"],\"rhs_contracting_dimensions\":[\"0\"…
161 // CHECK-SAME: rhs_contracting_dimensions = dense<0> : tensor<1xi64>
169 …_dimension_numbers\":{\"lhs_contracting_dimensions\":[\"1\"],\"rhs_contracting_dimensions\":[\"0\"…
186 // CHECK-SAME: rhs_contracting_dimensions = dense<0> : tensor<1xi64>
194 …_dimension_numbers\":{\"lhs_contracting_dimensions\":[\"1\"],\"rhs_contracting_dimensions\":[\"0\"…
506 // CHECK-SAME: rhs_contracting_dimensions = dense<1> : tensor<1xi64>},
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc295 c_exprs[dot_info_.dim_nums.rhs_contracting_dimensions(0)] = reduce_expr; in EmitLinalgMatmul()
560 int64 rhs_reduction_dimension = dim_nums.rhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm()
872 : rhs_shape.dimensions(1LL - dim_nums.rhs_contracting_dimensions(0)), in GetMatMultDims()
877 /*rhs_canonical=*/dim_nums.rhs_contracting_dimensions(0) == 0}; in GetMatMultDims()
886 hlo.dot_dimension_numbers().rhs_contracting_dimensions(0) != 0) { in ProfitableToMakeDotOperandColumnMajor()
996 bool rhs_canonical = dot_info.dim_nums.rhs_contracting_dimensions(0) == 0; in CanEmitTiledLlvmIrGemm()
1174 dot_info.dim_nums.rhs_contracting_dimensions(0) - num_batch_dims); in EmitBatchDotOperation()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dsimple.hlo71 rhs_contracting_dimensions: 0
Dimport.hlotxt331 …sor<1xi64>, rhs_batching_dimensions = dense<> : tensor<0xi64>, rhs_contracting_dimensions = dense<…
334 …sor<1xi64>, rhs_batching_dimensions = dense<> : tensor<0xi64>, rhs_contracting_dimensions = dense<…
337 …sor<1xi64>, rhs_batching_dimensions = dense<> : tensor<0xi64>, rhs_contracting_dimensions = dense<…
341 …sor<1xi64>, rhs_batching_dimensions = dense<> : tensor<0xi64>, rhs_contracting_dimensions = dense<…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td27 StructFieldAttr<"rhs_contracting_dimensions", I64ElementsAttr>
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py497 self.rhs_contracting_dimensions = []
522 dot_dims_proto.rhs_contracting_dimensions.extend(rhs_contract)
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc201 dimension_numbers.rhs_contracting_dimensions(i); in __anonbbc1c2d10202()
274 dimension_numbers.rhs_contracting_dimensions(), i) || in __anonbbc1c2d10202()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc246 rhs_buffer, rhs_shape, dim_nums.rhs_contracting_dimensions(0) == col_dim); in RunGemm()
Dir_emission_utils.cc122 rhs_shape.dimensions(dim_numbers.rhs_contracting_dimensions(0))); in IsMatrixMultiplication()

12