Home
last modified time | relevance | path

Searched refs:graph_output (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_test.cc34 Value* graph_output = graph.NewValue(); in TEST() local
36 ASSERT_TRUE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
39 EXPECT_THAT(graph.values(), UnorderedElementsAre(graph_input, graph_output)); in TEST()
41 EXPECT_THAT(graph.outputs(), UnorderedElementsAre(graph_output)); in TEST()
43 EXPECT_THAT(graph.FindOutputs(node->id), UnorderedElementsAre(graph_output)); in TEST()
45 EXPECT_THAT(graph.FindProducer(graph_output->id), ::testing::Eq(node)); in TEST()
46 EXPECT_THAT(graph.FindConsumers(graph_output->id), UnorderedElementsAre()); in TEST()
103 Value* graph_output = graph.NewValue(); in TEST() local
104 ASSERT_TRUE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
105 EXPECT_FALSE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
[all …]
/external/pytorch/docs/source/scripts/exportdb/
Dgenerate_example_rst.py82 graph_output = str(exported_program)
83 graph_output = re.sub(r" # File(.|\n)*?\n", "", graph_output)
84 graph_output = graph_output.replace("\n", "\n ")
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dremove_noop_test.cc215 Value* graph_output = graph.NewValue(); in TEST() local
230 ASSERT_TRUE(graph.SetProducer(consumer_node->id, graph_output->id).ok()); in TEST()
232 EXPECT_THAT(graph.outputs(), UnorderedElementsAre(graph_output)); in TEST()
241 EXPECT_THAT(graph.outputs(), UnorderedElementsAre(graph_output)); in TEST()
245 UnorderedElementsAre(graph_input, graph_output, value0)); in TEST()
254 Value* graph_output = graph.NewValue(); in TEST() local
272 ASSERT_TRUE(graph.SetProducer(consumer_node->id, graph_output->id).ok()); in TEST()
274 EXPECT_THAT(graph.outputs(), UnorderedElementsAre(graph_output)); in TEST()
283 EXPECT_THAT(graph.outputs(), UnorderedElementsAre(graph_output)); in TEST()
287 UnorderedElementsAre(graph_input, graph_output, value0)); in TEST()
/external/pytorch/torch/csrc/jit/passes/onnx/
Dnaming.cpp135 for (const auto* graph_output : graph->outputs()) { in IsGraphOutput() local
136 if (v == graph_output) { in IsGraphOutput()
Dshape_type_inference.cpp2280 Value* graph_output, in ONNXUpdateTypeFromTensor() argument
2285 graph_output, TensorType::create(output), graph_output->type()); in ONNXUpdateTypeFromTensor()
2287 graph_output->inferTypeFrom(output); in ONNXUpdateTypeFromTensor()
2354 auto graph_output = graph->outputs().at(outputs_index); in ONNXAssignOutputShape() local
2356 graph_output, graph_output->type(), ListType::create(elem_type)); in ONNXAssignOutputShape()
2449 Value* graph_output = graph->outputs().at(outputs_index); in ReplaceGraphOutputNoneWithOptional() local
2452 graph_output->replaceAllUsesAfterNodeWith(opt_node, opt_node->output()); in ReplaceGraphOutputNoneWithOptional()
2453 if (!graph_output->type()->cast<NoneType>()) { in ReplaceGraphOutputNoneWithOptional()
2454 opt_node->addInput(graph_output); in ReplaceGraphOutputNoneWithOptional()
2455 opt_node->copyMetadata(graph_output->node()); in ReplaceGraphOutputNoneWithOptional()
/external/pytorch/torch/csrc/dynamo/
Dpython_compiled_autograd.cpp583 for (const auto& graph_output : call.graph_output) { in _compiled_autograd_impl() local
584 int input_nr = graph_output.first; in _compiled_autograd_impl()
585 int output_index = graph_output.second; in _compiled_autograd_impl()
Dcompiled_autograd.h77 graph_output.emplace_back(input_nr, output_idx); in mark_output()
86 std::vector<std::pair<int, int>> graph_output; member
388 collect(t.graph_output); in collect()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc470 auto* graph_output = graph_view.GetNode("i"); in TEST_F() local
471 ASSERT_NE(graph_output, nullptr); in TEST_F()
472 ASSERT_EQ(graph_output->NumRegularFanins(), 1); in TEST_F()
474 graph_output, 0, in TEST_F()
/external/pytorch/torch/csrc/jit/runtime/static/
Dimpl.cpp423 for (auto* graph_output : block.outputs()) { in ManagedTensorRanges() local
424 auto* lifetime = getLifetime(graph_output); in ManagedTensorRanges()