- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 103 for SetAttr (0.17 sec)
-
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/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) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
builder.setInsertionPoint(execute_launch); auto copy_with_layout = BuildCopyWithLayout(execute_launch, compile_launch, get_layout, input, &builder); copy_with_layout->setAttr(kDeviceAttr, execute_launch.getDeviceAttr()); execute.setOperand(execute_arg_index, copy_with_layout); } // Performs transformation for replicated inputs. Returns true if this is a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
live_ins); func::FuncOp outlined_func = BuildFunction(live_ins.getArrayRef(), cluster_op, symbol_table, builder); cluster_op->setAttr( builder->getStringAttr(kFuncAttr), mlir::SymbolRefAttr::get(builder->getContext(), outlined_func.getName())); builder->setInsertionPoint(cluster_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc
FunctionFilter::FunctionFilterType type, OpBuilder& builder) { for (Operation& op : func.front()) { if (type == FunctionFilter::SKIP_TARGET_ANNOTATION) { op.setAttr(kSkipTargetAnnotation, builder.getUnitAttr()); } else if (type == FunctionFilter::INCLUDE_TARGET_ANNOTATION) { op.removeAttr(kSkipTargetAnnotation); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
return ConstBytesAttr::get(builder->getContext(), 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);
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/lite/utils/perception_ops_utils.cc
} results->push_back(value.getInt()); } return success(); } } // namespace LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() { func_.eraseBody(); func_.addEntryBlock(); func_->setAttr(kTFImplements, StringAttr::get(func_.getContext(), kMaxUnpooling)); OpBuilder builder(func_.getBody()); std::string custom_option_buffer; if (failed(CreateCustomOptions(custom_option_buffer))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0)