Home
last modified time | relevance | path

Searched refs:lhs_contracting_dims (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgemm_rewrite_test.cc86 ROOT dot_a = f32[2,2] dot(x_transposed, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
109 dot_a = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
135 dot_a = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
161 dot_a = c64[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
187 dot_a = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
210 dot = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
236 dot_a = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
265 dot_a = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
293 dot = f32[2,2] dot(x, y), lhs_contracting_dims={1}, rhs_contracting_dims={0}
Delemental_ir_emitter.hlo60 …ROOT %dot = f32[1,4,5]{2,1,0} dot(%arg0, %arg1), lhs_batch_dims={0}, lhs_contracting_dims={2}, rhs…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt23 …ROOT %dot.4 = f32[] dot(f32[4]{0} %add.42, f32[4]{0} %Arg_1.2), lhs_contracting_dims={0}, rhs_cont…
311 …dot.3 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={1}, rhs_contracting_dims={0}, operand_p…
314 …dot.4 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={1}, rhs_contracting_dims={0}, operand_p…
317 …%dot.5 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={1}, rhs_contracting_dims={0}, operand_…
321 ROOT %dot.6 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={1}, rhs_contracting_dims={0}
332 …dot.3 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={0}, rhs_contracting_dims={1}, operand_p…
335 …dot.4 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={0}, rhs_contracting_dims={1}, operand_p…
338 …%dot.5 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={0}, rhs_contracting_dims={1}, operand_…
342 ROOT %dot.6 = f32[] dot(Arg_0.1, Arg_1.2), lhs_contracting_dims={0}, rhs_contracting_dims={1}
Dfully_connected_reference_model.hlotxt79 …%dot.36 = f32[300,5] dot(%reshape.26, %constant.35), lhs_contracting_dims={1}, rhs_contracting_dim…
Dexport.mlir463 …// CHECK: dot(s32[3,4] %{{.*}}, s32[4,5] %{{.*}}), lhs_contracting_dims={1}, rhs_contracting_dims…
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2108 auto lhs_contracting_dims = dnums.lhs_contracting_dimensions(); in OptimizeDotOfReorderContractingDims() local
2114 std::swap(lhs_contracting_dims, rhs_contracting_dims); in OptimizeDotOfReorderContractingDims()
2142 CHECK_EQ(lhs_contracting_dims.size(), 1); in OptimizeDotOfReorderContractingDims()
2145 return p.second == lhs_contracting_dims[0]; in OptimizeDotOfReorderContractingDims()
2161 lhs_contracting_dims.Clear(); in OptimizeDotOfReorderContractingDims()
2164 lhs_contracting_dims.Add(i); in OptimizeDotOfReorderContractingDims()
2173 if (!is_iota(AsInt64Slice(lhs_contracting_dims))) { in OptimizeDotOfReorderContractingDims()
2182 !absl::c_linear_search(lhs_contracting_dims, i)) { in OptimizeDotOfReorderContractingDims()
2189 for (auto dim : lhs_contracting_dims) { in OptimizeDotOfReorderContractingDims()
2190 permutation.push_back(transpose_dims[dim] - lhs_contracting_dims[0]); in OptimizeDotOfReorderContractingDims()
[all …]
Dhlo_parser.cc2380 optional<std::vector<int64>> lhs_contracting_dims; in ParseInstructionRhs() local
2382 /*required=*/false, AttrTy::kBracedInt64List, &lhs_contracting_dims}; in ParseInstructionRhs()
2402 if (lhs_contracting_dims) { in ParseInstructionRhs()
2404 lhs_contracting_dims->begin(), lhs_contracting_dims->end()}; in ParseInstructionRhs()
Dhlo_parser_test.cc1504 …ROOT dot = f32[2]{0} dot(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_batch_dims={1}, … in CreateTestCases()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1513 std::vector<int64> lhs_contracting_dims; in PartitionBaseCase() local
1514 lhs_contracting_dims.reserve(lhs.base_shape().rank()); in PartitionBaseCase()
1516 lhs_contracting_dims.push_back(cd.lhs); in PartitionBaseCase()
1520 lhs_contracting_dims, lhs.state().collective_ops_creator, in PartitionBaseCase()
1623 std::vector<int64> lhs_contracting_dims; in PartitionBaseCase() local
1624 lhs_contracting_dims.reserve(lhs.base_shape().rank()); in PartitionBaseCase()
1626 lhs_contracting_dims.push_back(cd.lhs); in PartitionBaseCase()
1630 lhs_contracting_dims, lhs.state().collective_ops_creator, in PartitionBaseCase()
1825 std::vector<int64> lhs_contracting_dims; in PartitionDotGroupOnBatch() local
1827 lhs_contracting_dims.reserve(dims_mapping.contracting_dims.size()); in PartitionDotGroupOnBatch()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1206 auto lhs_contracting_dims = in matchAndRewrite() local
1218 if (lhs_contracting_dims.size() != 1 || lhs_contracting_dims[0] != 2) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt512 …ROOT %out = f32[1,4,5]{2,1,0} dot(%arg0, %arg1), lhs_batch_dims={0}, lhs_contracting_dims={2}, rhs…