Searched refs:num_inputs (Results 1 - 6 of 6) sorted by relevance

/frameworks/ml/nn/common/operations/
H A DConcatenation.cpp28 int num_inputs = inputShapes.size(); local
29 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs);
30 std::vector<tflite::Dims<4> > inputDims(num_inputs);
31 for (int i=0; i<num_inputs; i++) {
38 inputDataPtrs.data(), inputDimsPtr.data(), num_inputs, local
47 int num_inputs = inputShapes.size(); local
48 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs);
49 std::vector<tflite::Dims<4> > inputDims(num_inputs);
50 for (int i=0; i<num_inputs; i++) {
57 inputDataPtrs.data(), inputDimsPtr.data(), num_inputs, local
[all...]
H A DLSTMTest.cpp180 uint32_t num_inputs() const { return n_input_; } function in class:android::nn::wrapper::LSTMOpModel
366 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs());
368 float* batch0_start = lstm_input + i * lstm.num_inputs();
369 float* batch0_end = batch0_start + lstm.num_inputs();
472 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs());
474 float* batch0_start = lstm_input + i * lstm.num_inputs();
475 float* batch0_end = batch0_start + lstm.num_inputs();
1086 sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs());
1088 float* batch0_start = lstm_input[0] + i * lstm.num_inputs();
1089 float* batch0_end = batch0_start + lstm.num_inputs();
[all...]
H A DLSHProjection.cpp41 const int num_inputs = NumInputsWithValues(operation, operands); local
42 NN_CHECK(num_inputs == 3 || num_inputs == 4);
H A DRNN.cpp45 const int num_inputs = NumInputsWithValues(operation, operands); local
46 NN_CHECK(num_inputs == 5 || num_inputs == 6);
H A DSVDF.cpp60 const int num_inputs = NumInputsWithValues(operation, operands); local
62 NN_CHECK(num_inputs == 6 || num_inputs == 7);
/frameworks/ml/nn/common/
H A DOperationsUtils.cpp375 int num_inputs = inputShapes.size();
383 for (int i = 1; i < num_inputs; ++i) {

Completed in 108 milliseconds