Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for StridedSliceGrad (0.11 sec)

  1. 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)
  2. 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)
Back to top