Directory listing for /PyTorch/test/pytorch/third_party/fbgemm/fbgemm_gpu/codegen/

• __init__.template
• batch_index_select_dim0_cpu_host.cpp
• batch_index_select_dim0_host.cpp
• embedding_backward_code_generator.py
• embedding_backward_dense_host.cpp
• embedding_backward_dense_host_cpu.cpp
• embedding_backward_split_cpu_approx_template.cpp
• embedding_backward_split_cpu_template.cpp
• embedding_backward_split_grad_template.cu
• embedding_backward_split_host_cpu_template.cpp
• embedding_backward_split_host_template.cpp
• embedding_backward_split_indice_weights_template.cu
• embedding_backward_split_kernel_cta_template.cu
• embedding_backward_split_kernel_warp_template.cu
• embedding_backward_split_template.cu
• embedding_bounds_check.cu
• embedding_bounds_check_host.cpp
• embedding_bounds_check_host_cpu.cpp
• embedding_common_code_generator.py
• embedding_forward_quantized_cpu_template.cpp
• embedding_forward_quantized_host.cpp
• embedding_forward_quantized_host_cpu.cpp
• embedding_forward_quantized_split_lookup.cu
• embedding_forward_quantized_split_nbit_host_template.cu
• embedding_forward_quantized_split_nbit_kernel_template.cu
• embedding_forward_split_cpu.cpp
• embedding_forward_split_cpu.h
• embedding_forward_split_kernel_nobag_small_template.cu
• embedding_forward_split_kernel_template.cu
• embedding_forward_split_kernel_v2_template.cu
• embedding_forward_split_meta_template.cpp
• embedding_forward_split_template.cu
• embedding_forward_template_helpers.cuh
• embedding_op_registration.h
• embedding_ops_placeholder.cpp
• embedding_optimizer_code_generator.py
• embedding_optimizer_split_device_kernel_template.cuh
• embedding_optimizer_split_host_template.cpp
• embedding_optimizer_split_kernel_template.cu
• embedding_optimizer_split_template.cu
• lookup_args.py
• optimizer_args.py
• split_embedding_codegen_lookup_invoker.template
• split_embedding_optimizer_codegen.template