Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.h84 return compute_function_; in compute_function()
143 ComputeFunctionType compute_function_; variable
H A Dcpu_executable.cc82 compute_function_ = reinterpret_cast<ComputeFunctionType>(sym->getAddress()); in CpuExecutable()
84 << reinterpret_cast<void*>(compute_function_); in CpuExecutable()
192 compute_function_(nullptr, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
H A Dir_emitter.cc130 llvm::Argument* out_parameter = compute_function_->result_arg(); in EmitThreadLocalFunctionEpilogue()
212 llvm::Function* ir_function = compute_function_->function(); in EmitComputation()
232 compute_function_.reset(); in EmitComputation()
246 compute_function_.reset(new IrFunction(function_name, linkage, in InitializeIrFunction()
1577 *compute_function_->function()) / in CreateShardedVectorType()
1752 *compute_function_->function()) / in EmitVectorizedReduce()
2568 compute_function_->function()); in HandleWhile()
2584 compute_function_->function()); in HandleWhile()
2599 compute_function_->function()->getBasicBlockList().push_back(exit_bb); in HandleWhile()
3198 return compute_function_->profile_counters_arg(); in GetProfileCountersArgument()
[all …]
H A Dir_emitter.h495 std::unique_ptr<IrFunction> compute_function_; variable