Searched refs:resize_output_check (Results 1 – 6 of 6) sorted by relevance
36 TORCH_API bool resize_output_check(const Tensor& output, IntArrayRef shape);
40 bool resize_output_check(const Tensor& output, IntArrayRef shape) { in resize_output_check() function
3877 if (resize_output_check(out, B_.sizes())) { in linalg_solve_triangular_out()
4063 if (resize_output_check(out[i], tmp[i].sizes())) { in split_with_sizes_copy_out()
753 if (resize_output_check(out[i], out_shape)) { in split_with_sizes_copy_out_cuda()
123 if (resize_output_check(out, self.sizes())) { in batch_norm_elementwise()