Searched defs:model_shapes (Results 1 – 1 of 1) sorted by relevance
572 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()686 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()694 OpKernelContext* context, const CudnnRnnModelShapes& model_shapes, in CreateForwardAndBackwardIODescriptors()790 const CudnnRnnModelShapes& model_shapes, in DoForward()870 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()990 const CudnnRnnModelShapes& model_shapes, in ShouldUsePaddedIO()1096 const CudnnRnnModelShapes& model_shapes, in CreateRnnDescriptor()1123 const CudnnRnnModelShapes& model_shapes, in GetCachedRnnDescriptor()1504 CudnnRnnModelShapes model_shapes; in ComputeAndReturnAlgorithm() local1567 const CudnnRnnModelShapes& model_shapes, in MaybeAutotune()[all …]