SYMBOL INDEX (975 symbols across 88 files)

FILE: GMeshDiffusion/diffusion_configs/config_lower_occgrid_normalized.py
  function get_config (line 6) | def get_config():

FILE: GMeshDiffusion/diffusion_configs/config_upper_occgrid_normalized.py
  function get_config (line 6) | def get_config():

FILE: GMeshDiffusion/lib/dataset/gshell_dataset.py
  class GShellDataset (line 5) | class GShellDataset(Dataset):
    method __init__ (line 6) | def __init__(self, filepath_metafile, extension='pt'):
    method __len__ (line 14) | def __len__(self):
    method __getitem__ (line 17) | def __getitem__(self, idx):

FILE: GMeshDiffusion/lib/dataset/gshell_dataset_aug.py
  class GShellAugDataset (line 4) | class GShellAugDataset(Dataset):
    method __init__ (line 5) | def __init__(self, FLAGS, extension='pt'):
    method __len__ (line 17) | def __len__(self):
    method __getitem__ (line 20) | def __getitem__(self, idx):
    method collate (line 31) | def collate(data):

FILE: GMeshDiffusion/lib/diffusion/evaler.py
  function uncond_gen (line 15) | def uncond_gen(
  function slerp (line 78) | def slerp(z1, z2, alpha):
  function uncond_gen_interp (line 88) | def uncond_gen_interp(
  function cond_gen (line 183) | def cond_gen(

FILE: GMeshDiffusion/lib/diffusion/likelihood.py
  function get_div_fn (line 26) | def get_div_fn(fn):
  function get_likelihood_fn (line 40) | def get_likelihood_fn(sde, inverse_scaler, hutchinson_type='Rademacher',

FILE: GMeshDiffusion/lib/diffusion/losses.py
  function get_optimizer (line 25) | def get_optimizer(config, params):
  function optimization_manager (line 40) | def optimization_manager(config):
  function get_ddpm_loss_fn (line 60) | def get_ddpm_loss_fn(vpsde, train, loss_type='l2', pred_type='noise', us...
  function get_step_fn (line 194) | def get_step_fn(sde, train, optimize_fn=None, loss_type='l2', pred_type=...

FILE: GMeshDiffusion/lib/diffusion/models/ema.py
  class ExponentialMovingAverage (line 10) | class ExponentialMovingAverage:
    method __init__ (line 15) | def __init__(self, parameters, decay, use_num_updates=True):
    method update (line 32) | def update(self, parameters):
    method copy_to (line 54) | def copy_to(self, parameters):
    method store (line 67) | def store(self, parameters):
    method restore (line 77) | def restore(self, parameters):
    method state_dict (line 92) | def state_dict(self):
    method load_state_dict (line 96) | def load_state_dict(self, state_dict, device='cuda'):

FILE: GMeshDiffusion/lib/diffusion/models/functional.py
  function has_cuda (line 39) | def has_cuda():
  function has_half (line 43) | def has_half():
  function has_bfloat (line 47) | def has_bfloat():
  function has_gemm (line 51) | def has_gemm():
  function enable_tf32 (line 55) | def enable_tf32():
  function disable_tf32 (line 59) | def disable_tf32():
  function enable_tiled_na (line 63) | def enable_tiled_na():
  function disable_tiled_na (line 67) | def disable_tiled_na():
  function enable_gemm_na (line 71) | def enable_gemm_na():
  function disable_gemm_na (line 75) | def disable_gemm_na():
  class NeighborhoodAttention1DQKAutogradFunction (line 79) | class NeighborhoodAttention1DQKAutogradFunction(Function):
    method forward (line 82) | def forward(ctx, query, key, rpb, kernel_size, dilation):
    method backward (line 94) | def backward(ctx, grad_out):
  class NeighborhoodAttention1DAVAutogradFunction (line 107) | class NeighborhoodAttention1DAVAutogradFunction(Function):
    method forward (line 110) | def forward(ctx, attn, value, kernel_size, dilation):
    method backward (line 121) | def backward(ctx, grad_out):
  class NeighborhoodAttention2DQKAutogradFunction (line 133) | class NeighborhoodAttention2DQKAutogradFunction(Function):
    method forward (line 136) | def forward(ctx, query, key, rpb, kernel_size, dilation):
    method backward (line 150) | def backward(ctx, grad_out):
  class NeighborhoodAttention2DAVAutogradFunction (line 163) | class NeighborhoodAttention2DAVAutogradFunction(Function):
    method forward (line 166) | def forward(ctx, attn, value, kernel_size, dilation):
    method backward (line 177) | def backward(ctx, grad_out):
  class NeighborhoodAttention3DQKAutogradFunction (line 189) | class NeighborhoodAttention3DQKAutogradFunction(Function):
    method forward (line 192) | def forward(ctx, query, key, rpb, kernel_size_d, kernel_size, dilation...
    method backward (line 208) | def backward(ctx, grad_out):
  class NeighborhoodAttention3DAVAutogradFunction (line 223) | class NeighborhoodAttention3DAVAutogradFunction(Function):
    method forward (line 226) | def forward(ctx, attn, value, kernel_size_d, kernel_size, dilation_d, ...
    method backward (line 241) | def backward(ctx, grad_out):
  function natten1dqkrpb (line 255) | def natten1dqkrpb(query, key, rpb, kernel_size, dilation):
  function natten1dqk (line 261) | def natten1dqk(query, key, kernel_size, dilation):
  function natten1dav (line 267) | def natten1dav(attn, value, kernel_size, dilation):
  function natten2dqkrpb (line 273) | def natten2dqkrpb(query, key, rpb, kernel_size, dilation):
  function natten2dqk (line 279) | def natten2dqk(query, key, kernel_size, dilation):
  function natten2dav (line 285) | def natten2dav(attn, value, kernel_size, dilation):
  function natten3dqkrpb (line 291) | def natten3dqkrpb(query, key, rpb, kernel_size_d, kernel_size, dilation_...
  function natten3dqk (line 297) | def natten3dqk(query, key, kernel_size_d, kernel_size, dilation_d, dilat...
  function natten3dav (line 303) | def natten3dav(attn, value, kernel_size_d, kernel_size, dilation_d, dila...

FILE: GMeshDiffusion/lib/diffusion/models/layers.py
  class GroupNormFloat32 (line 28) | class GroupNormFloat32(nn.GroupNorm):
    method forward (line 29) | def forward(self, input):
  function get_act_fn (line 35) | def get_act_fn(act_name):
  function variance_scaling (line 49) | def variance_scaling(scale, mode, distribution,
  function default_init (line 83) | def default_init(scale=1.):
  class Dense (line 89) | class Dense(nn.Module):
    method __init__ (line 91) | def __init__(self):
  function conv1x1 (line 95) | def conv1x1(in_planes, out_planes, stride=1, bias=True, init_scale=1., p...
  function conv3x3 (line 102) | def conv3x3(in_planes, out_planes, stride=1, bias=True, dilation=1, init...
  function conv5x5 (line 110) | def conv5x5(in_planes, out_planes, stride=2, bias=True, dilation=1, init...
  function conv3x3_transposed (line 119) | def conv3x3_transposed(in_planes, out_planes, stride=2, bias=True, dilat...
  function conv5x5_transposed (line 127) | def conv5x5_transposed(in_planes, out_planes, stride=2, bias=True, dilat...
  function get_timestep_embedding (line 141) | def get_timestep_embedding(timesteps, embedding_dim, max_positions=10000):
  class AttnBlock (line 158) | class AttnBlock(nn.Module):
    method __init__ (line 160) | def __init__(self, channels, num_groups=32):
    method forward (line 168) | def forward(self, x):
  class Upsample (line 192) | class Upsample(nn.Module):
    method __init__ (line 193) | def __init__(self, channels, with_conv=False):
    method forward (line 199) | def forward(self, x, temb=None):
  class Downsample (line 207) | class Downsample(nn.Module):
    method __init__ (line 208) | def __init__(self, channels, with_conv=False):
    method forward (line 214) | def forward(self, x, temb=None):
  class ResBlock (line 227) | class ResBlock(nn.Module):
    method __init__ (line 229) | def __init__(self, act_fn, in_ch, out_ch=None, temb_dim=None, conv_sho...
    method forward (line 253) | def forward(self, x, temb=None):
  class AttnResBlock (line 272) | class AttnResBlock(ResBlock):
    method __init__ (line 274) | def __init__(self, act_fn, in_ch, out_ch, temb_dim=None, conv_shortcut...
    method forward (line 278) | def forward(self, x, temb=None):

FILE: GMeshDiffusion/lib/diffusion/models/normalization.py
  function get_normalization (line 22) | def get_normalization(config, conditional=False):
  class ConditionalBatchNorm3d (line 43) | class ConditionalBatchNorm3d(nn.Module):
    method __init__ (line 44) | def __init__(self, num_features, num_classes, bias=True):
    method forward (line 57) | def forward(self, x, y):
  class ConditionalInstanceNorm3d (line 68) | class ConditionalInstanceNorm3d(nn.Module):
    method __init__ (line 69) | def __init__(self, num_features, num_classes, bias=True):
    method forward (line 82) | def forward(self, x, y):
  class ConditionalVarianceNorm3d (line 93) | class ConditionalVarianceNorm3d(nn.Module):
    method __init__ (line 94) | def __init__(self, num_features, num_classes, bias=False):
    method forward (line 101) | def forward(self, x, y):
  class VarianceNorm3d (line 110) | class VarianceNorm3d(nn.Module):
    method __init__ (line 111) | def __init__(self, num_features, bias=False):
    method forward (line 118) | def forward(self, x):
  class ConditionalNoneNorm3d (line 126) | class ConditionalNoneNorm3d(nn.Module):
    method __init__ (line 127) | def __init__(self, num_features, num_classes, bias=True):
    method forward (line 139) | def forward(self, x, y):
  class NoneNorm3d (line 149) | class NoneNorm3d(nn.Module):
    method __init__ (line 150) | def __init__(self, num_features, bias=True):
    method forward (line 153) | def forward(self, x):
  class InstanceNorm3dPlus (line 157) | class InstanceNorm3dPlus(nn.Module):
    method __init__ (line 158) | def __init__(self, num_features, bias=True):
    method forward (line 170) | def forward(self, x):
  class ConditionalInstanceNorm3dPlus (line 186) | class ConditionalInstanceNorm3dPlus(nn.Module):
    method __init__ (line 187) | def __init__(self, num_features, num_classes, bias=True):
    method forward (line 200) | def forward(self, x, y):

FILE: GMeshDiffusion/lib/diffusion/models/unet3d_occgrid.py
  function str_to_class (line 33) | def str_to_class(classname):
  class UNet3D (line 38) | class UNet3D(nn.Module):
    method __init__ (line 39) | def __init__(self, config):
    method sequentially_call_module (line 142) | def sequentially_call_module(self, idx, x, temb=None):
    method forward (line 145) | def forward(self, x, labels):

FILE: GMeshDiffusion/lib/diffusion/models/utils.py
  function register_model (line 27) | def register_model(cls=None, *, name=None):
  function get_model (line 46) | def get_model(name):
  function get_sigmas (line 50) | def get_sigmas(config):
  function get_ddpm_params (line 63) | def get_ddpm_params(config):
  function create_model (line 88) | def create_model(config, use_parallel=True, ddp=False, rank=None):
  function get_model_fn (line 111) | def get_model_fn(model, train=False):
  function get_reg_fn (line 142) | def get_reg_fn(model, train=False):
  function get_score_fn (line 179) | def get_score_fn(sde, model, train=False, continuous=False, std_scale=Tr...
  function to_flattened_numpy (line 236) | def to_flattened_numpy(x):
  function from_flattened_numpy (line 241) | def from_flattened_numpy(x, shape):

FILE: GMeshDiffusion/lib/diffusion/sampling.py
  function register_predictor (line 37) | def register_predictor(cls=None, *, name=None):
  function register_corrector (line 56) | def register_corrector(cls=None, *, name=None):
  function get_predictor (line 75) | def get_predictor(name):
  function get_corrector (line 79) | def get_corrector(name):
  function get_sampling_fn (line 83) | def get_sampling_fn(config, sde, shape, inverse_scaler, eps, grid_mask=N...
  class Predictor (line 139) | class Predictor(abc.ABC):
    method __init__ (line 142) | def __init__(self, sde, score_fn, probability_flow=False):
    method update_fn (line 150) | def update_fn(self, x, t):
  class Corrector (line 164) | class Corrector(abc.ABC):
    method __init__ (line 167) | def __init__(self, sde, score_fn, snr, n_steps):
    method update_fn (line 175) | def update_fn(self, x, t):
  class EulerMaruyamaPredictor (line 190) | class EulerMaruyamaPredictor(Predictor):
    method __init__ (line 191) | def __init__(self, sde, score_fn, probability_flow=False):
    method update_fn (line 194) | def update_fn(self, x, t):
  class ReverseDiffusionPredictor (line 204) | class ReverseDiffusionPredictor(Predictor):
    method __init__ (line 205) | def __init__(self, sde, score_fn, probability_flow=False):
    method update_fn (line 208) | def update_fn(self, x, t):
  class AncestralSamplingPredictor (line 217) | class AncestralSamplingPredictor(Predictor):
    method __init__ (line 220) | def __init__(self, sde, score_fn, probability_flow=False):
    method vpsde_update_fn (line 226) | def vpsde_update_fn(self, x, t):
    method update_fn (line 236) | def update_fn(self, x, t):
  class NonePredictor (line 244) | class NonePredictor(Predictor):
    method __init__ (line 247) | def __init__(self, sde, score_fn, probability_flow=False):
    method update_fn (line 250) | def update_fn(self, x, t):
  class DDIMPredictor (line 254) | class DDIMPredictor(Predictor):
    method __init__ (line 255) | def __init__(self, sde, score_fn, probability_flow=False):
    method update_fn (line 259) | def update_fn(self, x, t, tprev=None):
  class LangevinCorrector (line 264) | class LangevinCorrector(Corrector):
    method __init__ (line 265) | def __init__(self, sde, score_fn, snr, n_steps):
    method update_fn (line 270) | def update_fn(self, x, t):
  class AnnealedLangevinDynamics (line 294) | class AnnealedLangevinDynamics(Corrector):
    method __init__ (line 300) | def __init__(self, sde, score_fn, snr, n_steps):
    method update_fn (line 305) | def update_fn(self, x, t):
  class NoneCorrector (line 329) | class NoneCorrector(Corrector):
    method __init__ (line 332) | def __init__(self, sde, score_fn, snr, n_steps):
    method update_fn (line 335) | def update_fn(self, x, t):
  function shared_predictor_update_fn (line 339) | def shared_predictor_update_fn(x, t, sde, model, predictor, probability_...
  function shared_corrector_update_fn (line 350) | def shared_corrector_update_fn(x, t, sde, model, corrector, continuous, ...
  function get_pc_sampler (line 361) | def get_pc_sampler(sde, shape, predictor, corrector, inverse_scaler, snr,
  function ddim_predictor_update_fn (line 508) | def ddim_predictor_update_fn(x, t, tprev, sde, model, predictor, probabi...
  function get_ddim_sampler (line 519) | def get_ddim_sampler(sde, shape, predictor, inverse_scaler, n_steps=1,

FILE: GMeshDiffusion/lib/diffusion/sde_lib.py
  class SDE (line 9) | class SDE(abc.ABC):
    method __init__ (line 12) | def __init__(self, N):
    method T (line 23) | def T(self):
    method sde (line 28) | def sde(self, x, t):
    method marginal_prob (line 32) | def marginal_prob(self, x, t):
    method prior_sampling (line 37) | def prior_sampling(self, shape):
    method prior_logp (line 42) | def prior_logp(self, z):
    method discretize (line 54) | def discretize(self, x, t):
    method reverse (line 73) | def reverse(self, score_fn, probability_flow=False):
  class VPSDE (line 209) | class VPSDE(SDE):
    method __init__ (line 210) | def __init__(self, beta_min=0.1, beta_max=20, N=1000):
    method T (line 234) | def T(self):
    method sde (line 237) | def sde(self, x, t):
    method marginal_prob (line 243) | def marginal_prob(self, x, t):
    method prior_sampling (line 249) | def prior_sampling(self, shape):
    method prior_logp (line 252) | def prior_logp(self, z):
    method discretize (line 258) | def discretize(self, x, t):

FILE: GMeshDiffusion/lib/diffusion/trainer.py
  function train (line 20) | def train(config):

FILE: GMeshDiffusion/lib/diffusion/trainer_ddp.py
  function train (line 22) | def train(config):

FILE: GMeshDiffusion/lib/diffusion/utils.py
  function restore_checkpoint (line 6) | def restore_checkpoint(ckpt_dir, state, device, strict=False, rank=None):
  function save_checkpoint (line 38) | def save_checkpoint(ckpt_dir, state):

FILE: GMeshDiffusion/main_diffusion.py
  function main (line 19) | def main(argv):

FILE: GMeshDiffusion/main_diffusion_ddp.py
  function main (line 20) | def main(argv):

FILE: GMeshDiffusion/metadata/save_tet_info.py
  function tet_to_grids (line 14) | def tet_to_grids(vertices, values_list, grid_size):

FILE: GMeshDiffusion/metadata/tet_to_cubic_grid_dataset.py
  function tet_to_grids (line 7) | def tet_to_grids(vertices, values_list, grid_size):

FILE: data/tets/generate_tets.py
  function generate_tetrahedron_grid_file (line 21) | def generate_tetrahedron_grid_file(res=32, root='..'):
  function convert_from_quartet_to_npz (line 31) | def convert_from_quartet_to_npz(quartetfile = 'cube_32_tet.tet', npzfile...

FILE: dataset/dataset.py
  class Dataset (line 12) | class Dataset(torch.utils.data.Dataset):
    method __init__ (line 14) | def __init__(self):
    method __len__ (line 17) | def __len__(self):
    method __getitem__ (line 20) | def __getitem__(self):
    method collate (line 23) | def collate(self, batch):

FILE: dataset/dataset_deepfashion.py
  function load_K_Rt_from_P (line 24) | def load_K_Rt_from_P(filename, P=None):
  function _load_img (line 48) | def _load_img(path):
  class DatasetDeepFashion (line 59) | class DatasetDeepFashion(Dataset):
    method __init__ (line 60) | def __init__(self, base_dir, FLAGS, examples=None):
    method _parse_frame (line 101) | def _parse_frame(self, idx):
    method __len__ (line 121) | def __len__(self):
    method __getitem__ (line 124) | def __getitem__(self, itr):

FILE: dataset/dataset_deepfashion_testset.py
  function load_K_Rt_from_P (line 24) | def load_K_Rt_from_P(filename, P=None):
  function _load_img (line 48) | def _load_img(path):
  function _load_mask (line 58) | def _load_mask(path):
  class DatasetDeepFashionTestset (line 67) | class DatasetDeepFashionTestset(Dataset):
    method __init__ (line 68) | def __init__(self, base_dir, FLAGS, examples=None):
    method _parse_frame (line 101) | def _parse_frame(self, idx):
    method __len__ (line 119) | def __len__(self):
    method __getitem__ (line 122) | def __getitem__(self, itr):

FILE: dataset/dataset_llff.py
  function _load_mask (line 20) | def _load_mask(fn):
  function _load_img (line 26) | def _load_img(fn):
  class DatasetLLFF (line 39) | class DatasetLLFF(Dataset):
    method __init__ (line 40) | def __init__(self, base_dir, FLAGS, examples=None):
    method _parse_frame (line 75) | def _parse_frame(self, idx):
    method __len__ (line 93) | def __len__(self):
    method __getitem__ (line 96) | def __getitem__(self, itr):

FILE: dataset/dataset_mesh.py
  class DatasetMesh (line 24) | class DatasetMesh(Dataset):
    method __init__ (line 26) | def __init__(self, ref_mesh, glctx, cam_radius, FLAGS, validate=False,...
    method _rotate_scene (line 54) | def _rotate_scene(self, itr):
    method _random_scene (line 65) | def _random_scene(self):
    method __len__ (line 89) | def __len__(self):
    method __getitem__ (line 92) | def __getitem__(self, itr):

FILE: dataset/dataset_nerf.py
  function _load_img (line 25) | def _load_img(path):
  class DatasetNERF (line 36) | class DatasetNERF(Dataset):
    method __init__ (line 37) | def __init__(self, cfg_path, FLAGS, examples=None):
    method _parse_frame (line 59) | def _parse_frame(self, cfg, idx):
    method __len__ (line 73) | def __len__(self):
    method __getitem__ (line 76) | def __getitem__(self, itr):

FILE: dataset/dataset_nerf_colmap.py
  function _load_img (line 25) | def _load_img(path):
  class DatasetNERF (line 34) | class DatasetNERF(Dataset):
    method __init__ (line 35) | def __init__(self, cfg_path, FLAGS, examples=None):
    method _parse_frame (line 57) | def _parse_frame(self, cfg, idx):
    method __len__ (line 73) | def __len__(self):
    method __getitem__ (line 76) | def __getitem__(self, itr):

FILE: denoiser/denoiser.py
  class BilateralDenoiser (line 21) | class BilateralDenoiser(torch.nn.Module):
    method __init__ (line 22) | def __init__(self, influence=1.0):
    method set_influence (line 26) | def set_influence(self, factor):
    method forward (line 31) | def forward(self, input):

FILE: geometry/embedding.py
  class Embedding (line 4) | class Embedding(nn.Module):
    method __init__ (line 5) | def __init__(self, in_channels, N_freqs, logscale=True):
    method forward (line 21) | def forward(self, x):

FILE: geometry/gshell_flexicubes.py
  class GShellFlexiCubes (line 16) | class GShellFlexiCubes:
    method __init__ (line 67) | def __init__(self, device="cuda", qef_reg_scale=1e-3, weight_scale=0.99):
    method construct_voxel_grid (line 103) | def construct_voxel_grid(self, res):
    method __call__ (line 136) | def __call__(self, x_nx3, s_n, nu_n, cube_fx8, res, beta_fx12=None, al...
    method _compute_reg_loss (line 232) | def _compute_reg_loss(self, vd, ue, edge_group_to_vd, vd_num_edges):
    method _normalize_weights (line 242) | def _normalize_weights(self, beta_fx12, alpha_fx8, gamma_f, surf_cubes):
    method _get_case_id (line 266) | def _get_case_id(self, occ_fx8, surf_cubes, res):
    method _identify_surf_edges (line 309) | def _identify_surf_edges(self, s_n, cube_fx8, surf_cubes):
    method _identify_surf_cubes (line 334) | def _identify_surf_cubes(self, s_n, cube_fx8):
    method _linear_interp (line 345) | def _linear_interp(self, edges_weight, edges_x):
    method _linear_interp_nonan (line 357) | def _linear_interp_nonan(self, edges_weight, edges_x):
    method _solve_vd_QEF (line 373) | def _solve_vd_QEF(self, p_bxnx3, norm_bxnx3, c_bx3=None):
    method _compute_vd (line 387) | def _compute_vd(self, x_nx3, surf_cubes_fx8, surf_edges, s_n, nu_n, ca...
    method _triangulate (line 487) | def _triangulate(self, s_n, surf_edges, vd, nu_d, nu_d_stopvgd, vd_gam...
    method _triangulate_msdf (line 554) | def _triangulate_msdf(self, vertices, faces, nu_n, nu_n_stopvgd):
    method _tetrahedralize (line 593) | def _tetrahedralize(

FILE: geometry/gshell_flexicubes_geometry.py
  function compute_sdf_reg_loss (line 33) | def compute_sdf_reg_loss(sdf, all_edges):
  class GShellFlexiCubesGeometry (line 45) | class GShellFlexiCubesGeometry(torch.nn.Module):
    method __init__ (line 46) | def __init__(self, grid_res, scale, FLAGS):
    method generate_edges (line 111) | def generate_edges(self):
    method getAABB (line 120) | def getAABB(self):
    method clamp_deform (line 124) | def clamp_deform(self):
    method map_uv2 (line 130) | def map_uv2(self, faces):
    method map_uv (line 136) | def map_uv(self, face_gidx, max_idx):
    method getMesh (line 166) | def getMesh(self, material, _training=False):
    method render (line 210) | def render(self, glctx, target, lgt, opt_material, bsdf=None, denoiser...
    method tick (line 237) | def tick(self, glctx, target, lgt, opt_material, loss_fn, iteration, d...

FILE: geometry/gshell_tets.py
  function auto_normals (line 9) | def auto_normals(v_pos, t_pos_idx):
  function compute_tangents (line 40) | def compute_tangents(v_pos, v_tex, v_nrm, t_pos_idx, t_tex_idx, t_nrm_idx):
  class GShell_Tets (line 80) | class GShell_Tets:
    method __init__ (line 81) | def __init__(self):
    method sort_edges (line 200) | def sort_edges(self, edges_ex2):
    method map_uv (line 210) | def map_uv(self, face_gidx, max_idx):
    method __call__ (line 245) | def __call__(self, pos_nx3, sdf_n, msdf_n, tet_fx4, output_watertight_...
    method marching_from_auggrid (line 447) | def marching_from_auggrid(self, pos_nx3, sdf_n, tet_fx4,

FILE: geometry/gshell_tets_geometry.py
  function compute_sdf_reg_loss (line 33) | def compute_sdf_reg_loss(sdf, all_edges):
  class GShellTetsGeometry (line 45) | class GShellTetsGeometry(torch.nn.Module):
    method __init__ (line 46) | def __init__(self, grid_res, scale, FLAGS, offset=None, tet_init_file=...
    method generate_edges (line 149) | def generate_edges(self):
    method getAABB (line 158) | def getAABB(self):
    method clamp_deform (line 162) | def clamp_deform(self):
    method getMesh_from_augmented_grid_withocc (line 167) | def getMesh_from_augmented_grid_withocc(self, material, sdf_sign, sdf_...
    method getMesh (line 191) | def getMesh(self, material):
    method render (line 230) | def render(self, glctx, target, lgt, opt_material, bsdf=None, denoiser...
    method tick (line 257) | def tick(self, glctx, target, lgt, opt_material, loss_fn, iteration, d...

FILE: geometry/mlp.py
  class MLP (line 7) | class MLP(nn.Module):
    method __init__ (line 8) | def __init__(self, n_freq=6, d_hidden=128, d_out=1, n_hidden=3, skip_i...
    method forward (line 32) | def forward(self, x):

FILE: render/light.py
  class EnvironmentLight (line 21) | class EnvironmentLight:
    method __init__ (line 27) | def __init__(self, base):
    method xfm (line 34) | def xfm(self, mtx):
    method parameters (line 37) | def parameters(self):
    method clone (line 40) | def clone(self):
    method clamp_ (line 43) | def clamp_(self, min=None, max=None):
    method update_pdf (line 46) | def update_pdf(self):
    method generate_image (line 62) | def generate_image(self, res):
  function _load_env_hdr (line 71) | def _load_env_hdr(fn, scale=1.0, res=None, trainable=False):
  function load_env (line 86) | def load_env(fn, scale=1.0, res=None, trainable=False):
  function save_env_map (line 93) | def save_env_map(fn, light):
  function create_trainable_env_rnd (line 102) | def create_trainable_env_rnd(base_res, scale=0.5, bias=0.25):

FILE: render/material.py
  function load_mtl (line 21) | def load_mtl(fn, clear_ks=True):
  function save_mtl (line 72) | def save_mtl(fn, material):
  function create_trainable (line 99) | def create_trainable(material):
  function get_parameters (line 106) | def get_parameters(material):
  function _upscale_replicate (line 117) | def _upscale_replicate(x, full_res):
  function merge_materials (line 122) | def merge_materials(materials, texcoords, tfaces, mfaces):

FILE: render/mesh.py
  class Mesh (line 20) | class Mesh:
    method __init__ (line 21) | def __init__(self, v_pos=None, t_pos_idx=None, v_nrm=None, t_nrm_idx=N...
    method copy_none (line 35) | def copy_none(self, other):
    method clone (line 55) | def clone(self):
  function load_mesh (line 79) | def load_mesh(filename, mtl_override=None, mtl_default=None, mtl_type_ov...
  function aabb (line 88) | def aabb(mesh):
  function aabb_clean (line 94) | def aabb_clean(mesh):
  function compute_edges (line 101) | def compute_edges(attr_idx, return_inverse=False):
  function compute_edge_to_face_mapping (line 123) | def compute_edge_to_face_mapping(attr_idx, return_inverse=False):
  function unit_size (line 158) | def unit_size(mesh):
  function center_by_reference (line 170) | def center_by_reference(base_mesh, ref_aabb, scale):
  function center_by_reference_noscale (line 177) | def center_by_reference_noscale(base_mesh, ref_aabb, scale=None):
  function center_with_global_aabb (line 183) | def center_with_global_aabb(base_mesh, ref_aabb, scale):
  function center_with_global_aabb_perdim (line 191) | def center_with_global_aabb_perdim(base_mesh, ref_aabb, scale):
  function scale_with_global_aabb (line 198) | def scale_with_global_aabb(base_mesh, ref_aabb, scale):
  function scale_with_global_aabb_perdim (line 204) | def scale_with_global_aabb_perdim(base_mesh, ref_aabb, scale):
  function auto_normals (line 212) | def auto_normals(imesh):
  function compute_tangents (line 243) | def compute_tangents(imesh, v_tng=None):

FILE: render/mlptexture.py
  class _MLP (line 18) | class _MLP(torch.nn.Module):
    method __init__ (line 19) | def __init__(self, cfg, loss_scale=1.0):
    method forward (line 33) | def forward(self, x):
    method _init_weights (line 37) | def _init_weights(m):
  class MLPTexture3D (line 47) | class MLPTexture3D(torch.nn.Module):
    method __init__ (line 48) | def __init__(self, AABB, channels = 3, internal_dims = 32, hidden = 2,...
    method sample (line 87) | def sample(self, texc):
    method clamp_ (line 101) | def clamp_(self):
    method cleanup (line 104) | def cleanup(self):

FILE: render/obj.py
  function _find_mat (line 21) | def _find_mat(materials, name):
  function load_obj (line 31) | def load_obj(filename, clear_ks=True, mtl_override=None, mtl_default=Non...
  function write_obj (line 143) | def write_obj(folder, mesh, save_material=True):

FILE: render/optixutils/c_src/accessor.h
  type T (line 122) | typedef T* PtrType;
  type T (line 128) | typedef T* __restrict__ PtrType;
  function C10_HOST_DEVICE (line 150) | C10_HOST_DEVICE index_t size(index_t i) const {
  function C10_HOST_DEVICE (line 153) | C10_HOST_DEVICE PtrType data() {
  function PtrTraits (line 184) | PtrTraits, index_t> operator[](index_t i) const {
  function C10_HOST_DEVICE (line 199) | C10_HOST_DEVICE T & operator[](index_t i) {
  function C10_HOST_DEVICE (line 203) | C10_HOST_DEVICE const T & operator[](index_t i) const {
  function C10_HOST (line 222) | C10_HOST GenericPackedTensorAccessorBase() {}
  function C10_HOST (line 224) | C10_HOST GenericPackedTensorAccessorBase(
  function C10_HOST_DEVICE (line 246) | C10_HOST_DEVICE index_t stride(index_t i) const {
  function C10_HOST_DEVICE (line 249) | C10_HOST_DEVICE index_t size(index_t i) const {
  function C10_HOST_DEVICE (line 252) | C10_HOST_DEVICE PtrType data() {
  function C10_HOST (line 270) | C10_HOST GenericPackedTensorAccessor() : GenericPackedTensorAccessorBase...
  function C10_DEVICE (line 323) | C10_DEVICE T & operator[](index_t i) {
  function C10_DEVICE (line 326) | C10_DEVICE const T& operator[](index_t i) const {

FILE: render/optixutils/c_src/bsdf.h
  function __device__ (line 21) | __device__ inline float fwdLambert(const float3 nrm, const float3 wi)
  function __device__ (line 26) | __device__ inline void bwdLambert(const float3 nrm, const float3 wi, flo...
  function __device__ (line 35) | __device__ inline float fwdFresnelSchlick(const float f0, const float f9...
  function __device__ (line 42) | __device__ inline void bwdFresnelSchlick(const float f0, const float f90...
  function __device__ (line 54) | __device__ inline float3 fwdFresnelSchlick(const float3 f0, const float3...
  function __device__ (line 61) | __device__ inline void bwdFresnelSchlick(const float3 f0, const float3 f...
  function __device__ (line 76) | __device__ inline float fwdNdfGGX(const float alphaSqr, const float cosT...
  function __device__ (line 83) | __device__ inline void bwdNdfGGX(const float alphaSqr, const float cosTh...
  function __device__ (line 98) | __device__ inline float fwdLambdaGGX(const float alphaSqr, const float c...
  function __device__ (line 107) | __device__ inline void bwdLambdaGGX(const float alphaSqr, const float co...
  function __device__ (line 122) | __device__ inline float fwdMaskingSmithGGXCorrelated(const float alphaSq...
  function __device__ (line 129) | __device__ inline void bwdMaskingSmithGGXCorrelated(const float alphaSqr...
  function __device__ (line 144) | __device__ float3 fwdPbrSpecular(const float3 col, const float3 nrm, con...
  function __device__ (line 164) | __device__ void bwdPbrSpecular(
  function __device__ (line 222) | __device__ void fwdPbrBSDF(const float3 kd, const float3 arm, const floa...
  function __device__ (line 238) | __device__ void bwdPbrBSDF(

FILE: render/optixutils/c_src/common.h
  function __device__ (line 14) | __device__ inline float3 fetch3(const T &tensor, U idx, Args... args) {
  function float3 (line 17) | inline float3 fetch3(const T &tensor) {
  function __device__ (line 22) | __device__ inline float2 fetch2(const T &tensor, U idx, Args... args) {
  function float2 (line 25) | inline float2 fetch2(const T &tensor) {

FILE: render/optixutils/c_src/denoising.h
  type BilateralDenoiserParams (line 12) | struct BilateralDenoiserParams

FILE: render/optixutils/c_src/envsampling/params.h
  type EnvSamplingParams (line 11) | struct EnvSamplingParams

FILE: render/optixutils/c_src/math_utils.h
  function T (line 13) | __inline__ T clamp(T x, T _min, T _max) { return min(_max, max(_min, x)); }
  function __device__ (line 14) | static __device__ inline float3 make_float3(float a) { return make_float...
  function __device__ (line 80) | static __device__ inline float sum(float3 a)
  function __device__ (line 85) | static __device__ inline float dot(float3 a, float3 b) { return a.x * b....
  function __device__ (line 87) | static __device__ inline void bwd_dot(float3 a, float3 b, float3& d_a, f...
  function __device__ (line 93) | static __device__ inline float luminance(const float3 rgb)
  function __device__ (line 98) | static __device__ inline float3 cross(float3 a, float3 b)
  function __device__ (line 107) | static __device__ inline void bwd_cross(float3 a, float3 b, float3 &d_a,...
  function __device__ (line 118) | static __device__ inline float3 reflect(float3 x, float3 n)
  function __device__ (line 123) | static __device__ inline void bwd_reflect(float3 x, float3 n, float3& d_...
  function __device__ (line 134) | static __device__ inline float3 safe_normalize(float3 v)
  function __device__ (line 140) | static __device__ inline void bwd_safe_normalize(const float3 v, float3&...
  function __device__ (line 155) | static __device__ inline void branchlessONB(const float3 &n, float3 &b1,...

FILE: render/optixutils/c_src/optix_wrapper.cpp
  function context_log_cb (line 43) | static void context_log_cb( unsigned int level, const char* tag, const c...
  function readSourceFile (line 49) | static bool readSourceFile( std::string& str, const std::string& filename )
  function getCuStringFromFile (line 63) | static void getCuStringFromFile( std::string& cu, const char* filename )
  function getPtxFromCuString (line 73) | static void getPtxFromCuString( std::string& ptx, const char* include_di...
  type SbtRecord (line 144) | struct SbtRecord
  function createPipeline (line 149) | void createPipeline(const OptixDeviceContext context, const std::string&...

FILE: render/optixutils/c_src/optix_wrapper.h
  type OptiXState (line 17) | struct OptiXState
  function class (line 30) | class OptiXStateWrapper

FILE: render/optixutils/c_src/torch_bindings.cpp
  function optix_build_bvh (line 37) | void optix_build_bvh(OptiXStateWrapper& stateWrapper,torch::Tensor grid_...
  function packed_accessor32 (line 118) | PackedTensorAccessor32<T, N> packed_accessor32(torch::Tensor tensor)
  function env_shade_fwd (line 123) | std::tuple<torch::Tensor, torch::Tensor> env_shade_fwd(
  function env_shade_bwd (line 190) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, t...
  function bilateral_denoiser_fwd (line 274) | torch::Tensor bilateral_denoiser_fwd(torch::Tensor col, torch::Tensor nr...
  function bilateral_denoiser_bwd (line 297) | torch::Tensor bilateral_denoiser_bwd(torch::Tensor col, torch::Tensor nr...
  function PYBIND11_MODULE (line 321) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: render/optixutils/include/internal/optix_7_device_impl.h
  function optixTrace (line 39) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 78) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 118) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 159) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 201) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 244) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 288) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 333) | void optixTrace( OptixTraversableHandle handle,
  function optixTrace (line 379) | void optixTrace( OptixTraversableHandle handle,
  function optixSetPayload_0 (line 427) | void optixSetPayload_0( unsigned int p )
  function optixSetPayload_1 (line 432) | void optixSetPayload_1( unsigned int p )
  function optixSetPayload_2 (line 437) | void optixSetPayload_2( unsigned int p )
  function optixSetPayload_3 (line 442) | void optixSetPayload_3( unsigned int p )
  function optixSetPayload_4 (line 447) | void optixSetPayload_4( unsigned int p )
  function optixSetPayload_5 (line 452) | void optixSetPayload_5( unsigned int p )
  function optixSetPayload_6 (line 457) | void optixSetPayload_6( unsigned int p )
  function optixSetPayload_7 (line 462) | void optixSetPayload_7( unsigned int p )
  function optixGetPayload_0 (line 468) | unsigned int optixGetPayload_0()
  function optixGetPayload_1 (line 475) | unsigned int optixGetPayload_1()
  function optixGetPayload_2 (line 482) | unsigned int optixGetPayload_2()
  function optixGetPayload_3 (line 489) | unsigned int optixGetPayload_3()
  function optixGetPayload_4 (line 496) | unsigned int optixGetPayload_4()
  function optixGetPayload_5 (line 503) | unsigned int optixGetPayload_5()
  function optixGetPayload_6 (line 510) | unsigned int optixGetPayload_6()
  function optixGetPayload_7 (line 517) | unsigned int optixGetPayload_7()
  function optixUndefinedValue (line 525) | unsigned int optixUndefinedValue()
  function float3 (line 532) | float3 optixGetWorldRayOrigin()
  function float3 (line 541) | float3 optixGetWorldRayDirection()
  function float3 (line 550) | float3 optixGetObjectRayOrigin()
  function float3 (line 559) | float3 optixGetObjectRayDirection()
  function optixGetRayTmin (line 568) | float optixGetRayTmin()
  function optixGetRayTmax (line 575) | float optixGetRayTmax()
  function optixGetRayTime (line 582) | float optixGetRayTime()
  function optixGetRayFlags (line 589) | unsigned int optixGetRayFlags()
  function optixGetRayVisibilityMask (line 596) | unsigned int optixGetRayVisibilityMask()
  function OptixTraversableHandle (line 603) | OptixTraversableHandle optixGetInstanceTraversableFromIAS( OptixTraversa...
  function optixGetTriangleVertexData (line 613) | void optixGetTriangleVertexData( OptixTraversableHandle gas,
  function optixGetLinearCurveVertexData (line 627) | void optixGetLinearCurveVertexData( OptixTraversableHandle gas,
  function optixGetQuadraticBSplineVertexData (line 641) | void optixGetQuadraticBSplineVertexData( OptixTraversableHandle gas,
  function optixGetCubicBSplineVertexData (line 656) | void optixGetCubicBSplineVertexData( OptixTraversableHandle gas,
  function OptixTraversableHandle (line 673) | OptixTraversableHandle optixGetGASTraversableHandle()
  function optixGetGASMotionTimeBegin (line 680) | float optixGetGASMotionTimeBegin( OptixTraversableHandle handle )
  function optixGetGASMotionTimeEnd (line 687) | float optixGetGASMotionTimeEnd( OptixTraversableHandle handle )
  function optixGetGASMotionStepCount (line 694) | unsigned int optixGetGASMotionStepCount( OptixTraversableHandle handle )
  function optixGetWorldToObjectTransformMatrix (line 701) | void optixGetWorldToObjectTransformMatrix( float m[12] )
  function optixGetObjectToWorldTransformMatrix (line 736) | void optixGetObjectToWorldTransformMatrix( float m[12] )
  function float3 (line 771) | float3 optixTransformPointFromWorldToObjectSpace( float3 point )
  function float3 (line 781) | float3 optixTransformVectorFromWorldToObjectSpace( float3 vec )
  function float3 (line 791) | float3 optixTransformNormalFromWorldToObjectSpace( float3 normal )
  function float3 (line 801) | float3 optixTransformPointFromObjectToWorldSpace( float3 point )
  function float3 (line 811) | float3 optixTransformVectorFromObjectToWorldSpace( float3 vec )
  function float3 (line 821) | float3 optixTransformNormalFromObjectToWorldSpace( float3 normal )
  function optixGetTransformListSize (line 831) | unsigned int optixGetTransformListSize()
  function OptixTraversableHandle (line 838) | OptixTraversableHandle optixGetTransformListHandle( unsigned int index )
  function OptixTransformType (line 845) | OptixTransformType optixGetTransformTypeFromHandle( OptixTraversableHand...
  function OptixStaticTransform (line 852) | const OptixStaticTransform* optixGetStaticTransformFromHandle( OptixTrav...
  function OptixSRTMotionTransform (line 859) | const OptixSRTMotionTransform* optixGetSRTMotionTransformFromHandle( Opt...
  function OptixMatrixMotionTransform (line 866) | const OptixMatrixMotionTransform* optixGetMatrixMotionTransformFromHandl...
  function optixGetInstanceIdFromHandle (line 873) | unsigned int optixGetInstanceIdFromHandle( OptixTraversableHandle handle )
  function OptixTraversableHandle (line 880) | OptixTraversableHandle optixGetInstanceChildFromHandle( OptixTraversable...
  function float4 (line 887) | const float4* optixGetInstanceTransformFromHandle( OptixTraversableHandl...
  function float4 (line 894) | const float4* optixGetInstanceInverseTransformFromHandle( OptixTraversab...
  function optixReportIntersection (line 901) | bool optixReportIntersection( float hitT, unsigned int hitKind )
  function optixReportIntersection (line 913) | bool optixReportIntersection( float hitT, unsigned int hitKind, unsigned...
  function optixReportIntersection (line 925) | bool optixReportIntersection( float hitT, unsigned int hitKind, unsigned...
  function optixReportIntersection (line 937) | bool optixReportIntersection( float hitT, unsigned int hitKind, unsigned...
  function optixReportIntersection (line 949) | bool optixReportIntersection( float        hitT,
  function optixReportIntersection (line 966) | bool optixReportIntersection( float        hitT,
  function optixReportIntersection (line 984) | bool optixReportIntersection( float        hitT,
  function optixReportIntersection (line 1003) | bool optixReportIntersection( float        hitT,
  function optixReportIntersection (line 1023) | bool optixReportIntersection( float        hitT,
  function optixGetAttribute_0 (line 1049) | unsigned int optixGetAttribute_0()
  function optixGetAttribute_1 (line 1054) | unsigned int optixGetAttribute_1()
  function optixGetAttribute_2 (line 1059) | unsigned int optixGetAttribute_2()
  function optixGetAttribute_3 (line 1064) | unsigned int optixGetAttribute_3()
  function optixGetAttribute_4 (line 1069) | unsigned int optixGetAttribute_4()
  function optixGetAttribute_5 (line 1074) | unsigned int optixGetAttribute_5()
  function optixGetAttribute_6 (line 1079) | unsigned int optixGetAttribute_6()
  function optixGetAttribute_7 (line 1084) | unsigned int optixGetAttribute_7()
  function optixTerminateRay (line 1091) | void optixTerminateRay()
  function optixIgnoreIntersection (line 1096) | void optixIgnoreIntersection()
  function optixGetPrimitiveIndex (line 1101) | unsigned int optixGetPrimitiveIndex()
  function optixGetSbtGASIndex (line 1108) | unsigned int optixGetSbtGASIndex()
  function optixGetInstanceId (line 1115) | unsigned int optixGetInstanceId()
  function optixGetInstanceIndex (line 1122) | unsigned int optixGetInstanceIndex()
  function optixGetHitKind (line 1129) | unsigned int optixGetHitKind()
  function OptixPrimitiveType (line 1136) | OptixPrimitiveType optixGetPrimitiveType(unsigned int hitKind)
  function optixIsBackFaceHit (line 1143) | bool optixIsBackFaceHit( unsigned int hitKind )
  function optixIsFrontFaceHit (line 1150) | bool optixIsFrontFaceHit( unsigned int hitKind )
  function OptixPrimitiveType (line 1156) | OptixPrimitiveType optixGetPrimitiveType()
  function optixIsBackFaceHit (line 1161) | bool optixIsBackFaceHit()
  function optixIsFrontFaceHit (line 1166) | bool optixIsFrontFaceHit()
  function optixIsTriangleHit (line 1171) | bool optixIsTriangleHit()
  function optixIsTriangleFrontFaceHit (line 1176) | bool optixIsTriangleFrontFaceHit()
  function optixIsTriangleBackFaceHit (line 1181) | bool optixIsTriangleBackFaceHit()
  function optixGetCurveParameter (line 1186) | float optixGetCurveParameter()
  function float2 (line 1191) | float2 optixGetTriangleBarycentrics()
  function uint3 (line 1198) | uint3 optixGetLaunchIndex()
  function uint3 (line 1207) | uint3 optixGetLaunchDimensions()
  function CUdeviceptr (line 1216) | CUdeviceptr optixGetSbtDataPointer()
  function optixThrowException (line 1223) | void optixThrowException( int exceptionCode )
  function optixThrowException (line 1232) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1241) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1250) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1259) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1268) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1277) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1286) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixThrowException (line 1295) | void optixThrowException( int exceptionCode, unsigned int exceptionDetai...
  function optixGetExceptionCode (line 1304) | int optixGetExceptionCode()
  function optixGetExceptionDetail_0 (line 1316) | unsigned int optixGetExceptionDetail_0()
  function optixGetExceptionDetail_1 (line 1321) | unsigned int optixGetExceptionDetail_1()
  function optixGetExceptionDetail_2 (line 1326) | unsigned int optixGetExceptionDetail_2()
  function optixGetExceptionDetail_3 (line 1331) | unsigned int optixGetExceptionDetail_3()
  function optixGetExceptionDetail_4 (line 1336) | unsigned int optixGetExceptionDetail_4()
  function optixGetExceptionDetail_5 (line 1341) | unsigned int optixGetExceptionDetail_5()
  function optixGetExceptionDetail_6 (line 1346) | unsigned int optixGetExceptionDetail_6()
  function optixGetExceptionDetail_7 (line 1351) | unsigned int optixGetExceptionDetail_7()
  function OptixTraversableHandle (line 1358) | OptixTraversableHandle optixGetExceptionInvalidTraversable()
  function optixGetExceptionInvalidSbtOffset (line 1365) | int optixGetExceptionInvalidSbtOffset()
  function OptixInvalidRayExceptionDetails (line 1372) | OptixInvalidRayExceptionDetails optixGetExceptionInvalidRay()
  function OptixParameterMismatchExceptionDetails (line 1388) | OptixParameterMismatchExceptionDetails optixGetExceptionParameterMismatch()
  function ReturnT (line 1411) | ReturnT optixDirectCall( unsigned int sbtIndex, ArgTypes... args )
  function ReturnT (line 1421) | ReturnT optixContinuationCall( unsigned int sbtIndex, ArgTypes... args )
  function uint4 (line 1431) | uint4 optixTexFootprint2D( unsigned long long tex, unsigned int texInfo,...
  function uint4 (line 1447) | uint4 optixTexFootprint2DGrad( unsigned long long tex,
  function uint4 (line 1474) | uint4

FILE: render/optixutils/include/internal/optix_7_device_impl_exception.h
  function namespace (line 40) | namespace optix_impl {

FILE: render/optixutils/include/internal/optix_7_device_impl_transformations.h
  function namespace (line 36) | namespace optix_impl {

FILE: render/optixutils/include/optix_7_types.h
  type CUdeviceptr (line 50) | typedef unsigned long long CUdeviceptr;
  type CUdeviceptr (line 53) | typedef unsigned int CUdeviceptr;
  type OptixDeviceContext_t (line 57) | struct OptixDeviceContext_t
  type OptixModule_t (line 60) | struct OptixModule_t
  type OptixProgramGroup_t (line 63) | struct OptixProgramGroup_t
  type OptixPipeline_t (line 66) | struct OptixPipeline_t
  type OptixDenoiser_t (line 69) | struct OptixDenoiser_t
  type OptixTraversableHandle (line 72) | typedef unsigned long long OptixTraversableHandle;
  type OptixVisibilityMask (line 75) | typedef unsigned int OptixVisibilityMask;
  type OptixResult (line 112) | typedef enum OptixResult
  type OptixDeviceProperty (line 156) | typedef enum OptixDeviceProperty
  type OptixDeviceContextValidationMode (line 225) | typedef enum OptixDeviceContextValidationMode
  type OptixDeviceContextOptions (line 234) | typedef struct OptixDeviceContextOptions
  type OptixGeometryFlags (line 249) | typedef enum OptixGeometryFlags
  type OptixHitKind (line 269) | typedef enum OptixHitKind
  type OptixIndicesFormat (line 278) | typedef enum OptixIndicesFormat
  type OptixVertexFormat (line 289) | typedef enum OptixVertexFormat
  type OptixTransformFormat (line 301) | typedef enum OptixTransformFormat
  type OptixBuildInputTriangleArray (line 310) | typedef struct OptixBuildInputTriangleArray
  type OptixPrimitiveType (line 381) | typedef enum OptixPrimitiveType
  type OptixPrimitiveTypeFlags (line 398) | typedef enum OptixPrimitiveTypeFlags
  type OptixBuildInputCurveArray (line 429) | typedef struct OptixBuildInputCurveArray
  type OptixAabb (line 480) | typedef struct OptixAabb
  type OptixBuildInputCustomPrimitiveArray (line 493) | typedef struct OptixBuildInputCustomPrimitiveArray
  type OptixBuildInputInstanceArray (line 538) | typedef struct OptixBuildInputInstanceArray
  type OptixBuildInputType (line 556) | typedef enum OptixBuildInputType
  type OptixBuildInput (line 575) | typedef struct OptixBuildInput
  type OptixInstanceFlags (line 606) | typedef enum OptixInstanceFlags
  type OptixInstance (line 638) | typedef struct OptixInstance
  type OptixBuildFlags (line 668) | typedef enum OptixBuildFlags
  type OptixBuildOperation (line 706) | typedef enum OptixBuildOperation
  type OptixMotionFlags (line 717) | typedef enum OptixMotionFlags
  type OptixMotionOptions (line 728) | typedef struct OptixMotionOptions
  type OptixAccelBuildOptions (line 747) | typedef struct OptixAccelBuildOptions
  type OptixAccelBufferSizes (line 767) | typedef struct OptixAccelBufferSizes
  type OptixAccelPropertyType (line 787) | typedef enum OptixAccelPropertyType
  type OptixAccelEmitDesc (line 799) | typedef struct OptixAccelEmitDesc
  type OptixAccelRelocationInfo (line 811) | typedef struct OptixAccelRelocationInfo
  type OptixStaticTransform (line 822) | typedef struct OptixStaticTransform
  type OptixMatrixMotionTransform (line 862) | typedef struct OptixMatrixMotionTransform
  type OptixSRTData (line 907) | typedef struct OptixSRTData
  type OptixSRTMotionTransform (line 944) | typedef struct OptixSRTMotionTransform
  type OptixTraversableType (line 967) | typedef enum OptixTraversableType
  type OptixPixelFormat (line 980) | typedef enum OptixPixelFormat
  type OptixImage2D (line 995) | typedef struct OptixImage2D
  type OptixDenoiserModelKind (line 1015) | typedef enum OptixDenoiserModelKind
  type OptixDenoiserOptions (line 1034) | typedef struct OptixDenoiserOptions
  type OptixDenoiserGuideLayer (line 1046) | typedef struct OptixDenoiserGuideLayer
  type OptixDenoiserLayer (line 1061) | typedef struct OptixDenoiserLayer
  type OptixDenoiserParams (line 1078) | typedef struct OptixDenoiserParams
  type OptixDenoiserSizes (line 1104) | typedef struct OptixDenoiserSizes
  type OptixRayFlags (line 1116) | typedef enum OptixRayFlags
  type OptixTransformType (line 1172) | typedef enum OptixTransformType
  type OptixTraversableGraphFlags (line 1183) | typedef enum OptixTraversableGraphFlags
  type OptixCompileOptimizationLevel (line 1204) | typedef enum OptixCompileOptimizationLevel
  type OptixCompileDebugLevel (line 1221) | typedef enum OptixCompileDebugLevel
  type OptixModuleCompileBoundValueEntry (line 1268) | typedef struct OptixModuleCompileBoundValueEntry {
  type OptixModuleCompileOptions (line 1280) | typedef struct OptixModuleCompileOptions
  type OptixProgramGroupKind (line 1302) | typedef enum OptixProgramGroupKind
  type OptixProgramGroupFlags (line 1326) | typedef enum OptixProgramGroupFlags
  type OptixProgramGroupSingleModule (line 1338) | typedef struct OptixProgramGroupSingleModule
  type OptixProgramGroupHitgroup (line 1351) | typedef struct OptixProgramGroupHitgroup
  type OptixProgramGroupCallables (line 1372) | typedef struct OptixProgramGroupCallables
  type OptixProgramGroupDesc (line 1385) | typedef struct OptixProgramGroupDesc
  type OptixProgramGroupOptions (line 1411) | typedef struct OptixProgramGroupOptions
  type OptixExceptionCodes (line 1418) | typedef enum OptixExceptionCodes
  type OptixExceptionFlags (line 1521) | typedef enum OptixExceptionFlags
  type OptixPipelineCompileOptions (line 1545) | typedef struct OptixPipelineCompileOptions
  type OptixPipelineLinkOptions (line 1581) | typedef struct OptixPipelineLinkOptions
  type OptixShaderBindingTable (line 1594) | typedef struct OptixShaderBindingTable
  type OptixStackSizes (line 1634) | typedef struct OptixStackSizes
  type OptixQueryFunctionTableOptions (line 1654) | typedef enum OptixQueryFunctionTableOptions
  type OptixResult (line 1662) | typedef OptixResult( OptixQueryFunctionTable_t )( int          abiId,
  type OptixBuiltinISOptions (line 1673) | typedef struct OptixBuiltinISOptions
  type OptixInvalidRayExceptionDetails (line 1685) | typedef struct OptixInvalidRayExceptionDetails
  type OptixParameterMismatchExceptionDetails (line 1700) | typedef struct OptixParameterMismatchExceptionDetails

FILE: render/optixutils/include/optix_denoiser_tiling.h
  type OptixUtilDenoiserImageTile (line 54) | struct OptixUtilDenoiserImageTile
  function optixUtilGetPixelStride (line 73) | inline unsigned int optixUtilGetPixelStride( const OptixImage2D& image )
  function OptixResult (line 118) | inline OptixResult optixUtilDenoiserSplitImage(
  function OptixResult (line 206) | inline OptixResult optixUtilDenoiserInvokeTiled(

FILE: render/optixutils/include/optix_function_table.h
  type OptixFunctionTable (line 55) | typedef struct OptixFunctionTable

FILE: render/optixutils/include/optix_stack_size.h
  function OptixResult (line 52) | inline OptixResult optixUtilAccumulateStackSizes( OptixProgramGroup prog...
  function OptixResult (line 86) | inline OptixResult optixUtilComputeStackSizes( const OptixStackSizes* st...
  function OptixResult (line 151) | inline OptixResult optixUtilComputeStackSizesDCSplit( const OptixStackSi...
  function OptixResult (line 212) | inline OptixResult optixUtilComputeStackSizesCssCCTree( const OptixStack...
  function OptixResult (line 263) | inline OptixResult optixUtilComputeStackSizesSimplePathTracer( OptixProg...

FILE: render/optixutils/include/optix_stubs.h
  function OptixResult (line 188) | inline OptixResult optixInitWithHandle( void** handlePtr )
  function OptixResult (line 224) | inline OptixResult optixInit( void )
  function OptixResult (line 235) | inline OptixResult optixUninitWithHandle( void* handle )
  function OptixResult (line 318) | inline OptixResult optixDeviceContextCreate( CUcontext fromContext, cons...
  function OptixResult (line 323) | inline OptixResult optixDeviceContextDestroy( OptixDeviceContext context )
  function OptixResult (line 328) | inline OptixResult optixDeviceContextGetProperty( OptixDeviceContext con...
  function OptixResult (line 333) | inline OptixResult optixDeviceContextSetLogCallback( OptixDeviceContext ...
  function OptixResult (line 341) | inline OptixResult optixDeviceContextSetCacheEnabled( OptixDeviceContext...
  function OptixResult (line 346) | inline OptixResult optixDeviceContextSetCacheLocation( OptixDeviceContex...
  function OptixResult (line 351) | inline OptixResult optixDeviceContextSetCacheDatabaseSizes( OptixDeviceC...
  function OptixResult (line 356) | inline OptixResult optixDeviceContextGetCacheEnabled( OptixDeviceContext...
  function OptixResult (line 361) | inline OptixResult optixDeviceContextGetCacheLocation( OptixDeviceContex...
  function OptixResult (line 366) | inline OptixResult optixDeviceContextGetCacheDatabaseSizes( OptixDeviceC...
  function OptixResult (line 371) | inline OptixResult optixModuleCreateFromPTX( OptixDeviceContext         ...
  function OptixResult (line 384) | inline OptixResult optixModuleDestroy( OptixModule module )
  function OptixResult (line 389) | inline OptixResult optixBuiltinISModuleGet( OptixDeviceContext          ...
  function OptixResult (line 399) | inline OptixResult optixProgramGroupCreate( OptixDeviceContext          ...
  function OptixResult (line 411) | inline OptixResult optixProgramGroupDestroy( OptixProgramGroup programGr...
  function OptixResult (line 416) | inline OptixResult optixProgramGroupGetStackSize( OptixProgramGroup prog...
  function OptixResult (line 421) | inline OptixResult optixPipelineCreate( OptixDeviceContext              ...
  function OptixResult (line 434) | inline OptixResult optixPipelineDestroy( OptixPipeline pipeline )
  function OptixResult (line 439) | inline OptixResult optixPipelineSetStackSize( OptixPipeline pipeline,
  function OptixResult (line 449) | inline OptixResult optixAccelComputeMemoryUsage( OptixDeviceContext     ...
  function OptixResult (line 458) | inline OptixResult optixAccelBuild( OptixDeviceContext            context,
  function OptixResult (line 477) | inline OptixResult optixAccelGetRelocationInfo( OptixDeviceContext conte...
  function OptixResult (line 483) | inline OptixResult optixAccelCheckRelocationCompatibility( OptixDeviceCo...
  function OptixResult (line 488) | inline OptixResult optixAccelRelocate( OptixDeviceContext              c...
  function OptixResult (line 501) | inline OptixResult optixAccelCompact( OptixDeviceContext      context,
  function OptixResult (line 511) | inline OptixResult optixConvertPointerToTraversableHandle( OptixDeviceCo...
  function OptixResult (line 519) | inline OptixResult optixSbtRecordPackHeader( OptixProgramGroup programGr...
  function OptixResult (line 524) | inline OptixResult optixLaunch( OptixPipeline                  pipeline,
  function OptixResult (line 536) | inline OptixResult optixDenoiserCreate( OptixDeviceContext context, Opti...
  function OptixResult (line 541) | inline OptixResult optixDenoiserCreateWithUserModel( OptixDeviceContext ...
  function OptixResult (line 546) | inline OptixResult optixDenoiserDestroy( OptixDenoiser handle )
  function OptixResult (line 551) | inline OptixResult optixDenoiserComputeMemoryResources( const OptixDenoi...
  function OptixResult (line 559) | inline OptixResult optixDenoiserSetup( OptixDenoiser denoiser,
  function OptixResult (line 572) | inline OptixResult optixDenoiserInvoke( OptixDenoiser                   ...
  function OptixResult (line 590) | inline OptixResult optixDenoiserComputeIntensity( OptixDenoiser       ha...
  function OptixResult (line 600) | inline OptixResult optixDenoiserComputeAverageColor( OptixDenoiser      ...

FILE: render/optixutils/ops.py
  function find_cl_path (line 25) | def find_cl_path():
  class _optix_env_shade_func (line 81) | class _optix_env_shade_func(torch.autograd.Function):
    method forward (line 85) | def forward(ctx, optix_ctx, mask, ro, gb_pos, gb_normal, gb_view_pos, ...
    method backward (line 101) | def backward(ctx, diff_grad, spec_grad):
  class _bilateral_denoiser_func (line 110) | class _bilateral_denoiser_func(torch.autograd.Function):
    method forward (line 112) | def forward(ctx, col, nrm, zdz, sigma):
    method backward (line 119) | def backward(ctx, out_grad):
  class OptiXContext (line 128) | class OptiXContext:
    method __init__ (line 129) | def __init__(self):
  function optix_build_bvh (line 133) | def optix_build_bvh(optix_ctx, verts, tris, rebuild):
  function optix_env_shade (line 141) | def optix_env_shade(optix_ctx, mask, ro, gb_pos, gb_normal, gb_view_pos,...
  function bilateral_denoiser (line 145) | def bilateral_denoiser(col, nrm, zdz, sigma):

FILE: render/optixutils/tests/filter_test.py
  function length (line 22) | def length(x: torch.Tensor, eps: float =1e-20) -> torch.Tensor:
  function safe_normalize (line 25) | def safe_normalize(x: torch.Tensor, eps: float =1e-20) -> torch.Tensor:
  function dot (line 28) | def dot(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
  class BilateralDenoiser (line 31) | class BilateralDenoiser(torch.nn.Module):
    method __init__ (line 32) | def __init__(self, sigma=1.0):
    method set_sigma (line 36) | def set_sigma(self, sigma):
    method forward (line 41) | def forward(self, input):
  function relative_loss (line 76) | def relative_loss(name, ref, cuda):
  function test_filter (line 84) | def test_filter():

FILE: render/regularizer.py
  function luma (line 16) | def luma(x):
  function value (line 18) | def value(x):
  function chroma_loss (line 21) | def chroma_loss(kd, color_ref, lambda_chroma):
  function shading_loss (line 28) | def shading_loss(diffuse_light, specular_light, color_ref, lambda_diffus...
  function material_smoothness_grad (line 46) | def material_smoothness_grad(kd_grad, ks_grad, nrm_grad, lambda_kd=0.25,...
  function image_grad (line 56) | def image_grad(buf, std=0.01):
  function avg_edge_length (line 68) | def avg_edge_length(v_pos, t_pos_idx):
  function laplace_regularizer_const (line 77) | def laplace_regularizer_const(v_pos, t_pos_idx):
  function normal_consistency (line 101) | def normal_consistency(v_pos, t_pos_idx):

FILE: render/render.py
  function interpolate (line 25) | def interpolate(attr, rast, attr_idx, rast_db=None):
  function shade (line 31) | def shade(
  function render_layer (line 199) | def render_layer(
  function render_mesh (line 325) | def render_mesh(
  function render_uv (line 449) | def render_uv(ctx, mesh, resolution, mlp_texture):

FILE: render/renderutils/bsdf.py
  function _dot (line 19) | def _dot(x, y):
  function _reflect (line 22) | def _reflect(x, n):
  function _safe_normalize (line 25) | def _safe_normalize(x):
  function _bend_normal (line 28) | def _bend_normal(view_vec, smooth_nrm, geom_nrm, two_sided_shading):
  function _perturb_normal (line 38) | def _perturb_normal(perturbed_nrm, smooth_nrm, smooth_tng, opengl):
  function bsdf_prepare_shading_normal (line 46) | def bsdf_prepare_shading_normal(pos, view_pos, perturbed_nrm, smooth_nrm...
  function bsdf_lambert (line 57) | def bsdf_lambert(nrm, wi):
  function bsdf_frostbite (line 64) | def bsdf_frostbite(nrm, wi, wo, linearRoughness):
  function bsdf_phong (line 85) | def bsdf_phong(nrm, wo, wi, N):
  function bsdf_fresnel_shlick (line 96) | def bsdf_fresnel_shlick(f0, f90, cosTheta):
  function bsdf_ndf_ggx (line 100) | def bsdf_ndf_ggx(alphaSqr, cosTheta):
  function bsdf_lambda_ggx (line 105) | def bsdf_lambda_ggx(alphaSqr, cosTheta):
  function bsdf_masking_smith_ggx_correlated (line 112) | def bsdf_masking_smith_ggx_correlated(alphaSqr, cosThetaI, cosThetaO):
  function bsdf_pbr_specular (line 117) | def bsdf_pbr_specular(col, nrm, wo, wi, alpha, min_roughness=0.08):
  function bsdf_pbr (line 136) | def bsdf_pbr(kd, arm, pos, nrm, view_pos, light_pos, min_roughness, BSDF):

FILE: render/renderutils/c_src/bsdf.h
  type LambertKernelParams (line 16) | struct LambertKernelParams
  type FrostbiteDiffuseKernelParams (line 24) | struct FrostbiteDiffuseKernelParams
  type FresnelShlickKernelParams (line 34) | struct FresnelShlickKernelParams
  type NdfGGXParams (line 43) | struct NdfGGXParams
  type MaskingSmithParams (line 51) | struct MaskingSmithParams
  type PbrSpecular (line 60) | struct PbrSpecular
  type PbrBSDF (line 72) | struct PbrBSDF

FILE: render/renderutils/c_src/common.cpp
  function dim3 (line 18) | dim3 getLaunchBlockSize(int maxWidth, int maxHeight, dim3 dims)
  function dim3 (line 56) | dim3 getWarpSize(dim3 blockSize)
  function dim3 (line 65) | dim3 getLaunchGridSize(dim3 blockSize, dim3 dims)

FILE: render/renderutils/c_src/common.h
  function dim3 (line 29) | static inline dim3 getWarpSize(dim3 blockSize)
  function __device__ (line 38) | __device__ static inline float clamp(float val, float mn, float mx) { re...

FILE: render/renderutils/c_src/cubemap.h
  type DiffuseCubemapKernelParams (line 16) | struct DiffuseCubemapKernelParams
  type SpecularCubemapKernelParams (line 23) | struct SpecularCubemapKernelParams
  type SpecularBoundsKernelParams (line 33) | struct SpecularBoundsKernelParams

FILE: render/renderutils/c_src/loss.h
  type TonemapperType (line 16) | enum TonemapperType
  type LossType (line 22) | enum LossType
  type LossKernelParams (line 30) | struct LossKernelParams

FILE: render/renderutils/c_src/mesh.h
  type XfmKernelParams (line 16) | struct XfmKernelParams

FILE: render/renderutils/c_src/normal.h
  type PrepareShadingNormalKernelParams (line 16) | struct PrepareShadingNormalKernelParams

FILE: render/renderutils/c_src/tensor.h
  type Tensor (line 20) | struct Tensor
  function store (line 65) | inline void store(unsigned int x, unsigned int y, unsigned int z, float ...
  function __device__ (line 70) | __device__ inline void store(unsigned int x, unsigned int y, unsigned in...
  function __device__ (line 79) | __device__ inline void store_grad(unsigned int x, unsigned int y, unsign...
  function __device__ (line 84) | __device__ inline void store_grad(unsigned int x, unsigned int y, unsign...

FILE: render/renderutils/c_src/torch_bindings.cpp
  function update_grid (line 100) | void update_grid(dim3 &gridSize, torch::Tensor x)
  function update_grid (line 108) | void update_grid(dim3& gridSize, torch::Tensor x, Ts&&... vs)
  function Tensor (line 116) | Tensor make_cuda_tensor(torch::Tensor val)
  function Tensor (line 130) | Tensor make_cuda_tensor(torch::Tensor val, dim3 outDims, torch::Tensor* ...
  function prepare_shading_normal_fwd (line 161) | torch::Tensor prepare_shading_normal_fwd(torch::Tensor pos, torch::Tenso...
  function prepare_shading_normal_bwd (line 203) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, t...
  function lambert_fwd (line 237) | torch::Tensor lambert_fwd(torch::Tensor nrm, torch::Tensor wi, bool fp16)
  function lambert_bwd (line 268) | std::tuple<torch::Tensor, torch::Tensor> lambert_bwd(torch::Tensor nrm, ...
  function frostbite_fwd (line 295) | torch::Tensor frostbite_fwd(torch::Tensor nrm, torch::Tensor wi, torch::...
  function frostbite_bwd (line 330) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> f...
  function fresnel_shlick_fwd (line 359) | torch::Tensor fresnel_shlick_fwd(torch::Tensor f0, torch::Tensor f90, to...
  function fresnel_shlick_bwd (line 392) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> fresnel_shlick_b...
  function ndf_ggx_fwd (line 420) | torch::Tensor ndf_ggx_fwd(torch::Tensor alphaSqr, torch::Tensor cosTheta...
  function ndf_ggx_bwd (line 451) | std::tuple<torch::Tensor, torch::Tensor> ndf_ggx_bwd(torch::Tensor alpha...
  function lambda_ggx_fwd (line 478) | torch::Tensor lambda_ggx_fwd(torch::Tensor alphaSqr, torch::Tensor cosTh...
  function lambda_ggx_bwd (line 509) | std::tuple<torch::Tensor, torch::Tensor> lambda_ggx_bwd(torch::Tensor al...
  function masking_smith_fwd (line 536) | torch::Tensor masking_smith_fwd(torch::Tensor alphaSqr, torch::Tensor co...
  function masking_smith_bwd (line 569) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> masking_smith_bw...
  function pbr_specular_fwd (line 597) | torch::Tensor pbr_specular_fwd(torch::Tensor col, torch::Tensor nrm, tor...
  function pbr_specular_bwd (line 635) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, t...
  function pbr_bsdf_fwd (line 666) | torch::Tensor pbr_bsdf_fwd(torch::Tensor kd, torch::Tensor arm, torch::T...
  function pbr_bsdf_bwd (line 707) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, t...
  function diffuse_cubemap_fwd (line 740) | torch::Tensor diffuse_cubemap_fwd(torch::Tensor cubemap)
  function diffuse_cubemap_bwd (line 769) | torch::Tensor diffuse_cubemap_bwd(torch::Tensor cubemap, torch::Tensor g...
  function specular_bounds (line 799) | torch::Tensor specular_bounds(int resolution, float costheta_cutoff)
  function specular_cubemap_fwd (line 826) | torch::Tensor specular_cubemap_fwd(torch::Tensor cubemap, torch::Tensor ...
  function specular_cubemap_bwd (line 859) | torch::Tensor specular_cubemap_bwd(torch::Tensor cubemap, torch::Tensor ...
  function LossType (line 895) | LossType strToLoss(std::string str)
  function image_loss_fwd (line 907) | torch::Tensor image_loss_fwd(torch::Tensor img, torch::Tensor target, st...
  function image_loss_bwd (line 941) | std::tuple<torch::Tensor, torch::Tensor> image_loss_bwd(torch::Tensor im...
  function xfm_fwd (line 971) | torch::Tensor xfm_fwd(torch::Tensor points, torch::Tensor matrix, bool i...
  function xfm_bwd (line 1006) | torch::Tensor xfm_bwd(torch::Tensor points, torch::Tensor matrix, torch:...
  function PYBIND11_MODULE (line 1034) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: render/renderutils/c_src/vec3f.h
  type vec3f (line 14) | struct vec3f
  function __device__ (line 38) | __device__ static inline float sum(vec3f a)
  function __device__ (line 43) | __device__ static inline vec3f cross(vec3f a, vec3f b)
  function __device__ (line 52) | __device__ static inline void bwdCross(vec3f a, vec3f b, vec3f &d_a, vec...
  function __device__ (line 63) | __device__ static inline float dot(vec3f a, vec3f b)
  function __device__ (line 68) | __device__ static inline void bwdDot(vec3f a, vec3f b, vec3f& d_a, vec3f...
  function __device__ (line 74) | __device__ static inline vec3f reflect(vec3f x, vec3f n)
  function __device__ (line 79) | __device__ static inline void bwdReflect(vec3f x, vec3f n, vec3f& d_x, v...
  function __device__ (line 90) | __device__ static inline vec3f safeNormalize(vec3f v)
  function __device__ (line 96) | __device__ static inline void bwdSafeNormalize(const vec3f v, vec3f& d_v...

FILE: render/renderutils/c_src/vec4f.h
  type vec4f (line 14) | struct vec4f

FILE: render/renderutils/loss.py
  function _tonemap_srgb (line 16) | def _tonemap_srgb(f, exposure=5):
  function _SMAPE (line 20) | def _SMAPE(img, target, eps=0.01):
  function _RELMSE (line 25) | def _RELMSE(img, target, eps=0.1):
  function image_loss_fn (line 30) | def image_loss_fn(img, target, loss, tonemapper):

FILE: render/renderutils/ops.py
  function _get_plugin (line 23) | def _get_plugin():
  class _fresnel_shlick_func (line 92) | class _fresnel_shlick_func(torch.autograd.Function):
    method forward (line 94) | def forward(ctx, f0, f90, cosTheta):
    method backward (line 100) | def backward(ctx, dout):
  function _fresnel_shlick (line 104) | def _fresnel_shlick(f0, f90, cosTheta, use_python=False):
  class _ndf_ggx_func (line 115) | class _ndf_ggx_func(torch.autograd.Function):
    method forward (line 117) | def forward(ctx, alphaSqr, cosTheta):
    method backward (line 123) | def backward(ctx, dout):
  function _ndf_ggx (line 127) | def _ndf_ggx(alphaSqr, cosTheta, use_python=False):
  class _lambda_ggx_func (line 137) | class _lambda_ggx_func(torch.autograd.Function):
    method forward (line 139) | def forward(ctx, alphaSqr, cosTheta):
    method backward (line 145) | def backward(ctx, dout):
  function _lambda_ggx (line 149) | def _lambda_ggx(alphaSqr, cosTheta, use_python=False):
  class _masking_smith_func (line 159) | class _masking_smith_func(torch.autograd.Function):
    method forward (line 161) | def forward(ctx, alphaSqr, cosThetaI, cosThetaO):
    method backward (line 167) | def backward(ctx, dout):
  function _masking_smith (line 171) | def _masking_smith(alphaSqr, cosThetaI, cosThetaO, use_python=False):
  class _prepare_shading_normal_func (line 184) | class _prepare_shading_normal_func(torch.autograd.Function):
    method forward (line 186) | def forward(ctx, pos, view_pos, perturbed_nrm, smooth_nrm, smooth_tng,...
    method backward (line 193) | def backward(ctx, dout):
  function prepare_shading_normal (line 197) | def prepare_shading_normal(pos, view_pos, perturbed_nrm, smooth_nrm, smo...
  class _lambert_func (line 235) | class _lambert_func(torch.autograd.Function):
    method forward (line 237) | def forward(ctx, nrm, wi):
    method backward (line 243) | def backward(ctx, dout):
  function lambert (line 247) | def lambert(nrm, wi, use_python=False):
  class _frostbite_diffuse_func (line 269) | class _frostbite_diffuse_func(torch.autograd.Function):
    method forward (line 271) | def forward(ctx, nrm, wi, wo, linearRoughness):
    method backward (line 277) | def backward(ctx, dout):
  function frostbite_diffuse (line 281) | def frostbite_diffuse(nrm, wi, wo, linearRoughness, use_python=False):
  class _pbr_specular_func (line 305) | class _pbr_specular_func(torch.autograd.Function):
    method forward (line 307) | def forward(ctx, col, nrm, wo, wi, alpha, min_roughness):
    method backward (line 314) | def backward(ctx, dout):
  function pbr_specular (line 318) | def pbr_specular(col, nrm, wo, wi, alpha, min_roughness=0.08, use_python...
  class _pbr_bsdf_func (line 344) | class _pbr_bsdf_func(torch.autograd.Function):
    method forward (line 346) | def forward(ctx, kd, arm, pos, nrm, view_pos, light_pos, min_roughness...
    method backward (line 354) | def backward(ctx, dout):
  function pbr_bsdf (line 358) | def pbr_bsdf(kd, arm, pos, nrm, view_pos, light_pos, min_roughness=0.08,...
  class _diffuse_cubemap_func (line 394) | class _diffuse_cubemap_func(torch.autograd.Function):
    method forward (line 396) | def forward(ctx, cubemap):
    method backward (line 402) | def backward(ctx, dout):
  function diffuse_cubemap (line 407) | def diffuse_cubemap(cubemap, use_python=False):
  class _specular_cubemap (line 416) | class _specular_cubemap(torch.autograd.Function):
    method forward (line 418) | def forward(ctx, cubemap, roughness, costheta_cutoff, bounds):
    method backward (line 425) | def backward(ctx, dout):
  function __ndfBounds (line 431) | def __ndfBounds(res, roughness, cutoff):
  function specular_cubemap (line 449) | def specular_cubemap(cubemap, roughness, cutoff=0.99, use_python=False):
  class _image_loss_func (line 466) | class _image_loss_func(torch.autograd.Function):
    method forward (line 468) | def forward(ctx, img, target, loss, tonemapper):
    method backward (line 475) | def backward(ctx, dout):
  function image_loss (line 479) | def image_loss(img, target, loss='l1', tonemapper='none', use_python=Fal...
  class _xfm_func (line 506) | class _xfm_func(torch.autograd.Function):
    method forward (line 508) | def forward(ctx, points, matrix, isPoints):
    method backward (line 514) | def backward(ctx, dout):
  function xfm_points (line 518) | def xfm_points(points, matrix, use_python=False):
  function xfm_vectors (line 536) | def xfm_vectors(vectors, matrix, use_python=False):

FILE: render/renderutils/tests/test_bsdf.py
  function relative_loss (line 20) | def relative_loss(name, ref, cuda):
  function test_normal (line 25) | def test_normal():
  function test_schlick (line 59) | def test_schlick():
  function test_ndf_ggx (line 85) | def test_ndf_ggx():
  function test_lambda_ggx (line 109) | def test_lambda_ggx():
  function test_masking_smith (line 132) | def test_masking_smith():
  function test_lambert (line 157) | def test_lambert():
  function test_frostbite (line 179) | def test_frostbite():
  function test_pbr_specular (line 207) | def test_pbr_specular():
  function test_pbr_bsdf (line 244) | def test_pbr_bsdf(bsdf):

FILE: render/renderutils/tests/test_loss.py
  function tonemap_srgb (line 20) | def tonemap_srgb(f):
  function l1 (line 23) | def l1(output, target):
  function relative_loss (line 30) | def relative_loss(name, ref, cuda):
  function test_loss (line 35) | def test_loss(loss, tonemapper):

FILE: render/renderutils/tests/test_mesh.py
  function tonemap_srgb (line 23) | def tonemap_srgb(f):
  function l1 (line 26) | def l1(output, target):
  function relative_loss (line 33) | def relative_loss(name, ref, cuda):
  function test_xfm_points (line 38) | def test_xfm_points():
  function test_xfm_vectors (line 58) | def test_xfm_vectors():

FILE: render/renderutils/tests/test_perf.py
  function test_bsdf (line 19) | def test_bsdf(BATCH, RES, ITR):

FILE: render/texture.py
  class texture2d_mip (line 20) | class texture2d_mip(torch.autograd.Function):
    method forward (line 22) | def forward(ctx, texture):
    method backward (line 26) | def backward(ctx, dout):
  class Texture2D (line 38) | class Texture2D:
    method __init__ (line 41) | def __init__(self, init, min_max=None):
    method sample (line 57) | def sample(self, texc, texc_deriv, filter_mode='linear-mipmap-linear'):
    method getRes (line 70) | def getRes(self):
    method getChannels (line 73) | def getChannels(self):
    method getMips (line 76) | def getMips(self):
    method parameters (line 82) | def parameters(self):
    method clamp_ (line 86) | def clamp_(self):
    method normalize_ (line 93) | def normalize_(self):
  function create_trainable (line 103) | def create_trainable(init, res=None, auto_mipmaps=True, min_max=None):
  function srgb_to_rgb (line 137) | def srgb_to_rgb(texture):
  function rgb_to_srgb (line 140) | def rgb_to_srgb(texture):
  function _load_mip2D (line 147) | def _load_mip2D(fn, lambda_fn=None, channels=None):
  function load_texture2D (line 155) | def load_texture2D(fn, lambda_fn=None, channels=None):
  function _save_mip2D (line 165) | def _save_mip2D(fn, mip, mipidx, lambda_fn):
  function save_texture2D (line 177) | def save_texture2D(fn, tex, lambda_fn=None):

FILE: render/util.py
  function dot (line 19) | def dot(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
  function reflect (line 22) | def reflect(x: torch.Tensor, n: torch.Tensor) -> torch.Tensor:
  function length (line 25) | def length(x: torch.Tensor, eps: float =1e-20) -> torch.Tensor:
  function safe_normalize (line 28) | def safe_normalize(x: torch.Tensor, eps: float =1e-20) -> torch.Tensor:
  function to_hvec (line 31) | def to_hvec(x: torch.Tensor, w: float) -> torch.Tensor:
  function ycocg2rgb (line 34) | def ycocg2rgb(ycocg):
  function hsv2rgb (line 41) | def hsv2rgb(image): # Based on https://kornia.readthedocs.io/en/latest/_...
  function pixel_grid (line 61) | def pixel_grid(width, height, center_x = 0.5, center_y = 0.5):
  function dilate (line 70) | def dilate(x, x_avg, mask, N):
  function _rgb_to_srgb (line 94) | def _rgb_to_srgb(f: torch.Tensor) -> torch.Tensor:
  function rgb_to_srgb (line 97) | def rgb_to_srgb(f: torch.Tensor) -> torch.Tensor:
  function _srgb_to_rgb (line 103) | def _srgb_to_rgb(f: torch.Tensor) -> torch.Tensor:
  function srgb_to_rgb (line 106) | def srgb_to_rgb(f: torch.Tensor) -> torch.Tensor:
  function reinhard (line 112) | def reinhard(f: torch.Tensor) -> torch.Tensor:
  function mse_to_psnr (line 122) | def mse_to_psnr(mse):
  function psnr_to_mse (line 126) | def psnr_to_mse(psnr):
  function get_miplevels (line 134) | def get_miplevels(texture: np.ndarray) -> float:
  function tex_2d (line 138) | def tex_2d(tex_map : torch.Tensor, coords : torch.Tensor, filter='neares...
  function cube_to_dir (line 149) | def cube_to_dir(s, x, y):
  function latlong_to_cubemap (line 158) | def latlong_to_cubemap(latlong_map, res):
  function cubemap_to_latlong (line 173) | def cubemap_to_latlong(cubemap, res):
  function scale_img_hwc (line 192) | def scale_img_hwc(x : torch.Tensor, size, mag='bilinear', min='area') ->...
  function scale_img_nhwc (line 195) | def scale_img_nhwc(x  : torch.Tensor, size, mag='bilinear', min='area') ...
  function avg_pool_nhwc (line 207) | def avg_pool_nhwc(x  : torch.Tensor, size) -> torch.Tensor:
  function segment_sum (line 216) | def segment_sum(data: torch.Tensor, segment_ids: torch.Tensor) -> torch....
  function fovx_to_fovy (line 235) | def fovx_to_fovy(fovx, aspect):
  function focal_length_to_fovy (line 238) | def focal_length_to_fovy(focal_length, sensor_height):
  function perspective (line 242) | def perspective(fovy=0.7854, aspect=1.0, n=0.1, f=1000.0, device=None):
  function perspective_offcenter (line 250) | def perspective_offcenter(fovy, fraction, rx, ry, aspect=1.0, n=0.1, f=1...
  function translate (line 274) | def translate(x, y, z, device=None):
  function rotate_x (line 280) | def rotate_x(a, device=None):
  function rotate_y (line 287) | def rotate_y(a, device=None):
  function rotate_z (line 294) | def rotate_z(a, device=None):
  function scale (line 301) | def scale(s, device=None):
  function lookAt (line 307) | def lookAt(eye, at, up):
  function random_rotation_translation (line 324) | def random_rotation_translation(t, device=None):
  function random_rotation (line 335) | def random_rotation(device=None):
  function lines_focal (line 350) | def lines_focal(o, d):
  function cosine_sample (line 361) | def cosine_sample(N, size=None):
  function bilinear_downsample (line 396) | def bilinear_downsample(x : torch.tensor) -> torch.Tensor:
  function bilinear_downsample (line 406) | def bilinear_downsample(x : torch.tensor, spp) -> torch.Tensor:
  function init_glfw (line 422) | def init_glfw():
  function display_image (line 440) | def display_image(image, title=None):
  function save_image (line 483) | def save_image(fn, x : np.ndarray) -> np.ndarray:
  function save_image_raw (line 492) | def save_image_raw(fn, x : np.ndarray):
  function load_image_raw (line 499) | def load_image_raw(fn) -> np.ndarray:
  function load_image (line 502) | def load_image(fn) -> np.ndarray:
  function time_to_text (line 511) | def time_to_text(x):
  function checkerboard (line 521) | def checkerboard(res, checker_size) -> np.ndarray:

FILE: train_gflexicubes_deepfashion.py
  function createLoss (line 52) | def createLoss(FLAGS):
  function prepare_batch (line 71) | def prepare_batch(target, bg_type='black'):
  function xatlas_uvmap (line 101) | def xatlas_uvmap(glctx, geometry, mat, FLAGS):
  function initial_guess_material (line 155) | def initial_guess_material(geometry, mlp, FLAGS, init_mat=None):
  function initial_guess_material_knownkskd (line 172) | def initial_guess_material_knownkskd(geometry, mlp, FLAGS, init_mat=None):
  function validate_itr (line 190) | def validate_itr(glctx, target, geometry, opt_material, lgt, FLAGS, deno...
  function validate (line 237) | def validate(glctx, geometry, opt_material, lgt, dataset_validate, out_d...
  function optimize_mesh (line 288) | def optimize_mesh(

FILE: train_gflexicubes_polycam.py
  function createLoss (line 52) | def createLoss(FLAGS):
  function prepare_batch (line 71) | def prepare_batch(target, bg_type='black'):
  function xatlas_uvmap (line 101) | def xatlas_uvmap(glctx, geometry, mat, FLAGS):
  function initial_guess_material (line 155) | def initial_guess_material(geometry, mlp, FLAGS, init_mat=None):
  function initial_guess_material_knownkskd (line 172) | def initial_guess_material_knownkskd(geometry, mlp, FLAGS, init_mat=None):
  function validate_itr (line 190) | def validate_itr(glctx, target, geometry, opt_material, lgt, FLAGS, deno...
  function validate (line 209) | def validate(glctx, geometry, opt_material, lgt, dataset_validate, out_d...
  function optimize_mesh (line 260) | def optimize_mesh(

FILE: train_gshelltet_deepfashion.py
  function createLoss (line 52) | def createLoss(FLAGS):
  function prepare_batch (line 71) | def prepare_batch(target, bg_type='black'):
  function xatlas_uvmap (line 101) | def xatlas_uvmap(glctx, geometry, mat, FLAGS):
  function initial_guess_material (line 155) | def initial_guess_material(geometry, mlp, FLAGS, init_mat=None):
  function initial_guess_material_knownkskd (line 172) | def initial_guess_material_knownkskd(geometry, mlp, FLAGS, init_mat=None):
  function validate_itr (line 190) | def validate_itr(glctx, target, geometry, opt_material, lgt, FLAGS, deno...
  function validate (line 227) | def validate(glctx, geometry, opt_material, lgt, dataset_validate, out_d...
  function optimize_mesh (line 278) | def optimize_mesh(

FILE: train_gshelltet_polycam.py
  function createLoss (line 52) | def createLoss(FLAGS):
  function prepare_batch (line 71) | def prepare_batch(target, bg_type='black'):
  function xatlas_uvmap (line 101) | def xatlas_uvmap(glctx, geometry, mat, FLAGS):
  function initial_guess_material (line 155) | def initial_guess_material(geometry, mlp, FLAGS, init_mat=None):
  function initial_guess_material_knownkskd (line 172) | def initial_guess_material_knownkskd(geometry, mlp, FLAGS, init_mat=None):
  function validate_itr (line 190) | def validate_itr(glctx, target, geometry, opt_material, lgt, FLAGS, deno...
  function validate (line 209) | def validate(glctx, geometry, opt_material, lgt, dataset_validate, out_d...
  function optimize_mesh (line 260) | def optimize_mesh(

FILE: train_gshelltet_synthetic.py
  function createLoss (line 51) | def createLoss(FLAGS):
  function prepare_batch (line 70) | def prepare_batch(target, bg_type='black'):
  function xatlas_uvmap (line 100) | def xatlas_uvmap(glctx, geometry, mat, FLAGS):
  function initial_guess_material (line 154) | def initial_guess_material(geometry, mlp, FLAGS, init_mat=None):
  function initial_guess_material_knownkskd (line 171) | def initial_guess_material_knownkskd(geometry, mlp, FLAGS, init_mat=None):
  function validate_itr (line 189) | def validate_itr(glctx, target, geometry, opt_material, lgt, FLAGS, deno...
  function validate (line 212) | def validate(glctx, geometry, opt_material, lgt, dataset_validate, out_d...
  function optimize_mesh (line 263) | def optimize_mesh(