Home
last modified time | relevance | path

Searched refs:TensorStorageType (Results 1 – 25 of 58) sorted by relevance

123

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_type_util.cc22 ObjectType ToObjectType(TensorStorageType type) { in ToObjectType()
24 case TensorStorageType::IMAGE_BUFFER: in ToObjectType()
25 case TensorStorageType::BUFFER: in ToObjectType()
27 case TensorStorageType::SINGLE_TEXTURE_2D: in ToObjectType()
28 case TensorStorageType::TEXTURE_2D: in ToObjectType()
29 case TensorStorageType::TEXTURE_ARRAY: in ToObjectType()
30 case TensorStorageType::TEXTURE_3D: in ToObjectType()
37 DataLayout ToDataLayout(TensorStorageType type) { in ToDataLayout()
39 case TensorStorageType::BUFFER: in ToDataLayout()
41 case TensorStorageType::IMAGE_BUFFER: in ToDataLayout()
[all …]
Denvironment.cc52 TensorStorageType storage_type) { in IsGpuSupportsStorageType()
54 case TensorStorageType::TEXTURE_2D: in IsGpuSupportsStorageType()
56 case TensorStorageType::BUFFER: in IsGpuSupportsStorageType()
58 case TensorStorageType::TEXTURE_ARRAY: in IsGpuSupportsStorageType()
60 case TensorStorageType::IMAGE_BUFFER: in IsGpuSupportsStorageType()
63 case TensorStorageType::TEXTURE_3D: in IsGpuSupportsStorageType()
65 case TensorStorageType::SINGLE_TEXTURE_2D: in IsGpuSupportsStorageType()
67 case TensorStorageType::UNKNOWN: in IsGpuSupportsStorageType()
157 std::vector<TensorStorageType> Environment::GetSupportedStorages() const { in GetSupportedStorages()
158 std::vector<TensorStorageType> storage_types; in GetSupportedStorages()
[all …]
Dtensor_test.cc90 absl::Status TensorTests(DataType data_type, TensorStorageType storage_type, in TensorTests()
130 ASSERT_OK(TensorTests(DataType::FLOAT32, TensorStorageType::BUFFER, &env_)); in TEST_F()
134 ASSERT_OK(TensorTests(DataType::FLOAT16, TensorStorageType::BUFFER, &env_)); in TEST_F()
139 TensorTests(DataType::FLOAT32, TensorStorageType::TEXTURE_2D, &env_)); in TEST_F()
144 TensorTests(DataType::FLOAT16, TensorStorageType::TEXTURE_2D, &env_)); in TEST_F()
149 TensorTests(DataType::FLOAT32, TensorStorageType::TEXTURE_3D, &env_)); in TEST_F()
154 TensorTests(DataType::FLOAT16, TensorStorageType::TEXTURE_3D, &env_)); in TEST_F()
159 TensorTests(DataType::FLOAT32, TensorStorageType::TEXTURE_ARRAY, &env_)); in TEST_F()
164 TensorTests(DataType::FLOAT16, TensorStorageType::TEXTURE_ARRAY, &env_)); in TEST_F()
169 TensorTests(DataType::FLOAT32, TensorStorageType::IMAGE_BUFFER, &env_)); in TEST_F()
[all …]
Dtensor.cc42 case TensorStorageType::BUFFER: in AllocateTensorMemory()
43 case TensorStorageType::IMAGE_BUFFER: { in AllocateTensorMemory()
57 case TensorStorageType::TEXTURE_2D: { in AllocateTensorMemory()
86 case TensorStorageType::TEXTURE_3D: { in AllocateTensorMemory()
115 case TensorStorageType::TEXTURE_ARRAY: { in AllocateTensorMemory()
146 case TensorStorageType::SINGLE_TEXTURE_2D: { in AllocateTensorMemory()
226 if (descriptor.storage_type == TensorStorageType::IMAGE_BUFFER) { in CreateTensor()
243 if (descriptor.storage_type == TensorStorageType::IMAGE_BUFFER) { in CreateTensorShared()
328 if (descriptor_.storage_type != TensorStorageType::BUFFER) { in GetGPUResources()
339 if (descriptor_.storage_type != TensorStorageType::TEXTURE_2D) { in GetGPUResources()
[all …]
Denvironment.h57 std::vector<TensorStorageType> GetSupportedStorages() const;
60 std::vector<TensorStorageType> GetSupportedStoragesWithHWZeroClampSupport()
62 bool IsSupported(TensorStorageType storage_type) const;
78 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info);
79 TensorStorageType GetStorageTypeWithMinimalMemoryConsumption(
Dtensor_type_util.h26 ObjectType ToObjectType(TensorStorageType type);
28 DataLayout ToDataLayout(TensorStorageType type);
30 TensorStorageType ToTensorStorageType(ObjectType object_type,
Dserialization.cc99 data::TensorStorageType ToFB(TensorStorageType type) { in ToFB()
101 case TensorStorageType::BUFFER: in ToFB()
102 return data::TensorStorageType::BUFFER; in ToFB()
103 case TensorStorageType::IMAGE_BUFFER: in ToFB()
104 return data::TensorStorageType::IMAGE_BUFFER; in ToFB()
105 case TensorStorageType::TEXTURE_2D: in ToFB()
106 return data::TensorStorageType::TEXTURE_2D; in ToFB()
107 case TensorStorageType::TEXTURE_ARRAY: in ToFB()
108 return data::TensorStorageType::TEXTURE_ARRAY; in ToFB()
109 case TensorStorageType::TEXTURE_3D: in ToFB()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dstorage_type_util.cc30 case TensorStorageType::BUFFER: { in CanCreateTensorWithShape()
37 case TensorStorageType::IMAGE_BUFFER: in CanCreateTensorWithShape()
40 case TensorStorageType::TEXTURE_3D: in CanCreateTensorWithShape()
50 case TensorStorageType::TEXTURE_ARRAY: in CanCreateTensorWithShape()
59 case TensorStorageType::TEXTURE_2D: in CanCreateTensorWithShape()
62 case TensorStorageType::SINGLE_TEXTURE_2D: in CanCreateTensorWithShape()
78 TensorStorageType SelectBestStorageType(const GpuInfo& gpu_info, in SelectBestStorageType()
80 const TensorStorageType& desired, in SelectBestStorageType()
88 return TensorStorageType::BUFFER; in SelectBestStorageType()
94 TensorDescriptor{data_type, TensorStorageType::IMAGE_BUFFER, in SelectBestStorageType()
[all …]
Dtensor_desc.cc57 std::string ToString(TensorStorageType type) { in ToString()
59 case TensorStorageType::UNKNOWN: in ToString()
61 case TensorStorageType::BUFFER: in ToString()
63 case TensorStorageType::TEXTURE_ARRAY: in ToString()
65 case TensorStorageType::TEXTURE_2D: in ToString()
67 case TensorStorageType::TEXTURE_3D: in ToString()
69 case TensorStorageType::SINGLE_TEXTURE_2D: in ToString()
71 case TensorStorageType::IMAGE_BUFFER: in ToString()
119 if (storage_type == TensorStorageType::BUFFER) { in GetGPUResources()
133 } else if (storage_type == TensorStorageType::SINGLE_TEXTURE_2D || in GetGPUResources()
[all …]
Dstorage_type_util.h34 TensorStorageType SelectBestStorageType(const GpuInfo& gpu_info,
36 const TensorStorageType& desired,
41 TensorStorageType tensor_storage_type);
Dtensor_desc.h35 enum class TensorStorageType { enum
47 TensorDescriptor(DataType dt, TensorStorageType st, Layout l) in TensorDescriptor()
96 TensorStorageType storage_type = TensorStorageType::UNKNOWN;
197 std::string ToString(TensorStorageType type);
Dtesting_util.h37 virtual std::vector<TensorStorageType> GetSupportedStorages() const = 0;
40 virtual std::vector<TensorStorageType>
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_spatial_tensor.cc63 case TensorStorageType::BUFFER: in AllocateTensorMemory()
64 case TensorStorageType::IMAGE_BUFFER: { in AllocateTensorMemory()
78 if (descriptor.storage_type == TensorStorageType::IMAGE_BUFFER) { in AllocateTensorMemory()
84 case TensorStorageType::TEXTURE_2D: { in AllocateTensorMemory()
105 case TensorStorageType::TEXTURE_3D: { in AllocateTensorMemory()
126 case TensorStorageType::TEXTURE_ARRAY: { in AllocateTensorMemory()
147 case TensorStorageType::SINGLE_TEXTURE_2D: in AllocateTensorMemory()
233 if (descriptor_.storage_type != TensorStorageType::BUFFER) { in GetGPUResources()
244 if (descriptor_.storage_type != TensorStorageType::TEXTURE_2D) { in GetGPUResources()
280 if (descriptor_.storage_type == TensorStorageType::BUFFER) { in GetGPUResources()
[all …]
Dmetal_spatial_tensor_test.mm34 using tflite::gpu::TensorStorageType;
100 static absl::Status TensorTests(DataType data_type, TensorStorageType storage_type,
141 auto status = TensorTests(DataType::FLOAT32, TensorStorageType::BUFFER, device);
147 auto status = TensorTests(DataType::FLOAT16, TensorStorageType::BUFFER, device);
153 auto status = TensorTests(DataType::FLOAT32, TensorStorageType::TEXTURE_2D, device);
159 auto status = TensorTests(DataType::FLOAT16, TensorStorageType::TEXTURE_2D, device);
165 auto status = TensorTests(DataType::FLOAT32, TensorStorageType::TEXTURE_3D, device);
171 auto status = TensorTests(DataType::FLOAT16, TensorStorageType::TEXTURE_3D, device);
177 auto status = TensorTests(DataType::FLOAT32, TensorStorageType::TEXTURE_ARRAY, device);
183 auto status = TensorTests(DataType::FLOAT16, TensorStorageType::TEXTURE_ARRAY, device);
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtest_util.cc48 std::vector<TensorStorageType> MetalExecutionEnvironment::GetSupportedStorages() in GetSupportedStorages()
50 return {TensorStorageType::BUFFER, TensorStorageType::IMAGE_BUFFER, in GetSupportedStorages()
51 TensorStorageType::TEXTURE_2D, TensorStorageType::TEXTURE_3D, in GetSupportedStorages()
52 TensorStorageType::TEXTURE_ARRAY}; in GetSupportedStorages()
57 std::vector<TensorStorageType>
59 return {TensorStorageType::TEXTURE_2D, TensorStorageType::TEXTURE_3D, in GetSupportedStoragesWithHWZeroClampSupport()
60 TensorStorageType::TEXTURE_ARRAY}; in GetSupportedStoragesWithHWZeroClampSupport()
Dtest_util.h41 std::vector<TensorStorageType> GetSupportedStorages() const override;
42 std::vector<TensorStorageType> GetSupportedStoragesWithHWZeroClampSupport()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.cc194 if ((src_storage == TensorStorageType::BUFFER || in SelectWinograd4x4To36()
195 src_storage == TensorStorageType::IMAGE_BUFFER) && in SelectWinograd4x4To36()
196 (dst_storage == TensorStorageType::BUFFER || in SelectWinograd4x4To36()
197 dst_storage == TensorStorageType::IMAGE_BUFFER)) { in SelectWinograd4x4To36()
212 if ((src_storage == TensorStorageType::BUFFER || in SelectWinograd36To4x4()
213 src_storage == TensorStorageType::IMAGE_BUFFER) && in SelectWinograd36To4x4()
214 (dst_storage == TensorStorageType::BUFFER || in SelectWinograd36To4x4()
215 dst_storage == TensorStorageType::IMAGE_BUFFER)) { in SelectWinograd36To4x4()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc131 {data_type, TensorStorageType::BUFFER, Layout::UNKNOWN}); in ConverterToConvWeights1x1OutX4Test()
165 {data_type, TensorStorageType::BUFFER, Layout::UNKNOWN}); in ConverterToConvWeights1x1OutX4UnalignedTest()
198 {data_type, TensorStorageType::BUFFER, Layout::UNKNOWN}); in ConverterToConvWeights1x1OutX2Test()
231 {data_type, TensorStorageType::BUFFER, Layout::UNKNOWN}); in ConverterToConvWeightsOutX2Test()
266 {data_type, TensorStorageType::BUFFER, Layout::UNKNOWN}); in ConverterToConvTransposedWeights4x4Test()
301 {data_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in ConverterToConvWeights4xTexturesTest()
303 {data_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in ConverterToConvWeights4xTexturesTest()
305 {data_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in ConverterToConvWeights4xTexturesTest()
307 {data_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in ConverterToConvWeights4xTexturesTest()
Ddepthwise_conv_3x3_stride_h2.cc31 const bool manual_clamp = src_tensor_type == TensorStorageType::BUFFER || in GetKernelDepthWiseConv3x3StrideH2()
32 src_tensor_type == TensorStorageType::IMAGE_BUFFER; in GetKernelDepthWiseConv3x3StrideH2()
104 if (src_tensor_type == TensorStorageType::BUFFER) { in GetKernelDepthWiseConv3x3StrideH2()
123 if (src_tensor_type == TensorStorageType::BUFFER) { in GetKernelDepthWiseConv3x3StrideH2()
131 } else if (src_tensor_type == TensorStorageType::IMAGE_BUFFER) { in GetKernelDepthWiseConv3x3StrideH2()
Ddepthwise_conv_3x3.cc64 const bool manual_clamp = src_tensor_type == TensorStorageType::BUFFER || in GenerateDepthwiseConvCode()
65 src_tensor_type == TensorStorageType::IMAGE_BUFFER; in GenerateDepthwiseConvCode()
145 if (src_tensor_type == TensorStorageType::BUFFER) { in GenerateDepthwiseConvCode()
171 if (src_tensor_type == TensorStorageType::BUFFER) { in GenerateDepthwiseConvCode()
181 } else if (src_tensor_type == TensorStorageType::IMAGE_BUFFER) { in GenerateDepthwiseConvCode()
Dwinograd_test_util.cc204 {TensorStorageType::BUFFER, TensorStorageType::IMAGE_BUFFER}) { in Winograd4x4To36Test()
275 {TensorStorageType::BUFFER, TensorStorageType::IMAGE_BUFFER}) { in Winograd36To4x4Test()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconv_buffer_1x1_test.cc55 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
57 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
92 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
94 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
Dcl_test.h47 std::vector<TensorStorageType> GetSupportedStorages() const override;
48 std::vector<TensorStorageType> GetSupportedStoragesWithHWZeroClampSupport()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dtensor.h52 using TensorStorageType = typename internal_tensor::StorageType<Type>::value;
59 TensorStorageType data;
88 tensor.data = typename Tensor<ShapeT, Type>::TensorStorageType(
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Ddw_convolution_selector.cc54 bool buffer_type = storage_type == TensorStorageType::BUFFER || in SelectDWConvolutionMali()
55 storage_type == TensorStorageType::IMAGE_BUFFER; in SelectDWConvolutionMali()

123