| 56 | |
| 57 | template <int Dims, typename Args> |
| 58 | TensorShape<Dims> GetOutputShape(const TensorShape<Dims> &in_sample_shape, const Args &args) { |
| 59 | TensorShape<Dims> out_sample_shape(args.shape); |
| 60 | return out_sample_shape; |
| 61 | } |
| 62 | |
| 63 | template <int Dims, typename Args> |
| 64 | TensorListShape<Dims> GetOutputShapes(const TensorListShape<Dims> &in_shapes, |
no outgoing calls
no test coverage detected