- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 31 for NamedAttribute (0.51 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/quantization/tensorflow/passes/add_dump_tensor_op.cc
log_dir_path_(std::move(log_dir_path)) {} private: SmallVector<NamedAttribute> CreateDumpAttributes( PatternRewriter &rewriter, const StringRef folder_name, const StringRef file_name, const bool enabled, const StringRef func_name, const StringRef node_name) const { SmallVector<NamedAttribute> dump_attributes{ rewriter.getNamedAttr("log_dir_path",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K 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/stablehlo/transforms/tfl_stablehlo_pass.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K 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)