| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_assignment.cc | 143 int stream_num = (hlo->opcode() == HloOpcode::kRng && in AssignStreams()
150 if (hlo->opcode() == HloOpcode::kRng && in AssignStreams()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 38 case PluginKind::kRng: in PluginKindString()
136 case PluginKind::kRng: in SetDefaultFactory()
158 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 | rng_expander.cc | 43 CHECK_EQ(rng->opcode(), HloOpcode::kRng); in ConvertSmallFpRngToF32Rng()
127 return instruction->opcode() == HloOpcode::kRng; in InstructionMatchesPattern()
|
| D | hlo_opcode_test.cc | 65 case HloOpcode::kRng: in TEST()
|
| D | hlo_constant_folding.cc | 55 instr->opcode() == HloOpcode::kRng) { in IsOrContainsIllegalInstr()
|
| D | reshape_mover.cc | 79 if (instruction->opcode() == HloOpcode::kRng && in CanTriviallyChangeShape()
161 case HloOpcode::kRng: { in UpdateOperand()
|
| D | hlo_opcode.h | 128 V(kRng, "rng", kHloOpcodeIsVariadic) \
|
| D | instruction_fusion.cc | 174 case HloOpcode::kRng: in IsExpensive()
|
| D | hlo_instruction.cc | 369 case HloOpcode::kRng: in CreateFromProto()
1528 case HloOpcode::kRng: in HasSideEffectNoRecurse()
1682 case HloOpcode::kRng: in CloneWithNewOperands()
2220 case HloOpcode::kRng: in IdenticalSlowPath()
3038 case HloOpcode::kRng: in IsFusible()
3224 case HloOpcode::kRng: in Visit()
|
| D | hlo_rematerialization_test.cc | 298 if (instruction->opcode() == HloOpcode::kRng) { in TEST_F()
|
| D | sharding_propagation.cc | 267 case HloOpcode::kRng: in PickRepresentativeOperand()
295 (instruction->opcode() != HloOpcode::kRng || is_spmd)) { in SupportSpatialPartitioning()
|
| D | hlo.proto | 118 // Only present for kRng.
|
| D | hlo_graph_dumper.cc | 971 case HloOpcode::kRng: in GetInstructionColor()
|
| D | hlo_parser.cc | 192 case HloOpcode::kRng: in CanInferShape()
2123 case HloOpcode::kRng: { in ParseInstructionRhs()
|
| D | layout_assignment.cc | 2308 case HloOpcode::kRng: in InstructionCanChangeLayout()
|
| D | space_to_batch_converter.cc | 672 if (hlo->opcode() == HloOpcode::kFusion || hlo->opcode() == HloOpcode::kRng || in IsTrivialElementwise()
|
| D | algebraic_simplifier.cc | 3903 if (operand->opcode() == HloOpcode::kRng && operand->user_count() == 1) { in HandleReshape()
5127 if (operand->opcode() == HloOpcode::kRng && operand->user_count() == 1) { in HandleTranspose()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_rng.cc | 286 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
317 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 | 146 instruction->shape().IsTuple() || opcode == HloOpcode::kRng || in GetTargetParallelTaskCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 584 case HloOpcode::kRng: { in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3736 if (hlo->HasSideEffectNoRecurse() && hlo->opcode() != HloOpcode::kRng) { in PreprocessSharding()
3756 if (hlo->opcode() == HloOpcode::kRng) { in PreprocessSharding()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 342 case HloOpcode::kRng: in IsConstantVisitor()
2195 return AddInstruction(std::move(instr), HloOpcode::kRng, parameters); in RngOpInternal()
|