Searched defs:saved_mean_data (Results 1 – 2 of 2) sorted by relevance
4534 DeviceMemory<U>* batch_var_data, DeviceMemory<U>* saved_mean_data, in DoFusedBatchNormActivationForwardImpl()4613 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward()4630 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward()4649 const DeviceMemory<U>& saved_mean_data, in DoFusedBatchNormActivationBackwardImpl()4731 const DeviceMemory<float>& saved_mean_data, in DoFusedBatchNormActivationBackward()4754 const DeviceMemory<float>& saved_mean_data, in DoFusedBatchNormActivationBackward()
820 auto saved_mean_data = (*saved_mean_tensor)->flat<U>().data(); in AllocateTFOutputs() local