Searched defs:shape5D (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 589 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() local
602 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedTensor() local
637 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in AllocateTensorMemory() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.cc | 431 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() local
444 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedBufferTensor() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | storage_type_util.cc | 191 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CanCreateTensorWithShape() local
|