Searched refs:obj_assignment (Results 1 – 2 of 2) sorted by relevance
46 const ObjectsAssignment<size_t>& obj_assignment) { in ObjectsToOffsets() argument47 size_t num_tensors = obj_assignment.object_ids.size(); in ObjectsToOffsets()48 size_t num_objects = obj_assignment.object_sizes.size(); in ObjectsToOffsets()54 result.total_size += obj_assignment.object_sizes[i]; in ObjectsToOffsets()57 result.offsets[i] = ids_to_offset[obj_assignment.object_ids[i]]; in ObjectsToOffsets()
39 const ObjectsAssignment<size_t>& obj_assignment);