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.h627 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data,
638 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data,
650 const DeviceMemory<float>& saved_var_data,
666 const DeviceMemory<float>& saved_var_data,
814 DeviceMemory<U>* saved_var_data,
828 const DeviceMemory<U>& saved_var_data,
Drocm_dnn.cc4843 DeviceMemory<U>* saved_var_data, in DoFusedBatchNormActivationForwardImpl() argument
4878 saved_var_data->opaque(), epsilon); in DoFusedBatchNormActivationForwardImpl()
4921 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward() argument
4927 saved_var_data, output_profile_result); in DoFusedBatchNormActivationForward()
4938 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward() argument
4944 saved_var_data, output_profile_result); in DoFusedBatchNormActivationForward()
4958 const DeviceMemory<U>& saved_var_data, DeviceMemory<T>* x_bn_backprop_data, in DoFusedBatchNormActivationBackwardImpl() argument
4993 saved_mean_data.opaque(), saved_var_data.opaque(), in DoFusedBatchNormActivationBackwardImpl()
5040 const DeviceMemory<float>& saved_var_data, in DoFusedBatchNormActivationBackward() argument
5049 saved_mean_data, saved_var_data, x_bn_backprop_data, scale_backprop_data, in DoFusedBatchNormActivationBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2589 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward() argument
2602 DeviceMemory<float>* saved_mean_data, DeviceMemory<float>* saved_var_data, in DoFusedBatchNormActivationForward() argument
2648 const DeviceMemory<float>& saved_var_data, in DoFusedBatchNormActivationBackward() argument
2666 const DeviceMemory<float>& saved_var_data, in DoFusedBatchNormActivationBackward() argument