FM / operators

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators

Name	Size	Mode	Actions
conv2d.h	1182	0644	edit dl rm
matmul.h	507	0644	edit dl rm
norm.h	350	0644	edit dl rm
operators.h	351	0644	edit dl rm
reduction.h	657	0644	edit dl rm
softmax.h	321	0644	edit dl rm
unary.h	306	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators/conv2d.h (1182B)

#pragma once

#include <torch/csrc/jit/tensorexpr/tensor.h>

namespace torch {
namespace jit {
namespace tensorexpr {

// An API to compute 2D depthwise convolutions with bias.
TORCH_API Tensor conv2d_depthwise(
    BufHandle input,
    BufHandle weight,
    BufHandle bias,
    int stride,
    int pad,
    int groups);

// An API to compute 2D depthwise convolutions without bias.
TORCH_API Tensor conv2d_depthwise(
    BufHandle input,
    BufHandle weight,
    int stride,
    int pad,
    int groups);

TORCH_API Tensor conv2d_depthwise(
    BufHandle input,
    BufHandle weight,
    BufHandle bias,
    ExprHandle N,
    ExprHandle C,
    ExprHandle H,
    ExprHandle W,
    ExprHandle K,
    ExprHandle CperG,
    ExprHandle R,
    ExprHandle S,
    ExprHandle stride,
    ExprHandle pad,
    ExprHandle groups);

TORCH_API Tensor conv2d_depthwise(
    BufHandle input,
    BufHandle weight,
    ExprHandle N,
    ExprHandle C,
    ExprHandle H,
    ExprHandle W,
    ExprHandle K,
    ExprHandle CperG,
    ExprHandle R,
    ExprHandle S,
    ExprHandle stride,
    ExprHandle pad,
    ExprHandle groups);

} // namespace tensorexpr
} // namespace jit
} // namespace torch

#pragma once #include namespace torch { namespace jit { namespace tensorexpr { // An API to compute 2D depthwise convolutions with bias. TORCH_API Tensor conv2d_depthwise( BufHandle input, BufHandle weight, BufHandle bias, int stride, int pad, int groups); // An API to compute 2D depthwise convolutions without bias. TORCH_API Tensor conv2d_depthwise( BufHandle input, BufHandle weight, int stride, int pad, int groups); TORCH_API Tensor conv2d_depthwise( BufHandle input, BufHandle weight, BufHandle bias, ExprHandle N, ExprHandle C, ExprHandle H, ExprHandle W, ExprHandle K, ExprHandle CperG, ExprHandle R, ExprHandle S, ExprHandle stride, ExprHandle pad, ExprHandle groups); TORCH_API Tensor conv2d_depthwise( BufHandle input, BufHandle weight, ExprHandle N, ExprHandle C, ExprHandle H, ExprHandle W, ExprHandle K, ExprHandle CperG, ExprHandle R, ExprHandle S, ExprHandle stride, ExprHandle pad, ExprHandle groups); } // namespace tensorexpr } // namespace jit } // namespace torch