Searched defs:EnableAllLayers (Results 1 – 1 of 1) sorted by relevance
1123 struct EnableAllLayers { struct1124 EnableAllLayers(Arena& A, Slice<DimEntry> levels) { in EnableAllLayers() function1144 ~EnableAllLayers() { in ~EnableAllLayers() argument1151 mpy::obj<Tensor> from_batched(Arena& A, at::Tensor batchedtensor, bool has_device) { in from_batched()1179 void inplace_update_layers(TensorRef batchtensor, Slice<DimEntry> levels) { in inplace_update_layers()1194 int64_t levels_start_{};1195 Slice<mpy::hdl<Dim>> levels_to_dim_;