Searched refs:ComputeTask (Results 1 – 5 of 5) sorted by relevance
40 class ComputeTask {42 ComputeTask() = default;45 ComputeTask(ComputeTask&& args) = default;46 ComputeTask& operator=(ComputeTask&& args) = default;47 ComputeTask(const ComputeTask&) = delete;48 ComputeTask& operator=(const ComputeTask&) = delete;55 absl::Status AddTask(ComputeTask* task);
65 void ComputeTask::Init(std::unique_ptr<GPUOperation>&& operation) { in Init()69 const OperationDef& ComputeTask::GetDefinition() const { in GetDefinition()73 bool ComputeTask::IsLinkable() const { return operation_->IsLinkable(); } in IsLinkable()75 absl::Status ComputeTask::AddTask(ComputeTask* task) { in AddTask()79 absl::Status ComputeTask::Compile(MetalDevice* device) { in Compile()92 absl::Status ComputeTask::CompileProgram(MetalDevice* device, in CompileProgram()190 absl::Status ComputeTask::UpdateParams() { in UpdateParams()217 void ComputeTask::Encode(id<MTLComputeCommandEncoder> encoder) { in Encode()230 void ComputeTask::SetSrcTensor(MetalSpatialTensor* tensor, int index) { in SetSrcTensor()236 void ComputeTask::SetDstTensor(MetalSpatialTensor* tensor, int index) { in SetDstTensor()[all …]
42 ComputeTask task;
43 class ComputeTask; variable155 friend class metal::ComputeTask;
92 ComputeTask gpu_task; in ExecuteGPUOperation()