Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dbuffer.h92 const int elements_count = size_ / sizeof(T); in ReadData() local
Dtexture2d.h110 const int elements_count = width_ * height_ * 4; in ReadData() local
Dtensor.cc237 const int elements_count = in WriteDataBHWDC() local
286 const int elements_count = in ReadDataBHWDC() local
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task_descriptor.cc54 RuntimeOptions::Precision destination_type, size_t elements_count) { in GetByteBufferConvertedResized()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconv_buffer.h91 const int elements_count = in UploadWeights() local
Dconv_buffer_1x1.h115 const int elements_count = in UploadWeights() local
Dconvolution_transposed_thin.h86 const int elements_count = in UploadWeights() local
Ddepth_wise_conv_3x3.h89 const int elements_count = texture_width * texture_height; in UploadWeightsAndBiases() local
Ddepth_wise_conv.h94 const int elements_count = kernel_x * kernel_y * dst_depth; in UploadWeights() local
Ddepth_wise_conv_3d.h96 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeights() local
Dconv_powervr.h148 const int elements_count = in UploadWeights() local
Dconvolution_transposed.h101 const int elements_count = kernel_x * kernel_y * src_depth * dst_depth * 4; in UploadWeights() local
Dconv_texture.h126 const int elements_count = texture_width * texture_height; in UploadWeights() local
Dconvolution_transposed_3d.h102 const int elements_count = in UploadWeights() local
Dconv_3d.h158 const int elements_count = 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.cc435 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
457 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
479 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
501 const int elements_count = 10; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtranspose_conv.cc282 uint elements_count; member