Searched defs:layer_output (Results 1 – 2 of 2) sorted by relevance
478 std::vector<at::Tensor> layer_output(num_directions); in mkldnn_rnn() local
1091 auto layer_output = layer(layer_input, *(hidden_it++), *(weight_it++)); in apply_layer_stack() local