- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for attrName (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
return true; } bool GetI32VectorFromDenseI64CompositeAttr( const DictionaryAttr& composite_attrs, const std::string& attr_name, std::vector<int32_t>* out_vec) { DenseIntElementsAttr attr; if (!EnsureAttribute<DenseIntElementsAttr>(composite_attrs, attr_name, &attr)) { return false; } return DenseI64AttrToI32Vector(attr, out_vec); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
for (auto& attr : func->getAttrs()) { StringRef attr_name = attr.getName().getValue(); if (attr_name.starts_with("tf_saved_model.")) { func->removeAttr(attr_name); } } auto iface = cast<FunctionOpInterface>(func.getOperation()); for (int i = 0; i < func.getNumArguments(); ++i) { for (auto& attr : iface.getArgAttrs(i)) { const StringAttr& attr_name = attr.getName();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
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/translate/export_tf_dialect_op.cc
// TODO(b/146937733): Don't use <void> here. llvm::StringRef attr_name = mlir::OpTrait::AttrSizedOperandSegments< void>::getOperandSegmentSizeAttr(); attrs_to_ignore.insert(attr_name.data()); } if (inst->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) { // TODO(b/146937733): Don't use <void> here. llvm::StringRef attr_name = mlir::OpTrait::AttrSizedResultSegments<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc
for (auto& attr : op->getAttrs()) { llvm::StringRef attr_name = attr.getName().getValue(); if (attribute_getter_map.find(attr_name.str()) != attribute_getter_map.end()) { auto attr_val = (attribute_getter_map[attr_name.str()])(rewriter, identifier_to_attr); attrs.push_back(rewriter.getNamedAttr(attr_name, attr_val)); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
void runOnOperation() override; }; LogicalResult SetOnceModuleAttribute(StringRef attr_name, 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) {
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/eager/c_api_unified_experimental.cc
} tsl::Set_TF_Status_from_Status(s, tracing_op->SetOpName(op_name)); } void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name, TF_DataType value, TF_Status* s) { Status status = unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value)); TF_SetStatus(s, static_cast<TF_Code>(status.code()), absl::StatusMessageAsCStr(status)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
} else { auto attr_name = compose_func.getArgAttrOfType<StringAttr>( arg.index(), kAttrArgumentNameAttr); auto attribute = op->getAttr(attr_name.getValue()); if (!attribute) { attribute = compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr); } if (!attribute && attr_name.getValue() == "out_type") {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
if (operand.index() >= call_op.getNumOperands()) { 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; }
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/lite/stablehlo/transforms/legalize_hlo.cc
} } auto check_i64_attr = [&](const std::string& attr_name) -> mlir::LogicalResult { if (!backend_config.contains(attr_name)) { return op.emitOpError() << "Missing " << attr_name << " attribute in backend_config"; } auto attr = backend_config.getAs<IntegerAttr>(attr_name); if (!attr || !attr.getType().isInteger(64)) { return op.emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)