Home
last modified time | relevance | path

Searched refs:CanCreateTensorWithShape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dtensor_desc.cc1397 absl::Status TensorDescriptor::CanCreateTensorWithShape( in CanCreateTensorWithShape() function in tflite::gpu::TensorDescriptor
1543 absl::Status TensorDescriptor::CanCreateTensorWithShape( in CanCreateTensorWithShape() function in tflite::gpu::TensorDescriptor
1546 return CanCreateTensorWithShape(gpu_info, shape5D); in CanCreateTensorWithShape()
1551 if (CanCreateTensorWithShape(gpu_info, shape).ok()) { in UpdateToSupportedStorageType()
1556 return CanCreateTensorWithShape(gpu_info, shape); in UpdateToSupportedStorageType()
1561 CanCreateTensorWithShape(gpu_info, shape).ok()) { in UpdateToSupportedStorageType()
1565 return CanCreateTensorWithShape(gpu_info, shape); in UpdateToSupportedStorageType()
H A Dtensor_desc.h118 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info,
121 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info,
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model.cc158 if (!tensor_desc.CanCreateTensorWithShape(gpu_info, shape).ok()) { in CheckExternalTensorDescription()
242 .CanCreateTensorWithShape(gpu_info, shape) in ReserveGraphTensors()