- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for arg_attrs (0.24 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) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
// TF op. LogicalResult CollectInputsAndAttributes( PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op, SmallVectorImpl<Value>* inputs, NamedAttrList* arg_attrs, llvm::StringMap<Attribute>* derived_attrs) const; // Uses the collected attribute values to derive all the output types. LogicalResult DeriveOutputTypes(Location loc, FunctionType signature,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
const auto file_prefix_attr = builder.getStringAttr(kTfFilePrefix); const auto arg_attrs = builder.getDictionaryAttr({builder.getNamedAttr( kTfSavedModelIndexPathAttr, builder.getArrayAttr({file_prefix_attr}))}); const int insert_idx = func_op.getNumArguments(); func_op.insertArgument(insert_idx, /*argType=*/filename_op_type, arg_attrs, NameLoc::get(file_prefix_attr));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
llvm::SmallVector<Type> arg_types; llvm::SmallVector<Type> res_types; llvm::SmallVector<DictionaryAttr> arg_attrs; llvm::SmallVector<DictionaryAttr> res_attrs; if (failed(FilterTfgSpecificArgResultAttributes( getContext(), ftype.getInputs(), graph_func.getAllArgAttrs(), arg_types, arg_attrs)) || failed(FilterTfgSpecificArgResultAttributes(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/c/c_api_function_test.cc
// Verify that FunctionDef ArgDef has attributes. ASSERT_EQ(func_->record->fdef().arg_attr_size(), 1); auto arg_attrs = func_->record->fdef().arg_attr().find(0); ASSERT_NE(arg_attrs, func_->record->fdef().arg_attr().end()); auto iter = arg_attrs->second.attr().find("_test_attr"); ASSERT_NE(iter, arg_attrs->second.attr().end()); EXPECT_EQ(iter->second.s(), "value"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
llvm::StringSet<> all_attrs; for (auto& attr : (*this)->getAttrs()) { all_attrs.insert(attr.getName().strref()); } for (const auto& operand : llvm::enumerate(getArgumentTypes())) { if (auto attr_name = getArgAttrOfType<StringAttr>( operand.index(), kAttrArgumentNameAttr)) { all_attrs.insert(attr_name.getValue()); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
const absl::InlinedVector<Node*, 4>& control_ret_nodes) { // Store the arg/return attributes as a list rather than uniqueuing during // construction. llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs; arg_attrs.resize(func.getNumArguments()); llvm::SmallVector<mlir::NamedAttrList, 4> ret_attrs; ret_attrs.resize(func.getNumResults()); auto set_attributes_on_func = [&](Node* node, int64_t index, bool is_arg) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
// Fetch inputs and outputs from the signature. llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs, res_attrs; main_op.getAllArgAttrs(arg_attrs); main_op.getAllResultAttrs(res_attrs); std::vector<std::string> sig_def_inputs = GetStringsFromDictionaryAttr(arg_attrs, kTfSavedModelIndexPathAttr); std::vector<std::string> sig_def_outputs =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/tfr_gen.py
arg_attr += '{{tfr.name="{}"'.format(attr_def.name) if attr_def.HasField('default_value'): default_val = _get_val_from_proto(arg_type, attr_def.default_value) arg_attr += ',tfr.default={}'.format(default_val) arg_attr += '}' idx += 1 arg_str = '{}: {}{}'.format(arg_name, arg_type, arg_attr) arg_list.append(arg_str)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 27 15:27:03 UTC 2022 - 55.8K bytes - Viewed (0)