Searched refs:HloRunner (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 37 HloRunner::CreateModuleFromString(const absl::string_view hlo_string, in CreateModuleFromString()
60 HloRunner::ReadModuleFromBinaryProtoFile(const std::string& filename, in ReadModuleFromBinaryProtoFile()
69 HloRunner::ReadModuleFromTextProtoFile(const std::string& filename, in ReadModuleFromTextProtoFile()
78 HloRunner::ReadModuleFromHloTextFile(const std::string& filename, in ReadModuleFromHloTextFile()
88 HloRunner::HloRunner(se::Platform* platform) { in HloRunner() function in xla::HloRunner
95 HloRunner::~HloRunner() {} in ~HloRunner()
97 StatusOr<ScopedShapedBuffer> HloRunner::TransferLiteralToDevice( in TransferLiteralToDevice()
110 StatusOr<std::vector<ScopedShapedBuffer>> HloRunner::TransferLiteralsToDevice( in TransferLiteralsToDevice()
122 StatusOr<std::vector<ScopedShapedBuffer>> HloRunner::TransferLiteralsToDevice( in TransferLiteralsToDevice()
132 StatusOr<Literal> HloRunner::TransferLiteralFromDevice( in TransferLiteralFromDevice()
[all …]
|
| D | hlo_runner.h | 45 class HloRunner {
83 explicit HloRunner(se::Platform* platform);
85 ~HloRunner();
|
| D | buffer_assignment_test.cc | 2550 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 211 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()
224 HloRunner::ReplicatedExecuteOptions options; in ExecuteReplicated()
332 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in RunAndCompare()
347 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in Run()
402 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in RunMultipleTimes()
461 HloRunner::ReadModuleFromHloTextFile(filename, GetDebugOptionsForTest()); in RunAndCompareFromFile()
474 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in RunAndCompareNoHloPasses()
488 HloRunner::ReadModuleFromHloTextFile(filename, GetDebugOptionsForTest()); in RunAndCompareNoHloPassesFromFile()
|
| D | token_hlo_test.cc | 149 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
187 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
197 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
|
| D | hlo_test_base.h | 311 HloRunner test_runner_;
312 HloRunner reference_runner_;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | interactive_graphviz.cc | 678 module = HloRunner::ReadModuleFromBinaryProtoFile( in RealMain()
682 module = HloRunner::ReadModuleFromHloTextFile( in RealMain()
|