Searched refs:fwOutputShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 210 Shape* fwOutputShape, Shape* bwOutputShape, in Prepare() argument
384 fwOutputShape->type = inputShape.type; in Prepare()
385 fwOutputShape->offset = inputShape.offset; in Prepare()
386 fwOutputShape->scale = inputShape.scale; in Prepare()
387 fwOutputShape->dimensions.resize(3); in Prepare()
388 fwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
389 fwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
390 fwOutputShape->dimensions[2] = params_.merge_outputs ? n_fw_output + n_bw_output : n_fw_output; in Prepare()
|
| D | BidirectionalSequenceRNN.cpp | 165 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped() local
186 fwOutputTransposed.resize(getNumberOfElements(fwOutputShape)); in executeTyped()
213 std::swap(fwOutputShape.dimensions[0], fwOutputShape.dimensions[1]); in executeTyped()
307 transposeFirstTwoDims(fwOutputTransposed.data(), fwOutputShape, in executeTyped()
|
| D | BidirectionalSequenceLSTM.h | 40 Shape* fwOutputShape, Shape* bwOutputShape, Shape* fwOutputActivationState,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1059 Shape fwOutputShape, bwOutputShape, fwOutputActivationStateShape, in executeOperation() local
1063 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1066 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()
|