Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for FusedBatchNormGrad (0.25 sec)

  1. tensorflow/cc/gradients/nn_grad_test.cc

      auto y = FusedBatchNormV3(scope_, x, scale, offset, mean, var, attrs);
      RunTest({x, scale, offset}, {shape, scale_shape, scale_shape}, {y.y},
              {shape});
    }
    
    INSTANTIATE_TEST_SUITE_P(
        FusedBatchNormGrad, FusedBatchNormGradTest,
        ::testing::Combine(::testing::Bool(), ::testing::Bool(),
                           ::testing::Values(TensorShape({2, 3, 4, 5}),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 22 20:45:22 UTC 2022
    - 15K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/nn_grad.cc

      grad_outputs->push_back(dx);
      return scope.status();
    }
    REGISTER_GRADIENT_OP("FractionalMaxPool", FractionalMaxPoolGradHelper);
    
    // Templated constructor for FusedBatchNormGrad[..]::Attrs.
    template <typename T>
    T FusedBatchNormGradAttrs(float epsilon, StringPiece data_format,
                              bool is_training) {
      T result;
      result.epsilon_ = epsilon;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 27 23:34:33 UTC 2022
    - 24.5K bytes
    - Viewed (0)
Back to top