Searched refs:kRng (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_assignment.cc | 138 int stream_num = (hlo->opcode() == HloOpcode::kRng && in AssignStreams()
145 if (hlo->opcode() == HloOpcode::kRng && in AssignStreams()
|
| D | instruction_fusion.cc | 199 if (producer->opcode() == HloOpcode::kRng) { in ShouldFuseInexpensiveChecks()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 35 case PluginKind::kRng: in PluginKindString()
133 case PluginKind::kRng: in SetDefaultFactory()
155 case PluginKind::kRng: in HasFactory()
|
| D | plugin.h | 42 kRng, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_element_type_converter_test.cc | 137 inst->opcode() == HloOpcode::kRng; in TEST_F()
163 if (inst->opcode() == HloOpcode::kRng) { in TEST_F()
|
| D | hlo_opcode_test.cc | 64 case HloOpcode::kRng: in TEST()
|
| D | reshape_mover.cc | 78 if (instruction->opcode() == HloOpcode::kRng && in CanTriviallyChangeShape()
160 case HloOpcode::kRng: { in UpdateOperand()
|
| D | hlo_constant_folding.cc | 55 instr->opcode() == HloOpcode::kRng) { in IsOrContainsIllegalInstr()
|
| D | hlo_opcode.h | 122 V(kRng, "rng", kHloOpcodeIsVariadic) \
|
| D | instruction_fusion.cc | 149 case HloOpcode::kRng: in IsExpensive()
|
| D | hlo_rematerialization_test.cc | 376 if (instruction->opcode() == HloOpcode::kRng) { in TEST_F()
|
| D | hlo_instruction.cc | 339 case HloOpcode::kRng: in CreateFromProto()
1243 case HloOpcode::kRng: in HasSideEffectNoRecurse()
1368 case HloOpcode::kRng: in CloneWithNewOperands()
1820 case HloOpcode::kRng: in IdenticalSlowPath()
2629 case HloOpcode::kRng: in Visit()
|
| D | hlo.proto | 112 // Only present for kRng.
|
| D | hlo_graph_dumper.cc | 955 case HloOpcode::kRng: in GetInstructionColor()
|
| D | algebraic_simplifier.cc | 2619 if (operand->opcode() == HloOpcode::kRng && operand->user_count() == 1) { in HandleReshape()
3380 if (operand->opcode() == HloOpcode::kRng && operand->user_count() == 1) { in HandleTranspose()
|
| D | layout_assignment.cc | 2069 case HloOpcode::kRng: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 1602 : HloInstruction(HloOpcode::kRng, shape), distribution_(distribution) { in HloRngInstruction()
|
| D | elemental_ir_emitter.cc | 2425 case HloOpcode::kRng: in MakeElementGenerator()
|
| D | hlo_parser.cc | 1420 case HloOpcode::kRng: { in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_rng.cc | 287 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
318 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_rng.cc | 260 cuda::kCudaPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_curand()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 144 opcode == HloOpcode::kOutfeed || opcode == HloOpcode::kRng || in GetTargetParallelTaskCount()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 226 case HloOpcode::kRng: in IsConstantVisitor()
1805 return AddInstruction(std::move(instr), HloOpcode::kRng, parameters); in RngOp()
|