Searched defs:dX_data (Results 1 – 3 of 3) sorted by relevance
905 T* dX_data = dX.defined() ? dX.data_ptr<T>() : nullptr; in GroupNormBackwardKernelImplInternal() local1056 T* dX_data, in ApplyInputGradientsChannelsLastColMov()1104 T* dX_data, in ApplyInputGradientsChannelsLastColMov()1161 T* dX_data, in ApplyInputGradientsChannelsLastRowMov()1197 T* dX_data, in ApplyInputGradientsChannelsLastRowMov()1385 T* dX_data = dX.defined() ? dX.data_ptr<T>() : nullptr; in GroupNormBackwardKernelImplChannelsLastInternal() local
191 T* dX_data, in layer_norm_backward_frame()306 T* dX_data, in layer_norm_backward_frame()514 T* dX_data = dX->defined() ? dX->template data_ptr<T>() : nullptr; in LayerNormBackwardKernelImplInternal() local
1158 T* dX_data = dX->defined() ? dX->template data_ptr<T>() : nullptr; in LayerNormBackwardKernelImplInternal() local