Searched defs:model_shapes (Results 1 – 1 of 1) sorted by relevance
565 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()643 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()651 OpKernelContext* context, const CudnnRnnModelShapes& model_shapes, in CreateForwardAndBackwardIODescriptors()730 const CudnnRnnModelShapes& model_shapes, in DoForward()789 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()977 const CudnnRnnModelShapes& model_shapes, in CreateRnnDescriptor()1002 const CudnnRnnModelShapes& model_shapes, in GetCachedRnnDescriptor()1282 CudnnRnnModelShapes model_shapes; in ComputeAndReturnAlgorithm() local1341 const CudnnRnnModelShapes& model_shapes, in MaybeAutoTune()1360 const CudnnRnnModelShapes& model_shapes, in AllocateOutputs()[all …]