Searched refs:inv_var_to_var (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 228 std::function<void()> inv_var_to_var) override;
241 std::function<void()> inv_var_to_var) override;
569 std::function<void()> inv_var_to_var);
|
| D | cuda_dnn.cc | 3189 std::function<void()> inv_var_to_var) { in DoBatchNormalizationForward() argument
3195 is_training, std::move(var_to_inv_var), std::move(inv_var_to_var)), in DoBatchNormalizationForward()
3210 std::function<void()> inv_var_to_var) { in DoBatchNormalizationForward() argument
3216 is_training, std::move(var_to_inv_var), std::move(inv_var_to_var)), in DoBatchNormalizationForward()
3232 std::function<void()> inv_var_to_var) { in DoBatchNormalizationForwardImpl() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 219 std::function<void()> inv_var_to_var) override;
232 std::function<void()> inv_var_to_var) override;
641 std::function<void()> inv_var_to_var);
|
| D | rocm_dnn.cc | 3007 std::function<void()> inv_var_to_var) { in DoBatchNormalizationForward() argument
3012 std::move(var_to_inv_var), std::move(inv_var_to_var)); in DoBatchNormalizationForward()
3026 std::function<void()> inv_var_to_var) { in DoBatchNormalizationForward() argument
3031 std::move(var_to_inv_var), std::move(inv_var_to_var)); in DoBatchNormalizationForward()
3046 std::function<void()> inv_var_to_var) { in DoBatchNormalizationForwardImpl() argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 351 std::function<void()> inv_var_to_var = [d, &batch_var_ptr, epsilon, in operator ()() local
366 std::move(inv_var_to_var)) in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1001 std::function<void()> inv_var_to_var) {
1018 std::function<void()> inv_var_to_var) {
|
| D | stream.h | 243 std::function<void()> inv_var_to_var);
264 std::function<void()> inv_var_to_var);
|
| D | stream.cc | 350 std::function<void()> inv_var_to_var) { in ThenBatchNormalizationForward() argument
359 std::move(inv_var_to_var))); in ThenBatchNormalizationForward()
400 std::function<void()> inv_var_to_var) { in ThenBatchNormalizationForward() argument
409 std::move(inv_var_to_var))); in ThenBatchNormalizationForward()
|