Home
last modified time | relevance | path

Searched refs:saved_var_data (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h575 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data,
586 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data,
598 const DeviceMemory<float>& saved_var_data,
614 const DeviceMemory<float>& saved_var_data,
739 DeviceMemory<U>* saved_var_data,
753 const DeviceMemory<U>& saved_var_data,
Drocm_dnn.cc4247 DeviceMemory<U>* saved_var_data, in DoFusedBatchNormActivationForwardImpl() argument
4282 saved_var_data->opaque(), epsilon); in DoFusedBatchNormActivationForwardImpl()
4325 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward() argument
4331 saved_var_data, output_profile_result); in DoFusedBatchNormActivationForward()
4342 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward() argument
4348 saved_var_data, output_profile_result); in DoFusedBatchNormActivationForward()
4362 const DeviceMemory<U>& saved_var_data, DeviceMemory<T>* x_bn_backprop_data, in DoFusedBatchNormActivationBackwardImpl() argument
4397 saved_mean_data.opaque(), saved_var_data.opaque(), in DoFusedBatchNormActivationBackwardImpl()
4444 const DeviceMemory<float>& saved_var_data, in DoFusedBatchNormActivationBackward() argument
4453 saved_mean_data, saved_var_data, x_bn_backprop_data, scale_backprop_data, in DoFusedBatchNormActivationBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2454 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data,
2467 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data,
2513 const DeviceMemory<float>& saved_var_data,
2531 const DeviceMemory<float>& saved_var_data,