Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceRNN.cpp33 constexpr uint32_t kTimeMajorParam = 6; variable
71 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped()
138 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare()
DBidirectionalSequenceRNN.cpp44 constexpr uint32_t kTimeMajorParam = 13; variable
110 int32_t timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped()
284 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
314 int32_t timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare()
DUnidirectionalSequenceLSTM.cpp71 constexpr uint32_t kTimeMajorParam = 23; variable
91 return context->getInputValue<bool>(kTimeMajorParam); in isTimeMajor()
177 kTimeMajorParam, in prepare()
DBidirectionalSequenceLSTM.h134 static constexpr int kTimeMajorParam = 52; variable
DBidirectionalSequenceLSTM.cpp151 params_.time_major = getScalarData<bool>(*GetInput(operation, operands, kTimeMajorParam)); in BidirectionalSequenceLSTM()