Searched refs:saved_shape (Results 1 – 2 of 2) sorted by relevance
179 TensorShape saved_shape; in RestoreTensor() local181 context, reader->HasTensor(tensor_name, &saved_shape, &type), in RestoreTensor()192 TensorShape output_shape(saved_shape); in RestoreTensor()193 TensorSlice slice_to_load(saved_shape.dims()); in RestoreTensor()202 context, parsed_shape.IsSameSize(saved_shape), in RestoreTensor()207 ", save file shape: ", saved_shape.DebugString())); in RestoreTensor()
145 TensorShape saved_shape(ssm->shape()); in Add()147 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape)); in Add()