Searched defs: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
|
| /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
639 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedTensor() local
653 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 | 548 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() local
561 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedBufferTensor() local
|