Searched defs:dbeta_data (Results 1 – 4 of 4) sorted by relevance
758 T* dbeta_data = dbeta.defined() ? dbeta.mutable_data_ptr<T>() : nullptr; in GroupNorm1dBackward() local929 T* dbeta_data = dbeta.defined() ? dbeta.mutable_data_ptr<T>() : nullptr; in GroupNormBackwardKernelImplInternal() local
1214 T* dbeta_data = dbeta->defined() ? dbeta->template data_ptr<T>() : nullptr; in LayerNormBackwardKernelImplInternal() local
907 PT* dbeta_data = dbeta.defined() ? dbeta.data_ptr<PT>() : nullptr; in GroupNormBackwardKernelImplInternal() local1387 PT* dbeta_data = dbeta.defined() ? dbeta.data_ptr<PT>() : nullptr; in GroupNormBackwardKernelImplChannelsLastInternal() local
516 T2* dbeta_data = dbeta->defined() ? dbeta->template data_ptr<T2>() : nullptr; in LayerNormBackwardKernelImplInternal() local