Home
last modified time | relevance | path

Searched refs:inv_var_to_var (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h228 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);
Dcuda_dnn.cc3189 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/
Drocm_dnn.h219 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);
Drocm_dnn.cc3007 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/
Dfused_batch_norm_op.cc351 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/
Ddnn.h1001 std::function<void()> inv_var_to_var) {
1018 std::function<void()> inv_var_to_var) {
Dstream.h243 std::function<void()> inv_var_to_var);
264 std::function<void()> inv_var_to_var);
Dstream.cc350 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()