## Ops Version : Alpha
| Op Name | Doc Test | Compatiable/Completeness Test | Exception | Performance Test |
| ------------------------- | ------------- | ----------------------------- | --------- | ---------------- |
| oneflow.autograd.backward | [oneflow.Tensor.backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L727) | [unsqueeze_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_unsqueeze.py#L54) | [non_requires_grad_tensor_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/exceptions/test_autograd.py#L24) | |
| oneflow.autograd.grad | [oneflow.Tensor.grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L753) | [adagrad_clip_grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_optim_adagrad.py#L213) | [non_requires_grad_tensor_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/exceptions/test_autograd.py#L24) | |
| oneflow.autograd.no_grad | | [no_grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_autograd_mode.py#L62) | | |
| oneflow.autograd.enable_grad | | [enable_grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_autograd_mode.py#L50) | | |
| oneflow.autograd.set_grad_enabled | | [set_grad_enabled](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_autograd_mode.py#L74) | | |
| oneflow.autograd.inference_mode | | [inference_mode](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_autograd_mode.py#L27) | | |
| oneflow.Tensor.grad | [oneflow.Tensor.grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L753) | [adagrad_clip_grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_optim_adagrad.py#L213) | [non_requires_grad_tensor_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/exceptions/test_autograd.py#L24) | |
| oneflow.Tensor.requires_grad | [oneflow.Tensor.requires_grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L800) | [requires_grad_tensor_inplace_and_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_autograd.py#L170) | [non_requires_grad_tensor_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/exceptions/test_autograd.py#L24) | |
| oneflow.Tensor.is_leaf | [oneflow.Tensor.is_leaf](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L767) | | | |
| oneflow.Tensor.backward | [oneflow.Tensor.backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L727) | [unsqueeze_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_unsqueeze.py#L54) | [non_requires_grad_tensor_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/exceptions/test_autograd.py#L24) | |
| oneflow.Tensor.detach | | [tensor_detach](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/tensor/test_tensor_part_2.py#L91) | | |
| oneflow.Tensor.register_hook | [oneflow.Tensor.register_hook](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L833) | [tensor_register_hook](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/tensor/test_tensor_part_1.py#L446) | | |
| oneflow.Tensor.retain_grad | [oneflow.Tensor.retain_grad](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L866) | [retain_grad_for_leaf_tensor](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_autograd.py#L178) | | |
| oneflow.autograd.Function.forward | | [eye_forward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_eye.py#L27) | | |
| oneflow.autograd.Function.backward | [oneflow.Tensor.backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/framework/docstr/tensor.py#L727) | [unsqueeze_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_unsqueeze.py#L54) | [non_requires_grad_tensor_backward](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/exceptions/test_autograd.py#L24) | |
| oneflow.autograd.Function.apply | | [module_apply](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_module.py#L161) | | |
| oneflow.autograd.autograd_function.FunctionAutoGradCaptureState.mark_non_differentiable | | | | |
| oneflow.autograd.autograd_function.FunctionAutoGradCaptureState.save_for_backward | | | | |
| oneflow.autograd.autograd_function.FunctionAutoGradCaptureState.saved_tensors | | | | |
| oneflow.cuda.is_available | | | | |
| oneflow.cuda.device_count | | | | |
| oneflow.cuda.current_device | | | | |
| oneflow.cuda.set_device | | | | |
| oneflow.cuda.synchronize | | | | |
| oneflow.cuda.manual_seed_all | | | | |
| oneflow.cuda.manual_seed | | [generator_manual_seed](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/oneflow/test/../../../python/oneflow/test/modules/test_generator.py#L72) | | |
| oneflow.cuda.HalfTensor | | | | |
| oneflow.cuda.FloatTensor | | | | |
| oneflow.cuda.DoubleTensor | | | | |
| oneflow.cuda.BoolTensor | | | | |
| oneflow.cuda.ByteTensor | | | | |
| oneflow.cuda.CharTensor | | | | |
| oneflow.cuda.IntTensor | | | | |
| oneflow.cuda.LongTensor | | | | |
| oneflow.cuda.empty_cache | | | | |
| oneflow.nn.functional.conv1d | [oneflow._C.conv1d](https://github.com/Oneflow-Inc/oneflow/blob/5d4308ecd3c72dafe29634f964a103694e4dea5b/python/onef
一个采用全新架构设计的工业级通用深度学习框架
需积分: 5 186 浏览量
2024-05-09
11:23:26
上传
评论
收藏 188.79MB ZIP 举报
一个采用全新架构设计的工业级通用深度学习框架 （2000个子文件） 
custom_eval_frame.c 2KB nn_functor.cpp 261KB math_functor.cpp 237KB array_functor.cpp 173KB nn_grad_functor.cpp 76KB operator.cpp 74KB replace_embedding_ops_pass.cpp 68KB plan_util.cpp 67KB rnn_functor.cpp 66KB tensor_functions.cpp 61KB model_update_kernels.cpp 59KB job_build_and_infer_ctx.cpp 57KB autograd.cpp 55KB boxing_collector.cpp 54KB sbp_infer_util.cpp 54KB Passes.cpp 50KB lazy_op_interpreter.cpp 49KB user_op.cpp 49KB stateful_opkernel.cpp 48KB histogram.cpp 48KB nn_graph.cpp 45KB MLIROneFlowTranslation.cpp 44KB insert_nccl_logical_op_pass.cpp 43KB oneflow.cpp 43KB random_functor.cpp 41KB conv_kernels.cpp 41KB reshape_user_op_util_test.cpp 41KB model_update_ops.cpp 40KB op_expr.cpp 40KB tensor_methods.cpp 40KB instructions_builder.cpp 40KB autotick.cpp 38KB group_conv_kernel.cpp 38KB sbp_node.cpp 37KB nccl_logical_fusion_kernel.cpp 37KB placement_sbp_util.cpp 36KB deform_conv_kernel.cpp 36KB OneFlowToTosa.cpp 35KB persistent_table.cpp 35KB sbp_graph.cpp 35KB fused_attention_functor.cpp 33KB light_actor.cpp 33KB tensor.cpp 33KB broadcast_elementwise_binary.cpp 33KB binary_functor.cpp 33KB activation_kernels.cpp 33KB fused_attention_ops.cpp 33KB logical_chain_pass.cpp 32KB actor.cpp 32KB multi_tensor_model_update_kernel.cpp 30KB Importer.cpp 30KB tf_pool_cpu_kernel.cpp 30KB dispatch_stateful_ops.cpp 29KB activation_functor.cpp 29KB global_cast.cpp 28KB nccl_logical_kernels.cpp 28KB intra_job_mem_sharing_util.cpp 28KB cudnn_conv_util.cpp 27KB nccl_logical_2d_sbp_kernels.cpp 26KB normalization_kernel.cpp 26KB max_pool_kernel.cpp 26KB model_update_kernel_util.cpp 25KB sbp_constructor.cpp 25KB higher_derivative_functor.cpp 25KB quantization_aware_training.cpp 25KB normalization_op.cpp 25KB comm_functor.cpp 24KB optimizer_placement_optimization_pass.cpp 24KB one_embedding_ops.cpp 24KB sbp_edge.cpp 24KB tensor_index.cpp 24KB UserOpConversion.cpp 24KB eager_ccl_kernel.cpp 23KB passes.cpp 23KB embedding_manager.cpp 23KB parallel_desc.cpp 23KB auto_memory.cpp 22KB eager_local_op_interpreter.cpp 22KB image_object_preprocess_kernels.cpp 22KB reshape_user_op_util.cpp 22KB multi_tensor_model_update_ops.cpp 22KB reduce_kernel.cpp 22KB conv_cudnn_kernels.cpp 22KB upsample_nearest_kernel.cpp 22KB slice_kernel.cpp 21KB conv_op.cpp 21KB avg_pool_kernel.cpp 20KB job_builder.cpp 20KB group_deconv_kernel.cpp 20KB eager_global_op_interpreter.cpp 20KB multi_tensor_model_update.cpp 20KB matmul_op.cpp 19KB quantization.cpp 19KB deconv_cpu_kernel.cpp 18KB tensor_api.cpp 18KB graph.cpp 18KB sbp_collector.cpp 18KB global_tensor_infer_cache.cpp 18KB memory_share_strategy.cpp 18KB cutlass_conv_tuner.cpp 18KB共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
