- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for StridedSliceGrad (0.11 sec)
-
tensorflow/cc/gradients/array_grad.cc
TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "shrink_axis_mask", &shrink_axis_mask)); grad_outputs->push_back( StridedSliceGrad(scope, x, begin, end, strides, grad_inputs[0], StridedSliceGrad::BeginMask(begin_mask) .EndMask(end_mask) .EllipsisMask(ellipsis_mask) .NewAxisMask(new_axis_mask)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
auto paddings = Const(scope_, {{1, 1}, {2, 2}}); TensorShape y_shape({2, 3}); auto y = MirrorPadGrad(scope_, x, paddings, "SYMMETRIC"); RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, StridedSliceGrad) { TensorShape x_shape({6, 4, 4}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); // y = x[2:6:2, 1:3, 1:3] auto y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0)