- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetI32VectorFromDenseI64CompositeAttr (0.75 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
data.w_in = op_type.getShape()[3]; std::vector<int32_t> kernel_size; GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "kernel_size", &kernel_size); data.kh = kernel_size[0]; data.kw = kernel_size[1]; std::vector<int32_t> padding; GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "padding", &padding); data.ph = padding[0]; data.pw = padding[1];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
std::transform(range.begin(), range.end(), ret.begin(), [](int64_t attr) { return static_cast<int32_t>(attr); }); *out_vec = std::move(ret); return true; } bool GetI32VectorFromDenseI64CompositeAttr( const DictionaryAttr& composite_attrs, const std::string& attr_name, std::vector<int32_t>* out_vec) { DenseIntElementsAttr attr; if (!EnsureAttribute<DenseIntElementsAttr>(composite_attrs, attr_name,
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
// with the name attr_name. If so, extracts its values and stores as a vector // of int32_t elements. // Note: This assumes the DenseIntElementsAttr has its values stored as int64_t. bool GetI32VectorFromDenseI64CompositeAttr( const DictionaryAttr& composite_attrs, const std::string& attr_name, std::vector<int32_t>* out_vec); // Get a DenseIntElementsAttr of type I64 and convert it to an I32 attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0)