Home
last modified time | relevance | path

Searched refs:AllocateTensorMemory (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor.h120 absl::Status AllocateTensorMemory(const CLContext& context, const BHWC& shape,
124 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape,
Dtensor.cc33 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() function
223 AllocateTensorMemory(context, shape, descriptor, nullptr, &mem)); in CreateTensor()
613 AllocateTensorMemory(*context, shape_, descriptor_, data_ptr, &memory)); in CreateFromDescriptor()
650 absl::Status AllocateTensorMemory(const CLContext& context, const BHWC& shape, in AllocateTensorMemory() function
654 return AllocateTensorMemory(context, shape5D, descriptor, nullptr, result); in AllocateTensorMemory()
657 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() function
660 return AllocateTensorMemory(context, shape, descriptor, nullptr, result); in AllocateTensorMemory()
Dapi.cc200 AllocateTensorMemory(env->context(), shape, desc, &cl_memory_)); in MaybeAllocateExternalObject()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_spatial_tensor.cc57 absl::Status AllocateTensorMemory(id<MTLDevice> device, const BHWDC& shape, in AllocateTensorMemory() function
160 RETURN_IF_ERROR(AllocateTensorMemory(device, shape, descriptor, nullptr, in CreateTensor()
517 RETURN_IF_ERROR(AllocateTensorMemory(device, shape_, descriptor_, data_ptr, in CreateFromDescriptor()