Searched defs:model_shapes (Results 1 – 1 of 1) sorted by relevance
573 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()687 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()695 OpKernelContext* context, const CudnnRnnModelShapes& model_shapes, in CreateForwardAndBackwardIODescriptors()791 const CudnnRnnModelShapes& model_shapes, in DoForward()871 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()991 const CudnnRnnModelShapes& model_shapes, in ShouldUsePaddedIO()1097 const CudnnRnnModelShapes& model_shapes, in CreateRnnDescriptor()1124 const CudnnRnnModelShapes& model_shapes, in GetCachedRnnDescriptor()1505 CudnnRnnModelShapes model_shapes; in ComputeAndReturnAlgorithm() local1569 const CudnnRnnModelShapes& model_shapes, in MaybeAutotune()[all …]