Home
last modified time | relevance | path

Searched refs:num_output (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dscan_dataset_serialization_test.py40 num_output = 5
41 self.run_core_tests(lambda: self._build_dataset(num_output), num_output)
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_lstm_split_inputs.cc74 int num_output = num_cell; in Run() local
110 base_name + "weight_i_r", num_cell, num_output, kernel, 0, num_input); in Run()
113 base_name + "weight_c_r", num_cell, num_output, kernel, in Run()
117 base_name + "weight_f_r", num_cell, num_output, kernel, in Run()
121 base_name + "weight_o_r", num_cell, num_output, kernel, in Run()
Didentify_lstm_merge_inputs.cc66 int num_output = in Run() local
72 CHECK_EQ(num_cell, num_output); in Run()
82 int weights_dim2 = num_input + num_output; in Run()
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dtake_test.py40 num_output = min(count, 10) if count != -1 else 10
42 dataset, [tuple(components[0][i:i + 1]) for i in range(num_output)])
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dstats_dataset_ops_test.py391 num_output = 100 // 16 + 1
394 num_output,
419 num_output = total_records // batch_size
421 num_output = total_records // batch_size + 1
425 num_output,
434 for _ in range(num_output):
Dstats_dataset_test_base.py277 num_output, argument
285 for i in range(num_output):
305 float(num_output),
/external/libaom/libaom/av1/encoder/
Dml.c18 void av1_nn_output_prec_reduce(float *const output, int num_output) { in av1_nn_output_prec_reduce() argument
22 for (int i = 0; i < num_output; i++) { in av1_nn_output_prec_reduce()
Dml.h76 void av1_nn_output_prec_reduce(float *const output, int num_output);
/external/webrtc/modules/audio_processing/
Daudio_processing_unittest.cc2043 int num_output; in TEST_P() member
2054 reverse_output_rate_, cf[i].num_input, cf[i].num_output, in TEST_P()
2065 file_direction ? cf[i].num_reverse_output : cf[i].num_output; in TEST_P()
2085 cf[i].num_output, cf[i].num_reverse_input, in TEST_P()
2092 cf[i].num_output, cf[i].num_output, in TEST_P()
2155 << cf[i].num_input << ", " << cf[i].num_output << ", " in TEST_P()
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc394 const int num_output = op_kernel_context->num_outputs(); in SplitInputTask() local
395 for (int i = 0; i < num_output; ++i) { in SplitInputTask()