Searched refs:num_inputs (Results 1 – 5 of 5) sorted by relevance
52 int num_inputs = inputShapes.size(); in concatenation() local53 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs); in concatenation()54 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation()55 for (int i = 0; i < num_inputs; i++) { in concatenation()62 inputDimsPtr.data(), num_inputs, outputData, convertShapeToDims(outputShape)); in concatenation()72 int num_inputs = inputShapes.size(); in concatenation() local73 std::vector<float> inputScales(num_inputs); in concatenation()74 std::vector<int32> inputOffsets(num_inputs); in concatenation()75 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs); in concatenation()76 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation()[all …]
184 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LSTMOpModel360 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()362 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()363 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()457 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()459 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()460 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()985 const int input_sequence_size = sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST()987 float* batch0_start = lstm_input[0] + i * lstm.num_inputs(); in TEST()988 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()[all …]
55 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() local57 NN_CHECK(num_inputs == 6 || num_inputs == 7); in Prepare()
53 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() local54 NN_CHECK(num_inputs == 6); in Prepare()
194 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LayerNormLSTMOpModel