| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | obfuscate_names_test.cc | 94 NodeDef* const_node = graph_def.add_node(); in TestManyNodes() local
95 const_node->set_name(strings::StrCat("const_node", i)); in TestManyNodes()
96 const_node->set_op("Const"); in TestManyNodes()
113 NodeDef* const_node = graph_def.add_node(); in TestNameClashes() local
114 const_node->set_name(strings::StrCat("1", i)); in TestNameClashes()
115 const_node->set_op("Const"); in TestNameClashes()
|
| D | sort_by_execution_order_test.cc | 84 NodeDef* const_node = graph_def.add_node(); in TestSimpleLinear() local
85 const_node->set_name("const_node"); in TestSimpleLinear()
86 const_node->set_op("Const"); in TestSimpleLinear()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 120 NodeDef* const_node = nullptr; in HandleConst() local
125 const_node = node; in HandleConst()
134 const_node = node_map_->GetNode(const_node_name); in HandleConst()
135 if (const_node == nullptr) { in HandleConst()
136 const_node = optimized_graph_->add_node(); in HandleConst()
137 const_node->set_name(const_node_name); in HandleConst()
138 const_node->set_op("Const"); in HandleConst()
139 const_node->set_device(node->device()); in HandleConst()
140 *const_node->mutable_attr() = node->attr(); in HandleConst()
141 node_map_->AddNode(const_node->name(), const_node); in HandleConst()
[all …]
|
| D | constant_folding.cc | 1331 NodeDef* const_node = &const_nodes[i]; in FoldNode() local
1332 VLOG(1) << "Generated constant node:\n" << const_node->DebugString(); in FoldNode()
1333 if (const_node->name().empty()) { in FoldNode()
1344 std::find(const_node->input().begin(), const_node->input().end(), in FoldNode()
1345 input) == const_node->input().end()) { in FoldNode()
1346 *const_node->add_input() = input; in FoldNode()
1351 std::find(const_node->input().begin(), const_node->input().end(), in FoldNode()
1352 fanin_of_input) == const_node->input().end()) { in FoldNode()
1353 *const_node->add_input() = fanin_of_input; in FoldNode()
1368 *node->mutable_input() = const_node->input(); in FoldNode()
[all …]
|
| D | layout_optimizer.cc | 885 auto const_node = AddNodePermConst(name, device, {0, 3, 1, 2}); in AddNodePermNHWCToNCHW() local
888 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNHWCToNCHW()
889 return const_node; in AddNodePermNHWCToNCHW()
895 auto const_node = AddNodePermConst( in AddNodePermNCHWToNHWC() local
900 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNCHWToNHWC()
901 return const_node; in AddNodePermNCHWToNHWC()
916 auto const_node = in GetOrAddNodePermNHWCToNCHW() local
918 const_name = const_node->name(); in GetOrAddNodePermNHWCToNCHW()
928 auto const_node = in GetOrAddNodePermNCHWToNHWC() local
930 const_name = const_node->name(); in GetOrAddNodePermNCHWToNHWC()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 153 NodeDef* const_node) { in ConvertVariableToConstant() argument
154 const_node->set_name(variable_node.name()); in ConvertVariableToConstant()
155 const_node->set_op("Const"); in ConvertVariableToConstant()
156 (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype"); in ConvertVariableToConstant()
158 (*const_node->mutable_attr())["value"].mutable_tensor()); in ConvertVariableToConstant()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op_test.cc | 44 NodeDef const_node; in PersistentMemoryTrackingTest() local
48 .Finalize(&const_node)); in PersistentMemoryTrackingTest()
61 cpu_allocator(), const_node, in PersistentMemoryTrackingTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_while_test.cc | 320 Node* const_node; in TEST() local
327 .Finalize(root.graph(), &const_node)); in TEST()
328 TF_ASSERT_OK(root.DoShapeInference(const_node)); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils_test.cc | 562 auto const_node = vectorized->node_def( in TEST() local
567 const_node.name()); in TEST()
649 auto const_node = find_const(2); in TEST() local
654 const_node->name()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 129 Node* const_node = g->AddNode(const_def, &s); in ReplaceArgUsageWithConstNode() local
150 *replace_def.mutable_input(usages[i].dst_input) = const_node->name(); in ReplaceArgUsageWithConstNode()
157 g->AddEdge(const_node, 0, replace_node, usages[i].dst_input); in ReplaceArgUsageWithConstNode()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 678 Node* const_node = in AddControlFlow() local
681 AddControlFlowInfo(const_node, src_frame, &cf_info); in AddControlFlow()
682 g->AddEdge(const_node, 0, child_loop.enter, 0); in AddControlFlow()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 394 NodeDef const_node; in MakeConstNodeDefFromTensorProto() local
395 const_node.set_name("const_from_shape"); in MakeConstNodeDefFromTensorProto()
396 const_node.set_op("Const"); in MakeConstNodeDefFromTensorProto()
397 auto* attr = const_node.mutable_attr(); in MakeConstNodeDefFromTensorProto()
401 return const_node; in MakeConstNodeDefFromTensorProto()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 862 Node *const_node = g->AddNode(const_def, &s); in TEST_F() local
871 g->AddEdge(const_node, 0, fn_node, 0); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 4225 const_node = tf.constant(
4228 const_node, shape=[2, 2], name="input")
4229 out = tf.equal(input_tensor, const_node, name="output")
|