Home
last modified time | relevance | path

Searched defs:grady (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/aten/src/ATen/native/mkldnn/
DGelu.cpp53 ideep::tensor grady = itensor_from_tensor(grad_output); in mkldnn_gelu_backward() local
DPrelu.cpp46 const ideep::tensor grady = itensor_from_mkldnn(grad_output); in mkldnn_prelu_backward() local
DRelu.cpp65 ideep::tensor grady = itensor_from_mkldnn(grad_output); in mkldnn_relu_backward() local
DLinear.cpp120 ideep::tensor& grady = itensor_from_mkldnn(grad_output_reshaped); in mkldnn_linear_backward_input() local
152 ideep::tensor& grady = itensor_from_mkldnn(grad_output_reshaped); in mkldnn_linear_backward_weights() local
DNormalization.cpp260 ideep::tensor& grady = itensor_from_mkldnn(grad_output); in mkldnn_batch_norm_backward() local
DPooling.cpp346 const ideep::tensor& grady = itensor_from_mkldnn(grad_output); in _mkldnn_pooling_backward() local
/external/pytorch/aten/src/ATen/native/
DActivation.cpp421 ideep::tensor grady = itensor_from_tensor(grad, /*from_const_data_ptr*/true); in TORCH_IMPL_FUNC() local