Searched refs:var_to_inv_var (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 227 std::function<const DeviceMemory<float>&()> var_to_inv_var,
240 std::function<const DeviceMemory<float>&()> var_to_inv_var,
568 std::function<const DeviceMemory<U>&()> var_to_inv_var,
|
| D | cuda_dnn.cc | 3188 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward() argument
3195 is_training, std::move(var_to_inv_var), std::move(inv_var_to_var)), in DoBatchNormalizationForward()
3209 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward() argument
3216 is_training, std::move(var_to_inv_var), std::move(inv_var_to_var)), in DoBatchNormalizationForward()
3231 bool is_training, std::function<const DeviceMemory<U>&()> var_to_inv_var, in DoBatchNormalizationForwardImpl() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 218 std::function<const DeviceMemory<float>&()> var_to_inv_var,
231 std::function<const DeviceMemory<float>&()> var_to_inv_var,
640 std::function<const DeviceMemory<U>&()> var_to_inv_var,
|
| D | rocm_dnn.cc | 3006 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward() argument
3012 std::move(var_to_inv_var), std::move(inv_var_to_var)); in DoBatchNormalizationForward()
3025 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward() argument
3031 std::move(var_to_inv_var), std::move(inv_var_to_var)); in DoBatchNormalizationForward()
3045 bool is_training, std::function<const DeviceMemory<U>&()> var_to_inv_var, in DoBatchNormalizationForwardImpl() argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 338 std::function<const DeviceMemory<U>&()> var_to_inv_var = in operator ()() local
365 is_training, std::move(var_to_inv_var), in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1000 std::function<const DeviceMemory<float>&()> var_to_inv_var,
1017 std::function<const DeviceMemory<float>&()> var_to_inv_var,
|
| D | stream.h | 242 std::function<const DeviceMemory<float> &()> var_to_inv_var,
263 std::function<const DeviceMemory<float> &()> var_to_inv_var,
|
| D | stream.cc | 349 std::function<const DeviceMemory<float> &()> var_to_inv_var, in ThenBatchNormalizationForward() argument
358 saved_inv_var, is_training, std::move(var_to_inv_var), in ThenBatchNormalizationForward()
399 std::function<const DeviceMemory<float> &()> var_to_inv_var, in ThenBatchNormalizationForward() argument
408 saved_inv_var, is_training, std::move(var_to_inv_var), in ThenBatchNormalizationForward()
|