- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 168 for setFAttr (0.13 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/infeed_ops_xla_adjust_layout.cc
op.getResultTypes().end()); if (!op->getAttr("layout")) { auto layout = mlir::GetTPUInfeedLayout(result_types, builder); if (failed(layout)) return; op->setAttr("layout", layout.value()); } } }; void InfeedsOpsXlaAdjustLayout::runOnOperation() { getOperation().walk(runOnInfeedOp); } } // anonymous namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MutatePass) void runOnOperation() override { mlir::OpBuilder builder(&getContext()); getOperation()->setAttr("tfl.random_attr", builder.getUnitAttr()); } }; class AlwaysFailPass : public mlir::PassWrapper<AlwaysFailPass, mlir::OperationPass<>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(AlwaysFailPass)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
} auto init = builder.create<func::FuncOp>( module.getLoc(), name, FunctionType::get(module.getContext(), /*inputs=*/{}, /*results=*/{})); // Make savedmodel verification happy. init->setAttr("tf_saved_model.exported_names", builder.getStrArrayAttr({name})); builder.setInsertionPointToStart(init.addEntryBlock()); for (func::FuncOp func : tf_saved_model::GetInitializerFunctions(module)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/c/python_api.cc
mutex_lock l(graph->mu); RecordMutation(graph, op, mutation_type); } void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input) { TF_AddOperationControlInput(graph, op, input); } void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, TF_Buffer* attr_value_proto, TF_Status* status) { TF_SetAttr(graph, op, attr_name, attr_value_proto, status); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 12 18:48:56 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/restore_function_name.cc
return; } auto function = symbol_table.lookup<func::FuncOp>(current_function_name.getValue()); if (function) { function.setName(original_function_name); } call_op->setAttr(TF::kStablehloEntryFunctionAttrName, FlatSymbolRefAttr::get(original_function_name)); } void RestoreFunctionNamePass::runOnOperation() { ModuleOp module_op = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/constant_op_device_assignment.cc
// device of the user operation. if (cloned_op_by_device.find(device_attr.getValue()) == cloned_op_by_device.end()) { mlir::Operation *new_op = builder.clone(*op.getOperation()); new_op->setAttr(kDeviceAttr, device_attr); cloned_op_by_device[device_attr.getValue()] = new_op; } // Update the user operation to use the result of the cloned ConstOp.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
const std::optional<DictionaryAttr>& devices) { auto result = region.walk([&](Operation* op) -> WalkResult { if (RequiresReplicaIDAttribute(op)) { op->setAttr(kReplicaIdAttr, builder.getI64IntegerAttr(replica_id)); return WalkResult::advance(); } if (auto placeholder = dyn_cast<TF::_TPUDeviceOrdinalPlaceholderOp>(op)) { int64_t device_ordinal;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
src/test/java/org/codelibs/fess/util/PrunedTagTest.java
} public void test_toString() { String tag = "tag", id = "id", css = "css", attrName = "attrName", attrValue = "attrValue"; PrunedTag prunedtag = new PrunedTag(tag); prunedtag.setAttr(attrName, attrValue); prunedtag.setId(id); prunedtag.setCss(css); assertEquals("PrunedTag [tag=" + tag + ", id=" + id + ", css=" + css + ", attrName=" + attrName + ", attrValue=" + attrValue + "]",
Registered: Wed Jun 12 13:08:18 UTC 2024 - Last Modified: Thu Feb 22 01:37:57 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
op_to_update->setAttr(kDeviceAttr, new_device_attr); PopulateDeviceForOpResults(*op_to_update, new_device_attr.getValue(), value_to_device); if (auto sink = llvm::dyn_cast<tf_executor::NextIterationSinkOp>(op_to_update)) { auto source = sink.GetSource(); source->setAttr(kDeviceAttr, new_device_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables.cc
} else { auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()(); is_variable_initialized = IsVariableInitialized(var_op, handle.device(), mgr, session); } var_op->setAttr("_is_initialized", BoolAttr::get(context, is_variable_initialized)); } return success(); } LogicalResult MarkInitializedVariablesInFunction(ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 19:14:56 UTC 2023 - 4.1K bytes - Viewed (0)