Home
last modified time | relevance | path

Searched refs:LinearStorageType (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_linear_desc.cc32 if (storage_type == LinearStorageType::BUFFER) { in GetGPUResources()
58 if (storage_type != LinearStorageType::BUFFER) { in PerformSelector()
78 if (storage_type == LinearStorageType::BUFFER) { in PerformReadSelector()
Dtensor_linear_desc.h30 enum class LinearStorageType { BUFFER, TEXTURE_2D }; enum
33 LinearStorageType storage_type;
Dstorage_type_util.cc138 LinearStorageType DeduceLinearStorageType( in DeduceLinearStorageType()
141 return LinearStorageType::BUFFER; in DeduceLinearStorageType()
143 return LinearStorageType::TEXTURE_2D; in DeduceLinearStorageType()
Dstorage_type_util.h40 LinearStorageType DeduceLinearStorageType(
Dserialization_base.fbs109 enum LinearStorageType : byte {
116 storage_type:LinearStorageType;
Dserialization_base_generated.h195 enum class LinearStorageType : int8_t { enum
202 inline const LinearStorageType (&EnumValuesLinearStorageType())[2] { in EnumValuesLinearStorageType()
203 static const LinearStorageType values[] = { in EnumValuesLinearStorageType()
204 LinearStorageType::BUFFER, in EnumValuesLinearStorageType()
205 LinearStorageType::TEXTURE_2D in EnumValuesLinearStorageType()
219 inline const char *EnumNameLinearStorageType(LinearStorageType e) { in EnumNameLinearStorageType()
220 …if (flatbuffers::IsOutRange(e, LinearStorageType::BUFFER, LinearStorageType::TEXTURE_2D)) return "… in EnumNameLinearStorageType()
1204 tflite::gpu::data::LinearStorageType storage_type() const { in storage_type()
1205 return static_cast<tflite::gpu::data::LinearStorageType>(GetField<int8_t>(VT_STORAGE_TYPE, 0)); in storage_type()
1240 void add_storage_type(tflite::gpu::data::LinearStorageType storage_type) { in add_storage_type()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.cc263 desc.storage_type = weights_are_buffer ? LinearStorageType::BUFFER in CreateDepthwiseConvolution2D()
264 : LinearStorageType::TEXTURE_2D; in CreateDepthwiseConvolution2D()
291 ? LinearStorageType::BUFFER in CreateDepthwiseConvolution2DDynamicWeights()
292 : LinearStorageType::TEXTURE_2D; in CreateDepthwiseConvolution2DDynamicWeights()
331 desc.storage_type = weights_are_buffer ? LinearStorageType::BUFFER in CreateDepthwiseConvolution3D()
332 : LinearStorageType::TEXTURE_2D; in CreateDepthwiseConvolution3D()
Dfully_connected.cc164 desc.storage_type = gpu_info.SupportsImages() ? LinearStorageType::TEXTURE_2D in CreateFullyConnected()
165 : LinearStorageType::BUFFER; in CreateFullyConnected()
Dconvolution_transposed_4x4.cc434 ? LinearStorageType::BUFFER in CreateConvolutionTransposed4x4()
435 : LinearStorageType::TEXTURE_2D; in CreateConvolutionTransposed4x4()
459 ? LinearStorageType::BUFFER in CreateConvolutionTransposed4x4DynamicWeights()
460 : LinearStorageType::TEXTURE_2D; in CreateConvolutionTransposed4x4DynamicWeights()
Dwinograd.cc432 desc.storage_type = LinearStorageType::TEXTURE_2D; in UploadBt()
508 bias_desc.storage_type = LinearStorageType::BUFFER; in CreateWinograd36To4x4()
651 desc.storage_type = LinearStorageType::TEXTURE_2D; in UploadAt()
705 desc.storage_type = LinearStorageType::TEXTURE_2D; in CreateWinograd36To4x4Tile4x1()
Dconvolution_transposed_3x3_thin.cc242 desc.storage_type = LinearStorageType::TEXTURE_2D; in CreateConvolutionTransposed3x3Thin()
264 desc.storage_type = LinearStorageType::TEXTURE_2D; in CreateConvolutionTransposed3x3ThinDynamicWeights()
Dconvolution_transposed_3x3.cc429 desc.storage_type = LinearStorageType::TEXTURE_2D; in CreateConvolutionTransposed3x3()
453 desc.storage_type = LinearStorageType::TEXTURE_2D; in CreateConvolutionTransposed3x3DynamicWeights()
Dconv_buffer_1x1.h161 desc.storage_type = LinearStorageType::BUFFER; in UploadBiases()
Dconv_constants.cc298 desc.storage_type = LinearStorageType::BUFFER; in CreateConvConstants()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dlinear_storage.cc64 if (storage_type_ == LinearStorageType::BUFFER) { in GetGPUResources()
80 if (storage_type_ == LinearStorageType::BUFFER) { in CreateFromTensorLinearDescriptor()
Dserialization.cc90 data::LinearStorageType ToFB(LinearStorageType type) { in ToFB()
92 case LinearStorageType::BUFFER: in ToFB()
93 return data::LinearStorageType::BUFFER; in ToFB()
94 case LinearStorageType::TEXTURE_2D: in ToFB()
95 return data::LinearStorageType::TEXTURE_2D; in ToFB()
184 LinearStorageType ToEnum(data::LinearStorageType type) { in ToEnum()
186 case data::LinearStorageType::BUFFER: in ToEnum()
187 return LinearStorageType::BUFFER; in ToEnum()
188 case data::LinearStorageType::TEXTURE_2D: in ToEnum()
189 return LinearStorageType::TEXTURE_2D; in ToEnum()
Dlinear_storage.h62 LinearStorageType storage_type_;
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dlinear_storage.cc69 if (storage_type_ == LinearStorageType::BUFFER) { in GetGPUResources()
88 if (storage_type_ == LinearStorageType::BUFFER) { in CreateFromTensorLinearDescriptor()
Dlinear_storage.h60 LinearStorageType storage_type_;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.cc183 desc0.storage_type = LinearStorageType::TEXTURE_2D; in CreateFCFCAdd()
190 desc1.storage_type = LinearStorageType::TEXTURE_2D; in CreateFCFCAdd()