Home
last modified time | relevance | path

Searched defs:conv_tbc (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionTBC.cpp18 Tensor conv_tbc(const Tensor& self, const Tensor& weight, const Tensor& bias, int64_t pad) { in conv_tbc() function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py329 conv_tbc = _add_docstr( variable
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py3293 def conv_tbc(g: jit_utils.GraphContext, input, weight, bias, pad): function
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py274 def conv_tbc(self, input): member in TestONNXRuntime.test_conv_tbc.ConvTBC