Searched refs:CpuMemory (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | api.h | 147 struct CpuMemory { struct
148 CpuMemory() = default;
149 CpuMemory(void* new_data, size_t new_size_bytes) in CpuMemory() argument
157 inline CpuMemory MakeCpuMemory(absl::Span<T> t) { in MakeCpuMemory() argument
158 CpuMemory m; in MakeCpuMemory()
165 inline CpuMemory MakeReadableCpuMemory(absl::Span<const T> t) { in MakeReadableCpuMemory()
166 CpuMemory m; in MakeReadableCpuMemory()
232 absl::variant<absl::monostate, OpenGlBuffer, OpenGlTexture, CpuMemory,
|
| D | api.cc | 40 ObjectType operator()(CpuMemory) const { return ObjectType::CPU_MEMORY; } in operator ()()
53 bool operator()(CpuMemory obj) const { in operator ()()
83 return absl::holds_alternative<CpuMemory>(obj); in IsObjectPresent()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | internal_api_samples.cc | 203 i, CpuMemory{tensor_ptr->data.data, tensor_ptr->bytes})); in RunModelSampleWithInternalAPI()
211 i, CpuMemory{output_tensors[i].data(), tensor_ptr->bytes})); in RunModelSampleWithInternalAPI()
327 i, CpuMemory{tensor_ptr->data.data, tensor_ptr->bytes})); in RunModelSampleWithInternalAPISerializedKernels()
335 i, CpuMemory{output_tensors[i].data(), tensor_ptr->bytes})); in RunModelSampleWithInternalAPISerializedKernels()
403 i, CpuMemory{tensor_ptr->data.data, tensor_ptr->bytes})); in RunModelSampleWithInternalAPISerialized()
411 i, CpuMemory{output_tensors[i].data(), tensor_ptr->bytes})); in RunModelSampleWithInternalAPISerialized()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | converter.cc | 318 auto cpu_input = absl::get_if<CpuMemory>(&input_obj); in Convert()
319 auto cpu_output = absl::get_if<CpuMemory>(&output_obj); in Convert()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | converter.cc | 491 auto cpu_input = absl::get_if<CpuMemory>(&input_obj); in Convert()
492 auto cpu_output = absl::get_if<CpuMemory>(&output_obj); in Convert()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api2.cc | 249 external_obj_ = CpuMemory{cpu_memory_.data(), cpu_memory_.size()}; in MaybeAllocateExternalObject()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | api.cc | 187 external_obj_ = CpuMemory{cpu_memory_.data(), cpu_memory_.size()}; in MaybeAllocateExternalObject()
|