Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/src/
DFullyConnected.cpp42 uint32_t input_n_elements = getNumberOfElements(input); in validateShapes() local
48 NN_RET_CHECK_EQ(input_n_elements % input_size, 0u); in validateShapes()
49 batch_size = input_n_elements / input_size; in validateShapes()
/packages/modules/NeuralNetworks/common/cpu_operations/
DFullyConnected.cpp63 uint32_t input_n_elements = getNumberOfElements(inputShape); in fullyConnectedFloat32() local
64 if (batch_size * batch_size == input_n_elements) { in fullyConnectedFloat32()