Searched defs:model_shapes (Results 1 – 1 of 1) sorted by relevance
570 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()684 CudnnRnnModelShapes* model_shapes) { in ExtractForwardInput()692 OpKernelContext* context, const CudnnRnnModelShapes& model_shapes, in CreateForwardAndBackwardIODescriptors()788 const CudnnRnnModelShapes& model_shapes, in DoForward()848 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()949 const CudnnRnnModelShapes& model_shapes, in ShouldUsePaddedIO()1055 const CudnnRnnModelShapes& model_shapes, in CreateRnnDescriptor()1082 const CudnnRnnModelShapes& model_shapes, in GetCachedRnnDescriptor()1463 CudnnRnnModelShapes model_shapes; in ComputeAndReturnAlgorithm() local1526 const CudnnRnnModelShapes& model_shapes, in MaybeAutoTune()[all …]