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()
|