Searched defs:model_types (Results 1 – 1 of 1) sorted by relevance
566 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()680 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()787 const CudnnModelTypes& model_types, in DoForward()848 const CudnnModelTypes& model_types, const CudnnRnnModelShapes& model_shapes, in DoBackward()1003 const CudnnModelTypes& model_types() const { return model_types_; } in model_types() function in tensorflow::CudnnRNNKernelCommon1943 const CudnnModelTypes& model_types, const Tensor** output, in ExtractBackwardInputs()