Searched refs:XlaCompiledCpuFunction (Results 1 – 8 of 8) sorted by relevance
126 XlaCompiledCpuFunction::RawFunction raw_function = in Compile()147 XlaCompiledCpuFunction::set_static_data_raw_function(&jit->static_data_, in Compile()149 XlaCompiledCpuFunction::set_static_data_buffer_infos( in Compile()151 XlaCompiledCpuFunction::set_static_data_num_buffers( in Compile()153 XlaCompiledCpuFunction::set_static_data_arg_index_table( in Compile()155 XlaCompiledCpuFunction::set_static_data_num_args( in Compile()157 XlaCompiledCpuFunction::set_static_data_num_variables(&jit->static_data_, in Compile()159 XlaCompiledCpuFunction::set_static_data_result_index(&jit->static_data_, in Compile()175 XlaCompiledCpuFunction::set_static_data_arg_names(&jit->static_data_, in Compile()177 XlaCompiledCpuFunction::set_static_data_variable_names( in Compile()[all …]
24 XlaCompiledCpuFunction::XlaCompiledCpuFunction(const StaticData& static_data, in XlaCompiledCpuFunction() function in tensorflow::XlaCompiledCpuFunction55 bool XlaCompiledCpuFunction::Run() { in Run()62 XlaCompiledCpuFunction::~XlaCompiledCpuFunction() { in ~XlaCompiledCpuFunction()94 int XlaCompiledCpuFunction::LookupArgIndex(const string& name) const { in LookupArgIndex()98 int XlaCompiledCpuFunction::LookupVariableIndex(const string& name) const { in LookupVariableIndex()106 int XlaCompiledCpuFunction::LookupResultIndex(const string& name) const { in LookupResultIndex()
50 class XlaCompiledCpuFunction {107 friend class XlaCompiledCpuFunction; variable120 explicit XlaCompiledCpuFunction(124 virtual ~XlaCompiledCpuFunction();126 XlaCompiledCpuFunction(const XlaCompiledCpuFunction&) = delete;127 XlaCompiledCpuFunction& operator=(const XlaCompiledCpuFunction&) = delete;
57 const XlaCompiledCpuFunction::StaticData& StaticData() const { in StaticData()68 XlaCompiledCpuFunction::StaticData static_data_;
178 XlaCompiledCpuFunction function(jit->StaticData()); in TEST()240 XlaCompiledCpuFunction function(jit->StaticData()); in TEST()
34 // XlaCompiledCpuFunction class with statically type-safe arg and result63 class MyClass final : public tensorflow::XlaCompiledCpuFunction {76 // Returns static data used to create an XlaCompiledCpuFunction.77 static const tensorflow::XlaCompiledCpuFunction::StaticData& StaticData() {78 static XlaCompiledCpuFunction::StaticData* kStaticData = [](){79 XlaCompiledCpuFunction::StaticData* data =80 new XlaCompiledCpuFunction::StaticData;102 : XlaCompiledCpuFunction(StaticData(), alloc_mode) {}
54 void zero_buffers(XlaCompiledCpuFunction* computation) { in zero_buffers()
125 XlaCompiledCpuFunction::AllocMode::RESULTS_PROFILES_AND_TEMPS_ONLY); in TEST()338 XlaCompiledCpuFunction::AllocMode::RESULTS_PROFILES_AND_TEMPS_ONLY); in TEST()588 XlaCompiledCpuFunction::AllocMode::RESULTS_PROFILES_AND_TEMPS_ONLY); in TEST()