Searched refs:SetSrcTensor (Results 1 – 4 of 4) sorted by relevance
64 void SetSrcTensor(MetalSpatialTensor* tensor, int index);
230 void ComputeTask::SetSrcTensor(MetalSpatialTensor* tensor, int index) { in SetSrcTensor() function in tflite::gpu::metal::ComputeTask
405 node.task.SetSrcTensor(GetTensor(src_ids[i]), i); in BindTensorsToOperations()667 task.SetSrcTensor(&it->second, i); in UpdatePreallocatedTensors()
96 gpu_task.SetSrcTensor(&src[i], i); in ExecuteGPUOperation()