Searched refs:restored_full_shape (Results 1 – 1 of 1) sorted by relevance
248 TensorShape restored_full_shape; in should_run_in_pool() local251 if (!reader->LookupTensorShape(tensor_name, &restored_full_shape).ok()) { in should_run_in_pool()255 return restored_full_shape.num_elements() > kLargeShapeThreshold; in should_run_in_pool()270 TensorShape restored_full_shape; in run() local272 reader->LookupTensorShape(tensor_name, &restored_full_shape)); in run()275 << restored_full_shape.num_elements(); in run()280 context->allocate_output(idx, restored_full_shape, &restored_tensor)); in run()292 if (!restored_full_shape.IsSameSize(parsed_full_shape)) { in run()297 restored_full_shape.DebugString()); in run()321 << restored_full_shape.num_elements(); in run()[all …]