Searched defs:saved_mean_data (Results 1 – 2 of 2) sorted by relevance
4246 DeviceMemory<U>* batch_var_data, DeviceMemory<U>* saved_mean_data, in DoFusedBatchNormActivationForwardImpl()4325 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward()4342 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward()4361 const DeviceMemory<U>& saved_mean_data, in DoFusedBatchNormActivationBackwardImpl()4443 const DeviceMemory<float>& saved_mean_data, in DoFusedBatchNormActivationBackward()4466 const DeviceMemory<float>& saved_mean_data, in DoFusedBatchNormActivationBackward()
803 auto saved_mean_data = (*saved_mean_tensor)->flat<T>().data(); in AllocateTFOutputs() local