Searched refs:shape5D (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | storage_type_util.cc | 74 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CanCreateTensorWithShape() local
75 return CanCreateTensorWithShape(gpu_info, shape5D, descriptor); in CanCreateTensorWithShape()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 626 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() local
627 return CreateTensor(context, shape5D, descriptor, nullptr, result); in CreateTensor()
639 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedTensor() local
640 return CreateTensorShared(context, shape5D, descriptor, memory, result); in CreateSharedTensor()
653 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in AllocateTensorMemory() local
654 return AllocateTensorMemory(context, shape5D, descriptor, nullptr, result); in AllocateTensorMemory()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.cc | 548 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() local
549 return CreateTensor(device, shape5D, descriptor, nullptr, nullptr, result); in CreateTensor()
561 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedBufferTensor() local
565 CreateTextureBuffer(buffer, shape5D, descriptor, &texture_buffer)); in CreateSharedBufferTensor()
567 *result = MetalSpatialTensor(buffer, texture_buffer, false, true, shape5D, in CreateSharedBufferTensor()
|