Hugging Face's logo

cranky-coder08
/
Phi2-Fine-Tuning

TensorBoard
Safetensors
Model card Files
xet
Metrics Community
Phi2-Fine-Tuning
File size: 1,176 Bytes 
c1af2fa
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
#pragma once
#include <array>
#include <cstdint>

namespace at {
class TensorBase;
}

namespace at::native {

void launch_grid_sampler_2d_forward_kernel(

    const TensorBase &output, const TensorBase &input, const TensorBase &grid,

    int64_t interpolation_mode, int64_t padding_mode, bool align_corners);

void launch_grid_sampler_3d_forward_kernel(

    const TensorBase &output, const TensorBase &input, const TensorBase &grid,

    int64_t interpolation_mode, int64_t padding_mode, bool align_corners);

void launch_grid_sampler_2d_backward_kernel(

    const TensorBase &grad_input, const TensorBase &grad_grid,

    const TensorBase &grad_output, const TensorBase &input,

    const TensorBase &grid, int64_t interpolation_mode, int64_t padding_mode,

    bool align_corners, std::array<bool, 2> output_mask);

void launch_grid_sampler_3d_backward_kernel(

    const TensorBase &grad_input, const TensorBase &grad_grid,

    const TensorBase &grad_output, const TensorBase &input,

    const TensorBase &grid, int64_t interpolation_mode, int64_t padding_mode,

    bool align_corners, std::array<bool, 2> output_mask);

}  // namespace at::native