Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for begin_values (0.16 sec)

  1. tensorflow/compiler/mlir/lite/utils/utils.h

    inline TypedAttr GetOffSet(Attribute begin, Attribute end) {
      auto begin_values = begin.cast<DenseElementsAttr>().getValues<int32_t>();
      auto end_values = end.cast<DenseElementsAttr>().getValues<int32_t>();
    
      SmallVector<int32_t> offsets;
      if (begin_values.size() == end_values.size()) {
        for (size_t i = 0; i < begin_values.size(); ++i) {
          offsets.push_back(end_values[i] - begin_values[i]);
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

      for (int i = 0, end = input_shape.size(); i < end; i++) {
        if (begin_values[i] < 0 ||
            (begin_values[i] + size_values[i] > input_shape[i])) {
          return CreateF32SplatConst(builder, size_shape, 0, location);
        }
      }
    
      // Create a dense constant op for slice's begin
      auto slice_i2c_begin =
          CreateI64DenseConst(builder, begin_shape, begin_values, location);
    
      // Create a dense constant op for slice's size
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

        auto create_slice_op = [&](int32_t axis_i, int32_t begin_i, int32_t size_i,
                                   Value input) {
          SmallVector<int64_t, 4> begin_values(input_rank, 0);
          begin_values[axis_i] = begin_i;
          auto begin_attr = DenseIntElementsAttr::get(axis_type, begin_values);
          auto begin =
              rewriter.create<ConstOp>(op->getLoc(), axis_type, begin_attr);
    
          SmallVector<int64_t, 4> output_shape;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

              loc, rewriter.getIndexType(), begin_value);
          begin_values.push_back(begin_value_casted);
        }
        auto index_ty = rewriter.getIndexType();
        auto start_indices = rewriter.create<tensor::FromElementsOp>(
            loc,
            tensorflow::GetTypeFromTFTensorShape(
                {static_cast<int64_t>(begin_values.size())}, index_ty),
            begin_values);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top