- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetBoolFromCompositeAttr (0.43 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
&attr)) { return false; } return DenseI64AttrToI32Vector(attr, out_vec); } std::optional<bool> GetBoolFromCompositeAttr( const DictionaryAttr& composite_attrs, llvm::StringRef attr_name) { auto attr = composite_attrs.get(attr_name); if (!attr) return std::nullopt; if (auto bool_attr = mlir::dyn_cast_or_null<BoolAttr>(attr)) {
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
bool DenseI64AttrToI32Vector(const DenseIntElementsAttr& dense_attr, std::vector<int32_t>* out_vec); // Gets boolean from composite attrs if it exists. std::optional<bool> GetBoolFromCompositeAttr( const DictionaryAttr& composite_attrs, llvm::StringRef attr_name); // Given a DictionaryAttr, checks if it has a DenseIntElementsAttr 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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
std::vector<int32_t> strides; GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "stride", &strides); data.sh = strides[0]; data.sw = strides[1]; data.ceil_mode = GetBoolFromCompositeAttr(composite_attrs, "ceil_mode").value(); data.h_out = CalculateSpatialOutDim(data.h_in, data.kh, data.ph, data.sh, data.ceil_mode);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0)