| 20 | namespace tensorflow { |
| 21 | |
| 22 | Status StatefulRandomShape(shape_inference::InferenceContext* c) { |
| 23 | using shape_inference::ShapeHandle; |
| 24 | // Check algorithm shape |
| 25 | ShapeHandle unused; |
| 26 | TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); |
| 27 | // Set output shape |
| 28 | ShapeHandle out; |
| 29 | TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); |
| 30 | c->set_output(0, out); |
| 31 | return Status::OK(); |
| 32 | } |
| 33 | |
| 34 | #define REGISTER_STATEFUL_OP(name, default_dtype) \ |
| 35 | REGISTER_OP(name) \ |
nothing calls this directly
no test coverage detected