Searched refs:num_inputs (Results 1 – 6 of 6) sorted by relevance
41 int num_inputs = inputShapes.size(); in concatenation() local42 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs); in concatenation()43 std::vector<tflite::Dims<4> > inputDims(num_inputs); in concatenation()44 for (int i = 0; i < num_inputs; i++) { in concatenation()51 inputDimsPtr.data(), num_inputs, outputData, convertShapeToDims(outputShape)); in concatenation()61 int num_inputs = inputShapes.size(); in concatenation() local62 std::vector<float> inputScales(num_inputs); in concatenation()63 std::vector<int32> inputOffsets(num_inputs); in concatenation()64 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs); in concatenation()65 std::vector<tflite::Dims<4> > inputDims(num_inputs); in concatenation()[all …]
180 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LSTMOpModel366 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()368 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()369 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()472 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()474 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()475 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()1086 sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST()1088 float* batch0_start = lstm_input[0] + i * lstm.num_inputs(); in TEST()1089 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()[all …]
42 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() local43 NN_CHECK(num_inputs == 3 || num_inputs == 4); in Prepare()
50 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() local51 NN_CHECK(num_inputs == 5 || num_inputs == 6); in Prepare()
51 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() local53 NN_CHECK(num_inputs == 6 || num_inputs == 7); in Prepare()
195 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LayerNormLSTMOpModel