Possibly convert HWIO to OIHW.

r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi

Create a DataProto from a dict of tensors. This assumes that 1. All the tensor in tensors have the same dim0 2. Only dim0 is the batch

Report the current GPU VRAM usage.

Build convolutional FFN module. Args: in_chs: Number of input channels. hidden_channels: Number of channels after exp

Creates a MobileNet-V3 model. Ref impl: ? Paper: https://arxiv.org/abs/1905.02244 Args: channel_multiplier: multiplier to number o

Return list of available model names, sorted alphabetically Args: filter - Wildcard filter string that works with fnmatch module

Creates an EfficientNet-Lite model. Ref impl: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet/lite Paper: https://

Get ulysses sequence parallel process group.

ResNet-V2 backbone helper

Union with another DataProto. Union batch and meta_info separately. Throw an error if - there are conflict keys in batch and they are