- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 387 for setAttr (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
auto new_key = StringAttr::get(ctx, old_key.str() + "_dtoh_0"); op->setAttr("key", new_key); } else if (isa<TF::XlaRecvFromHostOp>(op)) { StringRef old_key = op->getAttrOfType<StringAttr>("key").getValue(); auto new_key = StringAttr::get(ctx, old_key.str() + "_htod_0"); op->setAttr("key", new_key); } }); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
src/main/java/org/codelibs/fess/util/PrunedTag.java
Registered: Wed Jun 12 13:08:18 UTC 2024 - Last Modified: Thu Feb 22 01:37:57 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc
default_value); } string AttrView::SetterMethod() const { if (!attr_.is_list()) { return absl::StrCat("SetAttr", toUpperCamel(attr_.full_type())); } else { return absl::StrCat("SetAttr", toUpperCamel(attr_.base_type()), "List"); } } std::vector<string> AttrView::SetterArgs() const { if (attr_.full_type() == "string") {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 07:02:00 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
ConstOp zero = builder.create<ConstOp>(loc, zero_attr); zero->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true)); ConstOp shape = builder.create<ConstOp>(loc, shape_attr); shape->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true)); FillOp fill = builder.create<FillOp>(loc, shape, zero); fill->setAttr(kICIWeightDistributionMlirBridgeMarker,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
// Both the device attribute and compile_device_type attribute should be // propagated to the reduce function call. reduce_call->setAttr(kDeviceAttr, reduce_dataset->getAttrOfType<StringAttr>(kDeviceAttr)); reduce_call->setAttr( TF::kCompileDeviceTypeAttr, reduce_dataset->getAttrOfType<StringAttr>(TF::kCompileDeviceTypeAttr)); SmallVector<Value, 4> if_returns;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
if (auto device = res->getAttrOfType<StringAttr>("device")) { if (!device.getValue().empty()) result_id->setAttr("device", device); else result_id->setAttr("device", compile_device_op); } else if (compile_device_op) { result_id->setAttr("device", compile_device_op); } res.getOutput().replaceAllUsesWith(compile_op->getResult(0)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
new_call_op->setAttr(kEntryFuncAttrName, rewriter.getStringAttr(new_ref_func_name.getValue())); new_call_op->setAttrs(call_op->getAttrs()); new_call_op->removeAttr(rewriter.getStringAttr(kQuantTraitAttrName)); FlatSymbolRefAttr new_func_name_attr = FlatSymbolRefAttr::get(rewriter.getContext(), new_ref_func_name); new_call_op->setAttr(kEntryFuncAttrName, new_func_name_attr);
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/lite/experimental/tac/common/cost.h
namespace mlir { namespace TFL { namespace tac { // Cost attribute string on the TFL dialect. constexpr char kCost[] = "tac.cost"; inline void UpdateCost(Operation* op, float cost, OpBuilder* builder) { op->setAttr(kCost, builder->getF32FloatAttr(cost)); } // Get the cost annotated with kCost. inline bool GetCostOnOp(Operation* op, float* cost) { auto cost_type = op->getAttrOfType<FloatAttr>(kCost); if (cost_type == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 21 01:22:53 UTC 2021 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) return WalkResult::interrupt(); if (op->hasAttr(TF::kReplicationInfoAttr)) { op->setAttr(TF::kTpuReplicateAttr, op->getAttr(TF::kReplicationInfoAttr)); op->removeAttr(TF::kReplicationInfoAttr); op->removeAttr(TF::kCompileDeviceTypeAttr); } return mlir::WalkResult::advance(); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
auto func = builder.create<func::FuncOp>(module->getLoc(), kSessionInitFuncName, func_type); func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr({kSessionInitFuncName})); func->setAttr(kTfSavedModelInitializerTypeAttr, builder.getStringAttr(kTfSavedModelInitializerRestoreType)); func.setVisibility(mlir::func::FuncOp::Visibility::Public);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0)