Searched refs:CreateRng (Results 1 – 14 of 14) 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()
354 auto rng = builder.AddInstruction(HloInstruction::CreateRng( in TEST_F()
81 auto rng0 = builder.AddInstruction(HloInstruction::CreateRng(
397 static std::unique_ptr<HloInstruction> CreateRng(
340 instruction = CreateRng(shape, proto.distribution(), all_operands()); in CreateFromProto()665 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateRng( in CreateRng() function in xla::HloInstruction
1428 HloInstruction::CreateRng(shape, *distribution, operands)); in ParseInstructionRhs()
1978 HloInstruction::CreateRng(ShapeUtil::MakeShape(F32, {2, 2}), in TEST_F()
184 rng::RngSupport *CreateRng() override;
253 rng::RngSupport *HostExecutor::CreateRng() { in CreateRng() function in stream_executor::host::HostExecutor
347 virtual rng::RngSupport *CreateRng() { return nullptr; } in CreateRng() function
475 rng_.reset(implementation_->CreateRng()); in AsRng()
218 rng::RngSupport* CreateRng() override;
697 rng::RngSupport* GpuExecutor::CreateRng() { in CreateRng() function in stream_executor::gpu::GpuExecutor
816 rng::RngSupport* GpuExecutor::CreateRng() { in CreateRng() function in stream_executor::gpu::GpuExecutor