- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 168 for setFAttr (0.39 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
{StringAttr::get(context, "outputs"), StringAttr::get(context, absl::StrJoin(output_names, ","))}); auto dictAttr = DictionaryAttr::get(context, func_attrs); main_func->setAttr(StringAttr::get(context, kEntryFunctionAttr), dictAttr); main_func->setAttr( kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr({kImportModelDefaultGraphFuncName})); if (input_names.size() != main_func.getNumArguments() ||
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/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
OpBuilder builder(&ctx); module_op.walk([&builder](Operation* op) { if (op->hasAttrOfType<BoolAttr>("is_stateless") && ContainCalibrationStatisticsSaverOp(op)) { op->setAttr("is_stateless", builder.getBoolAttr(false)); } }); } std::unique_ptr<OperationPass<ModuleOp>> CreateInsertCalibrationStatisticsSaverPass( StringRef calibration_data_dir,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
// Do not append a UnitAttr for the "token" operand here to avoid // compilation failure when exporting the "layouts" attribute to a graph // node. Instead, add the UnitAttr during LegalizeTF pass. op->setAttr("layouts", layout.value()); return mlir::WalkResult::advance(); }); return !res.wasInterrupted(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
} } // Update input/output sharding attributes on tf_device.cluster_func op. cluster_func->setAttr(tensorflow::kInputShardingAttr, GetStrArrayAttr(builder, sharding_for_args)); cluster_func->setAttr(tensorflow::kOutputShardingAttr, GetStrArrayAttr(builder, sharding_for_rets)); return mlir::success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_update_embedding_enqueue_op_inputs.cc
mode_string_value)); auto outside_compilation_attr = embedding_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr); if (outside_compilation_attr) enqueue_mode->setAttr(kXlaOutsideCompilationAttr, outside_compilation_attr); mode_enqueue_operand.set(enqueue_mode); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc
auto func = builder.create<mlir::func::FuncOp>( ops[0]->getLoc(), dialect.str() + std::to_string(function_id), builder.getFunctionType(input_types, output_types)); func->setAttr("dialect", builder.getStringAttr(dialect)); auto block = func.addEntryBlock(); llvm::DenseSet<Operation*> all_operations; for (Operation* outer : ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/jit/node_matchers.h
input_matchers_ = std::move(inputs); } void set_control_deps(NodeSeqMatcher control_deps) { DCHECK(IsEmpty()); control_deps_ = std::move(control_deps); } void set_attr(AttrKeyValuePair attr) { DCHECK(IsEmpty()); attr_ = std::move(attr); } bool IsEmpty() const { return !name().has_value() && !op().has_value() && !inputs().has_value() &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
new_config.push_back(builder.getNamedAttr( kCalledIndexAttrName, builder.getI64IntegerAttr(it->second))); // Set the `tf.backend_config` attribute to the `new_config`. op->setAttr(kTfBackendConfigAttrName, builder.getDictionaryAttr(new_config)); return WalkResult::advance(); }); if (result.wasInterrupted()) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
return signalPassFailure(); } ModuleOp outlined_module = ModuleOp::create(getOperation().getLoc()); outlined_module->setAttrs(getOperation()->getAttrDictionary()); outlined_module->setAttr(SymbolTable::getSymbolAttrName(), StringAttr::get(ctx, kNestedModule)); symbol_table.insert(outlined_module); SymbolTable outlined_symbol_table(outlined_module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
if (!preceding_qcast) return failure(); auto new_qcast = rewriter.create<quantfork::QuantizeCastOp>( q_op.getLoc(), q_op.getType(), preceding_qcast.getArg()); new_qcast->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); q_op->replaceAllUsesWith(new_qcast); return success(); } }; class ConvertTFConstOpToArithConstOp : public OpRewritePattern<TF::ConstOp> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0)