Searched defs:scratchpad_tensor (Results 1 – 3 of 3) sorted by relevance
227 at::Tensor scratchpad_tensor = at::empty( in deconvolution() local313 at::Tensor scratchpad_tensor = at::empty( in deconvolution_backward_data() local417 at::Tensor scratchpad_tensor = at::empty( in deconvolution_backward_weights() local
240 at::Tensor scratchpad_tensor = at::empty( in convolution() local341 at::Tensor scratchpad_tensor = at::empty( in convolution_backward_weights() local432 at::Tensor scratchpad_tensor = at::empty( in convolution_backward_data() local
222 at::Tensor scratchpad_tensor = at::empty( in matmul() local