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