| 53 | |
| 54 | template <typename T> |
| 55 | std::vector<int64> AsInt64(const Tensor* tensor, int64 num_elements) { |
| 56 | std::vector<int64> ret(num_elements); |
| 57 | auto data = tensor->vec<T>(); |
| 58 | for (int64 i = 0; i < num_elements; ++i) { |
| 59 | ret[i] = data(i); |
| 60 | } |
| 61 | return ret; |
| 62 | } |
| 63 | |
| 64 | template <typename T> |
| 65 | Status PadKnown(InferenceContext* c, ShapeHandle input, |