Home
last modified time | relevance | path

Searched refs:num_output_values (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcount_ops.cc154 int num_output_values = GetOutputSize(max_value, maxlength_, minlength_); in Compute() local
155 OP_REQUIRES_OK(context, OutputSparse<W>(per_batch_counts, num_output_values, in Compute()
230 int num_output_values = GetOutputSize(max_value, maxlength_, minlength_); in Compute() local
231 OP_REQUIRES_OK(context, OutputSparse<W>(per_batch_counts, num_output_values, in Compute()
308 int num_output_values = GetOutputSize(max_value, maxlength_, minlength_); in Compute() local
309 OP_REQUIRES_OK(context, OutputSparse<W>(per_batch_counts, num_output_values, in Compute()
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h658 const int num_output_values = output_depth * num_output_pixels;
663 for (; i <= num_output_values - 16; i += 16) {
689 for (; i <= num_output_values - 4; i += 4) {
704 for (; i < num_output_values; i++) {
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h205 const int num_output_values = output_depth * num_output_pixels; in DepthwiseConvHybridGeneral() local
249 output_ptr += num_output_values; in DepthwiseConvHybridGeneral()
Ddepthwise_conv.h1777 const int num_output_values = output_depth * num_output_pixels;
1780 num_output_values, output_offset,
1784 output_ptr += num_output_values;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_float.h1066 const int num_output_values = output_depth * num_output_pixels;
1071 for (; i <= num_output_values - 16; i += 16) {
1087 for (; i <= num_output_values - 4; i += 4) {
1098 for (; i < num_output_values; i++) {
Ddepthwiseconv_uint8.h1845 const int num_output_values = output_depth * num_output_pixels;
1858 for (; i <= num_output_values - 16; i += 16) {
1906 for (; i <= num_output_values - 8; i += 8) {
1945 for (; i <= num_output_values - 4; i += 4) {
1975 for (; i < num_output_values; i++) {