- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for arg_attrs (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
// as we are replacing it with tf_saved_model.index_path. SmallVector<NamedAttribute> new_arg_attrs; for (auto arg_attr : arg_attrs) { if (arg_attr.getName() == "tf_saved_model.bound_input") continue; new_arg_attrs.emplace_back(arg_attr); } const NamedAttribute index_path_attr( builder.getStringAttr(kTfSavedModelIndexPathAttr), builder.getStrArrayAttr({index_path}));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
func::FuncOp main_func_op = GetMainFuncOp(*module_op); DictionaryAttr arg_attrs = main_func_op.getArgAttrDict(/*index=*/0); EXPECT_THAT(arg_attrs.get("tf_saved_model.bound_input"), IsNull()); const ArrayRef<Attribute> index_path_attrs = mlir::cast<ArrayAttr>(arg_attrs.get("tf_saved_model.index_path")) .getValue(); EXPECT_THAT(index_path_attrs, SizeIs(1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0)