Searched defs:x_inv_std (Results 1 – 4 of 4) sorted by relevance
65 GenericBatchNormGradState(Array x_mean, Array x_inv_std, Dtype beta_dtype) in GenericBatchNormGradState()69 const Array& x_inv_std() const { return x_inv_std_; } in x_inv_std() function
156 Array x_inv_std = EmptyLike(gamma_casted_cont, device); in Call() local240 const Array& x_inv_std = cuda_state.x_inv_std(); in Call() local
114 const Array& x_inv_std() const { return x_inv_std_; } in x_inv_std() function
225 …const Array& x_inv_std = generic_state.x_inv_std(); // Note: x_inv_std_ has the information of ep… in Call() local360 … const Array& x_inv_std = Reciprocal(Sqrt(x_var + eps)).AsType(interm_dtype, false); in BatchNorm() local