Home
last modified time | relevance | path

Searched refs:const_node (Results 1 – 25 of 40) sorted by relevance

12

/external/ComputeLibrary/src/graph/algorithms/
DTopologicalSort.cpp93 for(auto &const_node : g.nodes(NodeType::Const)) in bfs() local
95 if(const_node != EmptyNodeID) in bfs()
97 visited[const_node] = true; in bfs()
98 queue.push_back(const_node); in bfs()
148 for(auto &const_node : g.nodes(NodeType::Const)) in dfs() local
150 if(const_node != EmptyNodeID) in dfs()
152 visited[const_node] = true; in dfs()
153 stack.push(const_node); in dfs()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names_test.cc94 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()
Dsort_by_execution_order_test.cc84 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/pytorch/torch/csrc/jit/passes/onnx/
Dfixup_onnx_controlflow.cpp389 Node* const_node = nullptr; in InferShapeTypeForUninitializedOutput() local
393 const_node = graph->create(::c10::onnx::Constant, 1); in InferShapeTypeForUninitializedOutput()
397 const_node->t_(attr::value, at::zeros(size, elem_type)); in InferShapeTypeForUninitializedOutput()
398 const_node->output()->setType(other_output->type()); in InferShapeTypeForUninitializedOutput()
400 const_node->t_(attr::value, at::zeros({}, elem_type)); in InferShapeTypeForUninitializedOutput()
401 const_node->output()->setType( in InferShapeTypeForUninitializedOutput()
406 const_node = graph->create(::c10::onnx::SequenceEmpty, 1); in InferShapeTypeForUninitializedOutput()
411 const_node->i_(attr::dtype, onnx_type); in InferShapeTypeForUninitializedOutput()
412 const_node->output()->setType(other_output->type()); in InferShapeTypeForUninitializedOutput()
416 const_node->i_(attr::dtype, onnx_type); in InferShapeTypeForUninitializedOutput()
[all …]
Dscalar_type_analysis.cpp332 Node* const_node = n->owningGraph()->create(onnx::Constant); in UpdateScalarTypeForInputs() local
333 const_node->t_(attr::value, new_val); in UpdateScalarTypeForInputs()
334 const_node->insertBefore(n); in UpdateScalarTypeForInputs()
335 const_node->output()->setType(TensorType::create(new_val)); in UpdateScalarTypeForInputs()
336 const_node->copyMetadata(n); in UpdateScalarTypeForInputs()
337 n->replaceInputWith(input, const_node->output()); in UpdateScalarTypeForInputs()
Dunpack_quantized_weights.cpp212 Node* const_node = graph->create(Symbol::onnx("Constant")); in createIntTuple() local
213 const_node->is_(Symbol::attr("value"), is); in createIntTuple()
214 return const_node; in createIntTuple()
218 Node* const_node = graph->create(Symbol::onnx("Constant")); in createInt() local
219 const_node->i_(Symbol::attr("value"), i); in createInt()
220 return const_node; in createInt()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmake_padding_test.cc51 auto const_node = graph.NewNode(); in TEST() local
52 const_node->operation.type = ToString(OperationType::CONSTANT); in TEST()
57 const_node->operation.attributes = const_attr; in TEST()
61 ConnectTwoNodes(&graph, const_node, concat_node, &const_link).ok()); in TEST()
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dplaceholder_with_default.py34 const_node = tf.constant([1, 2, 2, 0],
38 const_node, shape=[2, 2], name="input")
39 out = tf.equal(input_tensor, const_node, name="output")
/external/executorch/devtools/debug_format/
Det_schema.py136 const_node = ValueNode(arg_name, val=str(arg))
137 nodes[arg_name] = const_node
140 const_node, module_mapping, node.meta
153 const_node = ValueNode(arg_name[0], val=arg)
154 nodes[arg_name[0]] = const_node
157 const_node, module_mapping, node.meta
165 const_node = ValueNode(arg_name[-1], val=str(arg))
166 nodes[arg_name[-1]] = const_node
169 const_node, module_mapping, node.meta
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc119 NodeDef* const_node = nullptr; in HandleConst() local
124 const_node = node; in HandleConst()
133 const_node = node_map_->GetNode(const_node_name); in HandleConst()
134 if (const_node == nullptr) { in HandleConst()
135 const_node = optimized_graph_->add_node(); in HandleConst()
136 const_node->set_name(const_node_name); in HandleConst()
137 const_node->set_op("Const"); in HandleConst()
138 const_node->set_device(node->device()); in HandleConst()
139 *const_node->mutable_attr() = node->attr(); in HandleConst()
140 node_map_->AddNode(const_node->name(), const_node); in HandleConst()
[all …]
Dconstant_folding.cc1520 NodeDef* const_node = &const_nodes[i]; in FoldNode() local
1521 VLOG(3) << "Generated constant node: " << SummarizeNodeDef(*const_node); in FoldNode()
1522 if (const_node->name().empty()) { in FoldNode()
1532 auto it = absl::c_find(const_node->input(), input); in FoldNode()
1533 return it != const_node->input().end(); in FoldNode()
1541 *const_node->add_input() = input; in FoldNode()
1550 *const_node->add_input() = fanin_of_input; in FoldNode()
1565 *node->mutable_input() = const_node->input(); in FoldNode()
1569 *node->mutable_attr() = const_node->attr(); in FoldNode()
1572 if (node_map_->GetNode(const_node->name())) { in FoldNode()
[all …]
Dloop_optimizer_test.cc189 const auto* const_node = view.GetNode("Const"); in TEST_F() local
190 ASSERT_NE(const_node, nullptr); in TEST_F()
191 const auto* const_node_node_def = const_node->node(); in TEST_F()
208 const auto* const_node = view.GetNode("Const"); in TEST_F() local
209 ASSERT_NE(const_node, nullptr); in TEST_F()
210 const auto* const_node_node_def = const_node->node(); in TEST_F()
Dremapper.cc1370 const auto* const_node = regular_node_view->node(); in FindMulAndMaximum() local
1371 if (const_node != nullptr && const_node->op() == "Const" && in FindMulAndMaximum()
1372 alpha_tensor.FromProto(const_node->attr().at("value").tensor())) { in FindMulAndMaximum()
2704 const auto* const_node = regular_node_view->node(); in ReplaceMulMaximumWithLeakyRelu() local
2705 if (const_node != nullptr && const_node->op() == "Const" && in ReplaceMulMaximumWithLeakyRelu()
2706 alpha_tensor.FromProto(const_node->attr().at("value").tensor())) { in ReplaceMulMaximumWithLeakyRelu()
/external/tensorflow/tensorflow/core/kernels/
Dconstant_op_test.cc48 NodeDef const_node; in PersistentMemoryTrackingTest() local
52 .Finalize(&const_node)); in PersistentMemoryTrackingTest()
65 cpu_allocator(), const_node, in PersistentMemoryTrackingTest()
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model.cc158 NodeDef* const_node) { in ConvertVariableToConstant() argument
159 const_node->set_name(variable_node.name()); in ConvertVariableToConstant()
160 const_node->set_op("Const"); in ConvertVariableToConstant()
161 (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype"); in ConvertVariableToConstant()
163 (*const_node->mutable_attr())["value"].mutable_tensor()); in ConvertVariableToConstant()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc144 TF_ASSIGN_OR_RETURN(Node * const_node, graph->AddNode(const_def)); in PropagateShapes()
145 graph->AddControlEdge(var_node, const_node); in PropagateShapes()
150 graph->AddControlEdge(const_node, e->dst()); in PropagateShapes()
156 graph->AddEdge(const_node, 0, dst, dst_input); in PropagateShapes()
/external/pytorch/torch/csrc/jit/passes/
Donnx.cpp344 Node* const_node = in NodeToONNX() local
346 const_node->t_(attr::value, value); in NodeToONNX()
347 const_node->output()->setType(TensorType::create(value)); in NodeToONNX()
351 const_node->copyMetadata(node); in NodeToONNX()
352 new_block->appendNode(const_node); in NodeToONNX()
353 ONNXShapeTypeInference(const_node, empty_params_dict, opset_version); in NodeToONNX()
354 auto py_output = py::cast(const_node->output()); in NodeToONNX()
/external/tensorflow/tensorflow/core/tfrt/utils/
Dgraph_partition.cc182 Node * const_node, in PrepareSubgraphForFunctionConversion()
190 node_info.node = const_node; in PrepareSubgraphForFunctionConversion()
193 auxiliary_output_node.emplace(const_node->name(), node_info); in PrepareSubgraphForFunctionConversion()
199 absl::StrCat("ret_", node_info.index, "/", const_node->name()), in PrepareSubgraphForFunctionConversion()
203 .Input(NodeBuilder::NodeOut(const_node->name(), 0, data_type)) in PrepareSubgraphForFunctionConversion()
206 subgraph->AddEdge(const_node, 0, ret_node, 0); in PrepareSubgraphForFunctionConversion()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.h243 virtual void set_const_value(NodeDef& const_node) { in set_const_value() argument
253 &(const_node)); in set_const_value()
255 const_value_.emplace(const_node); in set_const_value()
385 void set_const_value(NodeDef& const_node) override { in set_const_value() argument
467 void set_const_value(NodeDef& const_node) override { in set_const_value() argument
Ddtensor_device_util.cc752 NodeDef const_node = input->const_value().value(); in PrepareGraphForMlir() local
753 const_node.set_name(absl::StrCat("input_", i, "_const_value")); in PrepareGraphForMlir()
754 Node* const_value_n = graph->AddNode(const_node, &status); in PrepareGraphForMlir()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Drsqrt_builder.cc58 auto* const_node = graph_builder_->AddConstNodeWithData( in AddNumerator() local
67 numerator_out_ = TensorID(const_node->GetID(), 0); in AddNumerator()
/external/pytorch/test/onnx/
Dtest_pytorch_onnx_no_runtime.py705 const_node = [n for n in model.graph.node if n.op_type == "Constant"]
706 self.assertNotEqual(len(const_node), 0)
707 for node in const_node:
899 const_node = []
903 const_node.append(n)
907 self.assertNotEqual(len(const_node), 0)
910 for n in const_node:
1181 const_node = [n for n in onnx_model.graph.node if n.op_type == "Constant"]
1182 self.assertNotEqual(len(const_node), 0)
1184 for node in const_node:
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc187 TF_ASSIGN_OR_RETURN(Node * const_node, g->AddNode(const_def)); in ReplaceArgUsageWithConstNode()
190 ReplaceSrcOutputUsageWithNode(g, arg_node, 0, const_node)); in ReplaceArgUsageWithConstNode()
1094 TF_ASSIGN_OR_RETURN(Node * const_node, in RewriteTensorListWithConstElement()
1100 bwd_fbody->graph->AddEdge(const_node, 0, dst, dst_input); in RewriteTensorListWithConstElement()
/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_executor_test.cc417 Node* const_node = test::graph::Constant(g, i_t); in BM_const_identity() local
419 test::graph::Identity(g, const_node); in BM_const_identity()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc704 Node* const_node = in AddControlFlow() local
707 AddControlFlowInfo(const_node, src_frame, &cf_info); in AddControlFlow()
708 g->AddEdge(const_node, 0, child_loop.enter, 0); in AddControlFlow()

12