Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for begin_values (0.25 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)
Back to top