Searched defs:model_types (Results 1 – 1 of 1) sorted by relevance
568 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()682 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()789 const CudnnModelTypes& model_types, in DoForward()870 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()1044 const CudnnModelTypes& model_types() const { return model_types_; } in model_types() function in tensorflow::CudnnRNNKernelCommon1985 const CudnnModelTypes& model_types, const Tensor** output, in ExtractBackwardInputs()