Searched refs:slice_to_load (Results 1 – 1 of 1) sorted by relevance
193 TensorSlice slice_to_load(saved_shape.dims()); in RestoreTensor() local200 shape_spec, &parsed_shape, &slice_to_load, in RestoreTensor()221 reader->CopySliceData(tensor_name, slice_to_load, \ in RestoreTensor()