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.h79 return compute_function_; in compute_function()
145 ComputeFunctionType compute_function_; variable
Dcpu_executable.cc81 compute_function_ = reinterpret_cast<ComputeFunctionType>(sym->getAddress()); in CpuExecutable()
83 << reinterpret_cast<void*>(compute_function_); in CpuExecutable()
205 compute_function_(result_buffer, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
Dir_emitter.cc119 llvm::Argument* out_parameter = compute_function_->result_arg(); in EmitThreadLocalFunctionEpilogue()
191 llvm::Function* ir_function = compute_function_->function(); in EmitComputation()
205 compute_function_.reset(); in EmitComputation()
219 compute_function_.reset(new IrFunction(function_name, linkage, in InitializeIrFunction()
1481 *compute_function_->function()) / in CreateShardedVectorType()
1650 *compute_function_->function()) / in EmitVectorizedReduce()
2410 compute_function_->function()); in HandleWhile()
2424 compute_function_->function()); in HandleWhile()
2439 compute_function_->function()->getBasicBlockList().push_back(exit_bb); in HandleWhile()
3014 return compute_function_->profile_counters_arg(); in GetProfileCountersArgument()
[all …]
Dir_emitter.h454 std::unique_ptr<IrFunction> compute_function_; variable