Searched refs:SetObjectRef (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | compute_task.cc | 197 RETURN_IF_ERROR(metal_args_.SetObjectRef(operation_->src_tensors_names_[i], in UpdateParams()
206 RETURN_IF_ERROR(metal_args_.SetObjectRef(operation_->dst_tensors_names_[i], in UpdateParams()
233 metal_args_.SetObjectRef(operation_->src_tensors_names_[index], *tensor); in SetSrcTensor()
239 metal_args_.SetObjectRef(operation_->dst_tensors_names_[index], *tensor); in SetDstTensor()
|
| D | metal_arguments.h | 50 absl::Status SetObjectRef(const std::string& name, const GPUObject& object);
|
| D | metal_arguments.cc | 416 absl::Status MetalArguments::SetObjectRef(const std::string& name, in SetObjectRef() function in tflite::gpu::metal::MetalArguments
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.cc | 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()
|
| D | cl_arguments.h | 61 absl::Status SetObjectRef(const std::string& name, const GPUObject* object);
|
| D | cl_arguments.cc | 650 absl::Status CLArguments::SetObjectRef(const std::string& name, in SetObjectRef() function in tflite::gpu::cl::CLArguments
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | converter.cc | 51 RETURN_IF_ERROR(cl_args_.SetObjectRef("tensor", tensor)); in DispatchKernel()
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()
|