Searched refs:compute_shape_native_batch_norm_backward (Results 1 – 2 of 2) sorted by relevance
64 TORCH_API std::vector<torch::lazy::Shape> compute_shape_native_batch_norm_backward(const at::Tensor…
570 std::vector<torch::lazy::Shape> compute_shape_native_batch_norm_backward( in compute_shape_native_batch_norm_backward() function