Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 30 for StridedSlice (0.34 sec)

  1. tensorflow/cc/gradients/array_grad.cc

      grad_outputs->push_back(NoGradient());
      grad_outputs->push_back(NoGradient());
      grad_outputs->push_back(NoGradient());
      return scope.status();
    }
    REGISTER_GRADIENT_OP("StridedSlice", StridedSliceGradHelper);
    
    Status SliceGrad(const Scope& scope, const Operation& op,
                     const std::vector<Output>& grad_inputs,
                     std::vector<Output>* grad_outputs) {
    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/compiler/mlir/tensorflow/tests/tf-ops.mlir

      func.return %0 : tensor<1x1x100xi32>
    }
    
    // -----
    
    // Valid StridedSlice operation.
    func.func @testStridedSlice(%input: tensor<4x8xf32>, %begin: tensor<2xi64>, %end: tensor<2xi64>, %strides: tensor<2xi64>) -> tensor<?x?xf32> {
      %0 = "tf.StridedSlice"(%input, %begin, %end, %strides) : (tensor<4x8xf32>, tensor<2xi64>, tensor<2xi64>, tensor<2xi64>) -> tensor<?x?xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 23 14:40:35 UTC 2023
    - 236.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

    // CHECK-DAG:       %[[VAL_3:.*]] = "tf.Const"() <{value = dense<1> : tensor<2xi64>}> : () -> tensor<2xi64>
    // CHECK:           %[[VAL_4:.*]] = "tf.StridedSlice"(%[[VAL_0]], %[[VAL_1]], %[[VAL_2]], %[[VAL_3]])
    // CHECK-SAME:          <{begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64}>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        auto shape = rewriter.create<TF::ConstOp>(loc, shape_type, shape_attr);
    
        return rewriter.create<TF::ReshapeOp>(loc, result_type, filter, shape);
      }
    };
    
    // StridedSlice can have complicated attributes like begin_axis_mask,
    // end_axis_mask, ellipsis_axis_mask, new_axis_mask, shrink_axis_mask. These
    // masks will complicate the strided_slice computation logic, we can simplify
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir

      // CHECK-DAG: %[[SHAPE:.*]] = "tf.Shape"(%[[INPUT]]) : (tensor<?x?xf32>) -> tensor<2xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 05 18:35:42 UTC 2024
    - 92K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir

        %zero = "tf.Const"() {value = dense<0> : tensor<4xi32>} : () -> tensor<4xi32>
        %one = "tf.Const"() {value = dense<1> : tensor<4xi32>} : () -> tensor<4xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 17:24:10 UTC 2024
    - 167.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

          begin_indices[dim_index] = end_indices[dim_index];
        }
    
        rewriter.replaceOp(op, slices);
        return success();
      }
    };
    
    // Converts StridedSlice op to HLO Slice op along with Reverse op to handle
    // negative strides and Reshape op to update the output shape. Indices and
    // strides operands are converted to attributes with non-negative indexing.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

          ::llvm::SmallVectorImpl<int64_t> *slice_stride);
      }];
    }
    
    def TF_StridedSliceGradOp : TF_Op<"StridedSliceGrad", [Pure]> {
      let summary = "Returns the gradient of `StridedSlice`.";
    
      let description = [{
    Since `StridedSlice` cuts out pieces of its `input` which is size
    `shape`, its gradient will have the same shape (which is passed here
    as `shape`). The gradient will be zero in any element that the slice
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        SameOperandsAndResultsScale,
        TFL_OperandHasRankAtMost<0, 5>,
        TFL_OperandHasRank<1, 1>,
        TFL_OperandHasRank<2, 1>,
        TFL_OperandHasRank<3, 1>
      ]> {
      let summary = "StridedSlice Op";
    
      let description = [{
        Return a strided slice from `input`.
      }];
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top