- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 86 for StringAttr (0.12 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.h
int8_t input_zp_value, ArrayAttr strides, ArrayAttr dilations, StringAttr conv_padding, ArrayAttr explicit_paddings, Value &padding, int num_dims = 4);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_device_helper.cc
} // Returns true if operation does not have explicit device placement that would // prevent it from running on GPU device. bool CanUseGpuDevice(Operation *op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); if (!device_attr || device_attr.getValue().empty()) return true; DeviceNameUtils::ParsedName device; if (!DeviceNameUtils::ParseFullName(device_attr.getValue().str(), &device)) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 21 08:41:18 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
StringRef(content.data(), content.size())); } } // namespace void ConvertNMSPaddedFunc::RewriteFunc() { func_->setAttr(kTFImplements, StringAttr::get(func_.getContext(), kTfNMSPadded)); Value boxes = func_.getArgument(0); Value scores = func_.getArgument(1); Value max_output_size = func_.getArgument(2); Value iou_threshold = func_.getArgument(3);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc
} if (GetStableHloQuantConstraints(op)->has_same_scale_requirement) { return true; } const bool attr_enforced_quantizable = op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) && op->getAttrOfType<StringAttr>(kQuantTraitAttrName).getValue().str() == QuantTraitValues[QuantizationTrait::FullyQuantizable]; return attr_enforced_quantizable; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
TF::ReorderOpResultUses(launch_op); } std::string GetDevice(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return device_attr ? device_attr.getValue().str() : ""; } bool CanBeIgnoredInCluster(Operation* op) { auto device_attr = op->getAttrOfType<StringAttr>("device"); return !device_attr || device_attr.getValue().empty(); } void BuildClusters(Block& block, OpBuilder builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose_patterns.td
include "tensorflow/compiler/mlir/tfr/ir/tfr_ops.td" class Quantize<string value> : NativeCodeCall<"TFR::Quantize(" # value # ", $0, $1, $_builder)">; class HasStringAttr<string value> : AttrConstraint< CPred<"$_self.cast<StringAttr>().getValue() == \"" # value # "\"">>; def QuantActRangeNonePattern : Pattern< (TFR_TFRQuantActRangeOp (TFR_ConstOp HasStringAttr<"NONE">:$act), $scale, $zp),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/stablehlo_custom_call.cc
return failure(); } if (auto attr = mlir::dyn_cast<FlatSymbolRefAttr>(f)) { return attr; } op.emitOpError() << "'s attribute '" << kCalledFuncAttrName << "' is neither StringAttr nor FlatSymbolRefAttr"; return failure(); } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_analyzer_test_pass.cc
StringRef device; StringRef container; StringRef shared_name; }; ResourceKey GetResourceKey(TF::VarHandleOp var_handle_op) { ResourceKey resource_key; if (auto attr = var_handle_op->getAttrOfType<StringAttr>("device")) { resource_key.device = attr.getValue(); } resource_key.container = var_handle_op.getContainer(); resource_key.shared_name = var_handle_op.getSharedName(); return resource_key; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 00:21:29 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
return true; } // Returns the `shared_name` attribute value if exists. If not, returns an // empty string. StringRef GetSharedName(Operation* op) { if (!op->hasAttrOfType<StringAttr>(kSharedNameAttr)) return ""; return op->getAttrOfType<StringAttr>(kSharedNameAttr).getValue(); } // Checks if the HashTable is initialized. This function assumes that the // HashTable is initialized if it appears in the initializer since it can't
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc
void runOnOperation() override; private: bool ShouldPreserveOp(Operation* op); bool ShouldPreserveIsland(IslandOp island); void PruneGraph(GraphOp graph); llvm::SmallDenseSet<mlir::StringAttr, 4> ops_to_preserve_ids_; }; // Checks if a tf_executor.Graph can be pruned. // For TensorFlow V1.0 compatibility: when importing a graph without providing
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.6K bytes - Viewed (0)