Searched defs:mkldnn_rnn_layer (Results 1 – 4 of 4) sorted by relevance
27 std::tuple<Tensor, Tensor, Tensor, Tensor> mkldnn_rnn_layer( in mkldnn_rnn_layer() function216 std::tuple<Tensor, Tensor, Tensor, Tensor> mkldnn_rnn_layer(const Tensor& input, in mkldnn_rnn_layer() function
324 def mkldnn_rnn_layer( function
1074 def mkldnn_rnn_layer(match, *args, **kwargs): function
5973 def mkldnn_rnn_layer( function