/external/tensorflow/tensorflow/tools/graph_transforms/ |
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");
|
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/ |
constant_op_test.cc | 44 NodeDef const_node; local 48 .Finalize(&const_node)); 61 cpu_allocator(), const_node,
|
/external/tensorflow/tensorflow/cc/tools/ |
freeze_saved_model.cc | 153 NodeDef* const_node) { 154 const_node->set_name(variable_node.name()); 155 const_node->set_op("Const"); 156 (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype"); 158 (*const_node->mutable_attr())["value"].mutable_tensor());
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
loop_optimizer.cc | 120 NodeDef* const_node = nullptr; local 125 const_node = node; 134 const_node = node_map_->GetNode(const_node_name); 135 if (const_node == nullptr) { 136 const_node = optimized_graph_->add_node(); 137 const_node->set_name(const_node_name); 138 const_node->set_op("Const"); 139 const_node->set_device(node->device()); 140 *const_node->mutable_attr() = node->attr(); 141 node_map_->AddNode(const_node->name(), const_node) [all...] |
constant_folding.cc | 1331 NodeDef* const_node = &const_nodes[i]; local 2963 NodeDef* const_node = local [all...] |
layout_optimizer.cc | 885 auto const_node = AddNodePermConst(name, device, {0, 3, 1, 2}); local 895 auto const_node = AddNodePermConst( local 916 auto const_node = local 928 auto const_node = local [all...] |
/external/tensorflow/tensorflow/core/common_runtime/ |
lower_if_while_test.cc | 320 Node* const_node; local 327 .Finalize(root.graph(), &const_node)); 328 TF_ASSERT_OK(root.DoShapeInference(const_node));
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
vectorization_utils_test.cc | 562 auto const_node = vectorized->node_def( local 567 const_node.name()); 649 auto const_node = find_const(2); local 654 const_node->name()); [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/ |
tf2xla_util.cc | 129 Node* const_node = g->AddNode(const_def, &s); local 150 *replace_def.mutable_input(usages[i].dst_input) = const_node->name(); 157 g->AddEdge(const_node, 0, replace_node, usages[i].dst_input);
|
/external/tensorflow/tensorflow/core/graph/ |
graph_partition.cc | 678 Node* const_node = local 681 AddControlFlowInfo(const_node, src_frame, &cf_info); 682 g->AddEdge(const_node, 0, child_loop.enter, 0); [all...] |
/external/tensorflow/tensorflow/core/grappler/costs/ |
graph_properties.cc | 394 NodeDef const_node; local 395 const_node.set_name("const_from_shape"); 396 const_node.set_op("Const"); 397 auto* attr = const_node.mutable_attr(); 401 return const_node; [all...] |
/external/tensorflow/tensorflow/compiler/jit/ |
extract_outside_compilation_pass_test.cc | 862 Node *const_node = g->AddNode(const_def, &s); local [all...] |
/external/tensorflow/tensorflow/lite/testing/ |
generate_examples.py | [all...] |