- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for NamedAttribute (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
} return mlir::success(); } static mlir::LogicalResult ReformatOpAttributes( mlir::MLIRContext *context, llvm::ArrayRef<mlir::NamedAttribute> attrs, llvm::SmallVectorImpl<mlir::NamedAttribute> &output) { for (mlir::NamedAttribute attr : attrs) { if (attr.getName().strref().contains( mlir::tfg::TFGraphDialect::getDeviceAttrKey())) { tensorflow::DeviceNameUtils::ParsedName parsed_name;
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/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
// contraction, with two additions: the list of ops which have been fused // together; epsilon (only with FusedBatchNorm). std::vector<NamedAttribute> attrs = contraction->getAttrs(); ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops); attrs.push_back( NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr)); // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it // here.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
const ArrayRef<NamedAttribute> attributes, const ArrayRef<Operation*> ops) { // A map to find which operation an attribute belongs to. // The key for this map uses the entire NamedAttribute object, i.e. the // {attribute_name, attribute_value} pair. llvm::SmallDenseMap<NamedAttribute, Operation*> attr_to_op_map; for (Operation* op : ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
addOperations< #define GET_OP_LIST #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc.inc" >(); } static LogicalResult VerifyIndexPath(Operation *op, NamedAttribute named_attr) { auto attr = mlir::dyn_cast<ArrayAttr>(named_attr.getValue()); if (!attr) { return op->emitError() << "'" << kTfSavedModelIndexPathAttr
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.h
// NOLINTNEXTLINE Location loc, llvm::SmallVectorImpl<mlir::NamedAttribute> *attributes); // TODO(zichuanwei@): Populate Builtin_options_2 manual for now, should automate // these in the future void BuiltinOptions2ToAttributes( tflite::BuiltinOptions2Union op_union, mlir::Builder builder, llvm::SmallVectorImpl<mlir::NamedAttribute> &attributes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 21:00:09 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
// Create the TF op OperationState op_state(op.getLoc(), tf_op_full_name); op_state.addOperands(op.getOperands()); op_state.addTypes(op.getResultTypes()); SmallVector<NamedAttribute, 2> attrs; std::string parsed_op_name; tensorflow::NodeDef node_def; if (failed(ParseCustomOption(op.getCustomOption().getValue(), op.getLoc(), parsed_op_name, attrs, node_def))) {
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/tensorflow/utils/export_utils.cc
return node_def; } Status ConvertAttributes( const llvm::ArrayRef<mlir::NamedAttribute> attrs, const absl::flat_hash_set<absl::string_view>& attrs_to_ignore, bool remove_ref_type, AttrValueMap* values) { AttrValueMap func_call_attrs; for (const mlir::NamedAttribute& named_attr : attrs) { auto name_strref = named_attr.getName().str(); auto attr = named_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/tensorflow/translate/export_graphdef.cc
if (auto device_attr = func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) *node_def->mutable_device() = device_attr.getValue().str(); llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs = mlir::function_interface_impl::getArgAttrs(func, index); absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (auto device_attr = func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) *node_def->mutable_device() = device_attr.getValue().str(); llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs = mlir::function_interface_impl::getArgAttrs(func, index); absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
<< " but enclosing XlaCallModule has a function list of size " << function_list.size(); return WalkResult::interrupt(); } llvm::SmallVector<NamedAttribute> new_config; // Copy the attributes in the current config except `called_index`. for (auto attr : backend_config) { if (attr.getName() != kCalledIndexAttrName) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0)