Searched refs:HloRunner (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_runner.h40 class HloRunner { class in namespace:xla
42 HloRunner();
44 HloRunner(::perftools::gputools::Platform* platform);
46 ~HloRunner();
82 // HloRunner in a program without any backends linked in.
99 StatusOr<std::unique_ptr<Literal>> HloRunner::Execute(
H A Dhlo_runner.cc43 HloRunner::CreateModuleFromString(const tensorflow::StringPiece hlo_string,
67 HloRunner::ReadModuleFromBinaryProtoFile(const std::string& filename,
76 HloRunner::ReadModuleFromTextProtoFile(const std::string& filename,
85 HloRunner::ReadModuleFromHloTextFile(const std::string& filename,
97 struct HloRunner::EigenThreadPoolWrapper {
102 HloRunner::HloRunner() {} function in class:xla::HloRunner
104 HloRunner::HloRunner(se::Platform* platform) { function in class:xla::HloRunner
108 VLOG(1) << "Created HloRunner fo
[all...]
H A Dhlo_element_type_converter_test.cc29 return HloRunner::CreateModuleFromString(hlo_string,
H A Dcopy_insertion_test.cc1858 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest());
1959 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest());
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dhlo_test_base.h208 HloRunner test_runner_;
209 HloRunner reference_runner_;
H A Dhlo_test_base.cc219 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest());
233 HloRunner::ReadModuleFromHloTextFile(filename, GetDebugOptionsForTest());
247 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest());
261 HloRunner::ReadModuleFromHloTextFile(filename, GetDebugOptionsForTest());
H A Dmultioutput_fusion_test.cc200 HloRunner::CreateModuleFromString(testcase, GetDebugOptionsForTest())

Completed in 159 milliseconds