Searched refs:CanCreateTensorWithShape (Results 1 – 3 of 3) sorted by relevance
27 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info, in CanCreateTensorWithShape() function188 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info, in CanCreateTensorWithShape() function192 return CanCreateTensorWithShape(gpu_info, shape5D, descriptor); in CanCreateTensorWithShape()199 if (CanCreateTensorWithShape(gpu_info, shape, in SelectBestStorageType()207 return CanCreateTensorWithShape( in SelectBestStorageType()213 CanCreateTensorWithShape( in SelectBestStorageType()222 return CanCreateTensorWithShape( in SelectBestStorageType()229 CanCreateTensorWithShape( in SelectBestStorageType()238 return CanCreateTensorWithShape( in SelectBestStorageType()244 if (CanCreateTensorWithShape( in SelectBestStorageType()[all …]
29 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info,33 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info,
305 CanCreateTensorWithShape( in ReserveGraphTensors()