Searched refs:CanCreateTensorWithShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | storage_type_util.cc | 26 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWDC& shape, in CanCreateTensorWithShape() function
72 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWC& shape, in CanCreateTensorWithShape() function
75 return CanCreateTensorWithShape(gpu_info, shape5D, descriptor); in CanCreateTensorWithShape()
83 if (CanCreateTensorWithShape(gpu_info, shape, in SelectBestStorageType()
92 CanCreateTensorWithShape( in SelectBestStorageType()
103 CanCreateTensorWithShape( in SelectBestStorageType()
113 if (CanCreateTensorWithShape( in SelectBestStorageType()
|
| D | storage_type_util.h | 28 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWDC& shape,
31 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWC& shape,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 275 CanCreateTensorWithShape( in ReserveGraphTensors()
|