HomeSort by relevance Sort by last modified time
    Searched refs:const_node (Results 1 - 14 of 14) sorted by null

  /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...]

Completed in 1591 milliseconds