Searched refs:restored_full_shape (Results 1 – 1 of 1) sorted by relevance
247 TensorShape restored_full_shape; in should_run_in_pool() local250 if (!reader->LookupTensorShape(tensor_name, &restored_full_shape).ok()) { in should_run_in_pool()254 return restored_full_shape.num_elements() > kLargeShapeThreshold; in should_run_in_pool()269 TensorShape restored_full_shape; in run() local271 reader->LookupTensorShape(tensor_name, &restored_full_shape)); in run()274 << restored_full_shape.num_elements(); in run()279 context->allocate_output(idx, restored_full_shape, &restored_tensor)); in run()291 if (!restored_full_shape.IsSameSize(parsed_full_shape)) { in run()296 restored_full_shape.DebugString()); in run()342 TensorShape restored_full_shape; in RestoreTensorsV2() local[all …]