- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for DenseI64AttrToI32Vector (1.51 sec)
-
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) -
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)