Searched refs:bwOutputShape (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 136 Shape bwOutputShape; in executeTyped() local
138 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
157 bwOutputTransposed.resize(getNumberOfElements(bwOutputShape)); in executeTyped()
184 std::swap(bwOutputShape.dimensions[0], bwOutputShape.dimensions[1]); in executeTyped()
279 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
|
| D | BidirectionalSequenceLSTM.cpp | 209 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()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | BidirectionalSequenceLSTM.h | 38 Shape* bwOutputShape, Shape* fwOutputActivationState, Shape* fwOutputCellState,
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1004 Shape fwOutputShape, bwOutputShape, fwOutputActivationStateShape, in executeOperation() local
1008 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1015 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()
|