Searched refs:kTimeMajorParam (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceRNN.cpp | 38 constexpr uint32_t kTimeMajorParam = 6; variable
80 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped()
160 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare()
|
| D | BidirectionalSequenceRNN.cpp | 49 constexpr uint32_t kTimeMajorParam = 13; variable
161 const bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped()
361 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
385 bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 73 constexpr uint32_t kTimeMajorParam = 23; variable
98 return context->getInputValue<bool>(kTimeMajorParam); in isTimeMajor()
190 kTimeMajorParam, in prepare()
|
| D | BidirectionalSequenceLSTM.h | 136 static constexpr int kTimeMajorParam = 52; variable
|
| D | BidirectionalSequenceLSTM.cpp | 188 const auto& timeMajorOperand = *GetInput(operation, operands, kTimeMajorParam); in BidirectionalSequenceLSTM()
242 kTimeMajorParam, in Prepare()
263 const auto& timeMajorOperand = *GetInput(operation, operands, kTimeMajorParam); in Prepare()
|