Assumes that the handle_data is valid.
| 44 | |
| 45 | // Assumes that the handle_data is valid. |
| 46 | shape_inference::ShapeHandle GetElementShapeFromHandleData( |
| 47 | const std::vector<shape_inference::ShapeAndType>& shapes_and_types) { |
| 48 | return shapes_and_types[0].shape; |
| 49 | } |
| 50 | |
| 51 | REGISTER_OP("EmptyTensorList") |
| 52 | .Input("element_shape: shape_type") |