Searched refs:ConnectTwoNodes (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop_test.cc | 52 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, add_node, &temp).ok()); in TEST()
82 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, add_node, &temp).ok()); in TEST()
109 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, add_node, &temp).ok()); in TEST()
136 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, add_node, &temp).ok()); in TEST()
163 ASSERT_TRUE(ConnectTwoNodes(&graph, node_a, add_node, &temp_a).ok()); in TEST()
164 ASSERT_TRUE(ConnectTwoNodes(&graph, node_b, add_node, &temp_b).ok()); in TEST()
193 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, node_to_remove, &link).ok()); in TEST()
|
| D | make_fully_connected_test.cc | 82 ConnectTwoNodes(&graph, conv1x1_node0, conv4x4_node1, &link1).ok()); in TEST()
87 ConnectTwoNodes(&graph, conv4x4_node1, conv1x1_node2, &link2).ok()); in TEST()
|
| D | merge_padding_with_test.cc | 50 ASSERT_TRUE(ConnectTwoNodes(&graph, pad_node, conv_node, &temp).ok()); in TEST()
87 ASSERT_TRUE(ConnectTwoNodes(&graph, pad_node1, pad_node2, &temp1).ok()); in TEST()
95 ASSERT_TRUE(ConnectTwoNodes(&graph, pad_node2, conv_node, &temp2).ok()); in TEST()
|
| D | add_quant_adjustments_test.cc | 124 ASSERT_TRUE(ConnectTwoNodes(&graph, add1_node, quant_node, &link1).ok()); in TEST()
130 ASSERT_TRUE(ConnectTwoNodes(&graph, quant_node, add2_node, &link2).ok()); in TEST()
|
| D | fuse_add_to_conv_test.cc | 75 ASSERT_TRUE(ConnectTwoNodes(&graph, conv_node, add_node, &link1).ok()); in TEST()
210 ASSERT_TRUE(ConnectTwoNodes(&graph, add_node, conv_node, &link1).ok()); in TEST()
|
| D | make_padding_test.cc | 61 ConnectTwoNodes(&graph, const_node, concat_node, &const_link).ok()); in TEST()
|
| D | fuse_mul_to_conv_test.cc | 75 ASSERT_TRUE(ConnectTwoNodes(&graph, conv_node, mul_node, &link1).ok()); in TEST()
126 ASSERT_TRUE(ConnectTwoNodes(&graph, mul_node, conv_node, &link1).ok()); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | gpu_model_test_util.cc | 60 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, cos_node, &conv_output)); in TestLinkingConvolutionAndCosOp()
163 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, mul0_node, &conv_output)); in TestLinkingConvolution2InputMul2InputMul()
171 RETURN_IF_ERROR(ConnectTwoNodes(&graph, mul0_node, mul1_node, &mul0_output)); in TestLinkingConvolution2InputMul2InputMul()
302 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, mul0_node, &conv_output)); in TestLinkingConvolution2InputBroadcastMul2InputMul()
310 RETURN_IF_ERROR(ConnectTwoNodes(&graph, mul0_node, mul1_node, &mul0_output)); in TestLinkingConvolution2InputBroadcastMul2InputMul()
441 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, mul0_node, &conv_output)); in TestLinkingConvolution2InputMul2InputBroadcastMul()
449 RETURN_IF_ERROR(ConnectTwoNodes(&graph, mul0_node, mul1_node, &mul0_output)); in TestLinkingConvolution2InputMul2InputBroadcastMul()
580 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, mul0_node, &conv_output)); in TestLinkingConvolution2InputMul2InputMulCos()
588 RETURN_IF_ERROR(ConnectTwoNodes(&graph, mul0_node, mul1_node, &mul0_output)); in TestLinkingConvolution2InputMul2InputMulCos()
596 RETURN_IF_ERROR(ConnectTwoNodes(&graph, mul1_node, cos_node, &mul1_output)); in TestLinkingConvolution2InputMul2InputMulCos()
[all …]
|
| D | model.h | 257 absl::Status ConnectTwoNodes(GraphFloat32* graph, const Node* from_node,
|
| D | model.cc | 482 absl::Status ConnectTwoNodes(GraphFloat32* graph, const Node* from_node, in ConnectTwoNodes() function
|