Searched defs:x_bn_backprop_data (Results 1 – 2 of 2) sorted by relevance
2649 DeviceMemory<float>* x_bn_backprop_data, in DoFusedBatchNormActivationBackward()2667 DeviceMemory<Eigen::half>* x_bn_backprop_data, in DoFusedBatchNormActivationBackward()
4958 const DeviceMemory<U>& saved_var_data, DeviceMemory<T>* x_bn_backprop_data, in DoFusedBatchNormActivationBackwardImpl()5041 DeviceMemory<float>* x_bn_backprop_data, in DoFusedBatchNormActivationBackward()5064 DeviceMemory<Eigen::half>* x_bn_backprop_data, in DoFusedBatchNormActivationBackward()