- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for MaxPool3DGradHelper (0.17 sec)
-
tensorflow/cc/gradients/nn_grad.cc
grad_outputs->push_back(dx); grad_outputs->push_back(NoGradient()); grad_outputs->push_back(NoGradient()); return scope.status(); } REGISTER_GRADIENT_OP("MaxPoolV2", MaxPoolGradV2Helper); Status MaxPool3DGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { std::vector<int32> ksize;
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, MaxPool3DGradHelper) { TensorShape x_shape({1, 3, 3, 3, 1}); TensorShape y_shape({1, 1, 1, 1, 1}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); // Setup window and strides so that we only do one MaxPool3D.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 20:45:22 UTC 2022 - 15K bytes - Viewed (0)