Searched refs:graph_output (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_test.cc | 34 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/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop_test.cc | 215 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()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 500 auto* graph_output = graph_view.GetNode("i"); in TEST_F() local
501 ASSERT_NE(graph_output, nullptr); in TEST_F()
502 ASSERT_EQ(graph_output->NumRegularFanins(), 1); in TEST_F()
504 graph_output, 0, in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 153 for (const GraphTransferGraphOutputNodeInfo& graph_output : in SetupGraph() local
170 const TensorId tid = ParseTensorName(graph_output.name()); in SetupGraph()
|