Searched defs:mkldnn_batch_norm_backward (Results 1 – 1 of 1) sorted by relevance
32 std::tuple<Tensor, Tensor, Tensor> mkldnn_batch_norm_backward( in mkldnn_batch_norm_backward() function248 std::tuple<Tensor, Tensor, Tensor> mkldnn_batch_norm_backward(const Tensor& grad_output, in mkldnn_batch_norm_backward() function