Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for MaxPoolGradV2Helper (0.43 sec)

  1. tensorflow/cc/gradients/nn_grad.cc

          internal::MaxPoolGrad::DataFormat(data_format));
      grad_outputs->push_back(dx);
      return scope.status();
    }
    REGISTER_GRADIENT_OP("MaxPool", MaxPoolGradHelper);
    
    Status MaxPoolGradV2Helper(const Scope& scope, const Operation& op,
                               const std::vector<Output>& grad_inputs,
                               std::vector<Output>* grad_outputs) {
      string data_format;
      string padding;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 27 23:34:33 UTC 2022
    - 24.5K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/nn_grad_test.cc

      Tensor x_init_value = Tensor(DT_FLOAT, x_shape);
      SetRandomValuesForMaxPooling<float>(&x_init_value);
      RunTest(x, x_init_value, y, y_shape);
    }
    
    TEST_F(NNGradTest, MaxPoolGradV2Helper) {
      TensorShape x_shape({1, 2, 2, 1});
      TensorShape y_shape({1, 1, 1, 1});
      auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape));
      // Setup window and strides so that we only do one MaxPool.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 22 20:45:22 UTC 2022
    - 15K bytes
    - Viewed (0)
Back to top