- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for MaxPoolGradV2Helper (0.25 sec)
-
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) -
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)