Searched refs:num_inputs (Results 1 – 6 of 6) sorted by relevance
56 int num_inputs = inputShapes.size(); in concatenation() local57 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs); in concatenation()58 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation()59 for (int i = 0; i < num_inputs; i++) { in concatenation()66 inputDimsPtr.data(), num_inputs, outputData, convertShapeToDims(outputShape)); in concatenation()76 int num_inputs = inputShapes.size(); in concatenation() local77 std::vector<float> inputScales(num_inputs); in concatenation()78 std::vector<int32> inputOffsets(num_inputs); in concatenation()79 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs); in concatenation()80 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation()[all …]
183 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LSTMOpModel359 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()361 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()362 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()456 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()458 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()459 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()984 const int input_sequence_size = sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST()986 float* batch0_start = lstm_input[0] + i * lstm.num_inputs(); in TEST()987 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()
49 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() local50 NN_CHECK(num_inputs == 6); in Prepare()
195 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LayerNormLSTMOpModel
496 num_inputs -> { in setVolumeControlStateObservers()498 if (num_inputs != 0) in setVolumeControlStateObservers()500 IntStream.rangeClosed(1, num_inputs) in setVolumeControlStateObservers()