MCPcopy Create free account

hub / github.com/AI4Science-WestlakeU/wdno / functions

Functions1,208 in github.com/AI4Science-WestlakeU/wdno

⨍ Functions 1,208◇ Types & classes 150↳ Endpoints 24
↓ 1 callersFunctionpreprocess
(db)
burgers/ddpm_burgers/data_burgers_1d.py:32
↓ 1 callersMethodprogress
(self)
smoke/phi/model.py:125
↓ 1 callersMethodput
(self, dict, save=True)
smoke/phi/control/control_scene.py:45
↓ 1 callersMethodq_posterior
(self, x_start, x_t, t)
burgers/ddpm_burgers/diffusion_1d.py:196
↓ 1 callersMethodq_posterior
(self, x_start, x_t, t)
smoke/ddpm/diffusion_2d.py:714
↓ 1 callersMethodq_posterior
(self, x_start, x_t, t)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:669
↓ 1 callersMethodq_posterior
(self, x_start, x_t, t)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:677
↓ 1 callersFunctionrandom_word
(shape, min_count, max_count, margin=1, total_content=100, fontsize=10, y=40)
smoke/phi/control/voxelutil.py:43
↓ 1 callersMethodrange
(self)
smoke/phi/model.py:477
↓ 1 callersFunctionread_zipped_array
(filename)
smoke/phi/fluidformat.py:7
↓ 1 callersFunctionresample_tf
Resamples an N-dimensional tensor at the locations provided by sample_coords :param inputs: grid with dimensions (batch_size, spatial dimensions.
smoke/phi/math/tensorflow_backend.py:312
↓ 1 callersMethodreshape
(self, value, shape)
smoke/phi/math/tensorflow_backend.py:35
↓ 1 callersFunctionrestore_net
(scope, sess, graph, path)
smoke/phi/experimental.py:34
↓ 1 callersMethodrun
(self, dataloader)
smoke/inference_2d.py:261
↓ 1 callersFunctionrun_2d_Unet
(dataset, shape, ori_shape, args, RESCALER)
burgers/train_ddpm_burgers.py:185
↓ 1 callersMethodrun_action
(self, action)
smoke/phi/model.py:156
↓ 1 callersMethodrun_advect
(self, velocity, density)
smoke/phi/control/iksm.py:119
↓ 1 callersMethodrun_base_model
(self, state, wave_init, wave_control)
smoke/inference_2d.py:123
↓ 1 callersMethodrun_force
(self, initial_velocity, target_velocity, initial_density, real_target_density)
smoke/phi/control/iksm.py:122
↓ 1 callersMethodrun_ik
(self, initial_density, target_density)
smoke/phi/control/iksm.py:116
↓ 1 callersMethodrun_jerk
(self, initial_velocity, initial_force, next_force)
smoke/phi/control/iksm.py:132
↓ 1 callersMethodrun_model
state: not rescaled
smoke/inference_2d.py:235
↓ 1 callersMethodrun_super_model
(self, state, state_ori, wave_init, wave_control)
smoke/inference_2d.py:155
↓ 1 callersMethodsample
(self, cond = None, cond_scale = 1., batch_size = 16)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:721
↓ 1 callersMethodsample
(self, cond = None, cond_scale = 1., batch_size = 16)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:729
↓ 1 callersMethodsave
(self, f_name)
smoke/phi/tf/profiling.py:26
↓ 1 callersMethodsave
(self, dir)
smoke/phi/tf/flow.py:170
↓ 1 callersFunctionsave_acc
( acc, fname, make_dict_path: callable=lambda acc_dict, dict_args:{dict_args['model_n
burgers/ddpm_burgers/result_io.py:74
↓ 1 callersFunctionsave_eval_results
(save_values, model_i, fname, model_str='')
burgers/eval_ddpm_burgers.py:347
↓ 1 callersMethodsave_figures
(self, directory, fieldname, time, data, same_scale_data=None)
smoke/phi/viz/plot.py:54
↓ 1 callersMethodsave_model
(self)
smoke/phi/tf/model.py:209
↓ 1 callersMethodsave_results
(self, J_totals, J_targets, J_energys, mses, n_l2s)
smoke/inference_2d.py:297
↓ 1 callersFunctionseek_all_images
(img, channels = 3)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:793
↓ 1 callersFunctionseek_all_images
(img, channels = 3)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:801
↓ 1 callersMethodshape
(self, tensor)
smoke/phi/math/tensorflow_backend.py:153
↓ 1 callersMethodshape
(self, tensor)
smoke/phi/math/base.py:95
↓ 1 callersFunctionsigmoid_schedule
sigmoid schedule proposed in https://arxiv.org/abs/2212.11972 - Figure 8 better for images > 64x64, when used during training
burgers/ddpm_burgers/model_utils.py:113
↓ 1 callersFunctionslugify
Normalizes string, converts to lowercase, removes non-alpha characters, and converts spaces to hyphens.
smoke/phi/fluidformat.py:256
↓ 1 callersFunctionsm_resnet
(initial_density, target_density, training=False, trainable=True, reuse=tf.AUTO_REUSE)
smoke/phi/control/iksm.py:51
↓ 1 callersMethodsolve
(self, divergence, active_mask, fluid_mask, boundaries, accuracy, pressure_guess=None, **kwargs)
smoke/phi/solver/net.py:11
↓ 1 callersFunctionsolve_control
(density, velocity, target, remaining_time, level_control=False, velocity_generator=multi_scale_control_cnn_16
smoke/phi/control/control.py:169
↓ 1 callersMethodsolve_pressure
Calculates the pressure from the given velocity or velocity divergence using the specified solver. :param input: tensor containing the center
smoke/phi/flow.py:303
↓ 1 callersFunctionsolve_pressure_forward
(divergence, fluid_mask, max_iterations, guess, accuracy, boundaries)
smoke/phi/solver/spcg.py:34
↓ 1 callersFunctionsolve_pressure_tompson2
(divergence, level_control=False, constants_file=None, cubic=True)
smoke/phi/solver/net.py:32
↓ 1 callersMethodsolve_with_boundaries
See :func:`PressureSolver.solve`. Unlike the regular solve method, active_mask and fluid_mask are valid tensors which include one extra voxel at each
smoke/phi/solver/base.py:42
↓ 1 callersMethodsqrt
(self, x)
smoke/phi/math/tensorflow_backend.py:131
↓ 1 callersMethodstack
(self, values, axis=0)
smoke/phi/math/tensorflow_backend.py:23
↓ 1 callersMethodstart_tracing
(self, file)
smoke/phi/tf/flow.py:200
↓ 1 callersMethodstep
(self)
smoke/phi/model.py:133
↓ 1 callersMethodstep_ema
(self)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:938
↓ 1 callersMethodstep_ema
(self)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:946
↓ 1 callersMethodstop_tracing
(self)
smoke/phi/tf/flow.py:206
↓ 1 callersMethodsum
(self, value, axis=None)
smoke/phi/math/scipy_backend.py:48
↓ 1 callersMethodtarget_density
(self)
smoke/phi/control/control_scene.py:36
↓ 1 callersFunctiontensor_spatial_rank
(field)
smoke/phi/math/scipy_backend.py:188
↓ 1 callersFunctiontensor_to_coef_super
input: [N, more than 2*(1+3), padded, padded]
burgers/wave_trans.py:18
↓ 1 callersMethodtfstep
(self, optimizer=None)
smoke/phi/tf/model.py:136
↓ 1 callersMethodtile
(self, x, multiples)
smoke/phi/math/tensorflow_backend.py:175
↓ 1 callersMethodtime_to_keyframe
(self, index)
smoke/phi/control/control_scene.py:42
↓ 1 callersFunctionto_gauss_format
(tensor)
smoke/phi/experimental.py:202
↓ 1 callersFunctionto_tensorflow_scalar
(field, dimensions)
smoke/phi/solver/manta.py:60
↓ 1 callersFunctiontompson2_pressure
(div, constants_file=None)
smoke/phi/solver/net.py:19
↓ 1 callersMethodtrain
(self)
burgers/ddpm_burgers/train_diffusion.py:187
↓ 1 callersMethodtrain
(self)
smoke/ddpm/diffusion_2d.py:1257
↓ 1 callersFunctionunnormalize_img
(t)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:827
↓ 1 callersFunctionunnormalize_img
(t)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:835
↓ 1 callersMethodunstack
(self, tensor, axis=0)
smoke/phi/math/tensorflow_backend.py:162
↓ 1 callersMethodupdate_average
(self, old, new)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:119
↓ 1 callersMethodupdate_average
(self, old, new)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:126
↓ 1 callersMethodupdate_model_average
(self, ma_model, current_model)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:114
↓ 1 callersMethodupdate_model_average
(self, ma_model, current_model)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:121
↓ 1 callersMethodupdate_timeline
(self, chrome_trace)
smoke/phi/tf/profiling.py:13
↓ 1 callersFunctionupsample_flatten_dipole_2d_2x
(tensor, scaling="average")
smoke/phi/experimental.py:149
↓ 1 callersFunctionvideo_tensor_to_gif
(tensor, path, duration = 120, loop = 0, optimize = True)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:808
↓ 1 callersFunctionvideo_tensor_to_gif
(tensor, path, duration = 120, loop = 0, optimize = True)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:816
↓ 1 callersMethodview_batch
(self, fieldname, subrange=None)
smoke/phi/tf/model.py:201
↓ 1 callersMethodwarning
(self, item)
smoke/ddpm/utils.py:201
↓ 1 callersFunctionwrite_sim_frame
(simpath, arrays, fieldnames, index, check_same_dimensions=False)
smoke/phi/fluidformat.py:40
↓ 1 callersMethodwrite_sim_frame
(self, arrays, fieldnames, index, check_same_dimensions=False)
smoke/phi/fluidformat.py:148
↓ 1 callersFunctionwrite_zipped_array
(filename, array)
smoke/phi/fluidformat.py:15
FunctionCheckCudaErrorAux
smoke/phi/solver/cuda/src/pressure_solve_op.cu.cc:11
FunctionCheckCudaErrorAux
smoke/phi/solver/cuda/src/laplace_op.cu.cc:6
MethodCompute
smoke/phi/solver/cuda/src/laplace_op.cc:32
MethodCompute
smoke/phi/solver/cuda/src/pressure_solve_op.cc:59
MethodLaplaceMatrixOp
smoke/phi/solver/cuda/src/laplace_op.cc:28
MethodPressureSolveOp
smoke/phi/solver/cuda/src/pressure_solve_op.cc:53
Method__add__
(self, other)
smoke/phi/math/nd.py:550
Method__del__
(self)
smoke/phi/data/data.py:538
Method__enter__
(self)
smoke/phi/flow.py:150
Method__exit__
(self, exc_type, exc_val, exc_tb)
smoke/phi/flow.py:155
Method__getattr__
(self, item)
smoke/phi/control/control_scene.py:57
Method__getitem__
(self, item)
smoke/phi/control/sequences.py:67
Method__getitem__
(self, key)
smoke/phi/control/control_scene.py:54
Method__getitem__
(self, item)
smoke/phi/data/data.py:465
Method__getitem__
(self, item)
smoke/phi/data/data.py:523
Method__getitem__
(self, item)
smoke/phi/math/nd.py:580
Method__getitem__
(self, sim_id)
smoke/ddpm/data_2d.py:58
Method__getitem__
(self, sim_id)
smoke/ddpm/data_2d.py:156
Method__getitem__
(self, index)
smoke/video_diffusion_pytorch/video_diffusion_pytorch.py:870
Method__getitem__
(self, index)
smoke/video_diffusion_pytorch/video_diffusion_pytorch_conv3d.py:878
← previousnext →501–600 of 1,208, ranked by callers