Searched refs:CudnnRnnModelShapes (Results 1 – 1 of 1) sorted by relevance
498 struct CudnnRnnModelShapes { struct509 bool IsCompatibleWith(const CudnnRnnModelShapes& rhs) const { in IsCompatibleWith() argument526 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& in operator ()()545 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& lhs, in operator ()()546 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& rhs) in operator ()()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()[all …]