Home
last modified time | relevance | path

Searched refs:input_n_elements (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DFullyConnected.cpp67 uint32_t input_n_elements = getNumberOfElements(inputShape); in fullyConnectedFloat32() local
68 if (batch_size * batch_size == input_n_elements) { in fullyConnectedFloat32()
202 uint32_t input_n_elements = getNumberOfElements(input); in validateShapes() local
206 uint32_t batch_size = input_size == 0 ? 0 : input_n_elements / input_size; in validateShapes()
208 NN_RET_CHECK_EQ(input_size * batch_size, input_n_elements); in validateShapes()