Searched defs:dot_op (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_test.cc | 3928 auto dot_op = op::Dot(lhs, rhs); in TEST_F() local
3980 auto dot_op = op::Dot(lhs, rhs); in TEST_F() local
9988 const HloInstruction* dot_op = FindInstruction(module.get(), HloOpcode::kDot); in TEST_F() local
|
| /aosp_15_r20/external/mesa3d/src/amd/common/ |
| H A D | ac_nir_lower_ngg.c | 271 …nir_def *dot_op = !use_dot ? NULL : nir_ushr(b, nir_ushr(b, nir_imm_int(b, 0x01010101), shift), sh… in summarize_repack() local
284 …nir_def *dot_op = !use_dot ? NULL : nir_ushr(b, nir_ushr(b, nir_imm_int64(b, 0x0101010101010101), … in summarize_repack() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 1766 auto dot_op = cast<mhlo::DotOp>(old_op); in ConvertDotOp() local
|