Home
last modified time | relevance | path

Searched defs:elements_count (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dbuffer.h91 const int elements_count = size_ / sizeof(T); in ReadData() local
Dmetal_spatial_tensor.cc386 const int elements_count = in WriteDataBHWDC() local
452 const int elements_count = in ReadDataBHWDC() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dbuffer.h102 const int elements_count = size_ / sizeof(T); in ReadData() local
Dtexture2d.h119 const int elements_count = width_ * height_ * 4; in ReadData() local
Dtensor.cc490 const int elements_count = in WriteDataBHWDC() local
552 const int elements_count = in ReadDataBHWDC() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.h75 const int elements_count = kernel_x * kernel_y * dst_slices; in UploadWeightsForDWConv2D() local
150 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeightsForDWConv3D() local
Ddepthwise_conv_3x3.h83 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
Ddepthwise_conv_3x3_stride_h2.h77 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
Dconv_buffer_1x1.h134 const int elements_count = in UploadWeights() local
Dconvolution_transposed.h150 const int elements_count = in UploadWeights() local
Dfully_connected.h153 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
Dconv_powervr.h305 const int elements_count = weights.shape.d * weights.shape.h * in UploadWeights() local
/external/gemmlowp/meta/generators/
Dstreams_common.py57 def _GenerateLoadAggregateStore(emitter, registers, lanes_count, elements_count, argument
206 elements_count, aggregators, input_address, argument
/external/tensorflow/tensorflow/lite/micro/kernels/
Dactivations_test.cc290 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
312 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
334 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
356 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h130 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc908 int elements_count = shape.b * shape.w * shape.h * shape.d * aligned_channels; in UploadData() local