Home
last modified time | relevance | path

Searched refs:with_buffer (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/test/distributed/_composable/fsdp/
Dtest_fully_shard_init.py56 model = MLP(8, torch.device("cpu"), with_buffer=True)
80 model = MLP(8, torch.device("cpu"), with_buffer=True)
112 model = MLP(8, torch.device("cpu"), with_buffer=True)
213 model = nn.Sequential(*[MLP(8, with_buffer=True) for _ in range(3)])
226 model = nn.Sequential(*[MLP(8, with_buffer=True) for _ in range(2)])
237 model = nn.Sequential(*[MLP(8, with_buffer=True) for _ in range(5)])
598 model = nn.Sequential(MLP(mlp_dim, with_buffer=True), MLP(mlp_dim))
612 model = nn.Sequential(MLP(mlp_dim, with_buffer=True), MLP(mlp_dim))
633 model = MLP(mlp_dim, with_buffer=True)
685 model = nn.Sequential(MLP(mlp_dim, with_buffer=True), MLP(mlp_dim))
/external/minigbm/cros_gralloc/
Dcros_gralloc_driver.h49 void with_buffer(cros_gralloc_handle_t hnd,
Dcros_gralloc_driver.cc630 void cros_gralloc_driver::with_buffer(cros_gralloc_handle_t hnd, in with_buffer() function in cros_gralloc_driver
/external/minigbm/cros_gralloc/mapper_stablec/
DMapper.cpp300 mDriver->with_buffer(crosHandle, [&](cros_gralloc_buffer* crosBuffer) { in getStandardMetadata()
484 mDriver->with_buffer(crosHandle, [&](cros_gralloc_buffer* crosBuffer) { in setStandardMetadata()
631 mDriver->with_buffer( in dumpBuffer()
664 mDriver->with_buffer(crosHandle, [&](cros_gralloc_buffer* crosBuffer) { in getReservedRegion()
/external/rust/android-crates-io/crates/plotters/src/
Devcxr.rs82 let root = BitMapBackend::with_buffer(&mut buf, size).into_drawing_area(); in evcxr_bitmap_figure()
/external/minigbm/cros_gralloc/gralloc4/
DCrosGralloc4Mapper.cc424 mDriver->with_buffer(crosHandle, [&, this](cros_gralloc_buffer* crosBuffer) { in get()
618 mDriver->with_buffer(crosHandle, [&, this](cros_gralloc_buffer* crosBuffer) { in set()
972 mDriver->with_buffer(crosHandle, [&, this](cros_gralloc_buffer* crosBuffer) { in dumpBuffer()
1104 mDriver->with_buffer(crosHandle, [&](cros_gralloc_buffer* crosBuffer) { in getReservedRegion()
/external/pytorch/torch/testing/_internal/
Dcommon_fsdp.py854 with_buffer: bool = False,
860 if with_buffer: