Searched defs:model_types (Results 1 – 1 of 1) sorted by relevance
562 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() function in tensorflow::CudnnRNNKernelCommon1744 const CudnnModelTypes& model_types, const Tensor** output, in ExtractBackwardInputs()