Home
last modified time | relevance | path

Searched refs:compute_function_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.h83 return compute_function_; in compute_function()
147 ComputeFunctionType compute_function_; variable
Dcpu_executable.cc69 compute_function_ = in CpuExecutable()
72 << reinterpret_cast<void*>(compute_function_); in CpuExecutable()
188 compute_function_(result_buffer, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
Dir_emitter.cc142 llvm::Function* ir_function = compute_function_->function(); in EmitComputation()
146 compute_function_.reset(); in EmitComputation()
160 compute_function_.reset(new IrFunction(function_name, linkage, in InitializeIrFunction()
1516 *compute_function_->function()) / in CreateShardedVectorType()
2349 compute_function_->function()); in HandleWhile()
2363 compute_function_->function()); in HandleWhile()
2378 compute_function_->function()->getBasicBlockList().push_back(exit_bb); in HandleWhile()
2824 return compute_function_->profile_counters_arg(); in GetProfileCountersArgument()
2828 return compute_function_->buffer_table_arg(); in GetBufferTableArgument()
2832 return compute_function_->exec_run_options_arg(); in GetExecutableRunOptionsArgument()
[all …]
Dir_emitter.h441 std::unique_ptr<IrFunction> compute_function_; variable