Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DReduce.cpp50 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute() local
56 reinterpret_cast<const int32_t*>(inputShape.dimensions.data()), inputRank, in compute()
107 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
109 std::vector<bool> shouldReduce(inputRank); in prepare()
116 NN_RET_CHECK(handleNegativeAxis(inputRank, &axis)); in prepare()
124 for (uint32_t axis = 0; axis < inputRank; ++axis) { in prepare()
DLSTM.cpp386 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32() local
387 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat32()
390 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
391 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
393 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat32()
506 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat16() local
507 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat16()
510 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat16()
511 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat16()
513 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat16()
DUnidirectionalSequenceLSTM.cpp185 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
186 NN_RET_CHECK_EQ(inputRank, 3) << "Invalid input tensor rank: " << inputRank; in prepare()
190 const uint32_t inputSize = getSizeOfDimension(inputShape, inputRank - 1); in prepare()