Home
last modified time | relevance | path

Searched refs:bwOutputShape (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DBidirectionalSequenceLSTM.cpp209 Shape* fwOutputShape, Shape* bwOutputShape, in Prepare() argument
409 bwOutputShape->type = inputShape.type; in Prepare()
410 bwOutputShape->offset = inputShape.offset; in Prepare()
411 bwOutputShape->scale = inputShape.scale; in Prepare()
412 bwOutputShape->dimensions.resize(3); in Prepare()
413 bwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
414 bwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
415 bwOutputShape->dimensions[2] = n_bw_output; in Prepare()
DBidirectionalSequenceRNN.cpp165 Shape bwOutputShape; in executeTyped() local
167 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
186 bwOutputTransposed.resize(getNumberOfElements(bwOutputShape)); in executeTyped()
213 std::swap(bwOutputShape.dimensions[0], bwOutputShape.dimensions[1]); in executeTyped()
308 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
DBidirectionalSequenceLSTM.h38 Shape* bwOutputShape, Shape* fwOutputActivationState, Shape* fwOutputCellState,
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1003 Shape fwOutputShape, bwOutputShape, fwOutputActivationStateShape, in executeOperation() local
1007 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1014 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()