Searched defs:var_to_inv_var (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 338 std::function<const DeviceMemory<U>&()> var_to_inv_var = in operator ()() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3188 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward()
3209 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward()
3231 bool is_training, std::function<const DeviceMemory<U>&()> var_to_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3006 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward()
3025 std::function<const DeviceMemory<float>&()> var_to_inv_var, in DoBatchNormalizationForward()
3045 bool is_training, std::function<const DeviceMemory<U>&()> var_to_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 349 std::function<const DeviceMemory<float> &()> var_to_inv_var, in ThenBatchNormalizationForward()
399 std::function<const DeviceMemory<float> &()> var_to_inv_var, in ThenBatchNormalizationForward()
|