@@ -26,7 +26,7 @@ REGISTER_OP("PairwiseIou3D")
26
26
.SetShapeFn([](tensorflow::shape_inference::InferenceContext* c) {
27
27
c->set_output (
28
28
0 , c->MakeShape ({c->Dim (c->input (0 ), 0 ), c->Dim (c->input (1 ), 0 )}));
29
- return tensorflow ::Status ();
29
+ return absl ::Status ();
30
30
})
31
31
.Doc(R"doc(
32
32
Calculate pairwise IoUs between two set of 3D bboxes. Every bbox is represented
@@ -112,7 +112,7 @@ REGISTER_OP("NonMaxSuppression3D")
112
112
c->set_output (0 , output_shape);
113
113
c->set_output (1 , output_shape);
114
114
c->set_output (2 , output_shape);
115
- return tensorflow ::Status ();
115
+ return absl ::Status ();
116
116
})
117
117
.Doc(R"doc(
118
118
Greedily selects the top subset of 3D (7 DOF format) bounding boxes per class.
@@ -215,7 +215,7 @@ REGISTER_OP("SamplePoints")
215
215
c->set_output (1 , c->MakeShape ({batch_size, num_centers}));
216
216
c->set_output (2 , c->MakeShape ({batch_size, num_centers, num_neighbors}));
217
217
c->set_output (3 , c->MakeShape ({batch_size, num_centers, num_neighbors}));
218
- return :: tensorflow ::Status ();
218
+ return absl ::Status ();
219
219
})
220
220
.Doc(R"doc(
221
221
Sample points among 'points'.
0 commit comments