| /external/tensorflow/tensorflow/lite/toco/ |
| tooling_util_test.cc | 27 enum class Agreement { kBroadcast, kExtend, kBroadcastNotExtend, kNeither };
40 {Shape({3}), Shape({3}), Agreement::kBroadcast},
41 {Shape({256, 256, 3}), Shape({256, 256, 3}), Agreement::kBroadcast},
42 {Shape({256, 256, 3}), Shape({3}), Agreement::kBroadcast},
43 {Shape({8, 1, 6, 1}), Shape({7, 1, 5}), Agreement::kBroadcast},
44 {Shape({}), Shape({3}), Agreement::kBroadcast},
45 {Shape({}), Shape({3, 1}), Agreement::kBroadcast},
75 case Agreement::kBroadcast: {
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_constant_folding.cc | 104 if (instruction->opcode() == HloOpcode::kBroadcast ||
|
| bfloat16_support.cc | 83 case HloOpcode::kBroadcast:
|
| dynamic_padder.cc | 79 case HloOpcode::kBroadcast:
|
| hlo_opcode.h | 63 V(kBroadcast, "broadcast", 1) \
|
| reshape_mover.cc | 84 if (instruction->opcode() == HloOpcode::kBroadcast &&
171 case HloOpcode::kBroadcast: {
|
| instruction_fusion.cc | 64 case HloOpcode::kBroadcast:
183 if (operand->opcode() == HloOpcode::kBroadcast ||
|
| while_loop_invariant_code_motion.cc | 114 case HloOpcode::kBroadcast:
217 // kIota, kBroadcast or kConstant.
|
| algebraic_simplifier.cc | 70 case HloOpcode::kBroadcast:
[all...] |
| hlo_rematerialization_test.cc | 464 if (instruction->opcode() == HloOpcode::kBroadcast) {
|
| hlo_instruction.cc | 276 case HloOpcode::kBroadcast:
[all...] |
| indexed_array_analysis.cc | [all...] |
| hlo_graph_dumper.cc | [all...] |
| algebraic_simplifier_test.cc | [all...] |
| buffer_assignment_test.cc | [all...] |
| copy_insertion_test.cc | [all...] |
| layout_assignment.cc | [all...] |
| elemental_ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion.cc | 144 op2->opcode() != HloOpcode::kBroadcast) {
154 if (alpha->opcode() == HloOpcode::kBroadcast &&
183 if (producer->opcode() == HloOpcode::kBroadcast) {
|
| gpu_fusible.cc | 165 instr.opcode() == HloOpcode::kBroadcast ||
|
| elemental_ir_emitter.cc | 65 return operand->opcode() == HloOpcode::kBroadcast &&
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion.cc | 34 hlo.opcode() == HloOpcode::kBroadcast ||
|
| cpu_instruction_fusion_test.cc | 361 HloOpcode::kBroadcast, HloOpcode::kParameter, HloOpcode::kParameter,
381 {HloOpcode::kNegate, HloOpcode::kBroadcast, HloOpcode::kParameter});
758 {HloOpcode::kGather, HloOpcode::kAdd, HloOpcode::kBroadcast,
[all...] |
| /external/tensorflow/tensorflow/core/nccl/ |
| nccl_manager.cc | 358 AddParticipant(std::move(participant), context, kBroadcast,
364 AddParticipant(std::move(participant), context, kBroadcast,
521 if (collective->type == kBroadcast) {
584 case kBroadcast: {
|
| nccl_manager.h | 175 kBroadcast = 2,
|
