Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp102 const uint32_t numCells = getSizeOfDimension(inputToOutputShape, 0); in prepare() local
106 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells); in prepare()
112 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numCells); in prepare()
118 NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numCells); in prepare()
122 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numCells); in prepare()
128 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numCells); in prepare()
134 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numCells); in prepare()
138 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numCells); in prepare()
152 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numCells); in prepare()
158 NN_RET_CHECK_EQ(getSizeOfDimension(cellToForgetShape, 0), numCells); in prepare()
[all …]
DLSTM.cpp437 const uint32_t numCells = getSizeOfDimension(input_to_output_weights_shape, 0); in LSTMEvalFloat32() local
472 cell_state_in_buffer + batchSize * numCells); in LSTMEvalFloat32()
509 cell_state_out_buffer + batchSize * numCells); in LSTMEvalFloat32()
557 const uint32_t numCells = getSizeOfDimension(input_to_output_weights_shape, 0); in LSTMEvalFloat16() local
567 std::vector<float> input_to_input_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
571 std::vector<float> input_to_forget_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
573 std::vector<float> input_to_cell_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
575 std::vector<float> input_to_output_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
578 std::vector<float> recurrent_to_input_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
583 std::vector<float> recurrent_to_forget_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
[all …]
/packages/apps/Launcher3/src/com/android/launcher3/responsive/
DResponsiveSpecsProvider.kt74 numCells: Int, in <lambda>()
79 return CalculatedResponsiveSpec(aspectRatio, availableSpace, numCells, spec) in <lambda>()
99 numCells: Int, in <lambda>()
120 numCells, in <lambda>()