Searched refs:num_input (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_lstm_split_inputs.cc | 65 int num_input = model->GetArray(curr_op->inputs[LstmCellOperator::DATA_INPUT]) in Run() local
96 base_name + "weight_i_i", num_cell, num_input, kernel, 0, 0); in Run()
98 base_name + "weight_c_i", num_cell, num_input, kernel, in Run()
102 base_name + "weight_f_i", num_cell, num_input, kernel, num_cell * 2, 0); in Run()
105 base_name + "weight_o_i", num_cell, num_input, kernel, num_cell * 3, 0); in Run()
110 base_name + "weight_i_r", num_cell, num_output, kernel, 0, num_input); in Run()
114 num_cell, num_input); in Run()
118 num_cell * 2, num_input); in Run()
122 num_cell * 3, num_input); in Run()
|
| D | identify_lstm_merge_inputs.cc | 63 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run() local
82 int weights_dim2 = num_input + num_output; in Run()
106 num_input); in Run()
110 num_input); in Run()
114 num_cell * 2, num_input); in Run()
118 num_cell * 3, num_input); in Run()
|
| /external/mesa3d/src/gallium/auxiliary/tgsi/ |
| D | tgsi_aa_point.c | 48 unsigned num_input; // number of inputs member
73 ts->num_input++; in aa_decl()
115 texIn = ts->num_input++; in aa_prolog()
303 transform.num_input = 0; in tgsi_add_aa_point()
|
| /external/webrtc/modules/audio_processing/ |
| D | audio_processing_unittest.cc | 2042 int num_input; in TEST_P() member
2054 reverse_output_rate_, cf[i].num_input, cf[i].num_output, in TEST_P()
2084 reverse_output_rate_, cf[i].num_input, in TEST_P()
2155 << cf[i].num_input << ", " << cf[i].num_output << ", " in TEST_P()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1909 int num_input = n->num_inputs(); in MakeIdentityNodesForArgsToLift() local
1910 for (int arg_index = 0; arg_index < num_input; ++arg_index) { in MakeIdentityNodesForArgsToLift()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1359 int num_input; in HybridConv() local
1369 num_input = im2col_shape.FlatSize(); in HybridConv()
1377 num_input = im2col_shape.FlatSize(); in HybridConv()
1381 num_input = input_shape.FlatSize(); in HybridConv()
1396 const int gemm_input_rows = num_input / gemm_input_cols; in HybridConv()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2736 const int num_input = TrtTensorDimsNumElements(GetTestDims(dims_array)); in TEST_P() local
2738 num_input); in TEST_P()
2739 std::vector<float> input_data(num_input, 0); in TEST_P()
|