Searched refs:scale_ptr (Results 1 – 1 of 1) sorted by relevance
905 auto scale_ptr = StreamExecutorUtil::AsDeviceMemory<U>(scale); in operator ()() local957 x_ptr, scale_ptr, offset_ptr, estimated_mean_ptr, in operator ()()1098 auto scale_ptr = StreamExecutorUtil::AsDeviceMemory<U>(scale); in operator ()() local1129 y_backprop_ptr, x_ptr, scale_ptr, mean_ptr, inv_variance_ptr, in operator ()()