Home
last modified time | relevance | path

Searched refs:float4_size (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.h78 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsForDWConv2D() local
80 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsForDWConv2D()
94 desc.size = float4_size * elements_count; in UploadWeightsForDWConv2D()
153 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsForDWConv3D() local
155 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsForDWConv3D()
169 desc.size = float4_size * elements_count; in UploadWeightsForDWConv3D()
Dconvolution_transposed.h154 const int float4_size = f32_weights ? 16 : 8; in UploadWeights() local
155 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeights()
181 desc.size = float4_size * elements_count; in UploadWeights()
188 int sub_size = float4_size * texture_width * texture_height; in UploadWeights()
Ddepthwise_conv_3x3_stride_h2.h79 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsAndBiases() local
81 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsAndBiases()
96 desc.size = float4_size * elements_count; in UploadWeightsAndBiases()
Ddepthwise_conv_3x3.h85 const int float4_size = fp32_weights ? 16 : 8; in UploadWeightsAndBiases() local
87 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeightsAndBiases()
102 desc.size = float4_size * elements_count; in UploadWeightsAndBiases()
Dfully_connected.h156 const int float4_size = f32_weights ? 16 : 8; in UploadWeights() local
162 desc.size = float4_size * elements_count; in UploadWeights()
179 desc.data.resize(float4_size * elements_count); in UploadWeights()
Dconv_powervr.h309 const int float4_size = f32_weights ? 16 : 8; in UploadWeights() local
311 std::vector<uint8_t> data(float4_size * elements_count); in UploadWeights()
337 desc.size = float4_size * elements_count; in UploadWeights()
345 int sub_size = float4_size * texture_width * texture_height; in UploadWeights()
Dconv_buffer_1x1.h131 const int float4_size = f32_weights ? sizeof(float4) : sizeof(half4); in UploadWeights() local
141 desc.size = float4_size * elements_count; in UploadWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dlinear_storage.cc85 const int float4_size = desc.element_type == DataType::FLOAT32 in CreateFromTensorLinearDescriptor() local
88 return CreateCLBuffer(context->context(), depth_ * float4_size, read_only, in CreateFromTensorLinearDescriptor()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h133 const int float4_size = f32_weights ? 16 : 8; in UploadWeights() local
139 desc.size = float4_size * elements_count; in UploadWeights()
159 desc.data.resize(float4_size * elements_count); in UploadWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dlinear_storage.cc85 const int float4_size = desc.element_type == DataType::FLOAT32 in CreateFromTensorLinearDescriptor() local
94 length:depth_ * float4_size in CreateFromTensorLinearDescriptor()