| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gemm_rewrite_test.cc | 86 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}
|
| D | elemental_ir_emitter.hlo | 60 … %arg1), lhs_batch_dims={0}, lhs_contracting_dims={2}, rhs_batch_dims={0}, rhs_contracting_dims={1}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | import.hlotxt | 23 …32[] dot(f32[4]{0} %add.42, f32[4]{0} %Arg_1.2), lhs_contracting_dims={0}, rhs_contracting_dims={0}
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}
|
| D | fully_connected_reference_model.hlotxt | 79 ….36 = f32[300,5] dot(%reshape.26, %constant.35), lhs_contracting_dims={1}, rhs_contracting_dims={0}
|
| D | export.mlir | 463 …CHECK: dot(s32[3,4] %{{.*}}, s32[4,5] %{{.*}}), lhs_contracting_dims={1}, rhs_contracting_dims={0}
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2109 auto rhs_contracting_dims = dnums.rhs_contracting_dimensions(); in OptimizeDotOfReorderContractingDims() local
2114 std::swap(lhs_contracting_dims, rhs_contracting_dims); in OptimizeDotOfReorderContractingDims()
2235 CHECK_EQ(rhs_contracting_dims.size(), 1); in OptimizeDotOfReorderContractingDims()
2239 rhs_contracting_dims[0]); in OptimizeDotOfReorderContractingDims()
2253 rhs_contracting_dims.Resize(lhs_contracting_dims.size(), in OptimizeDotOfReorderContractingDims()
2254 rhs_contracting_dims[0]); in OptimizeDotOfReorderContractingDims()
2255 absl::c_iota(rhs_contracting_dims, rhs_contracting_dims[0]); in OptimizeDotOfReorderContractingDims()
2261 rhs_transpose_shape_dims.begin() + rhs_contracting_dims[0], in OptimizeDotOfReorderContractingDims()
2262 rhs_transpose_shape_dims.begin() + rhs_contracting_dims[0] + in OptimizeDotOfReorderContractingDims()
2263 rhs_contracting_dims.size()); in OptimizeDotOfReorderContractingDims()
[all …]
|
| D | hlo_parser.cc | 2383 optional<std::vector<int64>> rhs_contracting_dims; in ParseInstructionRhs() local
2385 /*required=*/false, AttrTy::kBracedInt64List, &rhs_contracting_dims}; in ParseInstructionRhs()
2406 if (rhs_contracting_dims) { in ParseInstructionRhs()
2408 rhs_contracting_dims->begin(), rhs_contracting_dims->end()}; in ParseInstructionRhs()
|
| D | hlo_parser_test.cc | 1504 …t(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_batch_dims={1}, rhs_contracting_dims={0} in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1208 auto rhs_contracting_dims = in matchAndRewrite() local
1222 if (rhs_contracting_dims.size() != 1 || rhs_contracting_dims[0] != 1) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1826 std::vector<int64> rhs_contracting_dims; in PartitionDotGroupOnBatch() local
1828 rhs_contracting_dims.reserve(dims_mapping.contracting_dims.size()); in PartitionDotGroupOnBatch()
1831 rhs_contracting_dims.push_back(dim.rhs); in PartitionDotGroupOnBatch()
1855 rhs, rhs_dims, rhs_contracting_dims, rhs_non_contracting_dims, in PartitionDotGroupOnBatch()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 512 … %arg1), lhs_batch_dims={0}, lhs_contracting_dims={2}, rhs_batch_dims={0}, rhs_contracting_dims={1}
|