Searched refs:SetDstTensor (Results 1 – 4 of 4) sorted by relevance
66 void SetDstTensor(MetalSpatialTensor* tensor, int index);
236 void ComputeTask::SetDstTensor(MetalSpatialTensor* tensor, int index) { in SetDstTensor() function in tflite::gpu::metal::ComputeTask
409 node.task.SetDstTensor(GetTensor(dst_ids[i]), i); in BindTensorsToOperations()674 task.SetDstTensor(&it->second, i); in UpdatePreallocatedTensors()
99 gpu_task.SetDstTensor(&dst[i], i); in ExecuteGPUOperation()