Searched refs:HloRunner (Results 1 – 7 of 7) sorted by relevance
37 HloRunner::HloRunner(se::Platform* platform, int intra_op_parallelism_threads) { in HloRunner() function in xla::HloRunner46 HloRunner::~HloRunner() {} in ~HloRunner()48 StatusOr<ScopedShapedBuffer> HloRunner::TransferLiteralToDevice( in TransferLiteralToDevice()61 StatusOr<std::vector<ScopedShapedBuffer>> HloRunner::TransferLiteralsToDevice( in TransferLiteralsToDevice()73 StatusOr<std::vector<ScopedShapedBuffer>> HloRunner::TransferLiteralsToDevice( in TransferLiteralsToDevice()83 StatusOr<Literal> HloRunner::TransferLiteralFromDevice( in TransferLiteralFromDevice()91 StatusOr<Literal> HloRunner::Execute(std::unique_ptr<HloModule> module, in Execute()106 StatusOr<Literal> HloRunner::ExecuteWithExecutable( in ExecuteWithExecutable()150 StatusOr<ExecutionOutput> HloRunner::ExecuteWithDeviceBuffers( in ExecuteWithDeviceBuffers()159 StatusOr<ExecutionOutput> HloRunner::ExecuteWithDeviceBuffers( in ExecuteWithDeviceBuffers()[all …]
46 class HloRunner : public HloRunnerInterface {53 explicit HloRunner(se::Platform* platform,56 ~HloRunner() override;
148 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()186 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()196 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
208 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()222 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()238 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()444 HloRunner::ReplicatedExecuteOptions options; in RunReplicated()537 HloRunner::ReadModuleFromHloTextFile(filename, GetDebugOptionsForTest()); in RunAndCompareFromFile()563 HloRunner::ReadModuleFromHloTextFile(filename, GetDebugOptionsForTest()); in RunAndCompareNoHloPassesFromFile()
314 HloRunner test_runner_;315 HloRunner reference_runner_;
346 HloRunner::ReplicatedExecuteOptions opts; in XLA_TEST_F()426 HloRunner::ReplicatedExecuteOptions opts; in XLA_TEST_F()
93 HloRunner runner(platform); in ExecuteOnPlatform()