| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper_test.cc | 85 RenderedGraphFormat::kDot)); in TEST_F()
109 RenderedGraphFormat::kDot)); in TEST_F()
123 DebugOptions(), RenderedGraphFormat::kDot)); in TEST_F()
142 DebugOptions(), RenderedGraphFormat::kDot)); in TEST_F()
161 DebugOptions(), RenderedGraphFormat::kDot)); in TEST_F()
|
| D | hlo_graph_dumper.h | 48 kDot, enumerator
|
| D | batch_dot_simplification.cc | 98 return instr->opcode() == HloOpcode::kDot; in Run()
|
| D | dot_decomposer.cc | 294 if (instruction->opcode() != HloOpcode::kDot) { in Run()
331 if (instruction->opcode() != HloOpcode::kDot) { in Run()
|
| D | dynamic_padder.cc | 60 case HloOpcode::kDot: { in ChooseIdentityValue()
|
| D | hlo_opcode.h | 82 V(kDot, "dot", 2) \
|
| D | hlo_matchers.h | 157 : HloMatcher(HloOpcode::kDot, /*operands=*/{lhs, rhs}), in HloDotWithContractingDimsMatcher()
393 ::xla::HloOpcode::kDot, {lhs_matcher, rhs_matcher})); in Dot()
|
| D | bfloat16_normalization_test.cc | 46 if (hlo.opcode() == HloOpcode::kDot) { in SupportsBF16Operand()
56 hlo.opcode() == HloOpcode::kDot || hlo.opcode() == HloOpcode::kTuple || in SupportsBF16Output()
|
| D | transpose_folding.cc | 38 if (HloOpcode::kDot != dot.opcode() || in CanFoldOperandsIntoDot()
|
| D | dump.cc | 218 render_graph(RenderedGraphFormat::kDot), opts); in DumpHloModuleImpl()
|
| D | instruction_fusion.cc | 133 case HloOpcode::kDot: in IsExpensive()
|
| D | hlo_instruction.cc | 555 case HloOpcode::kDot: { in CreateFromProto()
1387 case HloOpcode::kDot: in CloneWithNewOperands()
1837 case HloOpcode::kDot: in IdenticalSlowPath()
2535 case HloOpcode::kDot: in Visit()
2947 case HloOpcode::kDot: in OperandElementUse()
|
| D | tuple_points_to_analysis.cc | 745 operand->opcode() == HloOpcode::kDot; in CanShareOperandBufferWithUser()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion.cc | 122 if (producer->opcode() == HloOpcode::kDot || in ShouldFuseInexpensiveChecks()
124 producer->fused_expression_root()->opcode() == HloOpcode::kDot)) { in ShouldFuseInexpensiveChecks()
181 return fused_parameter_users[0]->opcode() == HloOpcode::kDot; in ShouldFuseInexpensiveChecks()
285 if (producer->opcode() == HloOpcode::kDot || in ChooseKind()
287 producer->fused_expression_root()->opcode() == HloOpcode::kDot)) { in ChooseKind()
|
| D | ir_emission_utils.cc | 66 CHECK_EQ(dot.opcode(), HloOpcode::kDot); in DotImplementedAsGemm()
88 if (hlo.opcode() == HloOpcode::kDot) { in ImplementedAsGemm()
98 if (dot->opcode() != HloOpcode::kDot) { in ImplementedAsGemm()
101 if (dot->opcode() == HloOpcode::kDot) { in ImplementedAsGemm()
|
| D | gemm_thunk.cc | 272 if (hlo_instruction.opcode() == HloOpcode::kDot) { in GetDimensionNumbers()
283 if (dot->opcode() != HloOpcode::kDot) { in GetDimensionNumbers()
286 CHECK_EQ(dot->opcode(), HloOpcode::kDot); in GetDimensionNumbers()
|
| D | gpu_layout_assignment.cc | 190 if (instruction->opcode() == HloOpcode::kDot && in AddBackendConstraints()
|
| D | ir_emitter_unnested.cc | 1776 if (inst->opcode() == HloOpcode::kDot) { in BuildGemmThunk()
1800 if (dot->opcode() != HloOpcode::kDot) { in BuildGemmThunk()
1820 if (dot->opcode() != HloOpcode::kDot) { in BuildGemmThunk()
1826 DCHECK(dot->opcode() == HloOpcode::kDot); in BuildGemmThunk()
2232 case HloOpcode::kDot: in MayPreventVectorization()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | debug_options_parsers_test.cc | 51 HloOpcode::kDot); in TEST()
66 HloOpcode::kDot); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 49 hlo->opcode() == HloOpcode::kDot && hlo_shape.dimensions_size() == 2 && in IsNonComplexMatrixVectorDot()
129 if (consumer->opcode() == HloOpcode::kDot) { in ShouldFuse()
|
| D | dot_op_emitter_internal.h | 44 CHECK_EQ(instr.opcode(), HloOpcode::kDot); in DotInfo()
|
| D | cpu_layout_assignment.cc | 102 } else if (instr.opcode() == HloOpcode::kDot) { in OperandsAndResultMustHaveRowMajorLayout()
|
| D | parallel_task_assignment.cc | 141 opcode == HloOpcode::kDot || opcode == HloOpcode::kSelectAndScatter || in GetTargetParallelTaskCount()
|
| D | dot_op_emitter.cc | 65 CHECK_EQ(instr.opcode(), HloOpcode::kDot); in DotInfo()
707 if (hlo.opcode() == HloOpcode::kDot && hlo.shape().dimensions_size() == 2 && in ProfitableToMakeDotOperandColumnMajor()
724 if (fusion_root_op->opcode() != HloOpcode::kDot) { in ProfitableToMakeDotOperandColumnMajor()
|
| D | cpu_instruction_fusion_test.cc | 669 {HloOpcode::kDot, HloOpcode::kAdd, HloOpcode::kParameter, in TEST_F()
682 {HloOpcode::kDot, HloOpcode::kAdd, HloOpcode::kParameter, in TEST_F()
|