Searched refs:final_output_shape (Results 1 – 1 of 1) sorted by relevance
488 Tensor final_output_shape(DT_INT64, TensorShape({output.dims()})); in Compute() local491 final_output_shape.vec<int64>().data()); in Compute()495 context->set_output(2, final_output_shape); in Compute()