- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for attr_value (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
*list->add_tensor() = tensor; } else if (auto attr = mlir::dyn_cast<mlir::FlatSymbolRefAttr>(a)) { AttrValue attr_val; TF_RETURN_IF_ERROR(ConvertAttribute(attr, &attr_val)); *list->add_func() = attr_val.func(); } else if (auto attr = mlir::dyn_cast<mlir::TypeAttr>(a)) { AttrValue attr_val; // For type attributes, we only propagate the element type. mlir::Type elt_type = attr.getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
for (const auto& name_and_value : node_def.attr()) { const std::string& attr_name = name_and_value.first; const tensorflow::AttrValue& attr_value = name_and_value.second; absl::StatusOr<Attribute> mlir_attr = tensorflow::ConvertAttributeValue(attr_value, &builder); if (!mlir_attr.ok()) { return emitError(loc, mlir_attr.status().message()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
IntegerAttr attr_value, Operation* op, ModuleOp& module) { const auto ex_attr_value = module->getAttrOfType<IntegerAttr>(attr_name); if (ex_attr_value == nullptr) { module->setAttr(attr_name, attr_value); return success(); } if (ex_attr_value == attr_value) { return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
const void* proto, size_t proto_len, TF_Status* status) { tensorflow::AttrValue attr_value; if (!attr_value.ParseFromArray(proto, proto_len)) { status->status = InvalidArgument( "Unparseable AttrValue proto passed to " "TF_FunctionSetAttrValueProto"); return; } auto fdef_or = func->record->mutable_fdef();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
} return true; } bool HasSymbolRefAttr(Operation* op) { for (const auto& attr : op->getAttrs()) { Attribute attr_value = attr.getValue(); if (mlir::isa<SymbolRefAttr>(attr_value)) { return true; } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) { if (!array_attr.empty() && mlir::isa<SymbolRefAttr>(*array_attr.begin())) { return true; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
const void* proto, size_t proto_len, TF_Status* status) { tensorflow::AttrValue attr_value; if (!attr_value.ParseFromArray(proto, proto_len)) { status->status = tensorflow::errors::InvalidArgument("Unparseable AttrValue proto"); return; } if (op == nullptr) { status->status = tensorflow::errors::InvalidArgument(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
auto attr_name = signature.getArgAttrOfType<StringAttr>( operand.index(), kAttrArgumentNameAttr); auto attr_value = signature.getArgAttr(operand.index(), kAttrArgumentDefaultAttr); arg_attrs->push_back( rewriter.getNamedAttr(attr_name.getValue(), attr_value)); continue; } // The index is valid for the call_op. Value input = call_op.getOperand(operand.index());
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/python/pywrap_quantize_model.cc
#include "tensorflow/compiler/mlir/quantization/tensorflow/python/type_casters.h" // IWYU pragma: keep #include "tensorflow/compiler/mlir/quantization/tensorflow/quantization_options.pb.h" #include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/protobuf/meta_graph.pb.h" namespace py = pybind11; namespace { using ::tensorflow::SignatureDef; using ::tensorflow::quantization::ExportedModel;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 06:33:29 UTC 2024 - 12K bytes - Viewed (0) -
tensorflow/c/kernels.cc
LIST_CASE(tensor, TF_ATTR_FUNC); #undef LIST_CASE break; case tensorflow::AttrValue::kPlaceholder: *list_size = -1; *total_size = -1; break; case tensorflow::AttrValue::kFunc: *list_size = -1; *total_size = -1; break; case tensorflow::AttrValue::VALUE_NOT_SET: status->status = absl::InvalidArgumentError(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
.begin())>::value>::type> Status SetTypeAttribute(absl::string_view name, ContainerT types, AttrValueMap* values) { AttrValue value; auto& type_list = *value.mutable_list(); for (auto type : types) { DataType dtype; TF_RETURN_IF_ERROR(ConvertScalarTypeToDataType(type, &dtype)); type_list.add_type(dtype); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0)