Searched refs:OpenClBuffer (Results 1 – 5 of 5) sorted by relevance
102 auto buffer = absl::get_if<OpenClBuffer>(&obj); in GetOpenCLMemory()267 auto output = absl::get_if<OpenClBuffer>(&output_obj); in Convert()362 auto input = absl::get_if<OpenClBuffer>(&input_obj); in Convert()433 auto buffer_input = absl::get_if<OpenClBuffer>(&input_obj); in Convert()434 auto buffer_output = absl::get_if<OpenClBuffer>(&output_obj); in Convert()441 absl::Status Copy(const OpenClBuffer& input, const OpenClBuffer& output) { in Copy()500 auto buffer_output = absl::get_if<OpenClBuffer>(&output_obj); in Convert()512 auto buffer_input = absl::get_if<OpenClBuffer>(&input_obj); in Convert()
99 struct OpenClBuffer { struct100 OpenClBuffer() = default;101 explicit OpenClBuffer(cl_mem new_memobj) : memobj(new_memobj) {} in OpenClBuffer() function233 OpenClBuffer, OpenClTexture, VulkanBuffer, VulkanTexture>;
28 ObjectType operator()(OpenClBuffer) const { in operator ()()49 bool operator()(OpenClBuffer obj) const { return obj.memobj; } in operator ()()89 return absl::holds_alternative<OpenClBuffer>(obj); in IsObjectPresent()
275 auto cl_mem = absl::get_if<OpenClBuffer>(&output_obj); in Convert()287 auto cl_mem = absl::get_if<OpenClBuffer>(&input_obj); in Convert()
204 external_obj_ = OpenClBuffer{cl_memory_.memory()}; in MaybeAllocateExternalObject()342 RETURN_IF_ERROR(tie_->SetExternalObject(OpenClBuffer{cl_object_.memory()})); in SetExternalObject()375 return OpenClBuffer{tensor.GetMemoryPtr()}; in TensorToObj()378 return OpenClBuffer{tensor.GetMemoryPtrForWriting()}; in TensorToObj()