| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/ |
| D | fuse_auto_input.cc | 62 for (auto input_value : graph->FindInputs(node->id)) { in ApplyToNode()
107 for (const auto& input : graph->FindInputs(node_to_fuse.first->id)) { in ApplyToNode()
114 for (const auto& input : graph->FindInputs(node->id)) { in ApplyToNode()
123 for (auto value : graph->FindInputs(node->id)) { in ApplyToNode()
146 auto super_inputs = graph->FindInputs(input->id); in ApplyToNode()
|
| D | fuse_inline.cc | 46 graph->FindInputs(node2->id).size() != 1 || in ApplyToNodesSequence()
53 graph->FindInputs(node2->id).size()) { in ApplyToNodesSequence()
|
| D | fuse_inplace.cc | 106 if (graph->FindInputs(node2->id).size() != 1 || in ApplyToNodesSequence()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_test.cc | 42 EXPECT_THAT(graph.FindInputs(node->id), UnorderedElementsAre(graph_input)); in TEST()
86 EXPECT_THAT(graph.FindInputs(node->id), in TEST()
93 EXPECT_THAT(graph.FindInputs(node->id), UnorderedElementsAre(graph_input2)); in TEST()
118 EXPECT_THAT(graph.FindInputs(node->id), ElementsAre(v0, v1, v2)); in TEST()
120 EXPECT_THAT(graph.FindInputs(node->id), ElementsAre(v0, v3, v2)); in TEST()
328 EXPECT_THAT(graph.FindInputs(n0->id), ElementsAre(v0, v1)); in TEST()
329 EXPECT_THAT(graph.FindInputs(n2->id), ElementsAre(v1, v3)); in TEST()
368 EXPECT_THAT(graph.FindInputs(node1->id), UnorderedElementsAre(graph_input)); in TEST()
369 EXPECT_THAT(graph.FindInputs(node2->id), UnorderedElementsAre(graph_input)); in TEST()
395 EXPECT_THAT(graph.FindInputs(node2->id), UnorderedElementsAre(value)); in TEST()
[all …]
|
| D | model.cc | 58 std::vector<Value*> GraphFloat32::FindInputs(NodeId id) const { in FindInputs() function in tflite::gpu::GraphFloat32
391 for (auto input : graph->FindInputs(to_remove->id)) { in RemovePrecedingNode()
403 for (auto input : graph->FindInputs(to_remove->id)) { in RemoveFollowingNode()
410 for (auto input : graph->FindInputs(to_remove->id)) { in RemoveFollowingNode()
421 const auto inputs = graph->FindInputs(simple_node->id); in RemoveSimpleNodeKeepInput()
444 const auto inputs = graph->FindInputs(simple_node->id); in RemoveSimpleNodeKeepOutput()
|
| D | model_transformer.cc | 110 graph_->FindProducer(graph_->FindInputs(first_in_sequence)[0]->id); in ApplyStartingWithNode()
173 for (auto input : graph_->FindInputs(next_node_in_sequence->id)) { in ApplyStartingWithNode()
|
| D | model.h | 97 std::vector<Value*> FindInputs(NodeId id) const;
|
| D | model_builder.cc | 273 for (auto input : graph->FindInputs(node->id)) { in Parse()
280 for (auto input : graph->FindInputs(node->id)) { in Parse()
397 graph->FindInputs(node->id)[0]->tensor.shape, &attr); in Parse()
492 graph->FindInputs(node->id)[0]->tensor.shape, &attr); in Parse()
558 auto input_value = graph->FindInputs(node->id)[0]; in Parse()
819 auto input = graph->FindInputs(node->id)[0]; in Parse()
1220 auto input_shape = graph->FindInputs(node->id)[0]->tensor.shape; in Parse()
1360 auto input_shape = graph->FindInputs(node->id)[0]->tensor.shape; in Parse()
2112 auto weights_shape = graph->FindInputs(node->id)[1]->tensor.shape; in Parse()
2121 graph->FindInputs(node->id)[0]->tensor.shape, &attr); in Parse()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 44 auto dw_inputs = graph.FindInputs(dw_node->id); in TryDepthwiseConvPlus1x1Conv()
64 if (graph.FindInputs(conv_node->id).size() != 1) { in TryDepthwiseConvPlus1x1Conv()
108 auto fc0_inputs = graph.FindInputs(fc0_node->id); in TryFCFCAdd()
127 auto add_inputs = graph.FindInputs(add_node->id); in TryFCFCAdd()
143 auto fc1_inputs = graph.FindInputs(fc1_node->id); in TryFCFCAdd()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | add_bias.cc | 56 if (graph->FindInputs(node->id).size() != 1) { in ApplyToNode()
73 if (graph->FindInputs(node->id).size() != 1) { in ApplyToNode()
|
| D | remove_noop.cc | 106 auto inputs = graph->FindInputs(node->id); in NewRemoveDegenerateUpsampling()
119 auto input_shape = graph->FindInputs(node->id)[0]->tensor.shape; in ApplyToNode()
|
| D | merge_padding_with.cc | 119 auto inputs = graph->FindInputs(node->id); in ApplyToNode()
124 const auto& input_shape = graph->FindInputs(node->id)[0]->tensor.shape; in ApplyToNode()
|
| D | make_fully_connected.cc | 49 auto inputs = graph->FindInputs(node->id); in ApplyToNode()
|
| D | global_pooling_to_reduce_op.cc | 56 auto inputs = graph->FindInputs(node->id); in ApplyToNode()
|
| D | make_padding.cc | 56 auto inputs = graph->FindInputs(node->id); in ApplyToNode()
|
| D | fuse_add_to_conv.cc | 57 if (graph->FindInputs(conv_node.id).size() != 1) { in ApplyToNodesSequence()
|
| D | fuse_mul_to_conv.cc | 44 if (graph->FindInputs(conv_node.id).size() != 1) { in ApplyToNodesSequence()
109 if (graph->FindInputs(conv_node.id).size() != 1) { in ApplyToNodesSequence()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | compiler.cc | 128 for (const auto& tensor : graph.FindInputs(node->id)) { in Compile()
226 for (auto ref : compiled_graph_.FindInputs(node->id)) { in Compile()
245 for (auto ref : compiled_graph_.FindInputs(node->id)) { in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 241 auto inputs = graph.FindInputs(node.id); in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 333 auto inputs = graph.FindInputs(node.id); in ConvertOperations()
|