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()
|