Home
last modified time | relevance | path

Searched refs:WeightsUploadType (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_4x4.cc28 ConvolutionTransposed4x4::WeightsUploadType GetBestWeightsUploadType( in GetBestWeightsUploadType()
30 ConvolutionTransposed4x4::WeightsUploadType weights_upload_type = in GetBestWeightsUploadType()
31 ConvolutionTransposed4x4::WeightsUploadType::GLOBAL_MEM; in GetBestWeightsUploadType()
34 ConvolutionTransposed4x4::WeightsUploadType::LOCAL_MEM_ASYNC; in GetBestWeightsUploadType()
37 ConvolutionTransposed4x4::WeightsUploadType::LOCAL_MEM_BY_THREADS; in GetBestWeightsUploadType()
40 ConvolutionTransposed4x4::WeightsUploadType::CONSTANT_MEM; in GetBestWeightsUploadType()
43 ConvolutionTransposed4x4::WeightsUploadType::GLOBAL_MEM; in GetBestWeightsUploadType()
73 WeightsUploadType weights_upload_type) { in GenerateConvolutionTransposedCode()
94 ConvolutionTransposed4x4::WeightsUploadType::CONSTANT_MEM in GenerateConvolutionTransposedCode()
104 ConvolutionTransposed4x4::WeightsUploadType::LOCAL_MEM_BY_THREADS || in GenerateConvolutionTransposedCode()
[all …]
Dconvolution_transposed_3x3.cc33 weights_upload_type_ = WeightsUploadType::LOCAL_MEM_ASYNC; in ConvolutionTransposed3x3()
35 weights_upload_type_ = WeightsUploadType::LOCAL_MEM_BY_THREADS; in ConvolutionTransposed3x3()
37 weights_upload_type_ = WeightsUploadType::CONSTANT_MEM; in ConvolutionTransposed3x3()
39 weights_upload_type_ = WeightsUploadType::GLOBAL_MEM; in ConvolutionTransposed3x3()
57 ConvolutionTransposed3x3::WeightsUploadType weights_upload_type, in GenerateConvolutionTransposedCode()
79 ConvolutionTransposed3x3::WeightsUploadType::CONSTANT_MEM in GenerateConvolutionTransposedCode()
91 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_BY_THREADS || in GenerateConvolutionTransposedCode()
93 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_ASYNC; in GenerateConvolutionTransposedCode()
130 ConvolutionTransposed3x3::WeightsUploadType::CONSTANT_MEM in GenerateConvolutionTransposedCode()
182 ConvolutionTransposed3x3::WeightsUploadType::LOCAL_MEM_BY_THREADS) { in GenerateConvolutionTransposedCode()
[all …]
Dconvolution_transposed_3x3.h61 enum class WeightsUploadType { enum
85 ConvolutionTransposed3x3::WeightsUploadType weights_upload_type,
89 WeightsUploadType weights_upload_type_;
Dconvolution_transposed_4x4.h63 enum class WeightsUploadType { enum
83 WeightsUploadType weights_upload_type);
89 WeightsUploadType weights_upload_type);
Dconv_metal.cc175 ConvolutionMetal::WeightsUploadType::CONSTANT_MEM in GenerateConvolution()
180 ConvolutionMetal::WeightsUploadType::LOCAL_MEM_BY_THREADS; in GenerateConvolution()
186 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD8_BROADCAST || in GenerateConvolution()
188 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD16_BROADCAST || in GenerateConvolution()
190 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD32_BROADCAST; in GenerateConvolution()
193 ConvolutionMetal::WeightsUploadType::PRIVATE_MEM_SIMD8_BROADCAST) { in GenerateConvolution()
195 } else if (params.weights_upload_type == ConvolutionMetal::WeightsUploadType:: in GenerateConvolution()
198 } else if (params.weights_upload_type == ConvolutionMetal::WeightsUploadType:: in GenerateConvolution()
699 ConvolutionMetal::WeightsUploadType::LOCAL_MEM_BY_THREADS; in GetConvParamsForA7A8()
751 ConvolutionMetal::WeightsUploadType::GLOBAL_MEM; in GetConvParamsForA7A8()
[all …]
Dconv_powervr.cc326 WeightsUploadType::LOCAL_MEM_ASYNC_SUBGROUP || in GetPossibleKernelWorkGroups()
328 WeightsUploadType::LOCAL_MEM_BY_THREADS || in GetPossibleKernelWorkGroups()
353 ConvPowerVR::WeightsUploadType::CONSTANT_MEM in GenerateConv()
441 ConvPowerVR::WeightsUploadType::LOCAL_MEM_BY_THREADS || in GenerateConv()
443 ConvPowerVR::WeightsUploadType::LOCAL_MEM_ASYNC_SUBGROUP; in GenerateConv()
455 ConvPowerVR::WeightsUploadType::CONSTANT_MEM in GenerateConv()
520 ConvPowerVR::WeightsUploadType::LOCAL_MEM_BY_THREADS) { in GenerateConv()
906 ConvPowerVR::WeightsUploadType::LOCAL_MEM_ASYNC_SUBGROUP) { in GenerateConv()
910 ConvPowerVR::WeightsUploadType::LOCAL_MEM_BY_THREADS) { in GenerateConv()
954 ConvPowerVR::WeightsUploadType::LOCAL_MEM_BY_THREADS) { in GenerateConv()
[all …]
Dconv_powervr.h64 enum class WeightsUploadType { enum
91 WeightsUploadType weights_upload_type;
101 return weights_upload_type != WeightsUploadType::TEXTURES_MEM_X4; in AreWeightsBuffer()
106 WeightsUploadType::PRIVATE_MEM_SIMD_BROADCAST; in IsPrivateMemBroadcast()
230 ConvPowerVR::WeightsUploadType::CONSTANT_MEM in UploadBias()
269 ConvPowerVR::WeightsUploadType::CONSTANT_MEM in UploadWeights()
Dconv_metal.h31 enum class WeightsUploadType { enum
49 WeightsUploadType weights_upload_type;