| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util_test.cc | 27 enum class Agreement { kBroadcast, kExtend, kBroadcastNotExtend, kNeither }; enumerator
40 {Shape({3}), Shape({3}), Agreement::kBroadcast}, in CreateShapePairs()
41 {Shape({256, 256, 3}), Shape({256, 256, 3}), Agreement::kBroadcast}, in CreateShapePairs()
42 {Shape({256, 256, 3}), Shape({3}), Agreement::kBroadcast}, in CreateShapePairs()
43 {Shape({8, 1, 6, 1}), Shape({7, 1, 5}), Agreement::kBroadcast}, in CreateShapePairs()
44 {Shape({}), Shape({3}), Agreement::kBroadcast}, in CreateShapePairs()
45 {Shape({}), Shape({3, 1}), Agreement::kBroadcast}, in CreateShapePairs()
75 case Agreement::kBroadcast: { in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion.cc | 144 op2->opcode() != HloOpcode::kBroadcast) { in ShouldFuseInexpensiveChecks()
154 if (alpha->opcode() == HloOpcode::kBroadcast && in ShouldFuseInexpensiveChecks()
183 if (producer->opcode() == HloOpcode::kBroadcast) { in ShouldFuseInexpensiveChecks()
|
| D | gpu_fusible.cc | 165 instr.opcode() == HloOpcode::kBroadcast || in IsLoopFusible()
|
| D | elemental_ir_emitter.cc | 65 return operand->opcode() == HloOpcode::kBroadcast && in IsFPLiteralWithValue()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 358 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastSend()
364 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastRecv()
521 if (collective->type == kBroadcast) { in RunCollective()
584 case kBroadcast: { in LoopKernelLaunches()
|
| D | nccl_manager.h | 175 kBroadcast = 2, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover.cc | 84 if (instruction->opcode() == HloOpcode::kBroadcast && in CanTriviallyChangeShape()
171 case HloOpcode::kBroadcast: { in UpdateOperand()
|
| D | hlo_constant_folding.cc | 104 if (instruction->opcode() == HloOpcode::kBroadcast || in Run()
|
| D | bfloat16_support.cc | 83 case HloOpcode::kBroadcast: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | dynamic_padder.cc | 79 case HloOpcode::kBroadcast: in ChooseIdentityValue()
|
| D | hlo_opcode.h | 63 V(kBroadcast, "broadcast", 1) \
|
| D | instruction_fusion.cc | 64 case HloOpcode::kBroadcast: in IsExpensive()
183 if (operand->opcode() == HloOpcode::kBroadcast || in EffectivelyAtMostUnary()
|
| D | while_loop_invariant_code_motion.cc | 114 case HloOpcode::kBroadcast: in NotWorthHoistingIndividually()
|
| D | algebraic_simplifier.cc | 70 case HloOpcode::kBroadcast: in IsAll()
2128 if (operand->opcode() == HloOpcode::kBroadcast) { in HandleBroadcast()
2384 TF_RET_CHECK(broadcast->opcode() == HloOpcode::kBroadcast); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2412 if (user_operand->opcode() == HloOpcode::kBroadcast && in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2426 if (user_operand->opcode() == HloOpcode::kBroadcast && in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
2624 if (HloOpcode::kBroadcast == reshape->operand(0)->opcode()) { in HandleReshape()
2842 if (slice->operand(0)->opcode() != HloOpcode::kBroadcast) { in HandleSlice()
|
| D | hlo_rematerialization_test.cc | 464 if (instruction->opcode() == HloOpcode::kBroadcast) { in TEST_F()
|
| D | hlo_instruction.cc | 276 case HloOpcode::kBroadcast: in CreateFromProto()
1362 case HloOpcode::kBroadcast: in CloneWithNewOperands()
1813 case HloOpcode::kBroadcast: in IdenticalSlowPath()
2605 case HloOpcode::kBroadcast: in Visit()
|
| D | indexed_array_analysis.cc | 865 HloOpcode::kBroadcast) { in ComputeArrayForElementwiseBinaryOp()
|
| D | algebraic_simplifier_test.cc | 1335 EXPECT_EQ(root->operand(0)->opcode(), HloOpcode::kBroadcast); in TEST_F()
3889 EXPECT_EQ(root->opcode(), HloOpcode::kBroadcast); in TEST_F()
3915 EXPECT_EQ(root->opcode(), HloOpcode::kBroadcast); in TEST_F()
3935 EXPECT_EQ(root->opcode(), HloOpcode::kBroadcast); in TEST_F()
3956 EXPECT_EQ(root->opcode(), HloOpcode::kBroadcast); in TEST_F()
|
| D | hlo_graph_dumper.cc | 982 case HloOpcode::kBroadcast: in GetInstructionColor()
|
| D | buffer_assignment_test.cc | 2025 ASSERT_EQ(bcast->opcode(), HloOpcode::kBroadcast); in TEST_F()
2092 ASSERT_EQ(bcast->opcode(), HloOpcode::kBroadcast); in TEST_F()
|
| D | layout_assignment.cc | 2049 case HloOpcode::kBroadcast: in InstructionCanChangeLayout()
|
| D | copy_insertion_test.cc | 840 ASSERT_EQ(bcast->opcode(), HloOpcode::kBroadcast); in TEST_F()
|
| D | hlo_instructions.cc | 872 : HloInstruction(HloOpcode::kBroadcast, shape), in HloBroadcastInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 34 hlo.opcode() == HloOpcode::kBroadcast || in CanBeLoopFused()
|
| D | cpu_instruction_fusion_test.cc | 361 HloOpcode::kBroadcast, HloOpcode::kParameter, HloOpcode::kParameter, in TEST_F()
381 {HloOpcode::kNegate, HloOpcode::kBroadcast, HloOpcode::kParameter}); in TEST_F()
758 {HloOpcode::kGather, HloOpcode::kAdd, HloOpcode::kBroadcast, in TEST_P()
|