TorchWrapper

class TorchWrapper.TorchWrapperOp(in_info, out_info, lua_fw_func, lua_bw_func=None, lua_file=None, name=None)[source]
make_node(*args)[source]
infer_shape(node, input_shapes)[source]
perform(node, inputs, output_storage)[source]
c_header_dirs()[source]
c_lib_dirs()[source]
c_libraries()[source]
c_compile_args()[source]
c_support_code()[source]
c_support_code_struct(node, name)[source]
c_init_code_struct(node, name, sub)[source]
c_cleanup_code_struct(node, name)[source]
c_code(node, name, inputs, outputs, sub)[source]
grad(inputs, output_grads)[source]
connection_pattern(node)[source]
class TorchWrapper.GpuTorchWrapperOp(in_info, out_info, lua_fw_func, lua_bw_func=None, lua_file=None, name=None)[source]
c_support_code()[source]
c_libraries()[source]