## 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/oneflow/test/../../../python/oneflow/framewo
OneFlow 是一个采用全新架构设计的工业级通用深度学习框架
需积分: 5 111 浏览量
2023-08-07
10:02:50
上传
评论
收藏 7.16MB ZIP 举报
OneFlow 是一个采用全新架构设计的工业级通用深度学习框架 （2000个子文件） 
custom_eval_frame.c 2KB nn_functor.cpp 244KB math_functor.cpp 232KB array_functor.cpp 169KB nn_grad_functor.cpp 74KB operator.cpp 72KB replace_embedding_ops_pass.cpp 66KB plan_util.cpp 65KB rnn_functor.cpp 65KB tensor_functions.cpp 60KB model_update_kernels.cpp 58KB job_build_and_infer_ctx.cpp 56KB autograd.cpp 54KB boxing_collector.cpp 53KB sbp_infer_util.cpp 53KB Passes.cpp 49KB lazy_op_interpreter.cpp 48KB user_op.cpp 48KB histogram.cpp 48KB stateful_opkernel.cpp 47KB nn_graph.cpp 44KB MLIROneFlowTranslation.cpp 42KB insert_nccl_logical_op_pass.cpp 42KB oneflow.cpp 42KB reshape_user_op_util_test.cpp 40KB conv_kernels.cpp 40KB random_functor.cpp 40KB model_update_ops.cpp 39KB op_expr.cpp 39KB tensor_methods.cpp 39KB instructions_builder.cpp 39KB autotick.cpp 38KB group_conv_kernel.cpp 37KB sbp_node.cpp 36KB nccl_logical_fusion_kernel.cpp 36KB placement_sbp_util.cpp 35KB deform_conv_kernel.cpp 35KB OneFlowToTosa.cpp 34KB persistent_table.cpp 34KB sbp_graph.cpp 34KB fused_attention_functor.cpp 33KB broadcast_elementwise_binary.cpp 33KB tensor.cpp 32KB light_actor.cpp 32KB binary_functor.cpp 32KB logical_chain_pass.cpp 32KB fused_attention_ops.cpp 32KB actor.cpp 32KB activation_kernels.cpp 31KB multi_tensor_model_update_kernel.cpp 30KB tf_pool_cpu_kernel.cpp 29KB Importer.cpp 29KB dispatch_stateful_ops.cpp 29KB global_cast.cpp 28KB nccl_logical_kernels.cpp 28KB intra_job_mem_sharing_util.cpp 27KB activation_functor.cpp 27KB cudnn_conv_util.cpp 26KB nccl_logical_2d_sbp_kernels.cpp 26KB normalization_kernel.cpp 26KB max_pool_kernel.cpp 25KB model_update_kernel_util.cpp 25KB sbp_constructor.cpp 24KB higher_derivative_functor.cpp 24KB quantization_aware_training.cpp 24KB normalization_op.cpp 24KB comm_functor.cpp 24KB optimizer_placement_optimization_pass.cpp 24KB one_embedding_ops.cpp 23KB sbp_edge.cpp 23KB tensor_index.cpp 23KB UserOpConversion.cpp 23KB eager_ccl_kernel.cpp 23KB passes.cpp 22KB embedding_manager.cpp 22KB parallel_desc.cpp 22KB eager_local_op_interpreter.cpp 22KB auto_memory.cpp 22KB image_object_preprocess_kernels.cpp 22KB reshape_user_op_util.cpp 22KB multi_tensor_model_update_ops.cpp 21KB reduce_kernel.cpp 21KB conv_cudnn_kernels.cpp 21KB upsample_nearest_kernel.cpp 21KB slice_kernel.cpp 20KB conv_op.cpp 20KB avg_pool_kernel.cpp 20KB job_builder.cpp 19KB group_deconv_kernel.cpp 19KB eager_global_op_interpreter.cpp 19KB multi_tensor_model_update.cpp 19KB matmul_op.cpp 19KB quantization.cpp 18KB deconv_cpu_kernel.cpp 18KB tensor_api.cpp 18KB graph.cpp 18KB sbp_collector.cpp 18KB global_tensor_infer_cache.cpp 17KB memory_share_strategy.cpp 17KB cutlass_conv_tuner.cpp 17KB共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
