Searched refs:compute_shape_grid_sampler_2d (Results 1 – 2 of 2) sorted by relevance
45 TORCH_API std::vector<torch::lazy::Shape> compute_shape_grid_sampler_2d(const at::Tensor & input, c…
861 std::vector<Shape> compute_shape_grid_sampler_2d( in compute_shape_grid_sampler_2d() function