Home
last modified time | relevance | path

Searched defs:model_shapes (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc572 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() local
1567 const CudnnRnnModelShapes& model_shapes, in MaybeAutotune()
[all …]