Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for DenseI64AttrToI32Vector (1.51 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc

                     [](int64_t attr) { return static_cast<int32_t>(attr); });
      return DenseIntElementsAttr::get(
          RankedTensorType::get(ret.size(), builder.getIntegerType(32)), ret);
    }
    
    bool DenseI64AttrToI32Vector(const DenseIntElementsAttr& dense_attr,
                                 std::vector<int32_t>* out_vec) {
      std::vector<int32_t> ret(dense_attr.getNumElements());
      auto range = dense_attr.getValues<int64_t>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.h

        *out_attr = content;
        return true;
      } else {
        return false;
      }
    }
    
    // Changes a DenseIntElementsAttr **containing I64** elements to an I32 Vector.
    bool DenseI64AttrToI32Vector(const DenseIntElementsAttr& dense_attr,
                                 std::vector<int32_t>* out_vec);
    
    // Gets boolean from composite attrs if it exists.
    std::optional<bool> GetBoolFromCompositeAttr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
Back to top