Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceRNN.cpp71 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped() local
77 if (!timeMajor) { in executeTyped()
107 if (!timeMajor) { in executeTyped()
138 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare() local
139 NN_RET_CHECK(timeMajor == 0 || timeMajor == 1); in prepare()
141 timeMajor ? getSizeOfDimension(input, 1) : getSizeOfDimension(input, 0); in prepare()
143 timeMajor ? getSizeOfDimension(input, 0) : getSizeOfDimension(input, 1); in prepare()
161 output.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
162 output.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()
DBidirectionalSequenceRNN.cpp110 int32_t timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped() local
129 if (!timeMajor) { in executeTyped()
236 if (!timeMajor) { in executeTyped()
314 int32_t timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare() local
316 timeMajor ? getSizeOfDimension(input, 1) : getSizeOfDimension(input, 0); in prepare()
318 timeMajor ? getSizeOfDimension(input, 0) : getSizeOfDimension(input, 1); in prepare()
362 fwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
363 fwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()
369 bwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
370 bwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()
DLSTM.cpp383 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat32() argument
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()
408 if (!timeMajor) { in LSTMEvalFloat32()
421 const float* inputData = timeMajor ? input_buffer : transposedInput.data(); in LSTMEvalFloat32()
423 hasAuxInput ? (timeMajor ? aux_input_buffer : transposedAuxInput.data()) : nullptr; in LSTMEvalFloat32()
424 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()
469 if (!timeMajor) { in LSTMEvalFloat32()
503 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat16()
510 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat16()
[all …]
DLSTM.h130 bool timeMajor = true, bool forwardSequence = true);
158 _Float16* scratch_buffer_buffer, bool timeMajor = true, bool forwardSequence = true);
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp1950 ANeuralNetworksOperandType timeMajor = boolScalar; in lstmBidirectionalSequence() local
2017 timeMajor, in lstmBidirectionalSequence()
2983 ANeuralNetworksOperandType timeMajor = boolScalar; in bidirectionlSequenceRNNTest() local
2989 fwWeights, bwWeights, activation, timeMajor, mergeOutputs}, in bidirectionlSequenceRNNTest()
3051 ANeuralNetworksOperandType timeMajor = intScalar; in unidirectionlSequenceRNNTest() local
3055 {input, weights, recurrentWeights, bias, hiddenState, activation, timeMajor}, {output}); in unidirectionlSequenceRNNTest()
3166 ANeuralNetworksOperandType timeMajor = boolScalar; in unidirectionalSequenceLSTMTest() local
3206 timeMajor, in unidirectionalSequenceLSTMTest()