Home
last modified time | relevance | path

Searched refs:SetSrcTensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
H A Dtest_util.cc80 gpu_task.SetSrcTensor(&src[i], i); in ExecuteGpuOperationInternal()
170 gpu_task.SetSrcTensor(&src[i], i); in ExecuteGPUOperation()
H A Dconv_test.mm335 gpu_task.SetSrcTensor(&src_gpu, 0);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dcompute_task.h70 void SetSrcTensor(MetalSpatialTensor* tensor, int index);
H A Dinference_context.cc372 node.task.SetSrcTensor(GetTensor(src_ids[i]), i); in BindTensorsToOperations()
782 node.task.SetSrcTensor(tensor_ptr, i); in SetTensor()
H A Dcompute_task.cc328 void ComputeTask::SetSrcTensor(MetalSpatialTensor* tensor, int index) { in SetSrcTensor() function in tflite::gpu::metal::ComputeTask
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dcl_operation.h71 absl::Status SetSrcTensor(int index, Tensor* tensor);
H A Dcl_operation.cc150 absl::Status ClOperation::SetSrcTensor(int index, Tensor* tensor) { in SetSrcTensor() function in tflite::gpu::cl::ClOperation
H A Dinference_context.cc710 RETURN_IF_ERROR(node.cl_operation.SetSrcTensor(i, tensor_ptr)); in SetTensor()