Searched refs:num_input (Results 1 – 6 of 6) 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
81 int weights_dim2 = num_input + num_output; in Run()
105 num_input); in Run()
109 num_input); in Run()
113 num_cell * 2, num_input); in Run()
117 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/webrtc/modules/audio_processing/test/ |
| D | audio_processing_unittest.cc | 2500 int num_input; in TEST_P() member
2516 reverse_output_rate_, cf[i].num_input, cf[i].num_output, in TEST_P()
2550 reverse_output_rate_, cf[i].num_input, in TEST_P()
2619 << cf[i].num_input << ", " << cf[i].num_output << ", " in TEST_P()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1266 int num_input; in HybridConv() local
1278 num_input = im2col_shape.FlatSize(); in HybridConv()
1282 num_input = input_shape.FlatSize(); in HybridConv()
1297 const int gemm_input_rows = num_input / gemm_input_cols; in HybridConv()
1352 int num_input; in HybridConvPerChannel() local
1361 num_input = im2col_shape.FlatSize(); in HybridConvPerChannel()
1365 num_input = input_shape.FlatSize(); in HybridConvPerChannel()
1372 const int gemm_input_rows = num_input / gemm_input_cols; in HybridConvPerChannel()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2066 const int num_input = TrtTensorDimsNumElements(GetTestDims(dims_array)); in TestConvertBiasAdd() local
2068 num_input); in TestConvertBiasAdd()
2071 {"input", ConstructTensor<CType>(num_input, CType(0))}}; in TestConvertBiasAdd()
2072 DataVec output_data{{"my_biasadd", ConstructTensor<CType>(num_input)}}; in TestConvertBiasAdd()
|