Searched defs:x_bn_backprop_data (Results 1 – 1 of 1) sorted by relevance
4650 const DeviceMemory<U>& saved_var_data, DeviceMemory<T>* x_bn_backprop_data, in DoFusedBatchNormActivationBackwardImpl()4733 DeviceMemory<float>* x_bn_backprop_data, in DoFusedBatchNormActivationBackward()4756 DeviceMemory<Eigen::half>* x_bn_backprop_data, in DoFusedBatchNormActivationBackward()