Searched refs:NumInputsWithValues (Results 1 – 6 of 6) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | LSHProjection.cpp | 42 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare() 59 NN_CHECK(NumInputsWithValues(operation, operands) == 3); in Prepare() 64 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 4); in Prepare()
|
D | Multinomial.cpp | 59 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 3); in Prepare()
|
D | SVDF.cpp | 51 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare()
|
D | RNN.cpp | 50 const int num_inputs = NumInputsWithValues(operation, operands); in Prepare()
|
D | LSTM.cpp | 298 NN_CHECK(NumInputsWithValues(operation, operands) >= 15 && in Prepare() 299 NumInputsWithValues(operation, operands) <= 27); in Prepare()
|
/frameworks/ml/nn/common/include/ |
D | CpuExecutor.h | 240 inline int NumInputsWithValues(const Operation &operation, in NumInputsWithValues() function
|