- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for mutable_node (0.14 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/graph_def.h
void, FuncT, tensorflow::NodeDef&>>> void MutateNodeDefs(tensorflow::GraphDef& graph_def, FuncT&& func) { for (tensorflow::NodeDef& node_def : *graph_def.mutable_node()) { func(node_def); } for (tensorflow::FunctionDef& function_def : *graph_def.mutable_library()->mutable_function()) { for (tensorflow::NodeDef& node_def : *function_def.mutable_node_def()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 28 18:38:06 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
absl::StrCat(node_def.name(), "@", func_name)); } } } } } // Upgrade nodes in the GraphDef. for (auto& node_def : *gdef.mutable_node()) { const OpDef* op_def = nullptr; TF_RETURN_IF_ERROR(flib_def->LookUpOpDef(node_def.op(), &op_def)); // TODO(b/197144710): improve the shared_name attr, each op may use the // shared_name differently.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 14:33:47 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model.cc
void GetNodeNameToNodeDefMap( GraphDef* graph_def, std::unordered_map<string, NodeDef*>* name_to_node_map) { for (size_t i = 0; i < graph_def->node_size(); i++) { NodeDef* node = graph_def->mutable_node(i); (*name_to_node_map)[node->name()] = node; } } // Strips off the tensor part of the tensor_name to get the node_name. const string GetNodeNameFromTensorName(string tensor_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/cc/training/queue_runner_test.cc
auto dequeue0 = QueueDequeue(root.WithOpName(kDequeueOp0), q0, {DataType::DT_FLOAT}); GraphDef graph_def; TF_EXPECT_OK(root.ToGraphDef(&graph_def)); for (auto& node : *graph_def.mutable_node()) { node.set_device("/cpu:0"); } SessionOptions sess_options; sess_options.config.mutable_graph_options()->set_build_cost_model(1); std::unique_ptr<Session> session(NewSession(sess_options));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Sep 21 06:27:51 UTC 2019 - 14.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
// ordered. void SortControlInputs(GraphDef* gdef) { int64_t num_nodes = gdef->node_size(); for (int64_t i = 0; i < num_nodes; ++i) { NodeDef* node = gdef->mutable_node(i); // Stable sort control inputs and leave the order of data inputs unchanged. std::stable_sort(node->mutable_input()->begin(), node->mutable_input()->end(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
// - Replacing LegacyFedInput nodes with Placeholder nodes if // convert_legacy_fed_inputs option is enabled. Status PreprocessGraphDef(const GraphImportConfig* specs, GraphDef* graph_def) { for (auto& node_def : *graph_def->mutable_node()) { // TODO(hinsu): Completely deprecate support for LegacyFedInput ops. One // solution could be have a tool to let users upgrade old serialized graphs. if (specs && specs->convert_legacy_fed_inputs &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc
testing::ZeroArgInputSignature(); *saved.mutable_output_signature() = testing::ZeroReturnOutputSignature(); FunctionDef func = FuncDefWithNumInputsOutputs(0, 0); *func.mutable_signature()->mutable_name() = func_name; { std::unique_ptr<TFConcreteFunction> result; Status status = internal::LoadTFConcreteFunction(saved, &func, {}, context(), &result);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
auto& attr = iter.second; if (attr.has_func()) { try_use_original_func_name(attr.mutable_func()->mutable_name()); } else if (attr.has_list()) { for (auto& func_attr : *attr.mutable_list()->mutable_func()) { try_use_original_func_name(func_attr.mutable_name()); } } } } Status Exporter::AddInstructionNode(Operation* inst) { std::unique_ptr<NodeDef> node_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
auto& attr = iter.second; if (attr.has_func()) { try_use_original_func_name(attr.mutable_func()->mutable_name()); } else if (attr.has_list()) { for (auto& func_attr : *attr.mutable_list()->mutable_func()) { try_use_original_func_name(func_attr.mutable_name()); } } } } Status Exporter::AddInstructionNode(Operation* inst) { std::unique_ptr<NodeDef> node_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
result.Append(std::move(pad)); } AppendBufferData(result); std::string result_str = std::string(std::move(result)); auto mutable_model = tflite::GetMutableModel(result_str.data()); bool ret = UpdateBufferOffsets(mutable_model); if (!ret) { return std::nullopt; } return result_str; } return std::string(result); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)