Fix compile (#10667)

This commit is contained in:
Dmitri Smirnov 2022-02-25 10:08:30 -08:00 committed by GitHub
parent 12c44bfc4e
commit 957dccb379
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -32,7 +32,7 @@ Status Concat<T>::Compute(OpKernelContext* ctx) const {
input_tensors.push_back(ctx->Input<Tensor>(i));
}
std::vector<int64_t> output_dims = input_tensors[0]->Shape().GetDims();
auto output_dims = input_tensors[0]->Shape().AsShapeVector();
// 'Concat' mode
if (!is_stack_) {