pytorch  1.8.2
About: PyTorch provides Tensor computation (like NumPy) with strong GPU acceleration and Deep Neural Networks (in Python) built on a tape-based autograd system. LTS (Long Term Support) release.
  Fossies Dox: pytorch-1.8.2.tar.gz  ("unofficial" and yet experimental doxygen-generated source code documentation)  
Namespaces | Functions
inference_lstm_op.h File Reference
#include <algorithm>
#include <sstream>
#include <unordered_map>
#include <vector>
#include "caffe2/core/blob_serialization.h"
#include "caffe2/core/export_caffe2_op_to_c10.h"
#include "caffe2/core/operator.h"
#include "caffe2/core/tensor.h"
#include "caffe2/utils/eigen_utils.h"
#include "caffe2/utils/math.h"
#include "lstm_utils.h"
Include dependency graph for inference_lstm_op.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  caffe2
 Copyright (c) 2016-present, Facebook, Inc.
 

Functions

 C10_DECLARE_EXPORT_CAFFE2_OP_TO_C10 (LSTMOp)
 

Function Documentation

◆ C10_DECLARE_EXPORT_CAFFE2_OP_TO_C10()

C10_DECLARE_EXPORT_CAFFE2_OP_TO_C10 ( LSTMOp  )

Variable Documentation

◆ b_hh

Tensor b_hh

Definition at line 58 of file inference_lstm_op.h.

Referenced by at::native::gru_cell(), at::native::lstm_cell(), at::native::rnn_relu_cell(), and at::native::rnn_tanh_cell().

◆ b_ih

Tensor b_ih

Definition at line 57 of file inference_lstm_op.h.

Referenced by at::native::gru_cell(), at::native::lstm_cell(), at::native::rnn_relu_cell(), and at::native::rnn_tanh_cell().

◆ batch_first_

bool batch_first_
protected

Definition at line 305 of file inference_lstm_op.h.

◆ bidirectional_

bool bidirectional_
protected

Definition at line 303 of file inference_lstm_op.h.

◆ cell_

LSTMCell cell_

Definition at line 145 of file inference_lstm_op.h.

◆ context

CPUContext* context

Definition at line 59 of file inference_lstm_op.h.

Referenced by caffe2::AddConstInput(), caffe2::AddInput(), caffe2::AddLinSpacedInput(), caffe2::AddNoiseInput(), caffe2::OnnxifiOp< Context >::adjustOutputBatchSizes(), at::native::vulkan::api::available(), caffe2::math::AxpyFixedSize(), BM_CUDAContextCreation(), BM_CUDAContextStreamAccess(), cl::Buffer::Buffer(), cl::BufferGL::BufferGL(), cl::BufferRenderGL::BufferRenderGL(), caffe2::BuildInitializationList(), cl::CommandQueue::CommandQueue(), caffe2::RowWiseSparseAdagradFusedWithSparseLengthsWeightedSumGradientOp< Tdata, T, TLengths, rowWiseAdagradT >::compute(), at::native::vulkan::api::context(), caffe2::OpWrapper< OpType, T >::DequantizeInput(), at::native::vulkan::api::dispatch_prologue(), caffe2::EventRecordCUDA(), caffe2::EventWaitCUDACUDA(), caffe2::executeGpuBinaryOpTest(), caffe2::python::TensorFeeder< Context >::FeedTensor(), caffe2::python::Int8TensorFetcher::Fetch(), caffe2::emulator::fill_with_type(), torch::distributed::rpc::RRefContext::getInstance(), torch::distributed::autograd::DistAutogradContainer::getOrCreateContext(), cl::Image1D::Image1D(), cl::Image1DArray::Image1DArray(), cl::Image1DBuffer::Image1DBuffer(), cl::Image2D::Image2D(), cl::Image2DArray::Image2DArray(), cl::Image3D::Image3D(), cl::ImageGL::ImageGL(), torch::jit::loadModule(), THMapAllocator::makeDataPtr(), THRefcountedMapAllocator::makeDataPtr(), THCIpcDeleter::makeDataPtr(), torch::distributed::autograd::DistAutogradContainer::newContext(), caffe2::opt::OptimizeForBackend(), cl::Program::Program(), caffe2::Event::Record(), caffe2::gloo::CreateCommonWorld< Context >::rendezvousWithMPI(), caffe2::gloo::CreateCommonWorld< Context >::rendezvousWithStore(), at::native::vulkan::Conv2dOpContext::run(), caffe2::gloo::BarrierOp< Context >::RunOnDevice(), caffe2::gloo::CreateCommonWorld< Context >::RunOnDevice(), cl::Sampler::Sampler(), caffe2::math::ScaleFixedSize(), caffe2::OnnxifiTransformer::SubnetToOnnxifiOpViaOnnx(), caffe2::TensorCPUFromValues(), caffe2::TEST(), cl::UserEvent::UserEvent(), and caffe2::Event::Wait().

◆ context_

CPUContext* context_

Definition at line 93 of file inference_lstm_op.h.

Referenced by caffe2::YellowFinOp< T, Context >::AfterApply(), caffe2::ConvPoolOpBase< Context >::AllocateAndCopy(), caffe2::PartitionOpBase::ApplyPartition(), caffe2::ImageInputOp< Context >::ApplyTransformOnGPU(), caffe2::ATenOp< Context >::assignToValue(), caffe2::blob(), caffe2::BucketizeOp< Context >::BucketizeOp(), compute_1d_tiled(), compute_2d(), compute_2d_tiled(), compute_3d_tiled(), compute_4d_tiled(), caffe2::BatchMomentsGradientOp< T, Context >::ComputeBatchMomentsGradientNCHW(), caffe2::BatchMomentsGradientOp< T, Context >::ComputeBatchMomentsGradientNHWC(), caffe2::BatchMomentsOp< T, Context >::ComputeBatchMomentsNCHW(), caffe2::LayerNormGradientOp< Context >::ComputeInternalGradients(), caffe2::InstanceNormGradientOp< T, Context >::ComputeMoments(), caffe2::SpatialBNGradientOp< Context >::ComputeMultiBatchScaleBiasGradientsAndFusedParams(), caffe2::LarsOp< T, Context >::ComputeNorms(), caffe2::GroupNormDNNLowPOp< T >::ComputeQuantizedInvStd(), caffe2::SpatialBNOp< Context >::ComputeRunningMomentsAndFusedParam(), caffe2::SpatialBNGradientOp< Context >::ComputeScaleBiasGradientsAndFusedParams(), caffe2::LayerNormOp< Context >::ComputeSigmaAndFusedParams(), c10::Error::context(), caffe2::ImageInputOp< Context >::CopyPrefetched(), caffe2::VideoInputOp< Context >::CopyPrefetched(), caffe2::EnsureCPUOutputOp< Context >::CopyWithContext(), caffe2::GroupNormDNNLowPOp< T >::DequantizedGroupMomentsNCHW(), caffe2::GroupNormDNNLowPOp< T >::DequantizedGroupMomentsNHWC(), caffe2::SafeDequeueBlobsOp< Context >::dequeueMany(), caffe2::SparseFtrlOp< T >::DoRun(), caffe2::SumFP16FP16AccOp< Context >::DoRunWithFloat(), caffe2::ReversePackedSegsOp< Context >::DoRunWithLengthType(), caffe2::BatchMatMulFP16FakeOp< Context, Engine, USE_ACC_FP16, USE_TMP_ACCUMULATOR, USE_CUSTOM_ACC32 >::DoRunWithType(), caffe2::Fp16FCAccOp< Context, Engine, USE_ACC_FP16, USE_TMP_ACCUMULATOR, ADD_BIAS_FIRST >::DoRunWithType(), caffe2::ArgOp< Context, Reducer >::DoRunWithType(), caffe2::BatchMatMulOp< Context, Engine >::DoRunWithType(), caffe2::CudnnConvGradientOp::DoRunWithType(), caffe2::BinaryElementwiseWithArgsGradientOp< NumericTypes, CPUContext, BinaryFunctorWithDefaultCtor< DivFunctor< CPUContext > >, SameTypeAsInput, SameTypeAsInput >::DoRunWithType(), caffe2::WhereOp< Context >::DoRunWithType(), caffe2::SumReduceLikeOp< Context >::DoRunWithType(), caffe2::UnaryElementwiseWithArgsOp< InputTypes, Context, Functor, OutputTypeMap >::DoRunWithType(), caffe2::BinaryElementwiseWithArgsOp< InputTypes, Context, Functor, OutputTypeMap >::DoRunWithType(), caffe2::BinaryElementwiseWithArgsGradientOp< InputTypes, Context, Functor, OutputTypeMap, GradientTypeMap >::DoRunWithType(), caffe2::ExpandOp< InputTypes, Context >::DoRunWithType(), caffe2::ExpandGradientOp< InputTypes, Context >::DoRunWithType(), caffe2::MergeSingleListFeatureTensorsOp< Context >::DoRunWithType(), caffe2::MergeSingleListOrMapFeatureTensorsGradientOp< Context >::DoRunWithType(), caffe2::MergeMultiScalarFeatureTensorsGradientOp< Context >::DoRunWithType(), caffe2::MergeMultiListFeatureTensorsOp< Context >::DoRunWithType(), caffe2::MergeMultiListOrMapFeatureTensorsGradientOp< Context >::DoRunWithType(), caffe2::FullyConnectedOp< Context, Engine, TransposeWeight >::DoRunWithType(), caffe2::FullyConnectedGradientOp< Context, Engine, TransposeWeight >::DoRunWithType(), caffe2::GatherRangesToDenseOp< Context >::DoRunWithType(), caffe2::HistogramOp< Context >::DoRunWithType(), caffe2::LayerNormOp< Context >::DoRunWithType(), caffe2::LayerNormGradientOp< Context >::DoRunWithType(), caffe2::LengthsPadOp< Context >::DoRunWithType(), caffe2::LSTMUnitOp< Context >::DoRunWithType(), caffe2::LSTMUnitGradientOp< Context >::DoRunWithType(), caffe2::MeanOp< Context >::DoRunWithType(), caffe2::MeanGradientOp< Context >::DoRunWithType(), caffe2::MishGradientOp< Context >::DoRunWithType(), caffe2::PackRNNSequenceOpBase< Context, Forward >::DoRunWithType(), caffe2::GatherByKeyOp::DoRunWithType(), caffe2::LengthsPartitionOp::DoRunWithType(), caffe2::PowOp< InputTypes, Context, Functor, TypeMap >::DoRunWithType(), caffe2::int8::Int8SliceOp::DoRunWithType(), caffe2::SumReduceDimsOp< Context, FIRSTDIMS, NORMALIZE >::DoRunWithType(), caffe2::ReduceOp< InputTypes, Context, Reducer >::DoRunWithType(), caffe2::ReduceGradientOp< InputTypes, Context, Reducer >::DoRunWithType(), caffe2::SumSqrElementsOp< Context >::DoRunWithType(), caffe2::RemoveDataBlocksOp< Context >::DoRunWithType(), caffe2::RMSNormGradientOp< Context >::DoRunWithType(), caffe2::RecurrentNetworkOp< Context >::DoRunWithType(), caffe2::RecurrentNetworkGradientOp< Context >::DoRunWithType(), caffe2::AccumulateInputGradientOp< Context >::DoRunWithType(), caffe2::ScaleBlobsOp< Context >::DoRunWithType(), caffe2::ScaleOp< Context >::DoRunWithType(), caffe2::SelfBinningHistogramOp< Context >::DoRunWithType(), caffe2::GatherPaddingOp< Context >::DoRunWithType(), caffe2::SliceOp< Context >::DoRunWithType(), caffe2::SliceGradientOp< Context >::DoRunWithType(), caffe2::SparseNormalizeOp< T, Context >::DoRunWithType(), caffe2::SparseToDenseMaskOp< Context >::DoRunWithType(), caffe2::SparseToDenseMaskGradientOp< Context >::DoRunWithType(), caffe2::SpatialBNOp< Context >::DoRunWithType(), caffe2::SpatialBNGradientOp< Context >::DoRunWithType(), caffe2::TileOp< Context >::DoRunWithType(), caffe2::SumOp< Context >::DoRunWithType(), caffe2::WeightedSumOp< Context >::DoRunWithType(), caffe2::WeightedSumGradientOp< Context >::DoRunWithType(), caffe2::ScatterOp< Context >::DoRunWithType(), caffe2::GatherRangesOp< Context >::DoRunWithType(), caffe2::LengthsGatherOp< Context >::DoRunWithType(), caffe2::FbFCPackedOperator< Context, Engine, T_W >::DoRunWithType(), caffe2::FullyConnectedFakeLowpFPOp< Q, Context, Engine, TransposeWeight >::DoRunWithType(), caffe2::FullyConnectedGradientFakeLowpFPOp< Q, Context, Engine, TransposeWeight >::DoRunWithType(), caffe2::GatherDNNLowPOp< T >::DoRunWithType(), caffe2::SparseAdadeltaOp< Context >::DoRunWithType(), caffe2::SparseAdamOp< T, Context >::DoRunWithType(), caffe2::SparseMomentumSGDUpdateOp< T, Context >::DoRunWithType(), caffe2::RowWiseSparseAdagradFusedWithSparseLengthsWeightedSumGradientOp< Tdata, T, TLengths, rowWiseAdagradT >::DoRunWithType(), caffe2::WeightScaleOp< Context >::DoRunWithType(), caffe2::BatchGatherGradientOp< Context >::DoRunWithType2(), caffe2::MergeSingleMapFeatureTensorsOp< Context >::DoRunWithType2(), caffe2::MergeMultiMapFeatureTensorsOp< Context >::DoRunWithType2(), caffe2::PackSegmentsOp< Context >::DoRunWithType2(), caffe2::UnpackSegmentsOp< Context >::DoRunWithType2(), caffe2::SparseToDenseOp< Context >::DoRunWithType2(), caffe2::SquareRootDivideOp< Context >::DoRunWithType2(), caffe2::ReshapeOp< F, Context >::DoRunWithTypeImpl(), caffe2::AbstractReduceFrontOrBackOp< T, Context, Reducer, FirstDim, InputAccessor >::DoRunWithValue(), caffe2::AbstractReduceFrontOrBackGradientOp< T, Context, ReducerGradient, FirstDim >::DoRunWithValue(), caffe2::AbstractSortedSegmentOp< T, SIndex, Context, Reducer, SparseFused, InputAccessor >::DoRunWithValue(), caffe2::AbstractSortedSegmentGradientOp< T, SIndex, Context, ReducerGradient >::DoRunWithValue(), caffe2::AbstractUnsortedSegmentOp< T, SIndex, Context, Reducer, SparseFused, InputAccessor >::DoRunWithValue(), caffe2::AbstractUnsortedSegmentGradientOp< T, SIndex, Context, ReducerGradient >::DoRunWithValue(), caffe2::AbstractLengthsOp< TData, TLengths, Context, Reducer, SparseFused, InputAccessor >::DoRunWithValue(), caffe2::AbstractLengthsGradientOp< T, TLengths, Context, ReducerGradient, GradientNeedIndices >::DoRunWithValue(), caffe2::AbstractLengthsWithMainInputGradientOp< Tembedding, T, TLengths, Context, ReducerGradient, SparseFused, GradientNeedIndices >::DoRunWithValue(), caffe2::AbstractLengthsWithMainInputAndForwardOutputGradientOp< T, TLengths, Context, ReducerGradient >::DoRunWithValue(), caffe2::ScatterWeightedSumOp< T, Context >::DoRunWithValue(), caffe2::ScatterAssignOp< Context >::DoScatterAssign(), caffe2::NumpyTileOp< Context >::DoTile(), caffe2::TileGradientOp< Context >::DoTileGradient(), caffe2::int8::Int8GivenTensorFillOp::Fill(), caffe2::int8::Int8GivenIntTensorFillOp::Fill(), caffe2::UniformFillOp< T, Context >::Fill(), caffe2::GaussianFillOp< T, Context >::Fill(), caffe2::XavierFillOp< T, Context >::Fill(), caffe2::MSRAFillOp< T, Context >::Fill(), caffe2::GivenTensorByteStringToUInt8FillOp< Context >::Fill(), caffe2::BatchSparseToDenseOp< T, Context >::FillInDenseValues(), caffe2::BatchDenseToSparseOp< T, Context >::FillInSparseValues(), caffe2::DiagonalFillOp< Context >::FillWithType(), caffe2::UniqueUniformFillOp< Context >::FillWithType(), caffe2::ConstantFillOp< Context >::FillWithType(), caffe2::GivenTensorFillOp< T, Context >::FillWithType(), caffe2::LayerNormGradientOp< Context >::GammaBetaBackward(), caffe2::RMSNormGradientOp< Context >::GammaBetaBackward(), caffe2::TTSparseLengthsSumOp< T, Context, Engine >::GatherAllRows(), caffe2::TileOp< Context >::GetArgFromTensor(), caffe2::TileGradientOp< Context >::GetArgFromTensor(), caffe2::SparseToDenseOp< Context >::GetOutputFirstDim(), caffe2::ConvDNNLowPOp< T, ReluFused >::Im2ColNHWC_(), caffe2::PackedInt8BGRANHWCToNCHWCStylizerPreprocessOp::initNoiseCPU(), caffe2::YellowFinOp< T, Context >::MovingAverage(), caffe2::Operator< Context >::Operator(), caffe2::TensorProtosDBInput< Context >::Prefetch(), caffe2::VideoInputOp< Context >::Prefetch(), caffe2::ConvOp< T, Context >::Run1x1ConvOnDeviceWithOrderNCHW(), caffe2::ConvOp< T, Context >::Run1x1ConvOnDeviceWithOrderNHWC(), caffe2::HSoftmaxGradientOp< T, Context >::RunBackwardSingle(), caffe2::HSoftmaxOp< T, Context >::RunForwardSingle(), caffe2::BatchSparseToDenseOp< T, Context >::RunOnDevice(), caffe2::BatchDenseToSparseOp< T, Context >::RunOnDevice(), caffe2::BooleanMaskOp< Context >::RunOnDevice(), caffe2::BucketizeOp< Context >::RunOnDevice(), caffe2::ConditionalOp< Context >::RunOnDevice(), caffe2::LabelCrossEntropyGradientOp< T, Context >::RunOnDevice(), caffe2::CrossEntropyGradientOp< T, Context >::RunOnDevice(), caffe2::SquaredL2DistanceOp< T, Context >::RunOnDevice(), caffe2::CosineSimilarityOp< T, Context >::RunOnDevice(), caffe2::CosineSimilarityGradientOp< T, Context >::RunOnDevice(), caffe2::DotProductOp< T, Context >::RunOnDevice(), caffe2::DotProductGradientOp< T, Context >::RunOnDevice(), caffe2::DotProductWithPaddingOp< T, Context >::RunOnDevice(), caffe2::DropoutOp< T, Context >::RunOnDevice(), caffe2::DropoutGradientOp< T, Context >::RunOnDevice(), caffe2::ElementwiseLinearGradientOp< T, Context, Engine >::RunOnDevice(), caffe2::HSoftmaxOp< T, Context >::RunOnDevice(), caffe2::HSoftmaxGradientOp< T, Context >::RunOnDevice(), caffe2::HSoftmaxSearchOp< T, Context >::RunOnDevice(), caffe2::Float16UniformFillOp< Context >::RunOnDevice(), caffe2::LengthsTileOp< Context >::RunOnDevice(), caffe2::BatchBucketOneHotOp< Context >::RunOnDevice(), caffe2::MaxReduceDimsOp< T, Context, FIRSTDIMS >::RunOnDevice(), caffe2::RoIAlignGradientOp< T, Context >::RunOnDevice(), caffe2::PadEmptySamplesOp< Context >::RunOnDevice(), caffe2::SoftmaxOp< T, Context >::RunOnDevice(), caffe2::SoftmaxGradientOp< T, Context >::RunOnDevice(), caffe2::SparseDropoutWithReplacementOp< Context >::RunOnDevice(), caffe2::SpatialSoftmaxWithLossOp< T, Context >::RunOnDevice(), caffe2::UpsampleBilinearGradientOp< T, Context >::RunOnDevice(), caffe2::WeightedSampleOp< T, Context >::RunOnDevice(), caffe2::ConcatAddMulReplaceNaNClipOp< Context >::RunOnDevice(), caffe2::NCCLAllreduceOp::RunOnDevice(), caffe2::NCCLBroadcastOp::RunOnDevice(), caffe2::NCCLReduceOp::RunOnDevice(), caffe2::NCCLAllGatherOp::RunOnDevice(), caffe2::NCCLReduceScatterOp::RunOnDevice(), caffe2::CTCOp< T, Context >::RunOnDevice(), caffe2::ElementwiseRTCOp::RunOnDevice(), caffe2::FullyConnectedOpDecomp< T, Context, Engine >::RunOnDevice(), caffe2::FullyConnectedDecompGradientOp< T, Context, Engine >::RunOnDevice(), caffe2::FullyConnectedOpPrune< T, Context, Engine >::RunOnDevice(), caffe2::FullyConnectedPruneGradientOp< T, Context, Engine >::RunOnDevice(), caffe2::FullyConnectedOp_SPARSE< T, Context, Engine >::RunOnDevice(), caffe2::TTContractionOp< T, Context, Engine >::RunOnDevice(), caffe2::TTContractionGradientOp< T, Context, Engine >::RunOnDevice(), caffe2::AccumulateOp< T, Context >::RunOnDevice(), caffe2::ConcatBatchMatMulBatchGatherOp< Context >::RunOnDevice(), caffe2::SplitOp< Context >::RunOnDevice(), caffe2::SplitByLengthsOp< Context >::RunOnDevice(), caffe2::ConcatOp< Context >::RunOnDevice(), caffe2::CopyOp< Context, DstContext, SrcContext >::RunOnDevice(), caffe2::SquaredL2DistanceGradientOp< T, Context >::RunOnDevice(), caffe2::DotProductWithPaddingGradientOp< T, Context >::RunOnDevice(), caffe2::FillerOp< Context >::RunOnDevice(), caffe2::FlattenOp< Context >::RunOnDevice(), caffe2::FlexibleTopKGradientOp< T, Context >::RunOnDevice(), caffe2::GRUUnitOp< T, Context >::RunOnDevice(), caffe2::GRUUnitGradientOp< T, Context >::RunOnDevice(), caffe2::Im2ColOp< T, Context >::RunOnDevice(), caffe2::Col2ImOp< T, Context >::RunOnDevice(), caffe2::LengthsSplitOp< Context >::RunOnDevice(), caffe2::TTSparseLengthsSumGradientOp< T, Context >::RunOnDevice(), caffe2::LengthsTopKGradientOp< T, Context >::RunOnDevice(), caffe2::MatMulOp< T, Context, Engine >::RunOnDevice(), caffe2::MaxOp< T, Context >::RunOnDevice(), caffe2::MinOp< T, Context >::RunOnDevice(), caffe2::MomentsOp< T, Context >::RunOnDevice(), caffe2::NGramFromCategoricalOp< F, T, Context >::RunOnDevice(), caffe2::NHWC2NCHWOp< T, Context >::RunOnDevice(), caffe2::NCHW2NHWCOp< T, Context >::RunOnDevice(), caffe2::PrependDimOp< Context >::RunOnDevice(), caffe2::MergeDimOp< Context >::RunOnDevice(), caffe2::int8::Int8ConcatOp::RunOnDevice(), caffe2::int8::Int8FlattenOp::RunOnDevice(), caffe2::PairWiseLossOp< T, Context >::RunOnDevice(), caffe2::PairWiseLossGradientOp< T, Context >::RunOnDevice(), caffe2::SumElementsOp< T, Context >::RunOnDevice(), caffe2::SumElementsIntOp< T, Context >::RunOnDevice(), caffe2::MaxReductionOp< T, Context, ROWWISE >::RunOnDevice(), caffe2::RecurrentParamAccessOp< T, mode >::RunOnDevice(), caffe2::RecurrentGradientOp< T >::RunOnDevice(), caffe2::AbstractSortedSegmentRangeOp< T, SIndex, Context, RangeReducer, InputAccessor >::RunOnDevice(), caffe2::AbstractSortedSegmentRangeGradientOp< T, SIndex, Context, RangeReducerGradient >::RunOnDevice(), caffe2::GatherPaddingOp< Context >::RunOnDevice(), caffe2::ShapeOp< Context >::RunOnDevice(), caffe2::SpaceToBatchOp< Context >::RunOnDevice(), caffe2::BatchToSpaceOp< Context >::RunOnDevice(), caffe2::TopKOp< T, Context >::RunOnDevice(), caffe2::TopKGradientOp< T, Context >::RunOnDevice(), caffe2::TTLinearOp< T, Context, Engine >::RunOnDevice(), caffe2::UnsafeCoalesceOp< Context >::RunOnDevice(), caffe2::FlattenToVecOp< Context >::RunOnDevice(), caffe2::ResizeLikeOp< Context >::RunOnDevice(), caffe2::SizeOp< Context >::RunOnDevice(), caffe2::AccumulateHistogramOp< T, Context >::RunOnDevice(), caffe2::VariableLengthSequencePaddingOp< T, Context >::RunOnDevice(), caffe2::WeightedMultiSamplingOp< Context >::RunOnDevice(), caffe2::ConcatDNNLowPOp< T >::RunOnDevice(), caffe2::FullyConnectedDNNLowPPackWeightOp::RunOnDevice(), caffe2::SafeEnqueueBlobsOp< Context >::RunOnDevice(), caffe2::SafeDequeueBlobsOp< Context >::RunOnDevice(), caffe2::WeightedSampleDequeueBlobsOp< Context >::RunOnDevice(), caffe2::EnqueueRebatchingQueueOp::RunOnDevice(), caffe2::DequeueRebatchingQueueOp::RunOnDevice(), caffe2::AdadeltaOp< Context >::RunOnDevice(), caffe2::AdagradOp< Context >::RunOnDevice(), caffe2::AdamOp< T, Context >::RunOnDevice(), caffe2::ClipTensorByScalingOp< Context >::RunOnDevice(), caffe2::FP16MomentumSGDUpdateOp< T, Context >::RunOnDevice(), caffe2::FP32MomentumSGDUpdateOp< T, Context >::RunOnDevice(), caffe2::FtrlOp< T, Context >::RunOnDevice(), caffe2::GFtrlOp< T, Context >::RunOnDevice(), caffe2::LearningRateAdaptionOp< T, Context >::RunOnDevice(), caffe2::LearningRateOp< T, Context >::RunOnDevice(), caffe2::MomentumSGDOp< T, Context >::RunOnDevice(), caffe2::MomentumSGDUpdateOp< T, Context >::RunOnDevice(), caffe2::RmsPropOp< T, Context >::RunOnDevice(), caffe2::StormOp< Context >::RunOnDevice(), caffe2::WngradOp< T, Context >::RunOnDevice(), caffe2::AffineChannelGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::AffineChannelOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ChannelShuffleOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ChannelShuffleGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::LRNOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::LRNGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::PoolOp< T, Context, Functor >::RunOnDeviceWithOrderNCHW(), caffe2::PoolGradientOp< T, Context, Functor >::RunOnDeviceWithOrderNCHW(), caffe2::PadImageOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::PadImageGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ResizeNearest3DGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ResizeNearestGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::MaxPoolRTCOp::RunOnDeviceWithOrderNCHW(), caffe2::MaxPoolGradientRTCOp::RunOnDeviceWithOrderNCHW(), caffe2::ConvOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ConvGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ConvTransposeOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ConvTransposeGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::DeformConvOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::DeformConvGradientOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ConvDNNLowPAcc16Op< ReluFused >::RunOnDeviceWithOrderNCHW(), caffe2::ConvDNNLowPOp< T, ReluFused >::RunOnDeviceWithOrderNCHW(), caffe2::GroupNormOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::LocallyConnectedOp< T, Context >::RunOnDeviceWithOrderNCHWImpl(), caffe2::LocallyConnectedGradientOp< T, Context >::RunOnDeviceWithOrderNCHWImpl(), caffe2::AffineChannelGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::AffineChannelOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ChannelShuffleOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ChannelShuffleGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::LRNOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::LRNGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::PoolOp< T, Context, Functor >::RunOnDeviceWithOrderNHWC(), caffe2::PoolGradientOp< T, Context, Functor >::RunOnDeviceWithOrderNHWC(), caffe2::PadImageGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ResizeNearestGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ChannelShuffleDNNLowPOp< T >::RunOnDeviceWithOrderNHWC(), caffe2::ConvOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ConvGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ConvTransposeOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::ConvTransposeGradientOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::GroupNormOp< T, Context >::RunOnDeviceWithOrderNHWC(), caffe2::LocallyConnectedOp< T, Context >::RunOnDeviceWithOrderNHWCImpl(), caffe2::LocallyConnectedGradientOp< T, Context >::RunOnDeviceWithOrderNHWCImpl(), caffe2::ConvPoolOpBase< Context >::SetBiasMultiplier(), caffe2::ConvPoolOpBase< Context >::SetDeviceTensor(), caffe2::TransposeOp< Context >::TransposeImpl(), and caffe2::Operator< Context >::XOutput().

◆ final_hidden

hidden_type final_hidden

Definition at line 99 of file inference_lstm_op.h.

◆ has_biases_

bool has_biases_
protected

Definition at line 304 of file inference_lstm_op.h.

◆ layer_

FullLSTMLayer layer_
private

Definition at line 185 of file inference_lstm_op.h.

◆ num_layers_

int64_t num_layers_
protected

Definition at line 302 of file inference_lstm_op.h.

◆ outputs

output_type outputs

Definition at line 98 of file inference_lstm_op.h.

◆ w_hh

Tensor w_hh

Definition at line 56 of file inference_lstm_op.h.

◆ w_ih

Tensor w_ih

Definition at line 55 of file inference_lstm_op.h.