Searched defs:model_types (Results 1 – 1 of 1) sorted by relevance
569 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()683 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()790 const CudnnModelTypes& model_types, in DoForward()871 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()1045 const CudnnModelTypes& model_types() const { return model_types_; } in model_types() function in tensorflow::CudnnRNNKernelCommon1987 const CudnnModelTypes& model_types, const Tensor** output, in ExtractBackwardInputs()