Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp209 Shape* fwOutputShape, Shape* bwOutputShape, in Prepare() argument
383 fwOutputShape->type = inputShape.type; in Prepare()
384 fwOutputShape->offset = inputShape.offset; in Prepare()
385 fwOutputShape->scale = inputShape.scale; in Prepare()
386 fwOutputShape->dimensions.resize(3); in Prepare()
387 fwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
388 fwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
389 fwOutputShape->dimensions[2] = params_.merge_outputs ? n_fw_output + n_bw_output : n_fw_output; in Prepare()
DBidirectionalSequenceRNN.cpp134 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped() local
155 fwOutputTransposed.resize(getNumberOfElements(fwOutputShape)); in executeTyped()
182 std::swap(fwOutputShape.dimensions[0], fwOutputShape.dimensions[1]); in executeTyped()
276 transposeFirstTwoDims(fwOutputTransposed.data(), fwOutputShape, in executeTyped()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceLSTM.h37 bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* fwOutputShape,
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1004 Shape fwOutputShape, bwOutputShape, fwOutputActivationStateShape, in executeOperation() local
1008 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1011 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()