Searched refs:CreateRng (Results 1 – 15 of 15) sorted by relevance
579 auto rng1 = builder.AddInstruction(HloInstruction::CreateRng( in TEST_F()582 auto rng2 = builder.AddInstruction(HloInstruction::CreateRng( in TEST_F()621 auto rng = builder.AddInstruction(HloInstruction::CreateRng( in TEST_F()
81 auto rng0 = builder.AddInstruction(HloInstruction::CreateRng(
276 auto rng = builder.AddInstruction(HloInstruction::CreateRng( in TEST_F()
533 static std::unique_ptr<HloInstruction> CreateRng(
370 instruction = CreateRng(shape, proto.distribution(), all_operands()); in CreateFromProto()851 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateRng( in CreateRng() function in xla::HloInstruction
2131 HloInstruction::CreateRng(shape, *distribution, operands)); in ParseInstructionRhs()
2516 HloInstruction::CreateRng(ShapeUtil::MakeShape(F32, {2, 2}), in TEST_F()
161 rng::RngSupport *CreateRng() override;
345 rng::RngSupport *HostExecutor::CreateRng() { in CreateRng() function in stream_executor::host::HostExecutor
350 virtual rng::RngSupport *CreateRng() { return nullptr; } in CreateRng() function
460 rng_.reset(implementation_->CreateRng()); in AsRng()
216 rng::RngSupport* CreateRng() override;
696 rng::RngSupport* GpuExecutor::CreateRng() { in CreateRng() function in stream_executor::gpu::GpuExecutor
788 rng::RngSupport* GpuExecutor::CreateRng() { in CreateRng() function in stream_executor::gpu::GpuExecutor
3078 return b_.AddInstruction(HloInstruction::CreateRng( in HandleRng()3089 auto rng = b_.AddInstruction(HloInstruction::CreateRng( in HandleRng()