| /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/openscreen/cast/common/channel/ |
| D | message_util.h | 121 kBroadcast, enumerator
180 case CastMessageType::kBroadcast: in CastMessageTypeToString()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover.cc | 85 if (instruction->opcode() == HloOpcode::kBroadcast && in CanTriviallyChangeShape()
172 case HloOpcode::kBroadcast: { in UpdateOperand()
|
| D | hlo_constant_folding.cc | 104 if (instruction->opcode() == HloOpcode::kBroadcast || in Run()
|
| D | bfloat16_support.cc | 84 case HloOpcode::kBroadcast: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | hlo_opcode.h | 60 V(kBroadcast, "broadcast", 1) \
|
| D | hlo_cse.cc | 140 case HloOpcode::kBroadcast: in CseHash()
|
| D | space_to_batch_converter.cc | 693 (old_producer->opcode() == HloOpcode::kBroadcast && in CanPropagate()
696 old_producer->opcode() == HloOpcode::kBroadcast && in CanPropagate()
1061 if (instr->opcode() == HloOpcode::kBroadcast) { in RewriteBroadcastTree()
1074 if (op->opcode() == HloOpcode::kBroadcast) { in IsBroadcastTree()
1102 CHECK_EQ(broadcast->opcode(), HloOpcode::kBroadcast); in IsBroadcastPropagatable()
1115 if (consumer->operand(i)->opcode() == HloOpcode::kBroadcast) { in SupportedOpForPropagation()
1217 if (consumer->operand(i)->opcode() == HloOpcode::kBroadcast) { in Propagate()
1239 if (consumer->operand(i)->opcode() == HloOpcode::kBroadcast) { in Propagate()
1317 HloOpcode::kBroadcast && in Propagate()
|
| D | sharding_propagation.cc | 236 case HloOpcode::kBroadcast: in PickRepresentativeOperand()
299 case HloOpcode::kBroadcast: in SupportSpatialPartitioning()
568 if (aggressiveness < 2 && inst.opcode() == HloOpcode::kBroadcast) { in CanPropagateThroughAtAgressiveLevel()
736 case HloOpcode::kBroadcast: { in InferShardingFromOperands()
1135 case HloOpcode::kBroadcast: { in GetShardingFromUser()
1410 if (aggressiveness < 2 && instruction->opcode() == HloOpcode::kBroadcast) { in InferShardingFromUsers()
|
| D | while_loop_invariant_code_motion.cc | 114 case HloOpcode::kBroadcast: in NotWorthHoistingIndividually()
|
| D | instruction_fusion.cc | 71 case HloOpcode::kBroadcast: in IsExpensive()
212 if (operand->opcode() == HloOpcode::kBroadcast || in EffectivelyAtMostUnary()
|
| D | hlo_replication_analysis.cc | 144 hlo->opcode() == HloOpcode::kBroadcast || // in DetermineHloInstructionIsReplicated()
|
| D | conditional_simplifier.cc | 118 case HloOpcode::kBroadcast: in TryRemoveConditional()
|
| D | algebraic_simplifier.cc | 76 case HloOpcode::kBroadcast: in IsAll()
3139 if (operand->opcode() == HloOpcode::kBroadcast) { in HandleBroadcast()
3396 pad->operand(0)->opcode() == HloOpcode::kBroadcast && in HandlePad()
3580 TF_RET_CHECK(broadcast->opcode() == HloOpcode::kBroadcast); in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
3587 return instruction->opcode() == HloOpcode::kBroadcast && in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
3592 return instruction->opcode() == HloOpcode::kBroadcast && in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
3908 if (HloOpcode::kBroadcast == reshape->operand(0)->opcode()) { in HandleReshape()
4216 if (slice->operand(0)->opcode() != HloOpcode::kBroadcast) { in HandleSlice()
|
| D | hlo_rematerialization_test.cc | 386 if (instruction->opcode() == HloOpcode::kBroadcast) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 447 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastSend()
453 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastRecv()
650 if (status.ok() && collective->type == kBroadcast && in RunCollective()
757 case kBroadcast: { in LoopKernelLaunches()
|
| D | nccl_manager.h | 205 kBroadcast = 2, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | schedule_aware_all_gather_cse.cc | 47 operand->operand(0)->opcode() == HloOpcode::kBroadcast in MayConsiderAsAllGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 32 hlo.opcode() == HloOpcode::kBroadcast || in CanBeLoopFused()
|
| D | parallel_task_assignment.cc | 153 opcode == HloOpcode::kBroadcast || opcode == HloOpcode::kConcatenate || in GetTargetParallelTaskCount()
|
| D | cpu_instruction_fusion_test.cc | 363 HloOpcode::kBroadcast, HloOpcode::kParameter, HloOpcode::kParameter, in TEST_F()
383 {HloOpcode::kNegate, HloOpcode::kBroadcast, HloOpcode::kParameter}); in TEST_F()
767 {HloOpcode::kGather, HloOpcode::kAdd, HloOpcode::kBroadcast, in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 352 return xla::HloOpcode::kBroadcast; in MhloToHloOpcode()
487 return xla::HloOpcode::kBroadcast; in MhloToHloOpcode()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 67 return operand->opcode() == HloOpcode::kBroadcast && in IsFPLiteralWithValue()
|
| D | gpu_fusible.cc | 220 instr.opcode() == HloOpcode::kBroadcast || in IsLoopFusible()
|
| /external/libwebm/webm_parser/include/webm/ |
| D | dom_types.h | 627 kBroadcast = 1, enumerator
|