- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 67 for NamedAttribute (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
// of the index path attribute. SmallVector<NamedAttribute> ReplaceBoundInputAttrWithIndexPathAttr( const ArrayRef<NamedAttribute> arg_attrs, const StringRef index_path, Builder& builder) { // Keep all other attributes except the tf_saved_model.bound_input attribute, // as we are replacing it with tf_saved_model.index_path. SmallVector<NamedAttribute> new_arg_attrs; for (auto arg_attr : arg_attrs) {
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/tensorflow/transforms/strip_saved_module_metadata.cc
void runOnOperation() override; }; bool ShouldStripAttr(NamedAttribute &namedAttr) { auto name = namedAttr.getName().strref(); return name.starts_with("tf_saved_model."); } void StripModule(Operation *module) { auto stripAttrs = llvm::to_vector<4>(llvm::make_filter_range( module->getAttrs(), [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); })); for (auto namedAttr : stripAttrs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_tf_attributes.cc
auto stripAttrs = llvm::to_vector<4>(llvm::make_filter_range( func->getAttrs(), [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); })); for (auto namedAttr : stripAttrs) { func->removeAttr(namedAttr.getName()); } for (int i = 0; i < func.getNumArguments(); ++i) { llvm::ArrayRef<mlir::NamedAttribute> attrs = mlir::function_interface_impl::getArgAttrs(func, i);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.td
// (NamedAttributeList (NamedAttr<"transpose_a"> $transpose_a)) def NamedAttributeList : NativeCodeCall<"llvm::SmallVector<NamedAttribute>{$0...}">; // Creates a NamedAttribute given its name and value. Essentially creates // a pair: {attribute_name, attribute_value}. class NamedAttr<string attr_name> : NativeCodeCall<"NamedAttribute{$_builder.getStringAttr(\"" # attr_name # "\"), $0}">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 00:32:20 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc
NamedAttribute num_layers_attr( num_layers_str, IntegerAttr::get(IntegerType::get(fn.getContext(), /*width=*/32), num_layers)); // current_layer_index Composite Attribute. mlir::StringAttr current_layer_str = builder.getStringAttr("layer_index"); NamedAttribute current_layer_attr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc
llvm::ArrayRef<mlir::NamedAttribute> attrs = mlir::function_interface_impl::getArgAttrs(f, i); for (NamedAttribute a : attrs) { if (a.getName().strref().starts_with("tf_saved_model.")) { f.removeArgAttr(i, a.getName()); } } } for (int i = 0; i < f.getNumResults(); ++i) { for (NamedAttribute a : f.getResultAttrs(i)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h
NamedAttribute named_attr) override; LogicalResult verifyRegionResultAttribute(Operation *op, unsigned region_index, unsigned result_index, NamedAttribute named_attr) override; LogicalResult verifyOperationAttribute(Operation *op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 18 03:21:34 UTC 2024 - 5.1K bytes - Viewed (0) -
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)