Searched refs:const_node (Results 1 - 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | obfuscate_names_test.cc | 94 NodeDef* const_node = graph_def.add_node(); local
95 const_node->set_name(strings::StrCat("const_node", i));
96 const_node->set_op("Const");
113 NodeDef* const_node = graph_def.add_node(); local
114 const_node->set_name(strings::StrCat("1", i));
115 const_node->set_op("Const");
|
| H A D | remove_ema_test.cc | 80 NodeDef const_node; local
81 const_node.set_op("Const");
82 const_node.set_name(node.name());
83 SetNodeAttr("dtype", DT_FLOAT, &const_node);
86 SetNodeTensorAttr<float>("value", tensor, &const_node);
87 *(frozen_graph_def.mutable_node()->Add()) = const_node;
|
| H A D | sort_by_execution_order_test.cc | 77 relu_node->add_input("const_node");
84 NodeDef* const_node = graph_def.add_node(); local
85 const_node->set_name("const_node");
86 const_node->set_op("Const");
96 EXPECT_EQ(0, order["const_node"]);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | constant_op_test.cc | 44 NodeDef const_node; local
48 .Finalize(&const_node));
61 cpu_allocator(), const_node,
|
| /external/tensorflow/tensorflow/cc/tools/ |
| H A D | freeze_saved_model.cc | 138 NodeDef* const_node) {
139 const_node->set_name(variable_node.name());
140 const_node->set_op("Const");
141 (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype");
143 (*const_node->mutable_attr())["value"].mutable_tensor());
136 ConvertVariableToConstant(const NodeDef& variable_node, const Tensor& variable_value, NodeDef* const_node) argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 1044 NodeDef* const_node = &const_nodes[i]; local
1045 if (const_node->name().empty()) {
1056 std::find(const_node->input().begin(), const_node->input().end(),
1057 input) == const_node->input().end()) {
1058 *const_node->add_input() = input;
1063 std::find(const_node->input().begin(), const_node->input().end(),
1064 fanin_of_input) == const_node->input().end()) {
1065 *const_node
[all...] |
| H A D | layout_optimizer.cc | 829 auto const_node = AddNodePermConst(name, device, {0, 3, 1, 2}); local
832 *const_node->add_input() = AsControlDependency(depended_node);
833 return const_node;
839 auto const_node = AddNodePermConst( local
844 *const_node->add_input() = AsControlDependency(depended_node);
845 return const_node;
860 auto const_node = local
862 const_name = const_node->name();
872 auto const_node = local
874 const_name = const_node
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_partition.cc | 662 Node* const_node = local
665 AddControlFlowInfo(const_node, src_frame, &cf_info);
666 g->AddEdge(const_node, 0, child_loop.enter, 0);
|
Completed in 330 milliseconds