Searched defs:elements_count (Results 1 – 14 of 14) sorted by relevance
57 def _GenerateLoadAggregateStore(emitter, registers, lanes_count, elements_count, argument206 elements_count, aggregators, input_address, argument
153 const int elements_count = kernel_x * kernel_y * dst_slices; in UploadWeightsForDWConv2D() local225 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeightsForDWConv3D() local
86 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
79 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
162 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
91 const int elements_count = size_ / sizeof(T); in ReadData() local
110 const int elements_count = size_ / sizeof(T); in ReadData() local
144 const int elements_count = src_depth * dst_depth * 4; in UploadWeights() local
430 uint32_t elements_count; in DeserializeElementsData() local
292 size_t elements_count = 0; in getOutputError() local
606 dex::u4 elements_count = dex::ReadULeb128(pptr); in ParseAnnotation() local
193 const int elements_count = in BufferToKernelLanguage() local
1313 int elements_count = in GetSizeInBytesForShape() local
655 const int64_t elements_count, in radix_sort_parallel()