Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dbuffer.h91 const int elements_count = size_ / sizeof(T); in ReadData() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dbuffer.h110 const int elements_count = size_ / sizeof(T); in ReadData() 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/delegates/gpu/common/tasks/
Ddepthwise_conv.h153 const int elements_count = kernel_x * kernel_y * dst_slices; in UploadWeightsForDWConv2D() local
225 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeightsForDWConv3D() local
Ddepthwise_conv_3x3.h86 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
Ddepthwise_conv_3x3_stride_h2.h79 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
Dfully_connected.h162 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h144 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Darguments.cc193 const int elements_count = in BufferToKernelLanguage() local
Dtensor_desc.cc1313 int elements_count = in GetSizeInBytesForShape() local
/external/pytorch/aten/src/ATen/native/cpu/
DScatterGatherKernel.cpp655 const int64_t elements_count, in radix_sort_parallel()