Searched refs:pad_node (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | merge_padding_with_test.cc | 40 auto pad_node = graph.NewNode(); in TEST() local
41 ASSERT_TRUE(graph.AddConsumer(pad_node->id, input->id).ok()); in TEST()
42 pad_node->operation.type = ToString(OperationType::PAD); in TEST()
46 pad_node->operation.attributes = attr; in TEST()
50 ASSERT_TRUE(ConnectTwoNodes(&graph, pad_node, conv_node, &temp).ok()); in TEST()
126 auto pad_node = graph.NewNode(); in TEST() local
127 pad_node->operation.type = ToString(OperationType::PAD); in TEST()
131 pad_node->operation.attributes = pad_attr; in TEST()
133 ASSERT_TRUE(graph.AddConsumer(pad_node->id, input0->id).ok()); in TEST()
134 ASSERT_TRUE(graph.SetProducer(pad_node->id, padded->id).ok()); in TEST()
[all …]
|
| D | make_padding_test.cc | 72 auto pad_node = graph.nodes()[0]; in TEST() local
73 ASSERT_EQ(ToString(OperationType::PAD), pad_node->operation.type); in TEST()
74 auto pad_attr = absl::any_cast<PadAttributes>(pad_node->operation.attributes); in TEST()
|
| D | merge_padding_with.cc | 55 Node* pad_node = sequence.front(); in ApplyToNodesSequence() local
59 absl::any_cast<PadAttributes>(pad_node->operation.attributes); in ApplyToNodesSequence()
71 absl::Status status = RemovePrecedingNode(graph, pad_node, op_node); in ApplyToNodesSequence()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 700 Node* pad_node = graph->AddNode(pad_def, &s); in CreatePadNode() local
701 pad_node->set_assigned_device_name(split_node->assigned_device_name()); in CreatePadNode()
704 graph->AddEdge(split_node, split_index, pad_node, 0); in CreatePadNode()
705 graph->AddEdge(paddings_node, 0, pad_node, 1); in CreatePadNode()
706 graph->AddControlEdge(control_predecessor, pad_node); in CreatePadNode()
707 return pad_node; in CreatePadNode()
723 Node * pad_node, in CreateSplitNode()
726 to_split_node = pad_node; in CreateSplitNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 1754 auto* pad_node = context.graph_view->GetNode("p"); in TEST_F() local
1755 ASSERT_NE(pad_node, nullptr); in TEST_F()
1756 ASSERT_EQ(pad_node->NumRegularFanins(), 2); in TEST_F()
1757 VerifyRegularFaninMatch(pad_node, 0, input_transpose_node->GetName(), 0); in TEST_F()
1758 VerifyRegularFaninMatch(pad_node, 1, padding_transpose_node->GetName(), 0); in TEST_F()
1764 VerifyRegularFaninMatch(output_transpose_node, 0, pad_node->GetName(), 0); in TEST_F()
|