Searched refs:graph_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_test.cc | 33 Value* graph_input = graph.NewValue(); in TEST() local
35 ASSERT_TRUE(graph.AddConsumer(node->id, graph_input->id).ok()); in TEST()
39 EXPECT_THAT(graph.values(), UnorderedElementsAre(graph_input, graph_output)); in TEST()
40 EXPECT_THAT(graph.inputs(), UnorderedElementsAre(graph_input)); in TEST()
42 EXPECT_THAT(graph.FindInputs(node->id), UnorderedElementsAre(graph_input)); in TEST()
44 EXPECT_THAT(graph.FindConsumers(graph_input->id), UnorderedElementsAre(node)); in TEST()
47 EXPECT_THAT(graph.FindProducer(graph_input->id), ::testing::Eq(nullptr)); in TEST()
54 Value* graph_input = graph.NewValue(); in TEST() local
57 ASSERT_TRUE(graph.AddConsumer(node->id, graph_input->id).ok()); in TEST()
69 Value* graph_input = graph.NewValue(); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop_test.cc | 214 Value* graph_input = graph.NewValue(); in TEST() local
225 ASSERT_TRUE(graph.AddConsumer(producer_node->id, graph_input->id).ok()); in TEST()
231 EXPECT_THAT(graph.inputs(), UnorderedElementsAre(graph_input)); in TEST()
240 EXPECT_THAT(graph.inputs(), UnorderedElementsAre(graph_input)); in TEST()
245 UnorderedElementsAre(graph_input, graph_output, value0)); in TEST()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | signature_serialization.py | 80 for signature_input_name, graph_input in zip(
85 graph_input.op.get_attr("_user_specified_name"))
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.cc | 143 for (const GraphTransferGraphInputNodeInfo& graph_input : in SetupGraph() local
146 FindNodeInfo(graph_input.name(), &graph_transfer_info); in SetupGraph()
|