Home
last modified time | relevance | path

Searched refs:constant_tensor_alignment (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/executorch/exir/_serialize/test/
H A Dtest_program.py149 self, constant_tensor_alignment: int
161 self.gen_blob_data(constant_tensor_alignment // 2, b"\x10\x11\x01"),
162 self.gen_blob_data(constant_tensor_alignment - 1, b"\x20\x22\x02"),
163 self.gen_blob_data(constant_tensor_alignment, b"\x30\x33\x03"),
164 self.gen_blob_data(constant_tensor_alignment + 1, b"\x40\x44\x04"),
173 constant_tensor_alignment=constant_tensor_alignment,
204 constant_tensor_alignment * 3 + (constant_tensor_alignment + 1),
215 … constant_tensor_alignment, # tensor[1] has size constant_tensor_alignment // 2. Round up.
216 constant_tensor_alignment
218 constant_tensor_alignment
[all …]
H A Dtest_flatbuffer.py111 constant_tensor_alignment: Optional[int] = None,
124 constant_tensor_alignment=constant_tensor_alignment,
141 SCHEMA_FILES, out_dir, constant_tensor_alignment=128
215 constant_tensor_alignment=1,
260 constant_tensor_alignment=bad_alignment,
/aosp_15_r20/external/executorch/exir/_serialize/
H A D_flatbuffer.py34 constant_tensor_alignment: Optional[int],
56 assert_valid_alignment(constant_tensor_alignment, "constant_tensor_alignment")
69 if constant_tensor_alignment and b"@executorch-tensor-alignment" in line:
70 lines.append(patch_alignment(line, constant_tensor_alignment))
137 constant_tensor_alignment: Optional[int] = None,
154 constant_tensor_alignment=constant_tensor_alignment,
259 constant_tensor_alignment: Optional[int] = None,
279 constant_tensor_alignment=constant_tensor_alignment,
H A D_program.py351 constant_tensor_alignment: Optional[int] = None,
376 if constant_tensor_alignment is None:
377 constant_tensor_alignment = ALIGNMENT
388 program.constant_buffer, tensor_alignment=constant_tensor_alignment
447 constant_tensor_alignment=constant_tensor_alignment,
/aosp_15_r20/external/executorch/exir/program/
H A D_program.py444 constant_tensor_alignment=config.constant_tensor_alignment,
471 constant_tensor_alignment: Optional[int] = None,
485 self._constant_tensor_alignment: Optional[int] = constant_tensor_alignment
494 constant_tensor_alignment=self._constant_tensor_alignment,
1417 constant_tensor_alignment=backend_config.constant_tensor_alignment,
/aosp_15_r20/external/executorch/exir/
H A Dlowered_backend_module.py142 constant_tensor_alignment: Optional[int] = None,
155 constant_tensor_alignment=constant_tensor_alignment,
/aosp_15_r20/external/executorch/exir/capture/
H A D_config.py77 constant_tensor_alignment: Optional[int] = None variable in ExecutorchBackendConfig
/aosp_15_r20/external/executorch/test/models/
H A Dexport_delegated_program.py118 constant_tensor_alignment=constant_tensor_alignemnt,