Searched refs:tensor_alignment (Results 1 – 5 of 5) sorted by relevance
312 tensor_alignment: Optional[int] = None,333 _padding_required(buffer_length, tensor_alignment)334 if tensor_alignment is not None388 program.constant_buffer, tensor_alignment=constant_tensor_alignment408 tensor_alignment=None, # data is copied at Method load so no need to align.
64 // in TensorMetadata.offset are aligned to tensor_alignment.65 tensor_alignment: uint32;
36 tensor_alignment: int
39 bool preserve_all_tensors, int tensor_alignment, in ArenaPlanner() argument46 tensor_alignment_(tensor_alignment) {} in ArenaPlanner()
53 bool preserve_all_tensors, int tensor_alignment,