Searched refs:kTransposeDot (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | liveness_util.cc | 222 HloInstruction::FusionKind::kTransposeDot); in CanShareOperandBufferWithUser()
327 HloInstruction::FusionKind::kTransposeDot); in CanShareOperandBufferWithUser()
|
| D | transpose_folding.cc | 92 instructions_to_fuse, HloInstruction::FusionKind::kTransposeDot); in FoldTransposeIntoDot()
|
| D | hlo_instruction_test.cc | 1105 {dot, reshape}, HloInstruction::FusionKind::kTransposeDot); in TEST_F()
1172 {dot, b_t}, HloInstruction::FusionKind::kTransposeDot); in TEST_F()
1250 {dot, reshape}, HloInstruction::FusionKind::kTransposeDot); in TEST_F()
|
| D | liveness_util_test.cc | 334 {dot, b_t}, HloInstruction::FusionKind::kTransposeDot); in TEST_F()
|
| D | hlo_instruction.cc | 771 case HloInstruction::FusionKind::kTransposeDot: in FusionNodeName()
2422 case FusionKind::kTransposeDot: in ToCategory()
3200 case HloInstruction::FusionKind::kTransposeDot: in ToString()
3219 return HloInstruction::FusionKind::kTransposeDot; in StringToFusionKind()
|
| D | hlo_instruction.h | 170 kTransposeDot, // Fused into a dot with transposed operands. enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 85 hlo.fusion_kind() == HloInstruction::FusionKind::kTransposeDot && in ImplementedAsGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 187 HloInstruction::FusionKind::kTransposeDot); in TEST_F()
191 HloInstruction::FusionKind::kTransposeDot); in TEST_F()
|
| D | dot_op_emitter.cc | 1086 hlo.fusion_kind() == HloInstruction::FusionKind::kTransposeDot && in PotentiallyImplementedAsEigenDot()
|
| D | ir_emitter.cc | 2064 if (fusion->fusion_kind() == HloInstruction::FusionKind::kTransposeDot) { in HandleFusion()
|