Searched refs:CudnnModelTypes (Results 1 – 1 of 1) sorted by relevance
479 struct CudnnModelTypes { struct562 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()639 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()729 const CudnnModelTypes& model_types, in DoForward()789 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()929 const CudnnModelTypes& model_types() const { return model_types_; } in model_types()1028 CudnnModelTypes model_types_;1744 const CudnnModelTypes& model_types, const Tensor** output, in ExtractBackwardInputs()