Searched refs:slice_info (Results 1 – 5 of 5) sorted by relevance
94 slice_info = primary._save_slice_info105 slice_info.full_name + "/" + real_slot_name,106 slice_info.full_shape[:n], slice_info.var_offset[:n],107 slice_info.var_shape[:n]))
465 slice_info = v._save_slice_info # pylint:disable=protected-access467 slice_name = slice_info.full_name468 elif slice_name != slice_info.full_name:470 (slice_name, slice_info.full_name))471 _set_checkpoint_initializer(v, ckpt_file, tensor_name, slice_info.spec)
263 slice_info = v._get_save_slice_info()265 if slice_info:267 full_shape=slice_info.full_shape, var_offset=slice_info.var_offset)
215 const auto& slice_info = stack[stack_idx.value()]; in CompleteSlice() local216 uint32_t slice_idx = slice_info.row; in CompleteSlice()
313 if any(slice_info is not None for slice_info in save_slice_info):