Home
last modified time | relevance | path

Searched refs:cl_args_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc141 cl_args_(std::move(operation.cl_args_)) {} in ClOperation()
147 cl_args_ = std::move(operation.cl_args_); in operator =()
163 RETURN_IF_ERROR(cl_args_.SetObjectRef(operation_->src_tensors_names_[i], in UpdateParams()
172 RETURN_IF_ERROR(cl_args_.SetObjectRef(operation_->dst_tensors_names_[i], in UpdateParams()
175 RETURN_IF_ERROR(operation_->BindArguments(&cl_args_)); in UpdateParams()
188 RETURN_IF_ERROR(cl_args_.Init( in Compile()
201 RETURN_IF_ERROR(cl_args_.Init(creation_context.GetGpuInfo(), in CompileDeserialized()
230 RETURN_IF_ERROR(cl_args_.Bind(kernel_.kernel())); in Tune()
Dcl_operation.h69 RETURN_IF_ERROR(cl_args_.Bind(kernel_.kernel())); in AddToQueue()
82 cl_args_.MoveObjectRefsOut(&operation_->args_); in MoveObjectRefsFromCLToGeneric()
85 cl_args_.MoveObjectRefsIn(&operation_->args_); in MoveObjectRefsFromGenericToCL()
87 void SyncScalarValues() { cl_args_.CopyScalarValues(&operation_->args_); } in SyncScalarValues()
92 CLArguments cl_args_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc51 RETURN_IF_ERROR(cl_args_.SetObjectRef("tensor", tensor)); in DispatchKernel()
53 cl_args_.Bind(kernel_.kernel(), kernel_.GetBindingCounter())); in DispatchKernel()
64 CLArguments cl_args_; member in tflite::gpu::cl::__anondaf6909e0111::OpenClConverterImpl
165 RETURN_IF_ERROR(cl_args_.Init(environment->device().GetInfo(), {}, nullptr, in Init()
185 RETURN_IF_ERROR(cl_args_.SetObjectRef("src_tensor", &src_tensor)); in Convert()
186 RETURN_IF_ERROR(cl_args_.SetObjectRef("dst_tensor", &dst_tensor)); in Convert()
187 RETURN_IF_ERROR(cl_args_.Bind(kernel_.kernel())); in Convert()
258 RETURN_IF_ERROR(cl_args_.Init(environment->device().GetInfo(), {}, nullptr, in Init()
353 RETURN_IF_ERROR(cl_args_.Init(environment->device().GetInfo(), {}, nullptr, in Init()