SYMBOL INDEX (25463 symbols across 3642 files) FILE: .github/scripts/set_initial_variables.py function create_one (line 4) | def create_one(name=None, allow_fail=None): function create_conda (line 15) | def create_conda(name=None): function print_github_action_output (line 19) | def print_github_action_output(name=None, value=None): function print_result (line 23) | def print_result(build_matrix=None, test_matrix=None, out=None): function check_include (line 42) | def check_include(include_key=None, matrix: dict = None): FILE: ci/build/ensure_img.py function check_and_download (line 9) | def check_and_download(tag, url): FILE: ci/check/lintutils.py function chunk (line 24) | def chunk(seq, n): function dechunk (line 41) | def dechunk(chunks): function run_parallel (line 49) | def run_parallel(cmds, **kwargs): function get_sources (line 75) | def get_sources(source_dir, exclude_globs=[]): function stdout_pathcolonline (line 93) | def stdout_pathcolonline(completed_process, filenames): FILE: ci/check/run_clang_format.py function split_and_print (line 28) | def split_and_print(prefix, text): function handle_stream (line 37) | async def handle_stream(stream, cb): function run_command (line 46) | async def run_command(cmd=None, dry=False, name=None): function chunks (line 62) | def chunks(lst, n): function check_version (line 68) | def check_version(bin): function download (line 77) | def download(dry=False): FILE: ci/check/run_clang_tidy.py function split_and_print (line 27) | def split_and_print(prefix, text): function handle_stream (line 36) | async def handle_stream(stream, cb): function run_command (line 45) | async def run_command(cmd=None, dry=False, name=None): function download (line 60) | def download(build_dir, dry=False) -> Optional[List[str]]: FILE: ci/check/run_cmake_format.py function gen_cmd (line 45) | def gen_cmd(file): FILE: ci/check/run_license_format.py function get_txt (line 25) | def get_txt(path: str): function check_file (line 34) | def check_file(path): function format_file (line 52) | def format_file(path): function do_check (line 68) | def do_check(x): function do_format (line 73) | def do_format(x): function glob_files (line 77) | def glob_files(path: str = None, excludes=None): FILE: ci/test/distributed_run.py function is_img_existing (line 26) | def is_img_existing(tag): function get_affiliations (line 40) | def get_affiliations(host): function resolve_hostname_hardcoded (line 48) | def resolve_hostname_hardcoded(host: str): function find_free_port (line 56) | def find_free_port(): function spawn_shell (line 63) | async def spawn_shell(cmd: str = None): function spawn_shell_ignoring_failure (line 69) | async def spawn_shell_ignoring_failure(cmd: str = None): function build_docker_img (line 74) | async def build_docker_img(remote_host=None, workspace_dir=None): function create_remote_workspace_dir (line 92) | async def create_remote_workspace_dir( function get_docker_cache_args (line 108) | def get_docker_cache_args(): function launch_remote_container (line 117) | async def launch_remote_container( function handle_cast (line 165) | def handle_cast(conn=None, cmd=None): function handle_call (line 173) | def handle_call(conn=None, cmd=None, response=None): class DockerAgent (line 183) | class DockerAgent: method __init__ (line 184) | def __init__( method __enter__ (line 219) | def __enter__(self): method run_bash_script_async (line 222) | def run_bash_script_async(self, bash_script=None, cmd=None): method __exit__ (line 268) | def __exit__(self, exc_type, exc_val, exc_tb): function fix_and_sync_libs (line 272) | async def fix_and_sync_libs(oneflow_internal_path=None, remote_hosts=None): function remove_containers_by_name (line 342) | async def remove_containers_by_name(remote_hosts=None, container_name=No... function get_remote_hosts (line 355) | def get_remote_hosts(args): function exit_handler (line 555) | def exit_handler(): FILE: ci/test/multi_launch.py function parse_args (line 38) | def parse_args(): function run_and_capture (line 103) | async def run_and_capture(cmd=None, prefix=None, **kwargs): function launch_multiple (line 120) | async def launch_multiple( function main (line 142) | def main(): FILE: ci/test/parallel_run.py function gen_cmds (line 13) | def gen_cmds(cmd=None, dir=None, doctest=False): function find_free_port (line 36) | def find_free_port(): function split_and_print (line 43) | def split_and_print(prefix, text): function everyN (line 51) | def everyN(l: list, n: int): function contains_oom_info (line 56) | def contains_oom_info(txt: str): function should_retry (line 60) | def should_retry(txt: str): function print_out (line 64) | def print_out(prefix: str = "", content: str = ""): function spawn_shell_and_check (line 69) | async def spawn_shell_and_check(cmd: str = None, gpu_id: int = -1, check... function run_cmds (line 92) | async def run_cmds( FILE: docker/package/manylinux/build_wheel.py function get_arg_env (line 9) | def get_arg_env(env_var_name: str, mode="run"): function get_proxy_build_args (line 20) | def get_proxy_build_args(): function get_proxy_env_args (line 31) | def get_proxy_env_args(): function build_img (line 42) | def build_img( function common_cmake_args (line 75) | def common_cmake_args(cache_dir=None, extra_oneflow_cmake_args=None): function get_build_dir_arg (line 90) | def get_build_dir_arg(cache_dir, oneflow_src_dir): function force_rm_dir (line 97) | def force_rm_dir(dir_to_clean): function create_tmp_bash_and_run (line 104) | def create_tmp_bash_and_run(docker_cmd, img, bash_cmd, bash_args, bash_w... function get_common_docker_args (line 136) | def get_common_docker_args( function get_python_dir (line 168) | def get_python_dir(inplace=True, oneflow_src_dir=None, cache_dir=None): function build_third_party (line 177) | def build_third_party( function get_python_bin (line 243) | def get_python_bin(version): function build_oneflow (line 254) | def build_oneflow( function is_img_existing (line 350) | def is_img_existing(tag): function build (line 464) | def build(): FILE: docs/source/conf.py function should_skip_member (line 203) | def should_skip_member(app, what, name, obj, skip, options): function setup (line 213) | def setup(app): FILE: oneflow/api/common/ir_pass.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/api/common/job_build_and_infer_ctx.h function namespace (line 23) | namespace oneflow { FILE: oneflow/api/common/sbp.h function namespace (line 25) | namespace oneflow { FILE: oneflow/api/common/variable_tensor_mgr.h function namespace (line 23) | namespace oneflow { FILE: oneflow/api/cpp/embedding/embedding.cpp type oneflow_api (line 19) | namespace oneflow_api { type embedding (line 20) | namespace embedding { function CreateKeyValueStore (line 22) | std::string CreateKeyValueStore(const std::string& key_value_store_o... function LoadSnapshot (line 35) | void LoadSnapshot(const std::string& snapshot_name, const std::strin... FILE: oneflow/api/cpp/embedding/embedding.h function namespace (line 21) | namespace oneflow_api { FILE: oneflow/api/cpp/env.cpp type oneflow_api (line 23) | namespace oneflow_api { function initialize (line 24) | void initialize() { function release (line 29) | void release() { FILE: oneflow/api/cpp/env.h function namespace (line 19) | namespace oneflow_api { FILE: oneflow/api/cpp/env_impl.cpp type oneflow_api (line 39) | namespace oneflow_api { function IsEnvInited (line 45) | inline bool IsEnvInited() { return of::Singleton ConvertToTensorTuple( function GetDeviceTag (line 97) | std::string GetDeviceTag(const Device& device) { return device.type(); } function Unzip (line 100) | const std::pair, std::vector> Unzip(const of::Hash... function Shape (line 110) | Shape OfShapeToOfApiShape(const of::Shape& of_shape) { function LoadOneEmbedding (line 117) | void LoadOneEmbedding(const std::string& model_path, const Device& dev... class Graph::GraphImpl (line 141) | class Graph::GraphImpl final { method GraphImpl (line 145) | GraphImpl(const GraphImpl& graph) = delete; method GraphImpl (line 146) | GraphImpl(GraphImpl&& graph) = default; method GraphImpl (line 150) | GraphImpl& operator=(const GraphImpl& graph) = delete; method GraphImpl (line 151) | GraphImpl& operator=(GraphImpl&& graph) = default; method set_batch_size (line 156) | void set_batch_size(int batch_size) { batch_size_ = batch_size; } function Graph (line 194) | Graph& Graph::operator=(Graph&& graph) noexcept { function InputOutputInfos (line 200) | InputOutputInfos Graph::GetInputInfos() { return graph_->GetInputInfos... function InputOutputInfos (line 202) | InputOutputInfos Graph::GetOutputInfos() { return graph_->GetOutputInf... function IValue (line 208) | IValue Graph::Forward(const IValue& inputs) { function Graph (line 232) | Graph Graph::Load(const std::string& model_path, const Device& device) { function InputOutputInfos (line 249) | InputOutputInfos Graph::GraphImpl::GetInputInfos() { return input_info... function InputOutputInfos (line 251) | InputOutputInfos Graph::GraphImpl::GetOutputInfos() { return output_in... FILE: oneflow/api/cpp/framework/graph.h function namespace (line 30) | namespace oneflow { function namespace (line 36) | namespace oneflow_api { FILE: oneflow/api/cpp/framework/ivalue.cpp type oneflow_api (line 19) | namespace oneflow_api { function Tensor (line 43) | const Tensor& IValue::ToTensor() const { FILE: oneflow/api/cpp/framework/ivalue.h function namespace (line 24) | namespace oneflow_api { FILE: oneflow/api/cpp/framework/shape.cpp type oneflow_api (line 20) | namespace oneflow_api { function ToOneflowDimVcetor (line 25) | of::DimVector ToOneflowDimVcetor(const std::vector& dim_vec) { function Shape (line 39) | Shape& Shape::operator=(const Shape& shape) { FILE: oneflow/api/cpp/framework/shape.h function namespace (line 22) | namespace oneflow { function namespace (line 28) | namespace oneflow_api { FILE: oneflow/api/cpp/framework/tensor.cpp type oneflow_api (line 29) | namespace oneflow_api { function Tensor (line 46) | Tensor& Tensor::operator=(const Tensor& tensor) { function Tensor (line 51) | Tensor& Tensor::operator=(Tensor&& tensor) noexcept { function Shape (line 57) | Shape Tensor::shape() const { function Device (line 62) | Device Tensor::device() const { function DType (line 67) | DType Tensor::dtype() const { return static_cast(tensor_->dtype... function Tensor (line 84) | Tensor Tensor::from_buffer(const void* buffer, const Shape& shape, con... FILE: oneflow/api/cpp/framework/tensor.h function namespace (line 24) | namespace oneflow { function namespace (line 33) | namespace oneflow_api { FILE: oneflow/api/cpp/nn/functional/activation.cpp type oneflow_api (line 19) | namespace oneflow_api { type nn (line 20) | namespace nn { function Tensor (line 25) | Tensor relu(const Tensor& tensor) { FILE: oneflow/api/cpp/nn/functional/activation.h function namespace (line 21) | namespace oneflow_api { FILE: oneflow/api/cpp/tests/api_test.cpp type oneflow_api (line 31) | namespace oneflow_api { function Shape (line 33) | Shape RandomShape() { function RandomData (line 42) | std::vector RandomData(size_t size) { function GetExeDir (line 57) | std::string GetExeDir() { FILE: oneflow/api/cpp/tests/api_test.h function namespace (line 22) | namespace oneflow_api { FILE: oneflow/api/cpp/tests/graph_test.cpp type oneflow_api (line 31) | namespace oneflow_api { function Graph (line 35) | inline Graph LoadGraph(const Device& device) { function Forward (line 41) | inline void Forward(Graph& graph, const Device& device, int expected_b... function TEST (line 60) | TEST(Api, graph_cpu_test) { function TEST (line 68) | TEST(Api, graph_gpu_test) { function TEST (line 75) | TEST(Api, graph_multi_gpu_test) { function TEST (line 87) | TEST(Api, graph_cpu_batching_test) { function TEST (line 96) | TEST(Api, graph_gpu_batching_test) { function TEST (line 104) | TEST(Api, graph_multi_device_test) { function TEST (line 119) | TEST(Api, graph_unload_test) { function TEST (line 148) | TEST(Api, graph_thread_test) { function TEST (line 162) | TEST(Api, graph_input_order_test) { function TEST (line 191) | TEST(Api, graph_input_output_infos_test) { FILE: oneflow/api/cpp/tests/ivalue_test.cpp type oneflow_api (line 23) | namespace oneflow_api { function TEST (line 31) | TEST(Api, ivalue) { function TEST (line 51) | TEST(Api, ivalue_tensor) { function TEST (line 66) | TEST(Api, ivalue_tensor_vector) { function TEST (line 85) | TEST(Api, ivalue_copy) { function TEST (line 107) | TEST(Api, ivalue_move) { FILE: oneflow/api/cpp/tests/nn_test.cpp type oneflow_api (line 22) | namespace oneflow_api { function Relu (line 29) | std::vector Relu(const std::vector& data) { function TestRelu (line 40) | void TestRelu() { function TEST (line 54) | TEST(Api, nn_relu) { function TEST (line 60) | TEST(Api, nn_relu_multithreading) { FILE: oneflow/api/cpp/tests/one_embedding_test.cpp type oneflow_api (line 19) | namespace oneflow_api { function TEST (line 22) | TEST(Api, embedding_test) { FILE: oneflow/api/cpp/tests/tensor_test.cpp type oneflow_api (line 20) | namespace oneflow_api { function TEST (line 22) | TEST(Api, device) { function TEST (line 39) | TEST(Api, tensor) { function TEST (line 58) | TEST(Api, tensor_from_buffer_and_copy_to) { function TEST (line 78) | TEST(Api, tensor_zeros) { FILE: oneflow/api/python/autograd/autograd.cpp type oneflow (line 36) | namespace oneflow { type autograd (line 37) | namespace autograd { function IsScalarTensor (line 41) | bool IsScalarTensor(const one::Tensor& tensor) { function CheckAndInitOutGrads (line 50) | Maybe CheckAndInitOutGrads(const one::TensorTuple&... function Backward (line 105) | Maybe Backward(const one::TensorTuple& outputs, co... function Grad (line 117) | Maybe Grad(const one::TensorTuple& outputs, const ... class PySavedTensorHook (line 136) | class PySavedTensorHook final : public one::SavedTensorHook { method PySavedTensorHook (line 138) | PySavedTensorHook(const py::function& pack_hook, const py::functio... method pack (line 141) | void pack(const std::shared_ptr& tensor) { method unpack (line 146) | std::shared_ptr unpack() { class PySavedTensorHookCreator (line 166) | class PySavedTensorHookCreator final : public one::SavedTensorHookCr... method new_saved_tensor_hook (line 168) | std::unique_ptr new_saved_tensor_hook() cons... method append_new_hooks (line 172) | void append_new_hooks(const py::function& pack_hook, const py::fun... method pop_hooks (line 175) | void pop_hooks() { FILE: oneflow/api/python/autograd/autograd_engine.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/api/python/autograd/autograd_function.cpp type oneflow (line 30) | namespace oneflow { function UnpackTensorTuple (line 35) | Maybe UnpackTensorTuple(const py::object& input) { function PackTensorTuple (line 62) | py::object PackTensorTuple(const one::TensorTuple& tp) { function PackPyFunctionToFType (line 73) | one::AutogradFunctionBase::FType PackPyFunctionToFType(const py::funct... type one (line 84) | namespace one { FILE: oneflow/api/python/autograd/autograd_function_state.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { function FunctionAutoGradCaptureState (line 27) | inline FunctionAutoGradCaptureState* CheckAndGetStateData(PyAutograd... function PyObject (line 45) | static PyObject* PyAutogradFunctionState_new(PyTypeObject* type, PyO... function PyAutogradFunctionState_dealloc (line 57) | static void PyAutogradFunctionState_dealloc(PyAutogradFunctionState*... function PyObject (line 63) | static PyObject* PyAutogradFunctionState_save_for_backward(PyObject*... function PyObject (line 78) | static PyObject* PyAutogradFunctionState_mark_non_differentiable(PyO... function PyObject (line 93) | static PyObject* PyAutogradFunctionState_is_data_valid(PyObject* sel... function PyObject (line 109) | static PyObject* PyAutogradFunctionState_saved_tensors(PyObject* sel... function PyObject (line 115) | static PyObject* PyAutogradFunctionState_get_dict(PyObject* self, Py... function PyObject (line 131) | PyObject* PyAutogradFunctionState_getattro(PyObject* self, PyObject*... function PyAutogradFunctionState_setattro (line 144) | int PyAutogradFunctionState_setattro(PyObject* self, PyObject* attr,... function PyObject (line 190) | PyObject* PyAutogradFunctionState_NewFromPtr( FILE: oneflow/api/python/autograd/autograd_function_state.h function namespace (line 25) | namespace oneflow { FILE: oneflow/api/python/autograd/autograd_mode.cpp type oneflow (line 24) | namespace oneflow { type autograd (line 26) | namespace autograd { FILE: oneflow/api/python/autograd/function_node.cpp type oneflow (line 26) | namespace oneflow { type FunctionNodeUtil (line 30) | struct FunctionNodeUtil final { method ToString (line 31) | static std::string ToString(const one::FunctionNode& func_node) { FILE: oneflow/api/python/caster/autograd_function_state.h function namespace (line 26) | namespace pybind11 { FILE: oneflow/api/python/caster/common.h function namespace (line 22) | namespace pybind11 { FILE: oneflow/api/python/caster/maybe.h function namespace (line 23) | namespace pybind11 { FILE: oneflow/api/python/caster/optional.h function namespace (line 24) | namespace pybind11 { FILE: oneflow/api/python/caster/size.h function shape_type_caster (line 48) | PYBIND11_NAMESPACE_BEGIN(detail) function handle (line 67) | handle cast(U* src, return_value_policy policy, handle parent) { function operator (line 72) | operator T*() { return value_.get(); } FILE: oneflow/api/python/caster/tensor.h function namespace (line 24) | namespace pybind11 { FILE: oneflow/api/python/caster/test.cpp type oneflow (line 21) | namespace oneflow { class A (line 23) | class A { method inc_x (line 25) | void inc_x() { x++; } method get_x (line 26) | int get_x() { return x; } function get_singleton_a (line 32) | std::shared_ptr get_singleton_a() { FILE: oneflow/api/python/deprecated.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/api/python/dlpack/converter.cpp type oneflow (line 26) | namespace oneflow { function ToOneFlowDevice (line 28) | Maybe> ToOneFlowDevice(const DLDevice& ctx) { function ToOneFlowDataType (line 38) | Maybe ToOneFlowDataType(const DLDataType& dtype) { function fromDLPack (line 83) | Maybe fromDLPack(const DLManagedTensor* src) { function ToDLDevice (line 132) | Maybe ToDLDevice(Symbol ofdevice) { function ToDLDataType (line 145) | Maybe ToDLDataType(DataType ofdtype) { type ATenDLMTensor (line 165) | struct ATenDLMTensor { function deleter (line 170) | void deleter(DLManagedTensor* arg) { delete static_cast toDLPack(const std::shared_ptr& s... function DLPack_Capsule_Destructor (line 209) | void DLPack_Capsule_Destructor(PyObject* data) { FILE: oneflow/api/python/dlpack/converter.h function namespace (line 19) | namespace oneflow { FILE: oneflow/api/python/dlpack/dlpack.h type DLDeviceType (line 62) | typedef enum { type DLDevice (line 112) | typedef struct { type DLDataTypeCode (line 125) | typedef enum { type DLDataType (line 157) | typedef struct { type DLTensor (line 175) | typedef struct { type DLManagedTensor (line 227) | typedef struct DLManagedTensor { FILE: oneflow/api/python/env/env.cpp type oneflow (line 34) | namespace oneflow { function RegisterCudaDeviceProperties (line 38) | void RegisterCudaDeviceProperties(py::module& m) { function SwitchToShuttingDownPhase (line 60) | Maybe SwitchToShuttingDownPhase(EnvGlobalObjectsScope* env, bool... FILE: oneflow/api/python/env/env.h function namespace (line 34) | namespace oneflow { FILE: oneflow/api/python/ep/cuda_matmul_mode.cpp type oneflow (line 24) | namespace oneflow { type ep (line 26) | namespace ep { FILE: oneflow/api/python/exception/exception.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/api/python/flags.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/api/python/framework/autocast.cpp type oneflow (line 24) | namespace oneflow { function is_nested_count_zero (line 31) | bool is_nested_count_zero() { return (*nested_count()) == 0; } function increase_nested_count (line 32) | void increase_nested_count() { (*nested_count())++; } function decrease_nested_count (line 33) | void decrease_nested_count() { (*nested_count())--; } class AutoCastMode (line 35) | class AutoCastMode { method AutoCastMode (line 39) | AutoCastMode(const std::string& device_type, Symbol dtype, bo... FILE: oneflow/api/python/framework/device.cpp type oneflow (line 27) | namespace oneflow { FILE: oneflow/api/python/framework/doc.cpp type oneflow (line 23) | namespace oneflow { function AddFunctionDoc (line 25) | py::object AddFunctionDoc(py::object f, const std::string& doc_string) { function ReplaceDoc (line 78) | py::object ReplaceDoc(py::object f, const std::string& doc_string) { FILE: oneflow/api/python/framework/dtype.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/api/python/framework/framework.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/api/python/framework/framework.h function namespace (line 34) | namespace oneflow { FILE: oneflow/api/python/framework/global_mode.cpp type oneflow (line 27) | namespace oneflow { FILE: oneflow/api/python/framework/id_util.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/api/python/framework/instructions_builder.cpp type oneflow (line 27) | namespace oneflow { function DeprecatedPhysicalRun (line 31) | Maybe DeprecatedPhysicalRun(const std::functionob_ty... function PyObject (line 64) | PyObject* PyMemoryFormat_New(MemoryFormat memory_format) { FILE: oneflow/api/python/framework/memory_format.h function namespace (line 25) | namespace oneflow { FILE: oneflow/api/python/framework/nn_graph.cpp type oneflow (line 33) | namespace oneflow { function APINNGraphAdditionalVarNames (line 35) | Maybe APINNGraphAdditionalVarNames(const std::shared_ptr APINNGraphAdditionalVarTensors(const std::shared_ptr... function APINNGraphGetCurrentSerializedJob (line 46) | Maybe APINNGraphGetCurrentSerializedJob(const std::shared_p... FILE: oneflow/api/python/framework/one_embedding.cpp type oneflow (line 28) | namespace oneflow { class OneEmbeddingHandler (line 30) | class OneEmbeddingHandler final { method OneEmbeddingHandler (line 32) | OneEmbeddingHandler(const std::string& key_value_store_option_string... method LoadSnapshot (line 40) | void LoadSnapshot(const std::string& snapshot_name) { method SaveSnapshot (line 49) | void SaveSnapshot(const std::string& snapshot_name) { method CreateKeyValueStore (line 59) | void CreateKeyValueStore(const embedding::KeyValueStoreOptions& key_... type embedding (line 74) | namespace embedding { class PersistentTableWriter (line 76) | class PersistentTableWriter { method PersistentTableWriter (line 79) | PersistentTableWriter() = default; class PersistentTableWriterImpl (line 87) | class PersistentTableWriterImpl : public PersistentTableWriter { method PersistentTableWriterImpl (line 90) | PersistentTableWriterImpl(const std::vector& paths, c... method Write (line 107) | void Write(const py::array& keys, const py::array& values) override { method Close (line 136) | void Close() override { CloseImpl(); } method CloseImpl (line 139) | void CloseImpl() { function NewPersistentTableWriter (line 156) | std::shared_ptr NewPersistentTableWriter( function NewPersistentTableWriter (line 168) | std::shared_ptr NewPersistentTableWriter( class PersistentTableReader (line 194) | class PersistentTableReader { method PersistentTableReader (line 197) | PersistentTableReader() = default; class PersistentTableReaderImpl (line 205) | class PersistentTableReaderImpl : public PersistentTableReader { method PersistentTableReaderImpl (line 209) | PersistentTableReaderImpl(const std::vector& paths, c... method Next (line 235) | std::tuple Next() override { method Close (line 255) | void Close() override { CloseImpl(); } method CloseImpl (line 258) | void CloseImpl() { function NewPersistentTableReader (line 276) | std::shared_ptr NewPersistentTableReader( function NewPersistentTableReader (line 288) | std::shared_ptr NewPersistentTableReader( FILE: oneflow/api/python/framework/op_builder.cpp type oneflow (line 27) | namespace oneflow { type one (line 29) | namespace one { FILE: oneflow/api/python/framework/op_expr.cpp type oneflow (line 30) | namespace oneflow { function PybindExportOpExpr (line 36) | py::class_> PybindExport... FILE: oneflow/api/python/framework/parallel_conf_util.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/api/python/framework/random_generator.cpp type oneflow (line 27) | namespace oneflow { function CreateGenerator (line 29) | Maybe CreateGenerator(const std::string& device_str) { function GetCudaDefaultGenerators (line 34) | py::tuple GetCudaDefaultGenerators() { FILE: oneflow/api/python/framework/scope_util.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/api/python/framework/session_util.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/api/python/framework/shut_down_util.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/api/python/framework/size.cpp type oneflow (line 24) | namespace oneflow { function PyObject (line 28) | static PyObject* TensorSize_repr(TensorSize* self) { function PyObject (line 42) | static PyObject* TensorSize_new(PyTypeObject* type, PyObject* args, Py... function Py_ssize_t (line 57) | static Py_ssize_t TensorSize_length(TensorSize* self) { function PyObject (line 61) | static PyObject* TensorSize_concat(TensorSize* self, PyObject* other) { function PyObject (line 71) | static PyObject* TensorSize_repeat(TensorSize* self, Py_ssize_t n) { function PyObject (line 81) | static PyObject* TensorSize_item(TensorSize* self, Py_ssize_t i) { function TensorSize_contains (line 85) | static int TensorSize_contains(TensorSize* self, PyObject* el) { function PyObject (line 100) | static PyObject* TensorSize_subscript(TensorSize* self, PyObject* item) { function PyObject (line 116) | static PyObject* TensorSize_numel(PyObject* self, PyObject* args) { function TensorSize_Check (line 168) | int TensorSize_Check(PyObject* p) { return p && p->ob_type == &TensorS... function PyObject (line 170) | PyObject* TensorSize_New(Py_ssize_t len) { return TensorSize_Type.tp_a... function PyObject (line 172) | PyObject* TensorSize_NewFromShape(const Shape& size) { function Shape (line 182) | Shape TensorSize_AsShape(PyObject* self) { FILE: oneflow/api/python/framework/size.h function namespace (line 24) | namespace oneflow { FILE: oneflow/api/python/framework/tensor.cpp type oneflow (line 43) | namespace oneflow { type one (line 44) | namespace one { type AllocType (line 62) | struct AllocType {} function PyObject (line 75) | PyObject* PyTensor_wrap(const std::shared_ptr& data, PyTensorObje... function PyTensor_tryResurrect (line 108) | bool PyTensor_tryResurrect(PyObject* py_tensor) { function PyTensorObject_init (line 125) | static int PyTensorObject_init(PyObject* self, PyObject* args, PyObj... function PyTensorObject_dealloc (line 134) | static void PyTensorObject_dealloc(PyObject* self) { function PyParameterObject_init (line 145) | static int PyParameterObject_init(PyObject* self, PyObject* args, Py... function Py_ssize_t (line 163) | static Py_ssize_t PyTensorObject_length(PyTensorObject* self) { function PyObject (line 168) | static PyObject* PyTensorObject_getitem(PyObject* self, Py_ssize_t i... function PyObject (line 176) | static PyObject* PyTensorObject_subscript(PyObject* self, PyObject* ... function PyObject (line 197) | static PyObject* PyTensorObject_storage_offset(PyObject* self, PyObj... function PyObject (line 203) | static PyObject* PyTensorObject_stride(PyObject* self, PyObject* unu... function PyObject (line 214) | static PyObject* PyTensorObject_is_contiguous(PyObject* self, PyObje... function PyObject (line 220) | static PyObject* PyTensorObject_is_view(PyObject* self, PyObject* un... function PyObject (line 230) | static PyObject* PyTensorObject_contiguous(PyObject* self, PyObject* u... function PyObject (line 236) | static PyObject* PyTensorObject_contiguous_(PyObject* self, PyObject* ... function PyObject (line 243) | static PyObject* PyTensorObject_pin_memory(PyObject* self, PyObject* u... function PyObject (line 249) | static PyObject* PyTensorObject_is_pinned(PyObject* self, PyObject* un... function PyObject (line 255) | static PyObject* PyTensorObject_offload(PyObject* self, PyObject* unus... function PyObject (line 263) | static PyObject* PyTensorObject_load(PyObject* self, PyObject* unused) { function PyObject (line 271) | static PyObject* PyTensorObject_is_offloaded(PyObject* self, PyObject*... function PyObject (line 277) | static PyObject* PyTensorObject_is_floating_point(PyObject* self, PyOb... function PyObject (line 287) | static PyObject* PyTensorObject_requires_grad_(PyObject* self, PyObject*... function PyObject (line 301) | static PyObject* PyTensorObject_retain_grad(PyObject* self, PyObject* un... function PyObject (line 309) | static PyObject* PyTensorObject_detach(PyObject* self, PyObject* unused) { function PyObject (line 315) | static PyObject* PyTensorObject_clone(PyObject* self, PyObject* unused) { function PyObject (line 321) | static PyObject* PyTensorObject_zero_(PyObject* self, PyObject* unused) { function RawSbpBToP (line 329) | std::vector> RawSbpBToP(Symbol nd_sbp) { function PyObject (line 341) | static PyObject* PyTensorObject_zero_grad(PyObject* self, PyObject* args... function PyObject (line 370) | static PyObject* PyTensorObject_register_hook(PyObject* self, PyObject* ... function PyObject (line 378) | static PyObject* PyTensorObject__register_post_grad_accumulation_hook(Py... function PyObject (line 387) | static PyObject* PyTensorObject_global_id(PyObject* self, PyObject* unus... function PyObject (line 394) | static PyObject* PyTensorObject_check_meta_consistency(PyObject* self, P... function PyObject (line 401) | static PyObject* PyTensorObject_data_ptr(PyObject* self, PyObject* unuse... function PyObject (line 411) | static PyObject* PyTensorObject_to_numpy(PyObject* self, PyObject* unuse... function PyObject (line 430) | static PyObject* PyTensorObject_item(PyObject* self, PyObject* unused) { function PyObject (line 448) | static PyObject* PyTensorObject_type(PyObject* self, PyObject* args, PyO... function CopyFromNumpyArray (line 492) | void CopyFromNumpyArray(ep::Stream* stream, function CopyToNumpyArray (line 500) | void CopyToNumpyArray(ep::Stream* stream, function PyObject (line 509) | static PyObject* PyTensorObject__copy_to_numpy(PyObject* self, PyObject*... function PyObject (line 516) | static PyObject* PyTensorObject__copy_from_numpy(PyObject* self, PyObjec... function PyObject (line 526) | static PyObject* PyTensorObject__register_storage_delete_hook(PyObject* ... function concat_method_def (line 534) | static std::vector concat_method_def(PyMethodDef methods[], function PyObject (line 583) | static PyObject* PyTensorObject_ndim(PyObject* self, void* unused) { function PyObject (line 587) | static PyObject* PyTensorObject_shape(PyObject* self, void* unused) { function PyObject (line 591) | static PyObject* PyTensorObject_dtype(PyObject* self, void* unused) { function PyObject (line 598) | static PyObject* PyTensorObject_is_cpu(PyObject* self, void* unused) { function PyObject (line 602) | static PyObject* PyTensorObject_is_cuda(PyObject* self, void* unused) { function PyObject (line 606) | static PyObject* PyTensorObject_grad(PyObject* self, void* unused) { function PyTensorObject_set_grad (line 612) | static int PyTensorObject_set_grad(PyObject* self, PyObject* grad, void*... function PyObject (line 625) | static PyObject* PyTensorObject_data(PyObject* self, void* unused) { function PyTensorObject_set_data (line 631) | static int PyTensorObject_set_data(PyObject* self, PyObject* data, void*... function PyObject (line 642) | static PyObject* PyTensorObject_ref_tensor(PyObject* self, void* unused) { function PyTensorObject_set_ref_tensor (line 648) | static int PyTensorObject_set_ref_tensor(PyObject* self, PyObject* ref, ... function PyObject (line 661) | static PyObject* PyTensorObject_ref_index(PyObject* self, void* unused) { function PyTensorObject_set_ref_index (line 665) | static int PyTensorObject_set_ref_index(PyObject* self, PyObject* index,... function PyObject (line 674) | static PyObject* PyTensorObject_grad_fn(PyObject* self, void* unused) { function PyObject (line 678) | static PyObject* PyTensorObject_is_leaf(PyObject* self, void* unused) { function PyObject (line 682) | static PyObject* PyTensorObject_requires_grad(PyObject* self, void* unus... function PyTensorObject_set_requires_grad (line 686) | static int PyTensorObject_set_requires_grad(PyObject* self, PyObject* re... function PyObject (line 696) | static PyObject* PyTensorObject_is_lazy(PyObject* self, void* unused) { function PyObject (line 700) | static PyObject* PyTensorObject_is_eager(PyObject* self, void* unused) { function PyObject (line 704) | static PyObject* PyTensorObject_is_global(PyObject* self, void* unused) { function PyObject (line 708) | static PyObject* PyTensorObject_is_local(PyObject* self, void* unused) { function PyObject (line 712) | static PyObject* PyTensorObject__tensor_buffer_shapes_and_dtypes(PyObjec... function PyObject (line 718) | static PyObject* PyTensorObject_device(PyObject* self, void* unused) { function PyObject (line 724) | static PyObject* PyTensorObject_placement(PyObject* self, void* unused) { function PyObject (line 730) | static PyObject* PyTensorObject_sbp(PyObject* self, void* unused) { function PyObject (line 767) | static PyObject* TensorMetaCls_call(PyObject* type, PyObject* args, PyOb... function TensorMetaCls_dealloc (line 771) | static void TensorMetaCls_dealloc(PyObject* type) { PyType_Type.tp_deall... function PyHeapTypeObject (line 773) | static PyHeapTypeObject* MakeTensorMetaclass() { function PyTypeObject (line 799) | static PyTypeObject* MakeTensorType() { function PyTypeObject (line 832) | static PyTypeObject* MakeParameterType() { function PyObject (line 855) | PyObject* PyTensor_New(const std::shared_ptr& data) { function PyObject (line 859) | PyObject* PyParameter_New(const std::shared_ptr& data) { function PyObject (line 863) | PyObject* PyParameter_New(const std::shared_ptr& data, bool requ... FILE: oneflow/api/python/framework/tensor.h function namespace (line 24) | namespace oneflow { FILE: oneflow/api/python/framework/tensor_functions.cpp type oneflow (line 34) | namespace oneflow { type one (line 35) | namespace one { function PyObject (line 42) | PyObject* concat_self(PyObject* self, PyObject* args) { function PyObject (line 49) | PyObject* ndarray_judgment_and_compatibility(PyObject* self, PyObjec... function PyObject (line 109) | static PyObject* PyTensorObject_nb_pow(PyObject* a, PyObject* b, PyO... function PyObject (line 139) | PyObject* PyTensorObject_nb_inplace_pow(PyObject* a, PyObject* b, Py... function PyObject (line 358) | static PyObject* PyTensorObject_byte(PyObject* self, PyObject* unuse... function PyObject (line 364) | static PyObject* PyTensorObject_dim(PyObject* self, PyObject* unused) { function PyObject (line 370) | static PyObject* PyTensorObject_nelement(PyObject* self, PyObject* u... function PyObject (line 376) | static PyObject* PyTensorObject_element_size(PyObject* self, PyObjec... function PyObject (line 382) | static PyObject* PyTensorObject_get_device(PyObject* self, PyObject*... function PyObject (line 391) | static PyObject* PyTensorObject_size(PyObject* self, PyObject* args,... function PyObject (line 409) | static PyObject* PyTensorObject_cast(PyObject* self, PyObject* args,... function PyObject (line 427) | static PyObject* PyTensorObject_diag(PyObject* self, PyObject* args,... function PyObject (line 439) | static PyObject* PyTensorObject_diagonal(PyObject* self, PyObject* a... function PyObject (line 453) | static PyObject* PyTensorObject_matmul(PyObject* self, PyObject* arg... function PyObject (line 468) | static PyObject* PyTensorObject_reshape(PyObject* self, PyObject* ar... function PyObject (line 478) | static PyObject* PyTensorObject_reshape_as(PyObject* self, PyObject*... function PyObject (line 491) | static PyObject* PyTensorObject_cpu(PyObject* self, PyObject* unused) { function PyObject (line 498) | static PyObject* PyTensorObject_cuda(PyObject* self, PyObject* args,... function PyObject (line 521) | static PyObject* PyTensorObject_var(PyObject* self, PyObject* args, ... function PyObject (line 552) | static PyObject* PyTensorObject_std(PyObject* self, PyObject* args, ... function PyObject (line 584) | static PyObject* PyTensorObject_softplus(PyObject* self, PyObject* a... function PyObject (line 597) | static PyObject* PyTensorObject_relu(PyObject* self, PyObject* unuse... function PyObject (line 603) | static PyObject* PyTensorObject_relu_(PyObject* self, PyObject* unus... function PyObject (line 643) | static PyObject* PyTensorObject_view(PyObject* self, PyObject* args,... function PyObject (line 653) | static PyObject* PyTensorObject_view_as(PyObject* self, PyObject* ar... function PyObject (line 666) | static PyObject* PyTensorObject_permute(PyObject* self, PyObject* ar... function PyObject (line 677) | static PyObject* PyTensorObject_transpose(PyObject* self, PyObject* ... function PyObject (line 691) | static PyObject* PyTensorObject_local_to_global(PyObject* self, PyOb... function PyObject (line 729) | static PyObject* PyTensorObject_global_to_global(PyObject* self, PyO... function PyObject (line 789) | static PyObject* PyTensorObject_to_global(PyObject* self, PyObject* ... function PyObject (line 804) | static PyObject* PyTensorObject_to_local(PyObject* self, PyObject* u... function PyObject (line 819) | static PyObject* PyTensorObject_type_as(PyObject* self, PyObject* ar... function PyObject (line 854) | static PyObject* PyTensorObject_new(PyObject* self, PyObject* args, ... function PyTensorObject_setitem (line 910) | int PyTensorObject_setitem(PyObject* self, PyObject* item, PyObject*... function PyObject (line 1179) | PyObject* PyTensorObject_richcompare(PyObject* self, PyObject* other... FILE: oneflow/api/python/framework/tensor_functions_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/api/python/framework/tensor_tuple.cpp type oneflow (line 26) | namespace oneflow { type one (line 27) | namespace one { type TensorTupleUtil (line 31) | struct TensorTupleUtil final { method ToString (line 32) | static std::string ToString(const TensorTuple& tensor_tuple) { method MergeFrom (line 44) | static void MergeFrom(std::shared_ptr& tensor_tuple, ... method AppendTensor (line 48) | static void AppendTensor(std::shared_ptr& tensor_tuple, FILE: oneflow/api/python/framework/tensortype.cpp type oneflow (line 30) | namespace oneflow { type one (line 31) | namespace one { function get_dtype_string (line 66) | static const std::string get_dtype_string(PyTensorType* tensortype) { function PyObject (line 75) | static PyObject* PyTensorTypeMetaCls_call(PyObject* self, PyObject* ... FILE: oneflow/api/python/framework/tensortype.h function namespace (line 24) | namespace oneflow { FILE: oneflow/api/python/framework/thread.cpp type oneflow (line 23) | namespace oneflow { class UsingThreadUidSet (line 27) | class UsingThreadUidSet final { method UsingThreadUidSet (line 29) | UsingThreadUidSet() method Get (line 35) | Maybe Get() { method Put (line 50) | Maybe Put(int64_t thread_uid) { function UsingThreadUidSet (line 65) | UsingThreadUidSet* MutUsingThreadUidSet() { method UsingThreadUidSet (line 29) | UsingThreadUidSet() method Get (line 35) | Maybe Get() { method Put (line 50) | Maybe Put(int64_t thread_uid) { FILE: oneflow/api/python/framework/thread.h function namespace (line 22) | namespace oneflow { FILE: oneflow/api/python/framework/typeinfo.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type is_floating_point_with_half (line 39) | struct is_floating_point_with_half : public std::false_type {} function PyGetVal (line 49) | typename std::enable_if::value, PyObj... function PyGetVal (line 54) | typename std::enable_if::value, PyObject*>::type... function PyObject (line 58) | PyObject* PyGetMaxVal(DataType datatype) { function PyObject (line 69) | PyObject* PyGetMinVal(DataType datatype) { function PyObject (line 104) | static PyObject* PyIInfo_new(PyTypeObject* self, PyObject* args, PyO... function PyObject (line 127) | static PyObject* PyFInfo_new(PyTypeObject* self, PyObject* args, PyO... function PyObject (line 150) | static PyObject* PyDInfo_bits(PyObject* self, void*) { function PyObject (line 157) | static PyObject* PyDInfo_min(PyObject* self, void*) { function PyObject (line 169) | static PyObject* PyDInfo_max(PyObject* self, void*) { function PyObject (line 181) | static PyObject* PyFInfo_resolution(PyObject* self, void*) { function PyObject (line 194) | static PyObject* PyFInfo_eps(PyObject* self, void*) { function PyObject (line 207) | static PyObject* PyFInfo_tiny(PyObject* self, void*) { function PyObject (line 220) | static PyObject* PyDInfo_dtype(PyObject* self, void*) { function PyObject (line 228) | static PyObject* PyIInfo_str(PyObject* self) { function PyObject (line 239) | static PyObject* PyFInfo_str(PyObject* self) { type PyGetSetDef (line 254) | struct PyGetSetDef type PyGetSetDef (line 262) | struct PyGetSetDef function init_info_type (line 273) | static void init_info_type() { FILE: oneflow/api/python/framework/typeinfo.h function namespace (line 25) | namespace oneflow { FILE: oneflow/api/python/framework/variable_tensor_mgr.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/api/python/functional/common.cpp type oneflow (line 39) | namespace oneflow { type one (line 40) | namespace one { type functional (line 41) | namespace functional { type detail (line 43) | namespace detail { function GetItemInPyScalarTensor (line 48) | Maybe GetItemInPyScalarTensor(PyObject* obj) { function isinstance_fast (line 55) | bool isinstance_fast(PyObject* obj) { function T (line 66) | const T& cast_fast(PyObject* obj) { function T (line 79) | const T& cast_fast(PyObject* obj) { function PySequenceCheck (line 89) | bool PySequenceCheck(PyObject* obj, const std::function> PyUnpackTensorSequence(PyObje... function PyTensorTupleCheck (line 140) | bool PyTensorTupleCheck(PyObject* obj) { return detail::isinstance... function PyUnpackTensorTuple (line 142) | std::shared_ptr PyUnpackTensorTuple(PyObject* obj) { function PyScalarCheck (line 147) | bool PyScalarCheck(PyObject* obj) { function Scalar (line 151) | Scalar PyUnpackScalar(PyObject* obj) { function PyScalarTensorCheck (line 174) | bool PyScalarTensorCheck(PyObject* obj) { function Scalar (line 183) | Scalar PyUnpackScalarTensor(PyObject* obj) { function PyDTypeCheck (line 230) | bool PyDTypeCheck(PyObject* obj) { return detail::isinstance_fast<... function PyUnpackDType (line 231) | Symbol PyUnpackDType(PyObject* obj) { return *detail::cast_... function PyLayoutCheck (line 234) | bool PyLayoutCheck(PyObject* obj) { return detail::isinstance_fast... function PyUnpackLayout (line 235) | Symbol PyUnpackLayout(PyObject* obj) { return *detail::cas... function PyMemoryFormatCheck (line 238) | bool PyMemoryFormatCheck(PyObject* obj) { return PyMemoryFormat_Ch... function MemoryFormat (line 239) | MemoryFormat PyUnpackMemoryFormat(PyObject* obj) { return PyMemory... function PyDTypeSequenceCheck (line 242) | bool PyDTypeSequenceCheck(PyObject* obj) { function PyUnpackDTypeSequence (line 245) | std::vector> PyUnpackDTypeSequence(PyObject* obj) { function PyShapeCheck (line 250) | bool PyShapeCheck(PyObject* obj) { return PyLongSequenceCheck(obj); } function Shape (line 252) | Shape PyUnpackShape(PyObject* obj) { function PyShapeSequenceCheck (line 266) | bool PyShapeSequenceCheck(PyObject* obj) { function PyUnpackShapeSequence (line 269) | std::vector PyUnpackShapeSequence(PyObject* obj) { function PyGeneratorCheck (line 274) | bool PyGeneratorCheck(PyObject* obj) { return detail::isinstance_f... function PyUnpackGenerator (line 275) | std::shared_ptr PyUnpackGenerator(PyObject* obj) { function PyDeviceCheck (line 280) | bool PyDeviceCheck(PyObject* obj) { return detail::isinstance_fast... function PyUnpackDevice (line 281) | Symbol PyUnpackDevice(PyObject* obj) { function PyParallelDescCheck (line 286) | bool PyParallelDescCheck(PyObject* obj) { function PyUnpackParallelDesc (line 289) | Symbol PyUnpackParallelDesc(PyObject* obj) { function PySbpParallelCheck (line 294) | bool PySbpParallelCheck(PyObject* obj) { return detail::isinstance... function PyUnpackSbpParallel (line 295) | Symbol PyUnpackSbpParallel(PyObject* obj) { function PySbpParallelSequenceCheck (line 300) | bool PySbpParallelSequenceCheck(PyObject* obj) { function PyUnpackSbpParallelSequence (line 303) | std::vector> PyUnpackSbpParallelSequence(PyObj... function PyTensorIndexCheck (line 309) | bool PyTensorIndexCheck(PyObject* obj) { function TensorIndex (line 314) | TensorIndex PyUnpackTensorIndex(PyObject* obj) { function PyOpExprCheck (line 403) | bool PyOpExprCheck(PyObject* obj) { return detail::isinstance_fast... function PyUnpackOpExpr (line 405) | std::shared_ptr PyUnpackOpExpr(PyObject* obj) { function PyUnpackLong (line 410) | Maybe PyUnpackLong(PyObject* py_obj) { FILE: oneflow/api/python/functional/common.h function namespace (line 43) | namespace oneflow { FILE: oneflow/api/python/functional/dispatch_stateful_ops.cpp type oneflow (line 28) | namespace oneflow { type one (line 29) | namespace one { type functional (line 30) | namespace functional { type impl (line 32) | namespace impl { function ONEFLOW_FUNCTION_LIBRARY (line 34) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/api/python/functional/function_def.h function namespace (line 26) | namespace oneflow { FILE: oneflow/api/python/functional/indexing.cpp type oneflow (line 29) | namespace oneflow { type one (line 30) | namespace one { type functional (line 31) | namespace functional { type detail (line 33) | namespace detail { function PySliceUnpack (line 35) | void PySliceUnpack(PyObject* object, Py_ssize_t* start, Py_ssize... function DataType (line 59) | DataType InferScalarType(PyObject* object) { function ParseScalar (line 88) | void ParseScalar(PyObject* object, char* data, const DataType& d... function RecursiveParseAndAssign (line 113) | void RecursiveParseAndAssign(PyObject* object, char* data, const... function ParseArrayToTensor (line 128) | void ParseArrayToTensor(PyObject* object, function Shape (line 142) | Shape InferArraySizes(PyObject* object) { function ConvertToIndexingTensor (line 158) | Maybe ConvertToIndexingTensor(PyObject* object) { function IndexItem (line 196) | IndexItem UnpackIndexItem(PyObject* object) { FILE: oneflow/api/python/functional/indexing.h function namespace (line 27) | namespace oneflow { FILE: oneflow/api/python/functional/python_arg.cpp type oneflow (line 35) | namespace oneflow { type one (line 36) | namespace one { type functional (line 37) | namespace functional { function Scalar (line 98) | Scalar PythonArg::ObjectAs() const { function Shape (line 141) | Shape PythonArg::ObjectAs() const { function TensorIndex (line 186) | TensorIndex PythonArg::ObjectAs() const { function PyObject (line 197) | PyObject* PythonArg::ObjectAs() const { function MemoryFormat (line 210) | MemoryFormat PythonArg::ObjectAs() const { FILE: oneflow/api/python/functional/python_arg.h function namespace (line 29) | namespace oneflow { FILE: oneflow/api/python/functional/python_arg_parser.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type functional (line 22) | namespace functional { FILE: oneflow/api/python/functional/python_arg_parser.h function namespace (line 27) | namespace oneflow { FILE: oneflow/api/python/functional/python_return_types.h function namespace (line 32) | namespace oneflow { FILE: oneflow/api/python/functional/tensor_api.cpp type oneflow (line 42) | namespace oneflow { type one (line 43) | namespace one { type functional (line 44) | namespace functional { type impl (line 46) | namespace impl { class TensorWithDataFunctor (line 48) | class TensorWithDataFunctor { class GlobalTensorWithDataFunctor (line 83) | class GlobalTensorWithDataFunctor { class TensorEmptyGenericCtorFunctor (line 111) | class TensorEmptyGenericCtorFunctor { class GlobalTensorEmptyGenericCtorFunctor (line 120) | class GlobalTensorEmptyGenericCtorFunctor { class TensorWithOtherGenericCtorFunctor (line 130) | class TensorWithOtherGenericCtorFunctor { class TensorWithDataGenericCtorFunctor (line 142) | class TensorWithDataGenericCtorFunctor { class GlobalTensorWithDataGenericCtorFunctor (line 172) | class GlobalTensorWithDataGenericCtorFunctor { class TensorWithShapeGenericCtorFunctor (line 202) | class TensorWithShapeGenericCtorFunctor { class GlobalTensorWithShapeGenericCtorFunctor (line 218) | class GlobalTensorWithShapeGenericCtorFunctor { class AssignLocalTensorFunctor (line 230) | class AssignLocalTensorFunctor { method AssignLocalTensorFunctor (line 232) | AssignLocalTensorFunctor() { function get_shape_or_stride_from_numpy (line 253) | static std::vector get_shape_or_stride_from_numpy(size_... class LocalTensorSharedDlPackDataFunctor (line 259) | class LocalTensorSharedDlPackDataFunctor { method LocalTensorSharedDlPackDataFunctor (line 261) | LocalTensorSharedDlPackDataFunctor() {} class LocalTensorSharedNumpyDataFunctor (line 282) | class LocalTensorSharedNumpyDataFunctor { method LocalTensorSharedNumpyDataFunctor (line 284) | LocalTensorSharedNumpyDataFunctor() {} function ONEFLOW_FUNCTION_LIBRARY (line 366) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/api/python/functional/value_types.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type functional (line 23) | namespace functional { function IsIntegralType (line 86) | bool IsIntegralType(ValueType type) { return type >= kINT32 && typ... function IsIntegralListType (line 87) | bool IsIntegralListType(ValueType type) { function IsFloatingType (line 90) | bool IsFloatingType(ValueType type) { return type >= kFLOAT && typ... function IsFloatingListType (line 91) | bool IsFloatingListType(ValueType type) { FILE: oneflow/api/python/functional/value_types.h function namespace (line 32) | namespace oneflow { function namespace (line 207) | namespace std { FILE: oneflow/api/python/gil_foreign_lock_helper.cpp type oneflow (line 24) | namespace oneflow { class GILForeignLockHelper (line 25) | class GILForeignLockHelper final : public ForeignLockHelper { method WithScopedRelease (line 26) | Maybe WithScopedRelease(const std::function()>& Ca... method WithScopedAcquire (line 36) | Maybe WithScopedAcquire(const std::function()>& Ca... FILE: oneflow/api/python/init.cpp type oneflow (line 29) | namespace oneflow { function Int2IntListMapContaining (line 36) | bool Int2IntListMapContaining(const Int2IntListMap& bigger, const Int2... function PYBIND11_MODULE (line 51) | PYBIND11_MODULE(_oneflow_internal, m) { FILE: oneflow/api/python/ir.cpp type oneflow (line 41) | namespace oneflow { FILE: oneflow/api/python/job_build/job_build_and_infer.cpp type oneflow (line 23) | namespace oneflow { function MarkVariableGradients (line 25) | Maybe MarkVariableGradients(const one::TensorTuple& variables, function MarkOutputGradients (line 42) | Maybe MarkOutputGradients(const one::TensorTuple& outputs, FILE: oneflow/api/python/job_build/job_build_and_infer.h function namespace (line 30) | namespace oneflow { FILE: oneflow/api/python/job_build/lazy_mode.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/api/python/multiprocessing/init.cpp type oneflow (line 34) | namespace oneflow { type multiprocessing (line 35) | namespace multiprocessing { function multiprocessing_init (line 39) | void multiprocessing_init() { function set_num_threads (line 57) | void set_num_threads(int num) { FILE: oneflow/api/python/multiprocessing/object_ptr.cpp class OFPointer (line 23) | class OFPointer FILE: oneflow/api/python/multiprocessing/object_ptr.h function explicit (line 27) | explicit OFPointer(T* ptr) noexcept : ptr(ptr){} function T (line 35) | T* get() { return ptr; } function T (line 36) | const T* get() const { return ptr; } function T (line 37) | T* release() { function operator (line 42) | operator T*() { return ptr; } FILE: oneflow/api/python/multiprocessing/shared_memory.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/api/python/of_api_registry.cpp type oneflow (line 18) | namespace oneflow { function SubModuleMap (line 26) | SubModuleMap* GetSubModuleMap() { FILE: oneflow/api/python/of_api_registry.h function namespace (line 29) | namespace oneflow { FILE: oneflow/api/python/profiler.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/api/python/registry/registry.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/api/python/remat/remat.cpp type oneflow (line 27) | namespace oneflow { function rematable_storage (line 30) | Maybe rematable_storage(const std::shared_... FILE: oneflow/api/python/rpc/ccl.cpp type oneflow (line 26) | namespace oneflow { function CpuBroadcast (line 29) | Maybe CpuBroadcast(py::bytes* in, int64_t root) { FILE: oneflow/api/python/rpc/rank_group.cpp type oneflow (line 27) | namespace oneflow { function CheckCurrentRankGroupConsistency (line 31) | Maybe CheckCurrentRankGroupConsistency() { FILE: oneflow/api/python/session/session.cpp type oneflow (line 26) | namespace oneflow { FILE: oneflow/api/python/stack_getter.cpp type oneflow (line 29) | namespace oneflow { FILE: oneflow/api/python/symbol/job_conf_symbol.cpp type oneflow (line 26) | namespace oneflow { function CreateJobConfSymbol (line 28) | Maybe CreateJobConfSymbol(int64_t symbol_id, const std::strin... FILE: oneflow/api/python/symbol/op_conf_symbol.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/api/python/symbol/placement_symbol.cpp type oneflow (line 36) | namespace oneflow { function GetDeviceCount (line 40) | int64_t GetDeviceCount(const std::string& device_name) { type PlacementSymbolExportUtil (line 44) | struct PlacementSymbolExportUtil { method CheckDeviceTag (line 45) | static Maybe CheckDeviceTag(const std::string& type) { method CreateParallelDesc (line 53) | static Maybe CreateParallelDesc( method CreateParallelDesc (line 67) | static Maybe CreateParallelDesc(const std::string& pro... method ParseAndFormatRanks (line 80) | static Maybe> ParseAndFormatRanks(const py:... method GetRanksShape (line 107) | static Maybe GetRanksShape(PyArrayObject* ranks) { method ParseAndFormatRanks (line 113) | static Maybe> ParseAndFormatRanks(PyArrayOb... method CreateParallelDescSymbol (line 135) | static Maybe> CreateParallelDescSymbol( method CreateParallelDescSymbol (line 143) | static Maybe> CreateParallelDescSymbol(const st... method CreateParallelDescSymbol (line 154) | static Maybe> CreateParallelDescSymbol(const st... method AllDevicePlacement (line 158) | static Maybe> AllDevicePlacement(const std::str... method GetPlacementRanks (line 185) | static Maybe GetPlacementRanks(const Symbol... FILE: oneflow/api/python/symbol/sbp_symbol.cpp type oneflow (line 30) | namespace oneflow { function MakeSplitSbpParallelList (line 34) | Maybe>> MakeSplitSbpParallelList(int m... function GetSplitSbpParallel (line 41) | Maybe> GetSplitSbpParallel(int axis) { function GetBroadcastSbpParallel (line 52) | Maybe> GetBroadcastSbpParallel() { function GetPartialSumSbpParallel (line 57) | Maybe> GetPartialSumSbpParallel() { function SbpGetState (line 62) | Maybe> SbpGetState(const Symbol> GetSbpFromState(const std::pair CreateScopeSymbol(int64_t symbol_id, const std::string& s... FILE: oneflow/api/python/utils/dataloader.cpp type oneflow (line 30) | namespace oneflow { function setSignalHandler (line 58) | static inline void setSignalHandler(int signal, void (*handler)(int, s... function handler_SIGTERM (line 86) | static void handler_SIGTERM(int sig, siginfo_t* info, void* ctx) { function set_worker_signal_handlers (line 98) | static void set_worker_signal_handlers() { function error_if_any_worker_fails (line 108) | static void error_if_any_worker_fails() { function utils_unpackLong (line 155) | inline int64_t utils_unpackLong(PyObject* obj) { function set_worker_pids (line 166) | static void set_worker_pids(py::args py_args) { function remove_worker_pids (line 193) | static void remove_worker_pids(py::args py_args) { function PyObject (line 209) | static PyObject* set_worker_signal_handlers(PyObject* module, PyObject... function PyObject (line 213) | static PyObject* set_worker_pids(PyObject* module, PyObject* _ignored)... function PyObject (line 215) | static PyObject* remove_worker_pids(PyObject* module, PyObject* _ignor... function PyObject (line 217) | static PyObject* error_if_any_worker_fails(PyObject* module, PyObject*... FILE: oneflow/api/python/utils/tensor_utils.cpp type oneflow (line 34) | namespace oneflow { type one (line 35) | namespace one { function EagerLocalTensorZeros (line 37) | Maybe EagerLocalTensorZeros(const std::shared_ptr& t) { function CopyFromNumpyArray (line 60) | void CopyFromNumpyArray(ep::Stream* stream, function CopyLocalTensorFromUntypedArray (line 69) | Maybe CopyLocalTensorFromUntypedArray(const std::shared_ptr RegisterTensorHook(const std::shared_ptr& self, function RegisterTensorPostGradAccumulationHook (line 113) | Maybe RegisterTensorPostGradAccumulationHook(const std::shared... function TensorGetPyTupleOfSbp (line 120) | Maybe TensorGetPyTupleOfSbp(const Tensor& tensor) { function MakeLocalTensorFromData (line 129) | Maybe MakeLocalTensorFromData(PyObject* data, const Optional... function GetAllBroadcastNdSbp (line 198) | Maybe> GetAllBroadcastNdSbp(size_t ndim) { function MakeGlobalTensorFromData (line 210) | Maybe MakeGlobalTensorFromData(PyObject* data, const Optiona... function MakeTensorFromOtherTensor (line 274) | Maybe MakeTensorFromOtherTensor(const std::shared_ptr MakeTensorFromOtherTensor(const std::shared_ptr MakeTensorFromOtherTensor(const std::shared_ptr& order, std::vector& topo_structs, function UpdateSat (line 244) | void UpdateSat(const std::vector& topo_structs, Straigh... function InitInOutTopoStructs (line 264) | void InitInOutTopoStructs(std::vector* topo_structs) { function ComputeLayer (line 297) | void ComputeLayer(std::vector* topo_structs) { function InitAllParameters (line 310) | void InitAllParameters(std::vector* topo_structs, function StraightenOpNodes (line 349) | void StraightenOpNodes(HashMap& op_node2t... function InitMemory (line 393) | void InitMemory(const OpGraph& op_graph, SbpGraph* sbp_graph, bool n... function StraightenSubGraph (line 514) | void StraightenSubGraph(const std::vector& sub_graph, function StraightenOpGraph (line 541) | void StraightenOpGraph(const OpGraph& op_graph, std::vector InitLog2() { FILE: oneflow/core/auto_parallel/binary_set.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/auto_parallel/boxing_collector.cpp type oneflow (line 36) | namespace oneflow { function DfsSetNdSbp (line 42) | void DfsSetNdSbp(const std::vector& id2sbp_parallel, int3... function SetNdSbpDim (line 57) | Maybe SetNdSbpDim(const NdSbp& nd_sbp, int32_t hierarchy_num) { function TotalNumSplit (line 79) | int32_t TotalNumSplit(const NdSbp& nd_sbp, const ParallelDesc& paralle... function AskSbpCombinationFor1DSbp (line 91) | Maybe AskSbpCombinationFor1DSbp(const NdSbp& sbp_producer, const... FILE: oneflow/core/auto_parallel/boxing_collector.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/auto_parallel/sbp_collector.cpp type oneflow (line 23) | namespace oneflow { type auto_parallel (line 25) | namespace auto_parallel { function IfIntersectAll (line 29) | bool IfIntersectAll( function FindUniqueSbpSets (line 40) | void FindUniqueSbpSets( function FindUniqueSbpGroups (line 58) | void FindUniqueSbpGroups( function No2SbpFromSameUniqueGroup (line 80) | bool No2SbpFromSameUniqueGroup(const BinarySet& bs, FILE: oneflow/core/auto_parallel/sbp_collector.h function namespace (line 33) | namespace oneflow { FILE: oneflow/core/auto_parallel/sbp_constructor.cpp type oneflow (line 33) | namespace oneflow { type auto_parallel (line 35) | namespace auto_parallel { function UpdateMemoryRatio (line 55) | double UpdateMemoryRatio() { FILE: oneflow/core/auto_parallel/sbp_constructor.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/auto_parallel/sbp_edge.cpp type oneflow (line 29) | namespace oneflow { type auto_parallel (line 30) | namespace auto_parallel { FILE: oneflow/core/auto_parallel/sbp_edge.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/auto_parallel/sbp_graph.cpp type oneflow (line 25) | namespace oneflow { type auto_parallel (line 26) | namespace auto_parallel { function SbpNode (line 36) | SbpNode* SbpGraph::GenerateNode() { FILE: oneflow/core/auto_parallel/sbp_graph.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/auto_parallel/sbp_node.cpp type oneflow (line 33) | namespace oneflow { type auto_parallel (line 34) | namespace auto_parallel { function SbpEdge (line 839) | SbpEdge* SbpNode::FindEdgeWithNode(const SbpNode* other_node) const { function NdSbpSignature (line 850) | const NdSbpSignature& SbpNode::FinalSbpSignature() const { FILE: oneflow/core/auto_parallel/sbp_node.h function namespace (line 32) | namespace oneflow { FILE: oneflow/core/auto_parallel/sbp_util.cpp type oneflow (line 23) | namespace oneflow { type auto_parallel (line 24) | namespace auto_parallel { function RequireSameSbp (line 27) | bool RequireSameSbp(const OpNode* consumer, const std::string& ibn) { FILE: oneflow/core/auto_parallel/sbp_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/autograd/autograd_captured_tensor.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/autograd/autograd_engine.cpp type oneflow (line 41) | namespace oneflow { type one (line 42) | namespace one { function GatherFunctionNodes (line 46) | void GatherFunctionNodes(FunctionNode* node, std::stack>& ... function CopyOrAccGrad (line 86) | Maybe CopyOrAccGrad(AutogradMeta* autograd_meta, bool autograd... function RawTouchGlobalTensor (line 108) | Maybe RawTouchGlobalTensor(const std::shared_ptr&... function CheckGlobalTensorsMeta (line 115) | Maybe CheckGlobalTensorsMeta(const TensorTuple& tensor_tuple) { function GetDebugGraphFileName (line 122) | std::string GetDebugGraphFileName(const std::string& mode, const std... type NodeFrame (line 356) | struct NodeFrame { method NodeFrame (line 357) | explicit NodeFrame(FunctionNode* node) : node_(node), next_functio... method FunctionNode (line 361) | FunctionNode* GetNextFunction() { function AutogradEngine (line 515) | AutogradEngine* GetThreadLocalAutogradEngine() { function AddAccumulateFunctionNode (line 520) | Maybe AddAccumulateFunctionNode(const std::shared_ptr&... FILE: oneflow/core/autograd/autograd_engine.h function namespace (line 31) | namespace oneflow { function class (line 98) | class AutogradEngine { function class (line 146) | class GraphTask final { function ClearEngine (line 183) | void ClearEngine() override{} FILE: oneflow/core/autograd/autograd_function.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { FILE: oneflow/core/autograd/autograd_function.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/autograd/autograd_meta.cpp type oneflow (line 25) | namespace oneflow { type one (line 27) | namespace one { function GetSbpTuple (line 38) | Maybe>&> GetSbpTuple(Symbol Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(BaseActivationCaptureState* ctx, const TensorT... class Silu (line 41) | class Silu : public BaseActivation { method Apply (line 43) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... class Mish (line 55) | class Mish : public BaseActivation { method Apply (line 57) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... class Selu (line 69) | class Selu : public BaseActivation { method Apply (line 71) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... class Softsign (line 83) | class Softsign : public BaseActivation { method Apply (line 85) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... class GeLU (line 97) | class GeLU : public BaseActivation { method Apply (line 99) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... class FastGeLU (line 111) | class FastGeLU : public BaseActivation { method Apply (line 113) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... type QuickGeluCaptureState (line 125) | struct QuickGeluCaptureState : public AutoGradCaptureState { class QuickGeLU (line 129) | class QuickGeLU : public OpExprGradFunction { method Init (line 131) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 133) | Maybe Capture(QuickGeluCaptureState* ctx, const TensorTuple&... method Apply (line 143) | Maybe Apply(const QuickGeluCaptureState* ctx, const TensorTu... type SquareReLUCaptureState (line 155) | struct SquareReLUCaptureState : public AutoGradCaptureState { class SquareReLU (line 159) | class SquareReLU : public OpExprGradFunction { method Init (line 161) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 163) | Maybe Capture(SquareReLUCaptureState* ctx, const TensorTuple... method Apply (line 173) | Maybe Apply(const SquareReLUCaptureState* ctx, const TensorT... class HardSigmoid (line 185) | class HardSigmoid : public BaseActivation { method Apply (line 187) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... type HardShrinkCaptureState (line 199) | struct HardShrinkCaptureState : public AutoGradCaptureState { class HardShrink (line 204) | class HardShrink : public OpExprGradFunction { method Init (line 206) | Maybe Init(const OpExpr& op) override { method Capture (line 213) | Maybe Capture(HardShrinkCaptureState* ctx, const TensorTuple... method Apply (line 225) | Maybe Apply(const HardShrinkCaptureState* ctx, const TensorT... class HardSwish (line 241) | class HardSwish : public BaseActivation { method Apply (line 243) | Maybe Apply(const BaseActivationCaptureState* ctx, const Ten... type ReLUCaptureState (line 256) | struct ReLUCaptureState : public AutoGradCaptureState { class ReLU (line 260) | class ReLU : public OpExprGradFunction { method Init (line 262) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 264) | Maybe Capture(ReLUCaptureState* ctx, const TensorTuple& inpu... method Apply (line 273) | Maybe Apply(const ReLUCaptureState* ctx, const TensorTuple& ... type LeakyReluCaptureState (line 286) | struct LeakyReluCaptureState : public AutoGradCaptureState { class LeakyRelu (line 291) | class LeakyRelu : public OpExprGradFunction { method Init (line 293) | Maybe Init(const OpExpr& op) override { method Capture (line 300) | Maybe Capture(LeakyReluCaptureState* ctx, const TensorTuple&... method Apply (line 312) | Maybe Apply(const LeakyReluCaptureState* ctx, const TensorTu... type SoftplusCaptureState (line 327) | struct SoftplusCaptureState : public AutoGradCaptureState { class Softplus (line 333) | class Softplus : public OpExprGradFunction { method Init (line 335) | Maybe Init(const OpExpr& op) override { method Capture (line 342) | Maybe Capture(SoftplusCaptureState* ctx, const TensorTuple& ... method Apply (line 353) | Maybe Apply(const SoftplusCaptureState* ctx, const TensorTup... type HardTanhCaptureState (line 369) | struct HardTanhCaptureState : public AutoGradCaptureState { class HardTanh (line 375) | class HardTanh : public OpExprGradFunction { method Init (line 377) | Maybe Init(const OpExpr& op) override { method Capture (line 384) | Maybe Capture(HardTanhCaptureState* ctx, const TensorTuple& ... method Apply (line 397) | Maybe Apply(const HardTanhCaptureState* ctx, const TensorTup... type EluCaptureState (line 413) | struct EluCaptureState : public AutoGradCaptureState { class Elu (line 418) | class Elu : public OpExprGradFunction { method Init (line 420) | Maybe Init(const OpExpr& op) override { method Capture (line 427) | Maybe Capture(EluCaptureState* ctx, const TensorTuple& input... method Apply (line 439) | Maybe Apply(const EluCaptureState* ctx, const TensorTuple& o... type CeluCaptureState (line 454) | struct CeluCaptureState : public AutoGradCaptureState { class Celu (line 459) | class Celu : public OpExprGradFunction { method Init (line 461) | Maybe Init(const OpExpr& op) override { method Capture (line 468) | Maybe Capture(CeluCaptureState* ctx, const TensorTuple& inpu... method Apply (line 480) | Maybe Apply(const CeluCaptureState* ctx, const TensorTuple& ... type SoftShrinkCaptureState (line 495) | struct SoftShrinkCaptureState : public AutoGradCaptureState { class SoftShrink (line 500) | class SoftShrink : public OpExprGradFunction { method Init (line 502) | Maybe Init(const OpExpr& op) override { method Capture (line 509) | Maybe Capture(SoftShrinkCaptureState* ctx, const TensorTuple... method Apply (line 521) | Maybe Apply(const SoftShrinkCaptureState* ctx, const TensorT... type PReLUCaptureState (line 537) | struct PReLUCaptureState : public AutoGradCaptureState { class PReLU (line 542) | class PReLU : public OpExprGradFunction { method Init (line 544) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 546) | Maybe Capture(PReLUCaptureState* ctx, const TensorTuple& inp... method Apply (line 557) | Maybe Apply(const PReLUCaptureState* ctx, const TensorTuple&... type ThresholdCaptureState (line 576) | struct ThresholdCaptureState : public AutoGradCaptureState { class Threshold (line 581) | class Threshold : public OpExprGradFunction { method Init (line 583) | Maybe Init(const OpExpr& op) override { method Capture (line 590) | Maybe Capture(ThresholdCaptureState* ctx, const TensorTuple&... method Apply (line 602) | Maybe Apply(const ThresholdCaptureState* ctx, const TensorTu... type FracCaptureState (line 618) | struct FracCaptureState : public AutoGradCaptureState { class Frac (line 622) | class Frac : public OpExprGradFunction { method Init (line 624) | Maybe Init(const OpExpr& op) override { method Capture (line 631) | Maybe Capture(FracCaptureState* ctx, const TensorTuple& inpu... method Apply (line 639) | Maybe Apply(const FracCaptureState* ctx, const TensorTuple& ... FILE: oneflow/core/autograd/gradient_funcs/adaptive_avg_pool.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type AdaptivePoolCaptureState (line 25) | struct AdaptivePoolCaptureState : public AutoGradCaptureState { class AdaptivePoolNdGrad (line 30) | class AdaptivePoolNdGrad : public OpExprGradFunction Init(const OpExpr& op) override { return AdaptivePoolN... class AdaptiveAvgPool2dGrad (line 81) | class AdaptiveAvgPool2dGrad final : public AdaptivePoolNdGrad { method Init (line 83) | Maybe Init(const OpExpr& op) override { return AdaptivePoolN... class AdaptiveAvgPool3dGrad (line 86) | class AdaptiveAvgPool3dGrad final : public AdaptivePoolNdGrad { method Init (line 88) | Maybe Init(const OpExpr& op) override { return AdaptivePoolN... FILE: oneflow/core/autograd/gradient_funcs/adaptive_max_pool.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type AdaptiveMaxPoolCaptureState (line 25) | struct AdaptiveMaxPoolCaptureState : public AutoGradCaptureState { class AdaptiveMaxPoolNdGrad (line 30) | class AdaptiveMaxPoolNdGrad : public OpExprGradFunction Init(const OpExpr& op) override { return AdaptiveMaxPo... class AdaptiveMaxPool2dGrad (line 80) | class AdaptiveMaxPool2dGrad final : public AdaptiveMaxPoolNdGrad { method Init (line 82) | Maybe Init(const OpExpr& op) override { return AdaptiveMaxPo... class AdaptiveMaxPool3dGrad (line 85) | class AdaptiveMaxPool3dGrad final : public AdaptiveMaxPoolNdGrad { method Init (line 87) | Maybe Init(const OpExpr& op) override { return AdaptiveMaxPo... FILE: oneflow/core/autograd/gradient_funcs/add_n.cpp type oneflow (line 18) | namespace oneflow { type one (line 19) | namespace one { type AddNCaptureState (line 21) | struct AddNCaptureState : public AutoGradCaptureState { class AddN (line 26) | class AddN : public OpExprGradFunction { method Init (line 28) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(AddNCaptureState* ctx, const TensorTuple& inpu... method Apply (line 40) | Maybe Apply(const AddNCaptureState* ctx, const TensorTuple& ... FILE: oneflow/core/autograd/gradient_funcs/affine_grid.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type AffineGridInterpState (line 24) | struct AffineGridInterpState : public AutoGradCaptureState { class AffineGrid (line 30) | class AffineGrid : public OpExprGradFunction { method Init (line 32) | Maybe Init(const OpExpr& op) override { method Capture (line 39) | Maybe Capture(AffineGridInterpState* ctx, const TensorTuple&... method Apply (line 51) | Maybe Apply(const AffineGridInterpState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/amp_white_identity.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type AmpIdentityType (line 23) | enum class AmpIdentityType { type AmpIdentityCaptureState (line 28) | struct AmpIdentityCaptureState : public AutoGradCaptureState {} class AmpIdentityGrad (line 31) | class AmpIdentityGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 39) | Maybe Capture(AmpIdentityCaptureState* ctx, const TensorTupl... method Apply (line 44) | Maybe Apply(const AmpIdentityCaptureState* ctx, const Tensor... FILE: oneflow/core/autograd/gradient_funcs/as_strided.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type AsStridedCaptureState (line 24) | struct AsStridedCaptureState : public AutoGradCaptureState { class AsStrided (line 31) | class AsStrided : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/avg_pool.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type AvgPoolCaptureState (line 28) | struct AvgPoolCaptureState : public AutoGradCaptureState { class AvgPoolNdGrad (line 41) | class AvgPoolNdGrad : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/batch_gather.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type BatchGatherCaptureState (line 23) | struct BatchGatherCaptureState : public AutoGradCaptureState { class BatchGather (line 28) | class BatchGather : public OpExprGradFunction { method Init (line 34) | Maybe Init(const OpExpr& op) override { method Capture (line 41) | Maybe Capture(BiasAddCaptureState* ctx, const TensorTuple& i... method Apply (line 51) | Maybe Apply(const BiasAddCaptureState* ctx, const TensorTupl... FILE: oneflow/core/autograd/gradient_funcs/binary_cross_entropy.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type BinaryCrossEntropyCaptureState (line 22) | struct BinaryCrossEntropyCaptureState : public AutoGradCaptureState { class BinaryCrossEntropy (line 28) | class BinaryCrossEntropy : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe... method Capture (line 42) | Maybe Capture(BroadcastBinaryCaptureState* ctx, const Tensor... class BroadcastAdd (line 59) | class BroadcastAdd : public BroadcastBinaryGrad { method Apply (line 61) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 84) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... class BroadcastSub (line 98) | class BroadcastSub : public BroadcastBinaryGrad { method Apply (line 100) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 124) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... class BroadcastMul (line 138) | class BroadcastMul : public BroadcastBinaryGrad { method Apply (line 140) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 167) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... class BroadcastDiv (line 187) | class BroadcastDiv : public BroadcastBinaryGrad { method Apply (line 189) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 212) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... class BroadcastPow (line 230) | class BroadcastPow : public BroadcastBinaryGrad { method Apply (line 232) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 247) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... class BroadcastMinMax (line 257) | class BroadcastMinMax : public BroadcastBinaryGrad { method Apply (line 259) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 315) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... class BroadcastMinimum (line 329) | class BroadcastMinimum : public BroadcastMinMax { method Init (line 331) | Maybe Init(const OpExpr& op) override { class BroadcastMaximum (line 338) | class BroadcastMaximum : public BroadcastMinMax { method Init (line 340) | Maybe Init(const OpExpr& op) override { class BroadcastFMod (line 350) | class BroadcastFMod : public BroadcastBinaryGrad { method Apply (line 352) | Maybe Apply(const BroadcastBinaryCaptureState* ctx, const Te... method SaveTensorForBackward (line 410) | Maybe SaveTensorForBackward(BroadcastBinaryCaptureState* ctx... FILE: oneflow/core/autograd/gradient_funcs/broadcast_like.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type BroadCastLikeCaptureState (line 23) | struct BroadCastLikeCaptureState : public AutoGradCaptureState { class BroadCastLike (line 30) | class BroadCastLike : public OpExprGradFunction { method Init (line 35) | Maybe Init(const OpExpr& op) override { method Capture (line 41) | Maybe Capture(CastCaptureState* ctx, const TensorTuple& inpu... method Apply (line 48) | Maybe Apply(const CastCaptureState* ctx, const TensorTuple& ... FILE: oneflow/core/autograd/gradient_funcs/clip_by_scalar.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type ClipByScalarCaptureState (line 22) | struct ClipByScalarCaptureState : public AutoGradCaptureState { class ClipByScalar (line 28) | class ClipByScalar : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(ClipByScalarCaptureState* ctx, const TensorTup... method Apply (line 57) | Maybe Apply(const ClipByScalarCaptureState* ctx, const Tenso... FILE: oneflow/core/autograd/gradient_funcs/clip_by_scalar_max.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type ClipByScalarMaxCaptureState (line 22) | struct ClipByScalarMaxCaptureState : public AutoGradCaptureState { class ClipByScalarMax (line 27) | class ClipByScalarMax : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 36) | Maybe Capture(ClipByScalarMaxCaptureState* ctx, const Tensor... method Apply (line 54) | Maybe Apply(const ClipByScalarMaxCaptureState* ctx, const Te... FILE: oneflow/core/autograd/gradient_funcs/clip_by_scalar_min.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type ClipByScalarMinCaptureState (line 22) | struct ClipByScalarMinCaptureState : public AutoGradCaptureState { class ClipByScalarMin (line 27) | class ClipByScalarMin : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 36) | Maybe Capture(ClipByScalarMinCaptureState* ctx, const Tensor... method Apply (line 54) | Maybe Apply(const ClipByScalarMinCaptureState* ctx, const Te... FILE: oneflow/core/autograd/gradient_funcs/combined_margin_loss.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type CombinedMarginLossCaptureState (line 24) | struct CombinedMarginLossCaptureState : public AutoGradCaptureState { class CombinedMarginLoss (line 34) | class CombinedMarginLoss : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 43) | Maybe Capture(CombinedMarginLossCaptureState* ctx, const Ten... method Apply (line 60) | Maybe Apply(const CombinedMarginLossCaptureState* ctx, const... FILE: oneflow/core/autograd/gradient_funcs/complex.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type BaseComplexCaptureState (line 22) | struct BaseComplexCaptureState : public AutoGradCaptureState { class RealGrad (line 27) | class RealGrad : public OpExprGradFunction { method Init (line 29) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(BaseComplexCaptureState* ctx, const TensorTupl... method Apply (line 39) | Maybe Apply(const BaseComplexCaptureState* ctx, const Tensor... class ImagGrad (line 51) | class ImagGrad : public OpExprGradFunction { method Init (line 53) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 55) | Maybe Capture(BaseComplexCaptureState* ctx, const TensorTupl... method Apply (line 63) | Maybe Apply(const BaseComplexCaptureState* ctx, const Tensor... class ConjPhysicalGrad (line 75) | class ConjPhysicalGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 79) | Maybe Capture(BaseComplexCaptureState* ctx, const TensorTupl... method Apply (line 87) | Maybe Apply(const BaseComplexCaptureState* ctx, const Tensor... FILE: oneflow/core/autograd/gradient_funcs/concat.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type ConcatCaptureState (line 25) | struct ConcatCaptureState : public AutoGradCaptureState { class Concat (line 31) | class Concat : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/conv.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type ConvolutionNdCaptureState (line 26) | struct ConvolutionNdCaptureState : public AutoGradCaptureState { class ConvolutionNd (line 42) | class ConvolutionNd : public OpExprGradFunction { method Init (line 33) | Maybe Init(const OpExpr& op) override { method Capture (line 39) | Maybe Capture(CopyCaptureState* ctx, const TensorTuple& inpu... method Apply (line 52) | Maybe Apply(const CopyCaptureState* ctx, const TensorTuple& ... FILE: oneflow/core/autograd/gradient_funcs/ctc_loss.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type CTCLossCaptureState (line 26) | struct CTCLossCaptureState : public AutoGradCaptureState { class CTCLoss (line 33) | class CTCLoss : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/cublas_fused_mlp.cpp type oneflow (line 28) | namespace oneflow { type one (line 30) | namespace one { type CublasFusedMLPCaptureState (line 32) | struct CublasFusedMLPCaptureState : public AutoGradCaptureState { class CublasFusedMLP (line 40) | class CublasFusedMLP : public OpExprGradFunction { method Init (line 30) | Maybe Init(const OpExpr& op) override { class CumsumGrad (line 41) | class CumsumGrad : public CumGrad { method Capture (line 43) | Maybe Capture(CumCaptureState* ctx, const TensorTuple& input... method Apply (line 53) | Maybe Apply(const CumCaptureState* ctx, const TensorTuple& o... class CumProdGrad (line 70) | class CumProdGrad : public CumGrad { method Capture (line 72) | Maybe Capture(CumCaptureState* ctx, const TensorTuple& input... method Apply (line 85) | Maybe Apply(const CumCaptureState* ctx, const TensorTuple& o... FILE: oneflow/core/autograd/gradient_funcs/deconv.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type DeConvolutionNdCaptureState (line 25) | struct DeConvolutionNdCaptureState : public AutoGradCaptureState { class DeConvolutionNd (line 37) | class DeConvolutionNd : public OpExprGradFunction { method Init (line 32) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 34) | Maybe Capture(DependCaptureState* ctx, const TensorTuple& in... method Apply (line 48) | Maybe Apply(const DependCaptureState* ctx, const TensorTuple... FILE: oneflow/core/autograd/gradient_funcs/det.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type DetCaptureState (line 24) | struct DetCaptureState : public AutoGradCaptureState { class Det (line 30) | class Det : public OpExprGradFunction { method Init (line 32) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 33) | Maybe Capture(DetCaptureState* ctx, const TensorTuple& input... method Apply (line 42) | Maybe Apply(const DetCaptureState* ctx, const TensorTuple& o... FILE: oneflow/core/autograd/gradient_funcs/diag.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type DiagCaptureState (line 23) | struct DiagCaptureState : public AutoGradCaptureState { class Diag (line 28) | class Diag : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/diagonal.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type DiagonalInterpState (line 23) | struct DiagonalInterpState : public AutoGradCaptureState { class Diagonal (line 28) | class Diagonal : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/dim_gather.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type DimGatherCaptureState (line 23) | struct DimGatherCaptureState : public AutoGradCaptureState { class DimGather (line 28) | class DimGather : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/dim_scatter.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type DimScatterCaptureState (line 24) | struct DimScatterCaptureState : public AutoGradCaptureState { type ScatterType (line 29) | enum class ScatterType { kUpdate, kAdd, kMultiply } class DimScatter (line 32) | class DimScatter : public OpExprGradFunction { class DimScatterUpdateScalar (line 103) | class DimScatterUpdateScalar : public OpExprGradFunction { method Init (line 31) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 33) | Maybe Capture(DotCaptureState* ctx, const TensorTuple& input... method Apply (line 44) | Maybe Apply(const DotCaptureState* ctx, const TensorTuple& o... FILE: oneflow/core/autograd/gradient_funcs/dropout.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type DropoutCaptureState (line 22) | struct DropoutCaptureState : public AutoGradCaptureState { class Dropout (line 28) | class Dropout : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/eager_ccl_broadcast.cpp type oneflow (line 22) | namespace oneflow { type one (line 24) | namespace one { function EagerCclReduce (line 28) | Maybe EagerCclReduce(Symbol parallel_... function FindOrCreatEagerCclReduceOpExpr (line 37) | Maybe FindOrCreatEagerCclReduceOpExpr(Symbol Init(const OpExpr& op) override { method Capture (line 65) | Maybe Capture(EagerCclBroadcastCaptureState* ctx, const Tens... method Apply (line 73) | Maybe Apply(const EagerCclBroadcastCaptureState* ctx, const ... FILE: oneflow/core/autograd/gradient_funcs/elementwise_minimum_maximum.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type ElementwiseXimumCaptureState (line 25) | struct ElementwiseXimumCaptureState : public AutoGradCaptureState { class ElementwiseXimumOp (line 30) | class ElementwiseXimumOp : public OpExprGradFunction Capture(ElementwiseXimumCaptureState* ctx, const Tenso... method Apply (line 41) | Maybe Apply(const ElementwiseXimumCaptureState* ctx, const T... class ElementwiseMinimum (line 63) | class ElementwiseMinimum : public ElementwiseXimumOp { method Init (line 65) | Maybe Init(const OpExpr& op) override { class ElementwiseMaximum (line 71) | class ElementwiseMaximum : public ElementwiseXimumOp { method Init (line 73) | Maybe Init(const OpExpr& op) override { FILE: oneflow/core/autograd/gradient_funcs/embedding.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type EmbeddingCaptureState (line 26) | struct EmbeddingCaptureState : public AutoGradCaptureState { class Embedding (line 32) | class Embedding : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/expand.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type ExpandCaptureState (line 23) | struct ExpandCaptureState : public AutoGradCaptureState { class Expand (line 30) | class Expand : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/fake_quantization.cpp type oneflow (line 18) | namespace oneflow { type one (line 19) | namespace one { type FakeQuantizationCaptureState (line 21) | struct FakeQuantizationCaptureState : public AutoGradCaptureState { class FakeQuantization (line 25) | class FakeQuantization : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 29) | Maybe Capture(FakeQuantizationCaptureState* ctx, const Tenso... method Apply (line 36) | Maybe Apply(const FakeQuantizationCaptureState* ctx, const T... FILE: oneflow/core/autograd/gradient_funcs/fft.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { type FftR2CCaptureState (line 27) | struct FftR2CCaptureState : public AutoGradCaptureState { class FftR2C (line 35) | class FftR2C : public OpExprGradFunction { method Init (line 37) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 39) | Maybe Capture(FftR2CCaptureState* ctx, const TensorTuple& in... method Apply (line 53) | Maybe Apply(const FftR2CCaptureState* ctx, const TensorTuple... type FftC2CCaptureState (line 96) | struct FftC2CCaptureState : public AutoGradCaptureState { class FftC2C (line 103) | class FftC2C : public OpExprGradFunction { method Init (line 105) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 107) | Maybe Capture(FftC2CCaptureState* ctx, const TensorTuple& in... method Apply (line 121) | Maybe Apply(const FftC2CCaptureState* ctx, const TensorTuple... type FftC2RCaptureState (line 134) | struct FftC2RCaptureState : public AutoGradCaptureState { class FftC2R (line 142) | class FftC2R : public OpExprGradFunction { method Init (line 144) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 146) | Maybe Capture(FftC2RCaptureState* ctx, const TensorTuple& in... method Apply (line 160) | Maybe Apply(const FftC2RCaptureState* ctx, const TensorTuple... FILE: oneflow/core/autograd/gradient_funcs/fill.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type FillCaptureState (line 25) | struct FillCaptureState : public AutoGradCaptureState { class Fill (line 30) | class Fill : public OpExprGradFunction { class FillTensor (line 63) | class FillTensor : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/flatten.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type FlattenCaptureState (line 26) | struct FlattenCaptureState : public AutoGradCaptureState { class Flatten (line 30) | class Flatten : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/flip.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FlipCaptureState (line 23) | struct FlipCaptureState : public AutoGradCaptureState { class Flip (line 28) | class Flip : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/fold.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FoldInterpState (line 23) | struct FoldInterpState : public AutoGradCaptureState { class Fold (line 32) | class Fold : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/fused_bias_add_dropout.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FusedBiasAddDropoutInterpState (line 23) | struct FusedBiasAddDropoutInterpState : public AutoGradCaptureState { class FusedBiasAddDropout (line 30) | class FusedBiasAddDropout : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(FusedBiasAddGeluInterpState* ctx, const Tensor... method Apply (line 51) | Maybe Apply(const FusedBiasAddGeluInterpState* ctx, const Te... FILE: oneflow/core/autograd/gradient_funcs/fused_bias_add_scale_mask_softmax_dropout.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type FusedBiasAddScaleMaskSoftmaxDropoutCaptureState (line 22) | struct FusedBiasAddScaleMaskSoftmaxDropoutCaptureState : public Auto... class FusedBiasAddScaleMaskSoftmaxDropoutGradFunction (line 34) | class FusedBiasAddScaleMaskSoftmaxDropoutGradFunction method Init (line 37) | Maybe Init(const OpExpr& op) override { method Capture (line 44) | Maybe Capture(FusedBiasAddScaleMaskSoftmaxDropoutCaptureStat... method Apply (line 74) | Maybe Apply(const FusedBiasAddScaleMaskSoftmaxDropoutCapture... FILE: oneflow/core/autograd/gradient_funcs/fused_center.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type FusedCenterCaptureState (line 23) | struct FusedCenterCaptureState : public AutoGradCaptureState { class FusedCenterGrad (line 27) | class FusedCenterGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(FusedCenterCaptureState* ctx, const TensorTupl... method Apply (line 43) | Maybe Apply(const FusedCenterCaptureState* ctx, const Tensor... FILE: oneflow/core/autograd/gradient_funcs/fused_cross_interaction.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { type FusedCrossFeatureInteractionInterpState (line 27) | struct FusedCrossFeatureInteractionInterpState : public AutoGradCapt... class FusedCrossFeatureInteraction (line 40) | class FusedCrossFeatureInteraction method Init (line 43) | Maybe Init(const OpExpr& op) override { method Capture (line 50) | Maybe Capture(FusedCrossFeatureInteractionInterpState* ctx, ... method Apply (line 70) | Maybe Apply(const FusedCrossFeatureInteractionInterpState* ctx, FILE: oneflow/core/autograd/gradient_funcs/fused_dot_feature_interaction.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type FusedDotFeatureInteractionCaptureState (line 26) | struct FusedDotFeatureInteractionCaptureState : public AutoGradCaptu... class FusedDotFeatureInteraction (line 37) | class FusedDotFeatureInteraction FILE: oneflow/core/autograd/gradient_funcs/fused_fast_gelu_mul.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type FusedFastGeluMulGradCaptureState (line 22) | struct FusedFastGeluMulGradCaptureState : public AutoGradCaptureState { class FusedFastGeluMulGrad (line 26) | class FusedFastGeluMulGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(FusedFastGeluMulGradCaptureState* ctx, const T... method Apply (line 42) | Maybe Apply(const FusedFastGeluMulGradCaptureState* ctx, con... FILE: oneflow/core/autograd/gradient_funcs/fused_get_boundding_boxes_coord.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FusedGetBounddingBoxesCoordGradCaptureState (line 24) | struct FusedGetBounddingBoxesCoordGradCaptureState : public AutoGrad... class FusedGetBounddingBoxesCoordGrad (line 28) | class FusedGetBounddingBoxesCoordGrad method Init (line 31) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 33) | Maybe Capture(FusedGetBounddingBoxesCoordGradCaptureState* c... method Apply (line 44) | Maybe Apply(const FusedGetBounddingBoxesCoordGradCaptureStat... FILE: oneflow/core/autograd/gradient_funcs/fused_get_ciou_diagonal_angle.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FusedCiouAngleCaptureState (line 24) | struct FusedCiouAngleCaptureState : public AutoGradCaptureState { class FusedGetCiouDiagonalAngleGrad (line 29) | class FusedGetCiouDiagonalAngleGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 33) | Maybe Capture(FusedCiouAngleCaptureState* ctx, const TensorT... method Apply (line 49) | Maybe Apply(const FusedCiouAngleCaptureState* ctx, const Ten... FILE: oneflow/core/autograd/gradient_funcs/fused_get_ciou_result.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FusedGetCiouResultGradCaptureState (line 23) | struct FusedGetCiouResultGradCaptureState : public AutoGradCaptureSt... class FusedGetCiouResultGrad (line 30) | class FusedGetCiouResultGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 34) | Maybe Capture(FusedGetCiouResultGradCaptureState* ctx, const... method Apply (line 51) | Maybe Apply(const FusedGetCiouResultGradCaptureState* ctx, c... FILE: oneflow/core/autograd/gradient_funcs/fused_get_convex_diagonal_squared.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type FusedGetConvexDiagonalSquaredCaptureState (line 23) | struct FusedGetConvexDiagonalSquaredCaptureState : public AutoGradCa... class FusedGetConvexDiagonalSquaredGrad (line 28) | class FusedGetConvexDiagonalSquaredGrad method Init (line 31) | Maybe Init(const OpExpr& op) override { method Capture (line 38) | Maybe Capture(FusedGetConvexDiagonalSquaredCaptureState* ctx... method Apply (line 52) | Maybe Apply(const FusedGetConvexDiagonalSquaredCaptureState*... FILE: oneflow/core/autograd/gradient_funcs/fused_get_intersection_area.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type FusedGetIntersectionAreaCaptureState (line 23) | struct FusedGetIntersectionAreaCaptureState : public AutoGradCapture... class FusedGetIntersectionAreaGrad (line 27) | class FusedGetIntersectionAreaGrad method Init (line 30) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 32) | Maybe Capture(FusedGetIntersectionAreaCaptureState* ctx, con... method Apply (line 44) | Maybe Apply(const FusedGetIntersectionAreaCaptureState* ctx,... FILE: oneflow/core/autograd/gradient_funcs/fused_get_iou.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type FusedGetIouGradCaptureState (line 24) | struct FusedGetIouGradCaptureState : public AutoGradCaptureState { class FusedGetIouGrad (line 29) | class FusedGetIouGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 38) | Maybe Capture(FusedGetIouGradCaptureState* ctx, const Tensor... method Apply (line 56) | Maybe Apply(const FusedGetIouGradCaptureState* ctx, const Te... FILE: oneflow/core/autograd/gradient_funcs/fused_glu.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FusedGluGradCaptureState (line 23) | struct FusedGluGradCaptureState : public AutoGradCaptureState { class FusedGluGrad (line 33) | class FusedGluGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(FusedGruCellGradCaptureState* ctx, const Tenso... method Apply (line 48) | Maybe Apply(const FusedGruCellGradCaptureState* ctx, const T... FILE: oneflow/core/autograd/gradient_funcs/fused_lstm_cell.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type FusedLstmCellGradCaptureState (line 24) | struct FusedLstmCellGradCaptureState : public AutoGradCaptureState { class FusedLstmCellGrad (line 29) | class FusedLstmCellGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(FusedLstmCellGradCaptureState* ctx, const Tens... method Apply (line 50) | Maybe Apply(const FusedLstmCellGradCaptureState* ctx, const ... FILE: oneflow/core/autograd/gradient_funcs/fused_matmul_bias.cpp type oneflow (line 20) | namespace oneflow { type one (line 22) | namespace one { type FusedMatmulBiasCaptureState (line 24) | struct FusedMatmulBiasCaptureState : public AutoGradCaptureState { class FusedMatmulBias (line 30) | class FusedMatmulBias : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/fused_scale_tril_softmax_mask_scale.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type FusedScaleTrilSoftmaxMaskScaleInterpState (line 25) | struct FusedScaleTrilSoftmaxMaskScaleInterpState : public AutoGradCa... class FusedScaleTrilSoftmaxMaskScale (line 32) | class FusedScaleTrilSoftmaxMaskScale FILE: oneflow/core/autograd/gradient_funcs/fused_self_attention.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type FusedSelfAttentionInterpState (line 23) | struct FusedSelfAttentionInterpState : public AutoGradCaptureState { class FusedSelfAttention (line 28) | class FusedSelfAttention : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(FusedSelfAttentionInterpState* ctx, const Tens... method Apply (line 48) | Maybe Apply(const FusedSelfAttentionInterpState* ctx, const ... FILE: oneflow/core/autograd/gradient_funcs/fused_weighted_sum.cpp type oneflow (line 18) | namespace oneflow { type one (line 19) | namespace one { type FusedWeightedSumCaptureState (line 21) | struct FusedWeightedSumCaptureState : public AutoGradCaptureState { class FusedWeightedSum (line 27) | class FusedWeightedSum : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(FusedWeightedSumCaptureState* ctx, const Tenso... method Apply (line 41) | Maybe Apply(const FusedWeightedSumCaptureState* ctx, const T... FILE: oneflow/core/autograd/gradient_funcs/gather.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type GatherCaptureState (line 24) | struct GatherCaptureState : public AutoGradCaptureState { class Gather (line 29) | class Gather : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/gather_nd.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type GatherNdCaptureState (line 22) | struct GatherNdCaptureState : public AutoGradCaptureState { class GatherNd (line 26) | class GatherNd : public OpExprGradFunction { method Init (line 28) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(GatherNdCaptureState* ctx, const TensorTuple& ... method Apply (line 42) | Maybe Apply(const GatherNdCaptureState* ctx, const TensorTup... FILE: oneflow/core/autograd/gradient_funcs/global_cast.cpp type oneflow (line 25) | namespace oneflow { type one (line 26) | namespace one { type CastGlobalCaptureState (line 28) | struct CastGlobalCaptureState : public AutoGradCaptureState { class LocalToGlobal (line 35) | class LocalToGlobal : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 45) | Maybe Capture(CastGlobalCaptureState* ctx, const TensorTuple... method Apply (line 53) | Maybe Apply(const CastGlobalCaptureState* ctx, const TensorT... class GlobalToLocal (line 77) | class GlobalToLocal : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 87) | Maybe Capture(CastGlobalCaptureState* ctx, const TensorTuple... method Apply (line 100) | Maybe Apply(const CastGlobalCaptureState* ctx, const TensorT... FILE: oneflow/core/autograd/gradient_funcs/global_to_global.cpp type oneflow (line 25) | namespace oneflow { type one (line 26) | namespace one { type GlobalToGlobalState (line 28) | struct GlobalToGlobalState : public AutoGradCaptureState { class GlobalToGlobalGradFunction (line 33) | class GlobalToGlobalGradFunction : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 42) | Maybe Capture(GlobalToGlobalState* ctx, const TensorTuple& i... method Apply (line 51) | Maybe Apply(const GlobalToGlobalState* ctx, const TensorTupl... FILE: oneflow/core/autograd/gradient_funcs/gradient_accumulation.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type GradAccRepeatCaptureState (line 23) | struct GradAccRepeatCaptureState : public AutoGradCaptureState { class GradAccRepeat (line 27) | class GradAccRepeat : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(GraphFeedAndFetchCaptureState* ctx, const Tens... method Apply (line 37) | Maybe Apply(const GraphFeedAndFetchCaptureState* ctx, const ... FILE: oneflow/core/autograd/gradient_funcs/grid_sample.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type GridSampleInterpState (line 24) | struct GridSampleInterpState : public AutoGradCaptureState { class GridSample (line 35) | class GridSample : public OpExprGradFunction { method Init (line 37) | Maybe Init(const OpExpr& op) override { method Capture (line 44) | Maybe Capture(GridSampleInterpState* ctx, const TensorTuple&... method Apply (line 62) | Maybe Apply(const GridSampleInterpState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/group_norm.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type GroupNormCaptureState (line 23) | struct GroupNormCaptureState : public AutoGradCaptureState { class GroupNorm (line 38) | class GroupNorm : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/identity.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type IdentityCaptureState (line 22) | struct IdentityCaptureState : public AutoGradCaptureState { class Identity (line 26) | class Identity : public OpExprGradFunction { method Init (line 28) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(IdentityCaptureState* ctx, const TensorTuple& ... method Apply (line 37) | Maybe Apply(const IdentityCaptureState* ctx, const TensorTup... FILE: oneflow/core/autograd/gradient_funcs/inv.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type InvCaptureState (line 23) | struct InvCaptureState : public AutoGradCaptureState { class Inv (line 27) | class Inv : public OpExprGradFunction { method Init (line 29) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(InvCaptureState* ctx, const TensorTuple& input... method Apply (line 36) | Maybe Apply(const InvCaptureState* ctx, const TensorTuple& o... FILE: oneflow/core/autograd/gradient_funcs/kl_div.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type KLDivLossCaptureState (line 22) | struct KLDivLossCaptureState : public AutoGradCaptureState { class KLDivLoss (line 28) | class KLDivLoss : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/l2_normalize.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type L2NormalizeCaptureState (line 25) | struct L2NormalizeCaptureState : public AutoGradCaptureState { class L2Normalize (line 31) | class L2Normalize : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/lerp.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type LerpCaptureState (line 24) | struct LerpCaptureState : public AutoGradCaptureState { type ScalarLerpCaptureState (line 27) | struct ScalarLerpCaptureState : public AutoGradCaptureState { class LerpGrad (line 32) | class LerpGrad : public OpExprGradFunction { method Init (line 34) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 36) | Maybe Capture(LerpCaptureState* ctx, const TensorTuple& inpu... method Apply (line 49) | Maybe Apply(const LerpCaptureState* ctx, const TensorTuple& ... class ScalarLerpGrad (line 69) | class ScalarLerpGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 78) | Maybe Capture(ScalarLerpCaptureState* ctx, const TensorTuple... method Apply (line 99) | Maybe Apply(const ScalarLerpCaptureState* ctx, const TensorT... FILE: oneflow/core/autograd/gradient_funcs/linalg_cross.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type LinalgCrossCaptureState (line 23) | struct LinalgCrossCaptureState : public AutoGradCaptureState { class LinalgCross (line 29) | class LinalgCross : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/masked_fill.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type MaskedFillCaptureState (line 24) | struct MaskedFillCaptureState : public AutoGradCaptureState { class MaskedFill (line 28) | class MaskedFill : public OpExprGradFunction { method Init (line 30) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(MaskedFillCaptureState* ctx, const TensorTuple... method Apply (line 40) | Maybe Apply(const MaskedFillCaptureState* ctx, const TensorT... FILE: oneflow/core/autograd/gradient_funcs/math_binary_op.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type BinaryMathCaptureState (line 25) | struct BinaryMathCaptureState : public AutoGradCaptureState { class BinaryMathOp (line 35) | class BinaryMathOp : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 38) | Maybe Capture(BinaryMathCaptureState* ctx, const TensorTuple... method Apply (line 47) | Maybe Apply(const BinaryMathCaptureState* ctx, const TensorT... FILE: oneflow/core/autograd/gradient_funcs/math_unary_op.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type UnaryMathCaptureState (line 25) | struct UnaryMathCaptureState : public AutoGradCaptureState { class UnaryMathBwdWithDyXOp (line 33) | class UnaryMathBwdWithDyXOp : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 36) | Maybe Capture(UnaryMathCaptureState* ctx, const TensorTuple&... method Apply (line 43) | Maybe Apply(const UnaryMathCaptureState* ctx, const TensorTu... class UnaryMathBwdWithDyYOp (line 56) | class UnaryMathBwdWithDyYOp : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 59) | Maybe Capture(UnaryMathCaptureState* ctx, const TensorTuple&... method Apply (line 66) | Maybe Apply(const UnaryMathCaptureState* ctx, const TensorTu... class UnaryMathBwdWithFillZeroOp (line 78) | class UnaryMathBwdWithFillZeroOp : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 81) | Maybe Capture(UnaryMathCaptureState* ctx, const TensorTuple&... method Apply (line 87) | Maybe Apply(const UnaryMathCaptureState* ctx, const TensorTu... class NegativeOp (line 127) | class NegativeOp : public OpExprGradFunction { method Init (line 128) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 130) | Maybe Capture(UnaryMathCaptureState* ctx, const TensorTuple&... method Apply (line 136) | Maybe Apply(const UnaryMathCaptureState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/matmul.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type MatmulCaptureState (line 26) | struct MatmulCaptureState : public AutoGradCaptureState { class Matmul (line 36) | class Matmul : public OpExprGradFunction { type BroadcastMatmulCaptureState (line 106) | struct BroadcastMatmulCaptureState : public AutoGradCaptureState { class BroadcastMatmul (line 119) | class BroadcastMatmul : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/max_unpool.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type MaxUnpoolCaptureState (line 24) | struct MaxUnpoolCaptureState : public AutoGradCaptureState { class MaxUnpoolNdGrad (line 33) | class MaxUnpoolNdGrad : public OpExprGradFunction { method Init (line 33) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 34) | Maybe Capture(MedianCaptureState* ctx, const TensorTuple& in... method Apply (line 43) | Maybe Apply(const MedianCaptureState* ctx, const TensorTuple... type MedianWithIndicesCaptureState (line 72) | struct MedianWithIndicesCaptureState : public AutoGradCaptureState { class MedianWithIndices (line 76) | class MedianWithIndices : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 79) | Maybe Capture(MedianWithIndicesCaptureState* ctx, const Tens... method Apply (line 88) | Maybe Apply(const MedianWithIndicesCaptureState* ctx, const ... FILE: oneflow/core/autograd/gradient_funcs/mode.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { type ModeCaptureState (line 27) | struct ModeCaptureState : public AutoGradCaptureState { class Mode (line 31) | class Mode : public OpExprGradFunction { method Init (line 33) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 34) | Maybe Capture(ModeCaptureState* ctx, const TensorTuple& inpu... method Apply (line 43) | Maybe Apply(const ModeCaptureState* ctx, const TensorTuple& ... FILE: oneflow/core/autograd/gradient_funcs/narrow.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type NarrowCaptureState (line 26) | struct NarrowCaptureState : public AutoGradCaptureState { class Narrow (line 34) | class Narrow : public OpExprGradFunction { method Init (line 36) | Maybe Init(const OpExpr& op) override { method Capture (line 43) | Maybe Capture(NarrowCaptureState* ctx, const TensorTuple& in... method Apply (line 62) | Maybe Apply(const NarrowCaptureState* ctx, const TensorTuple... FILE: oneflow/core/autograd/gradient_funcs/nll.cpp type oneflow (line 20) | namespace oneflow { type one (line 22) | namespace one { type NLLCaptureState (line 24) | struct NLLCaptureState : public AutoGradCaptureState { class NLLGradFunction (line 29) | class NLLGradFunction : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/noncontiguous_binary_op.cpp type oneflow (line 26) | namespace oneflow { type one (line 27) | namespace one { type NonContiguousBinaryOpCaptureState (line 29) | struct NonContiguousBinaryOpCaptureState : public AutoGradCaptureSta... class NonContiguousBinaryOp (line 36) | class NonContiguousBinaryOp : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 51) | Maybe Capture(NormalizationGradCaptureState* ctx, const Tens... method Apply (line 90) | Maybe Apply(const NormalizationGradCaptureState* ctx, const ... FILE: oneflow/core/autograd/gradient_funcs/normalization_add_relu.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type NormalizationAddReluGradCaptureState (line 24) | struct NormalizationAddReluGradCaptureState : public AutoGradCapture... class NormalizationAddReluGrad (line 45) | class NormalizationAddReluGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 54) | Maybe Capture(NormalizationAddReluGradCaptureState* ctx, con... method Apply (line 127) | Maybe Apply(const NormalizationAddReluGradCaptureState* ctx,... FILE: oneflow/core/autograd/gradient_funcs/one_embedding_fused_lookup.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type OneEmbeddingFusedLookupCaptureState (line 23) | struct OneEmbeddingFusedLookupCaptureState : public AutoGradCaptureS... class OneEmbeddingFusedLookup (line 33) | class OneEmbeddingFusedLookup : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 37) | Maybe Capture(OneEmbeddingFusedLookupCaptureState* ctx, cons... method Apply (line 50) | Maybe Apply(const OneEmbeddingFusedLookupCaptureState* ctx, ... FILE: oneflow/core/autograd/gradient_funcs/padding.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type PadNdCaptureState (line 23) | struct PadNdCaptureState : public AutoGradCaptureState { class PadNd (line 28) | class PadNd : public OpExprGradFunction { method Init (line 30) | Maybe Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(PadNdCaptureState* ctx, const TensorTuple& inp... class ReflectionPadNd (line 53) | class ReflectionPadNd : public PadNd { method Apply (line 55) | Maybe Apply(const PadNdCaptureState* ctx, const TensorTuple&... class ReplicationPadNd (line 67) | class ReplicationPadNd : public PadNd { method Apply (line 69) | Maybe Apply(const PadNdCaptureState* ctx, const TensorTuple&... type ConstantPadNdCaptureState (line 81) | struct ConstantPadNdCaptureState : public AutoGradCaptureState { class ConstantPadNd (line 86) | class ConstantPadNd : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 95) | Maybe Capture(ConstantPadNdCaptureState* ctx, const TensorTu... method Apply (line 108) | Maybe Apply(const ConstantPadNdCaptureState* ctx, const Tens... FILE: oneflow/core/autograd/gradient_funcs/partial_fc_sample.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type PartialFCSampleState (line 23) | struct PartialFCSampleState : public AutoGradCaptureState { class PartialFCSample (line 29) | class PartialFCSample : public OpExprGradFunction { type ReduceProdOpInterpState (line 69) | struct ReduceProdOpInterpState : public AutoGradCaptureState { class ReduceProdOp (line 74) | class ReduceProdOp : public OpExprGradFunction { method Init (line 33) | Maybe Init(const OpExpr& op) override { method Capture (line 39) | Maybe Capture(ReshapeCaptureState* ctx, const TensorTuple& i... method Apply (line 45) | Maybe Apply(const ReshapeCaptureState* ctx, const TensorTupl... class ReshapeLikeGrad (line 54) | class ReshapeLikeGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 62) | Maybe Capture(ReshapeCaptureState* ctx, const TensorTuple& i... method Apply (line 71) | Maybe Apply(const ReshapeCaptureState* ctx, const TensorTupl... FILE: oneflow/core/autograd/gradient_funcs/rms_norm.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type RMSNormCaptureState (line 22) | struct RMSNormCaptureState : public AutoGradCaptureState { class RMSNormGrad (line 30) | class RMSNormGrad : public OpExprGradFunction { method Init (line 32) | Maybe Init(const OpExpr& op) override { return Maybe::... FILE: oneflow/core/autograd/gradient_funcs/roi_align.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type RoiAlignCaptureState (line 23) | struct RoiAlignCaptureState : public AutoGradCaptureState { class RoiAlign (line 32) | class RoiAlign : public OpExprGradFunction { method Init (line 34) | Maybe Init(const OpExpr& op) override { method Capture (line 41) | Maybe Capture(RoiAlignCaptureState* ctx, const TensorTuple& ... method Apply (line 58) | Maybe Apply(const RoiAlignCaptureState* ctx, const TensorTup... FILE: oneflow/core/autograd/gradient_funcs/roll.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type RollCaptureState (line 23) | struct RollCaptureState : public AutoGradCaptureState { class Roll (line 29) | class Roll : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/rrelu.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type RReluCaptureState (line 22) | struct RReluCaptureState : public AutoGradCaptureState { class RRelu (line 31) | class RRelu : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/scalar_add.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type ScalarAddCaptureState (line 22) | struct ScalarAddCaptureState : public AutoGradCaptureState { class ScalarAdd (line 26) | class ScalarAdd : public OpExprGradFunction { method Init (line 28) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(ScalarAddCaptureState* ctx, const TensorTuple&... method Apply (line 37) | Maybe Apply(const ScalarAddCaptureState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/scalar_div.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type ScalarDivCaptureState (line 24) | struct ScalarDivCaptureState : public AutoGradCaptureState { class ScalarDiv (line 29) | class ScalarDiv : public OpExprGradFunction { method Init (line 31) | Maybe Init(const OpExpr& op) override { method Capture (line 38) | Maybe Capture(ScalarDivCaptureState* ctx, const TensorTuple&... method Apply (line 53) | Maybe Apply(const ScalarDivCaptureState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/scalar_floordiv.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type ScalarFloorDivCaptureState (line 24) | struct ScalarFloorDivCaptureState : public AutoGradCaptureState { class ScalarFloorDiv (line 28) | class ScalarFloorDiv : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 32) | Maybe Capture(ScalarFloorDivCaptureState* ctx, const TensorT... method Apply (line 39) | Maybe Apply(const ScalarFloorDivCaptureState* ctx, const Ten... FILE: oneflow/core/autograd/gradient_funcs/scalar_fmod.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type ScalarFModGradCaptureState (line 23) | struct ScalarFModGradCaptureState : public AutoGradCaptureState { class ScalarFModGrad (line 27) | class ScalarFModGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(ScalarFModGradCaptureState* ctx, const TensorT... method Apply (line 38) | Maybe Apply(const ScalarFModGradCaptureState* ctx, const Ten... FILE: oneflow/core/autograd/gradient_funcs/scalar_mul.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type ScalarMulCaptureState (line 23) | struct ScalarMulCaptureState : public AutoGradCaptureState { class ScalarMul (line 28) | class ScalarMul : public OpExprGradFunction { method Init (line 30) | Maybe Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(ScalarMulCaptureState* ctx, const TensorTuple&... method Apply (line 52) | Maybe Apply(const ScalarMulCaptureState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/scalar_pow.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type ScalarPowCaptureState (line 23) | struct ScalarPowCaptureState : public AutoGradCaptureState { class ScalarPow (line 28) | class ScalarPow : public OpExprGradFunction { method Init (line 30) | Maybe Init(const OpExpr& op) override { method Capture (line 37) | Maybe Capture(ScalarPowCaptureState* ctx, const TensorTuple&... method Apply (line 55) | Maybe Apply(const ScalarPowCaptureState* ctx, const TensorTu... class ScalarReversePow (line 71) | class ScalarReversePow : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 80) | Maybe Capture(ScalarPowCaptureState* ctx, const TensorTuple&... method Apply (line 98) | Maybe Apply(const ScalarPowCaptureState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/scalar_truncdiv.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type ScalarTruncDivCaptureState (line 24) | struct ScalarTruncDivCaptureState : public AutoGradCaptureState { class ScalarTruncDiv (line 28) | class ScalarTruncDiv : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 32) | Maybe Capture(ScalarTruncDivCaptureState* ctx, const TensorT... method Apply (line 39) | Maybe Apply(const ScalarTruncDivCaptureState* ctx, const Ten... FILE: oneflow/core/autograd/gradient_funcs/scaled_dot_product_attention.cpp type oneflow (line 25) | namespace oneflow { type one (line 27) | namespace one { type ScaledDotProductFlashAttentionCaptureState (line 29) | struct ScaledDotProductFlashAttentionCaptureState : public AutoGradC... class ScaledDotProductFlashAttention (line 44) | class ScaledDotProductFlashAttention method Init (line 47) | Maybe Init(const OpExpr& op) override { method Capture (line 54) | Maybe Capture(ScaledDotProductFlashAttentionCaptureState* ct... method Apply (line 73) | Maybe Apply(const ScaledDotProductFlashAttentionCaptureState... FILE: oneflow/core/autograd/gradient_funcs/scatter_nd.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type ScatterNdCaptureState (line 22) | struct ScatterNdCaptureState : public AutoGradCaptureState { class ScatterNd (line 26) | class ScatterNd : public OpExprGradFunction { method Init (line 28) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 30) | Maybe Capture(ScatterNdCaptureState* ctx, const TensorTuple&... method Apply (line 41) | Maybe Apply(const ScatterNdCaptureState* ctx, const TensorTu... FILE: oneflow/core/autograd/gradient_funcs/select_top_n.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type SelectTopNCaptureState (line 25) | struct SelectTopNCaptureState : public AutoGradCaptureState { class SelectTopN (line 31) | class SelectTopN : public OpExprGradFunction { method Init (line 33) | Maybe Init(const OpExpr& op) override { return Maybe::... method Capture (line 35) | Maybe Capture(SelectTopNCaptureState* ctx, const TensorTuple... method Apply (line 46) | Maybe Apply(const SelectTopNCaptureState* ctx, const TensorT... FILE: oneflow/core/autograd/gradient_funcs/slice.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type SliceCaptureState (line 25) | struct SliceCaptureState : public AutoGradCaptureState { class Slice (line 32) | class Slice : public OpExprGradFunction { method Init (line 34) | Maybe Init(const OpExpr& op) override { method Capture (line 41) | Maybe Capture(SliceCaptureState* ctx, const TensorTuple& inp... method Apply (line 54) | Maybe Apply(const SliceCaptureState* ctx, const TensorTuple&... type SliceUpdateCaptureState (line 66) | struct SliceUpdateCaptureState : public AutoGradCaptureState { class SliceUpdate (line 76) | class SliceUpdate : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 86) | Maybe Capture(SliceUpdateCaptureState* ctx, const TensorTupl... method Apply (line 106) | Maybe Apply(const SliceUpdateCaptureState* ctx, const Tensor... FILE: oneflow/core/autograd/gradient_funcs/smooth_l1_loss.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type SmoothL1LossCaptureState (line 24) | struct SmoothL1LossCaptureState : public AutoGradCaptureState { class SmoothL1Loss (line 30) | class SmoothL1Loss : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 39) | Maybe Capture(SmoothL1LossCaptureState* ctx, const TensorTup... method Apply (line 54) | Maybe Apply(const SmoothL1LossCaptureState* ctx, const Tenso... FILE: oneflow/core/autograd/gradient_funcs/softmax.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type SoftmaxCaptureState (line 23) | struct SoftmaxCaptureState : public AutoGradCaptureState { class Softmax (line 27) | class Softmax : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/softmax_cross_entropy.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { type SoftmaxCrossEntropyGradState (line 22) | struct SoftmaxCrossEntropyGradState : public AutoGradCaptureState { class SoftmaxCrossEntropy (line 26) | class SoftmaxCrossEntropy : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 41) | Maybe Capture(SparseCrossEntropyCaptureState* ctx, const Ten... method Apply (line 54) | Maybe Apply(const SparseCrossEntropyCaptureState* ctx, const... FILE: oneflow/core/autograd/gradient_funcs/sparse_softmax_cross_entropy.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type SparseSoftmaxCrossEntropyCaptureState (line 25) | struct SparseSoftmaxCrossEntropyCaptureState : public AutoGradCaptur... class SparseSoftmaxCrossEntropy (line 29) | class SparseSoftmaxCrossEntropy : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/squeeze.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type SqueezeCaptureState (line 25) | struct SqueezeCaptureState : public AutoGradCaptureState { class Squeeze (line 29) | class Squeeze : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/stack.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type StackCaptureState (line 25) | struct StackCaptureState : public AutoGradCaptureState { class Stack (line 31) | class Stack : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/tensor_scalar_binary.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type TensorScalarCaptureState (line 23) | struct TensorScalarCaptureState : public AutoGradCaptureState { class TensorScalarAddOrSub (line 28) | class TensorScalarAddOrSub : public OpExprGradFunction Apply(const TensorScalarCaptureState* ctx, const Tenso... class TensorScalarSub (line 67) | class TensorScalarSub : public TensorScalarAddOrSub { method Apply (line 69) | Maybe Apply(const TensorScalarCaptureState* ctx, const Tenso... class TensorScalarMul (line 88) | class TensorScalarMul : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 31) | Maybe Capture(TensorScatterNdUpdateCaptureState* ctx, const ... method Apply (line 46) | Maybe Apply(const TensorScatterNdUpdateCaptureState* ctx, co... FILE: oneflow/core/autograd/gradient_funcs/tf_pool.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type TFPoolCaptureState (line 28) | struct TFPoolCaptureState : public AutoGradCaptureState { class TFPoolNdGrad (line 42) | class TFPoolNdGrad : public OpExprGradFunction { class TFMaxPoolNdGrad (line 105) | class TFMaxPoolNdGrad final : public TFPoolNdGrad { method Init (line 107) | Maybe Init(const OpExpr& op) override { return TFPoolNdGrad:... class TFAvgPoolNdGrad (line 114) | class TFAvgPoolNdGrad final : public TFPoolNdGrad { method Init (line 116) | Maybe Init(const OpExpr& op) override { return TFPoolNdGrad:... FILE: oneflow/core/autograd/gradient_funcs/to_contiguous.cpp type oneflow (line 18) | namespace oneflow { type one (line 19) | namespace one { type ToContiguousCaptureState (line 21) | struct ToContiguousCaptureState : public AutoGradCaptureState { class ToContiguous (line 25) | class ToContiguous : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 29) | Maybe Capture(ToContiguousCaptureState* ctx, const TensorTup... method Apply (line 36) | Maybe Apply(const ToContiguousCaptureState* ctx, const Tenso... FILE: oneflow/core/autograd/gradient_funcs/transpose.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type TransposeCaptureState (line 25) | struct TransposeCaptureState : public AutoGradCaptureState { class Transpose (line 30) | class Transpose : public OpExprGradFunction { function FOR_RANGE (line 65) | FOR_RANGE(int32_t, i, 0, ctx->perm.size()) { grad_perm.at(ctx->perm.... FILE: oneflow/core/autograd/gradient_funcs/tril.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type TrilCaptureState (line 23) | struct TrilCaptureState : public AutoGradCaptureState { class Tril (line 28) | class Tril : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/triu.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type TriuCaptureState (line 23) | struct TriuCaptureState : public AutoGradCaptureState { class Triu (line 28) | class Triu : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/trunc.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type TruncCaptureState (line 23) | struct TruncCaptureState : public AutoGradCaptureState { class Trunc (line 27) | class Trunc : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/two_stage_reduce.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type ReduceMode (line 24) | enum class ReduceMode : int32_t { type ReduceDeviceCaptureState (line 29) | struct ReduceDeviceCaptureState : public AutoGradCaptureState { class ReduceDevice (line 37) | class ReduceDevice : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 46) | Maybe Capture(ReduceDeviceCaptureState* ctx, const TensorTup... method Apply (line 59) | Maybe Apply(const ReduceDeviceCaptureState* ctx, const Tenso... type ReduceGlobalCaptureState (line 84) | struct ReduceGlobalCaptureState : public AutoGradCaptureState { class ReduceGlobal (line 93) | class ReduceGlobal : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 102) | Maybe Capture(ReduceGlobalCaptureState* ctx, const TensorTup... method Apply (line 117) | Maybe Apply(const ReduceGlobalCaptureState* ctx, const Tenso... FILE: oneflow/core/autograd/gradient_funcs/unfold.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type UnfoldInterpState (line 23) | struct UnfoldInterpState : public AutoGradCaptureState { class Unfold (line 33) | class Unfold : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/unfold_tensor.cpp type oneflow (line 20) | namespace oneflow { type one (line 22) | namespace one { type UnfoldTensorCaptureState (line 24) | struct UnfoldTensorCaptureState : public AutoGradCaptureState { class UnfoldTensor (line 31) | class UnfoldTensor : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/upsample.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type UpsampleCaptureState (line 24) | struct UpsampleCaptureState : public AutoGradCaptureState { class Upsample (line 33) | class Upsample : public OpExprGradFunction { type UpsampleNearest2DCaptureState (line 82) | struct UpsampleNearest2DCaptureState : public AutoGradCaptureState { class UpsampleNearest2D (line 90) | class UpsampleNearest2D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 94) | Maybe Capture(UpsampleNearest2DCaptureState* ctx, const Tens... method Apply (line 111) | Maybe Apply(const UpsampleNearest2DCaptureState* ctx, const ... type UpsampleBilinear2DCaptureState (line 130) | struct UpsampleBilinear2DCaptureState : public AutoGradCaptureState { class UpsampleBilinear2D (line 139) | class UpsampleBilinear2D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 143) | Maybe Capture(UpsampleBilinear2DCaptureState* ctx, const Ten... method Apply (line 161) | Maybe Apply(const UpsampleBilinear2DCaptureState* ctx, const... type UpsampleLinear1DCaptureState (line 180) | struct UpsampleLinear1DCaptureState : public AutoGradCaptureState { class UpsampleLinear1D (line 188) | class UpsampleLinear1D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 192) | Maybe Capture(UpsampleLinear1DCaptureState* ctx, const Tenso... method Apply (line 209) | Maybe Apply(const UpsampleLinear1DCaptureState* ctx, const T... type UpsampleNearest1DCaptureState (line 228) | struct UpsampleNearest1DCaptureState : public AutoGradCaptureState { class UpsampleNearest1D (line 235) | class UpsampleNearest1D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 239) | Maybe Capture(UpsampleNearest1DCaptureState* ctx, const Tens... method Apply (line 255) | Maybe Apply(const UpsampleNearest1DCaptureState* ctx, const ... type UpsampleBicubic2DCaptureState (line 274) | struct UpsampleBicubic2DCaptureState : public AutoGradCaptureState { class UpsampleBicubic2D (line 283) | class UpsampleBicubic2D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 287) | Maybe Capture(UpsampleBicubic2DCaptureState* ctx, const Tens... method Apply (line 305) | Maybe Apply(const UpsampleBicubic2DCaptureState* ctx, const ... type UpsampleNearest3DCaptureState (line 323) | struct UpsampleNearest3DCaptureState : public AutoGradCaptureState { class UpsampleNearest3D (line 332) | class UpsampleNearest3D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 336) | Maybe Capture(UpsampleNearest3DCaptureState* ctx, const Tens... method Apply (line 354) | Maybe Apply(const UpsampleNearest3DCaptureState* ctx, const ... type UpsampleTrilinear3DCaptureState (line 373) | struct UpsampleTrilinear3DCaptureState : public AutoGradCaptureState { class UpsampleTrilinear3D (line 383) | class UpsampleTrilinear3D : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 387) | Maybe Capture(UpsampleTrilinear3DCaptureState* ctx, const Te... method Apply (line 406) | Maybe Apply(const UpsampleTrilinear3DCaptureState* ctx, cons... FILE: oneflow/core/autograd/gradient_funcs/variance.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type VarianceState (line 26) | struct VarianceState : public AutoGradCaptureState { method VarianceState (line 27) | VarianceState() : requires_grad(false), unbiased(true), keepdim(fa... class Variance (line 34) | class Variance : public OpExprGradFunction { FILE: oneflow/core/autograd/gradient_funcs/vector_matrix_product.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type VectorMatrixProductCaptureState (line 26) | struct VectorMatrixProductCaptureState : public AutoGradCaptureState { class VectorMatrixProduct (line 33) | class VectorMatrixProduct : public OpExprGradFunction { FILE: oneflow/core/autograd/higher_order_gradient_funcs/activation.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { type BaseActivationGradGradCaptureState (line 27) | struct BaseActivationGradGradCaptureState : public AutoGradCaptureSt... class NoParamActivationGradGrad (line 36) | class NoParamActivationGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 40) | Maybe Capture(BaseActivationGradGradCaptureState* ctx, const... method Apply (line 57) | Maybe Apply(const BaseActivationGradGradCaptureState* ctx, c... type HardShrinkGradGradCaptureState (line 88) | struct HardShrinkGradGradCaptureState : public AutoGradCaptureState { class HardShrinkGradGrad (line 94) | class HardShrinkGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 102) | Maybe Capture(HardShrinkGradGradCaptureState* ctx, const Ten... method Apply (line 118) | Maybe Apply(const HardShrinkGradGradCaptureState* ctx, const... type SoftShrinkGradGradCaptureState (line 134) | struct SoftShrinkGradGradCaptureState : public AutoGradCaptureState { class SoftShrinkGradGrad (line 140) | class SoftShrinkGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 148) | Maybe Capture(SoftShrinkGradGradCaptureState* ctx, const Ten... method Apply (line 164) | Maybe Apply(const SoftShrinkGradGradCaptureState* ctx, const... type ReluGradGradCaptureState (line 180) | struct ReluGradGradCaptureState : public AutoGradCaptureState { class ReluGradGrad (line 185) | class ReluGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 188) | Maybe Capture(ReluGradGradCaptureState* ctx, const TensorTup... method Apply (line 200) | Maybe Apply(const ReluGradGradCaptureState* ctx, const Tenso... type LeakyReluGradGradCaptureState (line 212) | struct LeakyReluGradGradCaptureState : public AutoGradCaptureState { class LeakyReluGradGrad (line 218) | class LeakyReluGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 227) | Maybe Capture(LeakyReluGradGradCaptureState* ctx, const Tens... method Apply (line 244) | Maybe Apply(const LeakyReluGradGradCaptureState* ctx, const ... type SoftplusGradGradCaptureState (line 259) | struct SoftplusGradGradCaptureState : public AutoGradCaptureState { class SoftplusGradGrad (line 266) | class SoftplusGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 275) | Maybe Capture(SoftplusGradGradCaptureState* ctx, const Tenso... method Apply (line 293) | Maybe Apply(const SoftplusGradGradCaptureState* ctx, const T... type HardTanhGradGradCaptureState (line 314) | struct HardTanhGradGradCaptureState : public AutoGradCaptureState { class HardTanhGradGrad (line 321) | class HardTanhGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 329) | Maybe Capture(HardTanhGradGradCaptureState* ctx, const Tenso... method Apply (line 346) | Maybe Apply(const HardTanhGradGradCaptureState* ctx, const T... type EluGradGradCaptureState (line 363) | struct EluGradGradCaptureState : public AutoGradCaptureState { class EluGradGrad (line 369) | class EluGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 378) | Maybe Capture(EluGradGradCaptureState* ctx, const TensorTupl... method Apply (line 395) | Maybe Apply(const EluGradGradCaptureState* ctx, const Tensor... class CeluGradGrad (line 415) | class CeluGradGrad : public EluGradGrad { method Apply (line 417) | Maybe Apply(const EluGradGradCaptureState* ctx, const Tensor... type PReluGradGradCaptureState (line 434) | struct PReluGradGradCaptureState : public AutoGradCaptureState { class PReluGradGrad (line 443) | class PReluGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 447) | Maybe Capture(PReluGradGradCaptureState* ctx, const TensorTu... method Apply (line 463) | Maybe Apply(const PReluGradGradCaptureState* ctx, const Tens... type ThresholdGradGradCaptureState (line 497) | struct ThresholdGradGradCaptureState : public AutoGradCaptureState { class ThresholdGradGrad (line 503) | class ThresholdGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 512) | Maybe Capture(ThresholdGradGradCaptureState* ctx, const Tens... method Apply (line 529) | Maybe Apply(const ThresholdGradGradCaptureState* ctx, const ... FILE: oneflow/core/autograd/higher_order_gradient_funcs/avg_pool.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type AdaptiveAvgPoolNDGradGradCaptureState (line 25) | struct AdaptiveAvgPoolNDGradGradCaptureState : public AutoGradCaptur... class AdaptiveAvgPoolNdNdGradGrad (line 33) | class AdaptiveAvgPoolNdNdGradGrad method Init (line 36) | Maybe Init(const OpExpr& op) override { method Capture (line 43) | Maybe Capture(AdaptiveAvgPoolNDGradGradCaptureState* ctx, co... method Apply (line 71) | Maybe Apply(const AdaptiveAvgPoolNDGradGradCaptureState* ctx... type AvgPoolGradGradCaptureState (line 98) | struct AvgPoolGradGradCaptureState : public AutoGradCaptureState { class AvgPoolNdGradGrad (line 111) | class AvgPoolNdGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 120) | Maybe Capture(AvgPoolGradGradCaptureState* ctx, const Tensor... method Apply (line 140) | Maybe Apply(const AvgPoolGradGradCaptureState* ctx, const Te... FILE: oneflow/core/autograd/higher_order_gradient_funcs/binary_cross_entropy_loss.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type BinaryCrossEntropyGradGradCaptureState (line 23) | struct BinaryCrossEntropyGradGradCaptureState : public AutoGradCaptu... class BinaryCrossEntropyGradGrad (line 30) | class BinaryCrossEntropyGradGrad FILE: oneflow/core/autograd/higher_order_gradient_funcs/binary_cross_entropy_with_logits.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { type BinaryCrossEntropyWithLogitsGradGradCaptureState (line 23) | struct BinaryCrossEntropyWithLogitsGradGradCaptureState : public Aut... class BinaryCrossEntropyWithLogitsGradGrad (line 31) | class BinaryCrossEntropyWithLogitsGradGrad FILE: oneflow/core/autograd/higher_order_gradient_funcs/binary_cross_entropy_with_logits_reduce_mean.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type BinaryCrossEntropyWithLogitsReduceMeanGradGradCaptureState (line 24) | struct BinaryCrossEntropyWithLogitsReduceMeanGradGradCaptureState : ... class BinaryCrossEntropyWithLogitsReduceMeanGradGrad (line 34) | class BinaryCrossEntropyWithLogitsReduceMeanGradGrad FILE: oneflow/core/autograd/higher_order_gradient_funcs/conv.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type ConvDataGradGradCaptureState (line 26) | struct ConvDataGradGradCaptureState : public AutoGradCaptureState { class ConvDataGradGrad (line 41) | class ConvDataGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 45) | Maybe Capture(DivGradGradCaptureState* ctx, const TensorTupl... method Apply (line 65) | Maybe Apply(const DivGradGradCaptureState* ctx, const Tensor... FILE: oneflow/core/autograd/higher_order_gradient_funcs/kl_div_loss.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type KLDivLossGradGradCaptureState (line 24) | struct KLDivLossGradGradCaptureState : public AutoGradCaptureState { class KLDivLossGradGrad (line 34) | class KLDivLossGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 38) | Maybe Capture(UnaryMathGradGradState* ctx, const TensorTuple... method Apply (line 48) | Maybe Apply(const UnaryMathGradGradState* ctx, const TensorT... class UnaryMathGradGradWithZeroDDX (line 62) | class UnaryMathGradGradWithZeroDDX : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 64) | Maybe Capture(UnaryMathGradGradState* ctx, const TensorTuple... method Apply (line 73) | Maybe Apply(const UnaryMathGradGradState* ctx, const TensorT... FILE: oneflow/core/autograd/higher_order_gradient_funcs/matmul.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type BroadcastMatmulGradBGradCaptureState (line 26) | struct BroadcastMatmulGradBGradCaptureState : public AutoGradCapture... class BroadcastMatmulGradBGrad (line 34) | class BroadcastMatmulGradBGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 42) | Maybe Capture(BroadcastMatmulGradBGradCaptureState* ctx, con... method Apply (line 57) | Maybe Apply(const BroadcastMatmulGradBGradCaptureState* ctx,... FILE: oneflow/core/autograd/higher_order_gradient_funcs/max_pool.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { type MaxPoolGradGradCaptureState (line 24) | struct MaxPoolGradGradCaptureState : public AutoGradCaptureState { class MaxPoolNdGradGrad (line 30) | class MaxPoolNdGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 34) | Maybe Capture(MaxPoolGradGradCaptureState* ctx, const Tensor... method Apply (line 47) | Maybe Apply(const MaxPoolGradGradCaptureState* ctx, const Te... FILE: oneflow/core/autograd/higher_order_gradient_funcs/nll_loss.cpp type oneflow (line 20) | namespace oneflow { type one (line 22) | namespace one { type NLLCaptureState (line 24) | struct NLLCaptureState : public AutoGradCaptureState { class NLLLossGradGrad (line 31) | class NLLLossGradGrad : public OpExprGradFunction { FILE: oneflow/core/autograd/higher_order_gradient_funcs/pow.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { type PowXGradGradCaptureState (line 26) | struct PowXGradGradCaptureState : public AutoGradCaptureState { class PowXGradGrad (line 36) | class PowXGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 39) | Maybe Capture(PowXGradGradCaptureState* ctx, const TensorTup... method Apply (line 56) | Maybe Apply(const PowXGradGradCaptureState* ctx, const Tenso... type PowYGradGradCaptureState (line 97) | struct PowYGradGradCaptureState : public AutoGradCaptureState { class PowYGradGrad (line 108) | class PowYGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { return Maybe::... method Capture (line 112) | Maybe Capture(PowYGradGradCaptureState* ctx, const TensorTup... method Apply (line 131) | Maybe Apply(const PowYGradGradCaptureState* ctx, const Tenso... FILE: oneflow/core/autograd/higher_order_gradient_funcs/scalar_pow.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type ScalarPowGradGradCaptureState (line 25) | struct ScalarPowGradGradCaptureState : public AutoGradCaptureState { class ScalarPowGradGrad (line 31) | class ScalarPowGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 40) | Maybe Capture(ScalarPowGradGradCaptureState* ctx, const Tens... method Apply (line 61) | Maybe Apply(const ScalarPowGradGradCaptureState* ctx, const ... class ScalarReversePowGradGrad (line 91) | class ScalarReversePowGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 100) | Maybe Capture(ScalarPowGradGradCaptureState* ctx, const Tens... method Apply (line 121) | Maybe Apply(const ScalarPowGradGradCaptureState* ctx, const ... FILE: oneflow/core/autograd/higher_order_gradient_funcs/slice.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type SliceGradGradCaptureState (line 26) | struct SliceGradGradCaptureState : public AutoGradCaptureState { class SliceGradGrad (line 32) | class SliceGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 41) | Maybe Capture(SliceGradGradCaptureState* ctx, const TensorTu... method Apply (line 52) | Maybe Apply(const SliceGradGradCaptureState* ctx, const Tens... FILE: oneflow/core/autograd/higher_order_gradient_funcs/smooth_l1_loss.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type SmoothL1LossGradGradCaptureState (line 25) | struct SmoothL1LossGradGradCaptureState : public AutoGradCaptureState { class SmoothL1LossGradGrad (line 35) | class SmoothL1LossGradGrad : public OpExprGradFunction Init(const OpExpr& op) override { method Capture (line 44) | Maybe Capture(SmoothL1LossGradGradCaptureState* ctx, const T... method Apply (line 64) | Maybe Apply(const SmoothL1LossGradGradCaptureState* ctx, con... FILE: oneflow/core/autograd/higher_order_gradient_funcs/softmax.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { type SoftmaxGradGradCaptureState (line 25) | struct SoftmaxGradGradCaptureState : public AutoGradCaptureState { class SoftmaxGradGrad (line 30) | class SoftmaxGradGrad : public OpExprGradFunction RawCheckAsymmetricBroadcast(Symbol in, Symbol... function CalBroadcastRoot (line 51) | Maybe CalBroadcastRoot(Symbol src_parallel_desc, function EagerCclBroadcast (line 81) | Maybe EagerCclBroadcast(Symbol parallel... function AsymmetricBroadcast (line 96) | Maybe AsymmetricBroadcast(const std::shared_ptr RawReplaceInDeviceType(DeviceType device_type) { function RawReplaceOutDeviceType (line 36) | Maybe RawReplaceOutDeviceType(DeviceType device_type) { function RawFlattenHierarchy (line 54) | Maybe> RawFlattenHierarchy(Symbol pla... function RawFlattenInHierarchy (line 75) | Maybe RawFlattenInHierarchy() { function RawUnflattenHierarchy (line 83) | Maybe> RawUnflattenHierarchy(Symbol i... function RawUnflattenInHierarchy (line 99) | Maybe RawUnflattenInHierarchy() { function RawUnflattenOutHierarchy (line 107) | Maybe RawUnflattenOutHierarchy() { function GetAllPartialSumNdSbp (line 126) | Maybe> GetAllPartialSumNdSbp(int64_t ndim) { function RawReplaceNdSbpWithPartialSum (line 136) | Maybe> RawReplaceNdSbpWithPartialSum(Symbol RawOutPlacementAndPartialSum() { function GetAllBroadcastNdSbp (line 160) | Maybe> GetAllBroadcastNdSbp(int64_t ndim) { function RawReplaceNdSbpWithBroadcast (line 170) | Maybe> RawReplaceNdSbpWithBroadcast(Symbol RawInPlacementAndBroadcast() { function RawOutPlacementAndBroadcast (line 187) | Maybe RawOutPlacementAndBroadcast() { function GetSplitNdSbp (line 204) | Maybe> GetSplitNdSbp(int64_t axis) { function RawInPlacementAndSplit (line 212) | Maybe RawInPlacementAndSplit(int64_t axis) { function RawOutPlacementAndSplit (line 221) | Maybe RawOutPlacementAndSplit(int64_t axis) { function GetFisrtDeviceOfPlacement (line 239) | Maybe> GetFisrtDeviceOfPlacement(Symbol RawInFirstDeviceAndAllBroadcast() { function RawOutFirstDeviceAndAllBroadcast (line 266) | Maybe RawOutFirstDeviceAndAllBroadcast() { function RawPlacementAndRepeatFirstSbp (line 285) | Maybe> RawPlacementAndRepeatFirstSbp(Symbol RawInPlacementAndRepeatFirstSbp() { FILE: oneflow/core/boxing/boxing_dividor_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/boxing/boxing_interpreter_status.cpp type oneflow (line 22) | namespace oneflow { function RawMakeBoxingInterpreterStatus (line 26) | Maybe RawMakeBoxingInterpreterStatus(const st... function RawMakeComposedBoxingInterpreterStatus (line 35) | Maybe RawMakeComposedBoxingInterpreterStatus( function RawGetNdSbpRouting (line 75) | Maybe RawGetNdSbpRouting(Symbol src_placed_n... function RawGetPlacementRouting (line 87) | Maybe RawGetPlacementRouting( function RawGetBoxingDesc (line 100) | Maybe RawGetBoxingDesc(Symbol> s... FILE: oneflow/core/boxing/boxing_interpreter_status.h function std (line 69) | const std::string& boxing_routing() const; FILE: oneflow/core/boxing/ccl_boxing_function.cpp type oneflow (line 24) | namespace oneflow { class EagerBoxingKernelRegContext (line 28) | class EagerBoxingKernelRegContext final : public user_op::KernelRegCon... method EagerBoxingKernelRegContext (line 30) | explicit EagerBoxingKernelRegContext(DeviceType device_type) : devic... method DeviceType (line 33) | DeviceType device_type() const override { return device_type_; } method ParallelContext (line 34) | const ParallelContext& parallel_ctx() const override { PRINT_BUG_PRO... function RawCheckCclKernelRegistered (line 57) | Maybe RawCheckCclKernelRegistered(const std::string& op_type_nam... function RawCheckCclP2B (line 65) | Maybe RawCheckCclP2B(Symbol in, Symbol... function RawCheckCclP2S (line 84) | Maybe RawCheckCclP2S(Symbol in, Symbol... function RawCheckCclS2B (line 105) | Maybe RawCheckCclS2B(Symbol in, Symbol... function RawCheckCclS2S (line 127) | Maybe RawCheckCclS2S(Symbol in, Symbol... function CclP2B (line 156) | Maybe CclP2B(const std::shared_ptr& tensor, ... function CclP2S (line 170) | Maybe CclP2S(const std::shared_ptr& tensor, ... function CclS2B (line 185) | Maybe CclS2B(const std::shared_ptr& tensor, ... function CclS2S (line 199) | Maybe CclS2S(const std::shared_ptr& tensor, ... FILE: oneflow/core/boxing/cuda_copy_boxing_interpreter.cpp type oneflow (line 23) | namespace oneflow { function IgnoringDeviceTypeEqual (line 27) | Maybe IgnoringDeviceTypeEqual(Symbol lhs, Symbol CheckCopyH2D(Symbol in, Symbol out, function CheckCopyD2H (line 44) | Maybe CheckCopyD2H(Symbol in, Symbol out, function CopyBoxingFunction (line 55) | Maybe CopyBoxingFunction(const std::shared_ptr CheckEagerBoxingDataType(DataType val) { function RawGetBoxingFunction (line 68) | Maybe RawGetBoxingFunction(const std::string& method_... function GetBoxingFunction (line 81) | Maybe GetBoxingFunction(const std::string& method_nam... function RegisterBoxingFunction (line 87) | void RegisterBoxingFunction(const std::string& method_name, const Boxi... function BoxingExpr (line 181) | Maybe BoxingExpr(const std::string& boxing_name) { function BoxingExpr (line 187) | Maybe BoxingExpr(const std::shared_ptr& b... function BoxingExpr (line 192) | Maybe BoxingExpr(const std::shared_ptr& b... function BoxingExpr (line 198) | Maybe BoxingExpr(const std::shared_ptr& b... function BoxingExpr (line 204) | Maybe BoxingExpr(const std::shared_ptr& b... function OptionalBoxing (line 218) | Maybe OptionalBoxing(const std::string& boxing_mame) { FILE: oneflow/core/boxing/eager_boxing_interpreter.h function namespace (line 28) | namespace oneflow { function class (line 93) | class BoxingExprIf { function explicit (line 114) | explicit AtomicBoxingExpr(const std::string& boxing_name) FILE: oneflow/core/boxing/eager_boxing_interpreter_mgr.cpp type oneflow (line 23) | namespace oneflow { function IgnoringDeviceTypeEqual (line 27) | Maybe IgnoringDeviceTypeEqual(Symbol lhs, Symbol OptionalCudaCopy(const std::shared_ptr OptionalCpuCopy(const std::shared_ptr SymmetricOneDimSxToBBoxingExpr() { function SymmetricOneDimPToSxBoxingExpr (line 53) | Maybe SymmetricOneDimPToSxBoxingExpr() { function SymmetricCyclicNDimToNDimBoxingExpr (line 58) | Maybe SymmetricCyclicNDimToNDimBoxingExpr() { function SymmetricNDimToNDimBoxingExpr (line 67) | Maybe SymmetricNDimToNDimBoxingExpr() { function SymmetricOneDimToNDimBoxingExpr (line 72) | Maybe SymmetricOneDimToNDimBoxingExpr() { function SymmetricNDimToOneDimBoxingExpr (line 77) | Maybe SymmetricNDimToOneDimBoxingExpr() { function NToOneBoxingExpr (line 83) | Maybe NToOneBoxingExpr() { function OneToNBoxingExpr (line 93) | Maybe OneToNBoxingExpr() { function SymmetricOneDimXToBBoxingExpr (line 102) | Maybe SymmetricOneDimXToBBoxingExpr() { function ASymmetricOneDimXToBBoxingExpr (line 109) | Maybe ASymmetricOneDimXToBBoxingExpr() { function GenericBoxingExpr (line 115) | Maybe GenericBoxingExpr() { function RawMainBoxingExpr (line 132) | Maybe RawMainBoxingExpr() { function GetBoxingInterpreter (line 169) | Maybe GetBoxingInterpreter(Symbol in_nd... FILE: oneflow/core/boxing/eager_boxing_interpreter_mgr.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/boxing/eager_boxing_logger.cpp type oneflow (line 22) | namespace oneflow { class NullEagerBoxingLogger (line 26) | class NullEagerBoxingLogger final : public EagerBoxingLogger { method NullEagerBoxingLogger (line 29) | NullEagerBoxingLogger() = default; method Log (line 32) | void Log(const BoxingInterpreterStatus& status, const std::string& p... class NaiveEagerBoxingLogger (line 35) | class NaiveEagerBoxingLogger final : public EagerBoxingLogger { method NaiveEagerBoxingLogger (line 38) | NaiveEagerBoxingLogger() = default; method Log (line 41) | void Log(const BoxingInterpreterStatus& status, const std::string& p... function EagerBoxingLogger (line 49) | const EagerBoxingLogger* CreateEagerBoxingLogger() { FILE: oneflow/core/boxing/eager_boxing_logger.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/boxing/flatten_hierarchy.cpp type oneflow (line 22) | namespace oneflow { function RawCheckFlattenHierarchy (line 27) | Maybe RawCheckFlattenHierarchy(Symbol in, Symbol FlattenHierarchy(const std::shared_ptr... FILE: oneflow/core/boxing/generic_symmetric_nd_sbp_boxing.cpp type oneflow (line 27) | namespace oneflow { function RawIsAllBroadcastNdSbpAfterDim (line 31) | bool RawIsAllBroadcastNdSbpAfterDim(Symbol nd_sbp, int dim) { function GetBroadcastSbp (line 41) | Maybe> GetBroadcastSbp() { function CalcLogicalShape4Axis (line 50) | Maybe CalcLogicalShape4Axis(const Shape& logical_shape, int axis, function CalcTheFirstDiffAxisBetweenTwoNdSbp (line 84) | Maybe CalcTheFirstDiffAxisBetweenTwoNdSbp(Symbol in_nd_sbp... function Apply1DBoxing (line 94) | Maybe Apply1DBoxing(const std::shared_ptr& i... function RawCheckGenericSymmetricNdSbpBoxing (line 107) | Maybe RawCheckGenericSymmetricNdSbpBoxing(Symbol in... function GenericSymmetricNdSbpBoxing (line 122) | Maybe GenericSymmetricNdSbpBoxing(const std::shared_ptr RawCheckIdentity(Symbol in, Symbol GetIdentity(const std::shared_ptr& ten... FILE: oneflow/core/boxing/naive_1_to_p_boxing.cpp type oneflow (line 23) | namespace oneflow { function NdSbpIsAllPartialSum (line 27) | bool NdSbpIsAllPartialSum(Symbol nd_sbp) { function RawCheckNaive1ToP (line 35) | Maybe RawCheckNaive1ToP(Symbol in, Symbol Naive1ToP(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_b_to_1_boxing.cpp type oneflow (line 24) | namespace oneflow { function RawCheckNaiveBTo1 (line 28) | Maybe RawCheckNaiveBTo1(Symbol in, Symbol NaiveBTo1(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_b_to_s_boxing.cpp type oneflow (line 23) | namespace oneflow { function RawIsSplitSbp (line 27) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawIsBroadcastSbp (line 31) | bool RawIsBroadcastSbp(Symbol sbp_parallel) { function RawCheckNaiveBToS (line 38) | Maybe RawCheckNaiveBToS(Symbol in, Symbol NaiveBToS(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_p_to_b_boxing.cpp type oneflow (line 23) | namespace oneflow { function RawIsPartialSumSbp (line 27) | bool RawIsPartialSumSbp(Symbol sbp_parallel) { function RawIsBroadcastSbp (line 33) | bool RawIsBroadcastSbp(Symbol sbp_parallel) { function RawCheckNaivePToB (line 40) | Maybe RawCheckNaivePToB(Symbol in, Symbol NaivePToB(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_p_to_s_boxing.cpp type oneflow (line 23) | namespace oneflow { function RawIsPartialSumSbp (line 27) | bool RawIsPartialSumSbp(Symbol sbp_parallel) { function RawIsSplitSbp (line 33) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawCheckNaivePToS (line 38) | Maybe RawCheckNaivePToS(Symbol in, Symbol NaivePToS(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_s_to_b_boxing.cpp type oneflow (line 23) | namespace oneflow { function RawIsSplitSbp (line 27) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawIsBroadcastSbp (line 31) | bool RawIsBroadcastSbp(Symbol sbp_parallel) { function RawCheckNaiveSToB (line 38) | Maybe RawCheckNaiveSToB(Symbol in, Symbol NaiveSToB(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_s_to_p_boxing.cpp type oneflow (line 23) | namespace oneflow { function RawIsSplitSbp (line 27) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawIsPartialSumSbp (line 31) | bool RawIsPartialSumSbp(Symbol sbp_parallel) { function RawCheckNaiveSToP (line 38) | Maybe RawCheckNaiveSToP(Symbol in, Symbol NaiveSToP(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/naive_s_to_s_boxing.cpp type oneflow (line 23) | namespace oneflow { function RawIsSplitSbp (line 27) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawCheckNaiveSToS (line 32) | Maybe RawCheckNaiveSToS(Symbol in, Symbol NaiveSToS(const std::shared_ptr& tenso... FILE: oneflow/core/boxing/nd_sbp_dim_reduce_boxing.cpp type oneflow (line 25) | namespace oneflow { function RawInOutPlacedNdSbpDimReduce (line 29) | Maybe, Symbol>> RawInOutPl... function RawCheckParallelDimReduce (line 48) | Maybe RawCheckParallelDimReduce(Symbol in, Symbol ParallelDimReduce(const std::shared_ptr RawCheckNaiveOneToOne(Symbol in, Symbol NaiveOneToOne(const std::shared_ptr& t... FILE: oneflow/core/boxing/slice_boxing_util.cpp type oneflow (line 23) | namespace oneflow { type private_details (line 25) | namespace private_details { function PreprocessInputTensor4SliceBoxing (line 27) | Maybe PreprocessInputTensor4SliceBoxing(const std::shar... function PostprocessOutputTensor4SliceBoxing (line 44) | Maybe PostprocessOutputTensor4SliceBoxing(const std::sh... FILE: oneflow/core/boxing/slice_boxing_util.h type class (line 25) | enum class function namespace (line 27) | namespace private_details { function EagerSliceBoxingType (line 43) | enum class EagerSliceBoxingType : unsigned int { FILE: oneflow/core/boxing/symmetric_acyclic_nd_sbp_boxing.cpp type oneflow (line 28) | namespace oneflow { function ReinterpterGlobalTensor (line 32) | Maybe ReinterpterGlobalTensor(const std::shared_ptr Apply1DBoxing(const std::shared_ptr& i... function RawCheckSymmetricAcyclicNdSbpBoxing (line 59) | Maybe RawCheckSymmetricAcyclicNdSbpBoxing(Symbol in... function SymmetricAcyclicNdSbpBoxing (line 75) | Maybe SymmetricAcyclicNdSbpBoxing(const std::shared_ptr RawCheckSymmetricBToP(Symbol in, Symbol SymmetricBToP(const std::shared_ptr& t... FILE: oneflow/core/boxing/symmetric_b_to_s_boxing.cpp type oneflow (line 26) | namespace oneflow { function RawIsBroadcastSbp (line 30) | bool RawIsBroadcastSbp(Symbol sbp_parallel) { function RawIsSplitSbp (line 36) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawCheckSymmetricB2S (line 41) | Maybe RawCheckSymmetricB2S(Symbol in, Symbol SymmetricB2S(const std::shared_ptr& te... FILE: oneflow/core/boxing/symmetric_s_to_p_boxing.cpp type oneflow (line 25) | namespace oneflow { function RawIsSplitSbp (line 29) | bool RawIsSplitSbp(Symbol sbp_parallel) { return sbp_para... function RawIsPartialSumSbp (line 33) | bool RawIsPartialSumSbp(Symbol sbp_parallel) { function EagerSymmetricSToP (line 39) | Maybe EagerSymmetricSToP(Symbol paralle... function RawCheckSymmetricSToP (line 53) | Maybe RawCheckSymmetricSToP(Symbol in, Symbol SymmetricSToP(const std::shared_ptr& t... FILE: oneflow/core/boxing/unflatten_hierarchy.cpp type oneflow (line 22) | namespace oneflow { function RawCheckUnflattenHierarchy (line 27) | Maybe RawCheckUnflattenHierarchy(Symbol in, Symbol<... function UnflattenHierarchy (line 60) | Maybe UnflattenHierarchy(const std::shared_ptr InitBroadcastRankHeap(std::vector* ranks, const... function CpuBroadcast (line 52) | Maybe CpuBroadcast(const void* in, void* out, size_t buffer_si... function CpuSend (line 83) | Maybe CpuSend(const void* in, size_t buffer_size, int64_t dst) { function CpuRecv (line 101) | Maybe CpuRecv(void* out, size_t buffer_size, int64_t src) { FILE: oneflow/core/ccl/ccl.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/comm_network/comm_network.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/core/comm_network/comm_network.h function namespace (line 27) | namespace oneflow { function virtual (line 83) | virtual ~CommNetIf() {} function UnRegisterMemory (line 92) | void UnRegisterMemory(void* token) override { function HashSet (line 101) | const HashSet& mem_descs() { return mem_descs_; } FILE: oneflow/core/comm_network/epoll/epoll_comm_network.cpp type oneflow (line 27) | namespace oneflow { function sockaddr_in (line 33) | sockaddr_in GetSockAddr(const std::string& addr, uint16_t port) { function SockListen (line 42) | int SockListen(int listen_sockfd, int32_t* listen_port, int32_t total_... function GenPortKey (line 71) | std::string GenPortKey(int64_t machine_id) { return "EpollPort/" + std... function PushPort (line 72) | void PushPort(int64_t machine_id, uint16_t port) { function ClearPort (line 75) | void ClearPort(int64_t machine_id) { function PullPort (line 78) | uint16_t PullPort(int64_t machine_id) { function SocketMemDesc (line 118) | SocketMemDesc* EpollCommNet::NewMemDesc(void* ptr, size_t byte_size) { function FOR_RANGE (line 182) | FOR_RANGE(int32_t, idx, 0, src_machine_count) { function FOR_RANGE (line 198) | FOR_RANGE(int64_t, machine_id, 0, total_machine_num) { function SocketHelper (line 203) | SocketHelper* EpollCommNet::GetSocketHelper(int64_t machine_id) { FILE: oneflow/core/comm_network/epoll/epoll_comm_network.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/io_event_poller.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/io_event_poller.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_helper.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_helper.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_memory_desc.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_message.h function namespace (line 36) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_read_helper.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_read_helper.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_write_helper.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/core/comm_network/epoll/socket_write_helper.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/comm_network/ibverbs/ibverbs_comm_network.cpp type oneflow (line 26) | namespace oneflow { function GenTokensMsgKey (line 30) | std::string GenTokensMsgKey(int64_t machine_id) { function GenConnInfoKey (line 34) | std::string GenConnInfoKey(int64_t src_machine_id, int64_t dst_machine... function IBVForkInit (line 38) | void IBVForkInit() { function ParseUserDevicePort (line 46) | void ParseUserDevicePort(std::string* device_name, int* port) { function FOR_RANGE (line 198) | FOR_RANGE(int32_t, i, 0, found_wc_num) { FILE: oneflow/core/comm_network/ibverbs/ibverbs_comm_network.h function namespace (line 29) | namespace oneflow { FILE: oneflow/core/comm_network/ibverbs/ibverbs_memory_desc.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/core/comm_network/ibverbs/ibverbs_memory_desc.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/comm_network/ibverbs/ibverbs_qp.cpp type oneflow (line 26) | namespace oneflow { type ibv_port_attr (line 35) | struct ibv_port_attr function FOR_RANGE (line 63) | FOR_RANGE(size_t, i, 0, recv_msg_buf_.size()) { recv_msg_buf_.at(i) = ... function FOR_RANGE (line 154) | FOR_RANGE(size_t, i, 0, block_num) { function ActorMsgMR (line 263) | ActorMsgMR* IBVerbsQP::GetOneSendMsgMRFromBuf() { function WorkRequestId (line 271) | WorkRequestId* IBVerbsQP::NewWorkRequestId() { FILE: oneflow/core/comm_network/ibverbs/ibverbs_qp.h function namespace (line 24) | namespace oneflow { type WorkRequestId (line 55) | struct WorkRequestId { type IBVerbsCommNetRMADesc (line 62) | struct IBVerbsCommNetRMADesc function class (line 64) | class IBVerbsQP final { FILE: oneflow/core/common/array_ref.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/auto_registration_factory.h function namespace (line 21) | namespace oneflow { function Base (line 94) | Base* NewObj(Key k, Args&&... args) { function IsClassRegistered (line 105) | bool IsClassRegistered(Key k, Args&&... args) { FILE: oneflow/core/common/balanced_splitter.cpp type oneflow (line 18) | namespace oneflow { function Range (line 29) | Range BalancedSplitter::At(int64_t idx) const { function Range (line 44) | Range BalancedSplitter::At(int64_t first_idx, int64_t last_idx) const { FILE: oneflow/core/common/balanced_splitter.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/common/balanced_splitter_test.cpp type oneflow (line 19) | namespace oneflow { function TEST (line 21) | TEST(BalancedSplitter, split_20_to_6_part) { function TEST (line 31) | TEST(BalancedSplitter, split_2_to_3_part) { function TEST (line 38) | TEST(BalancedSplitter, GetRangeIndexForVal) { FILE: oneflow/core/common/bfloat16.h function namespace (line 24) | namespace oneflow { function bfloat16 (line 201) | inline bfloat16 operator*(bfloat16 a, int32_t b) { return a * static_cas... function bfloat16 (line 206) | inline bfloat16 operator*(int32_t a, bfloat16 b) { return static_cast* channel, Range range) { function CallFromReceiverThread (line 28) | void CallFromReceiverThread(std::vector* visit, Channel* cha... FILE: oneflow/core/common/check.cpp type oneflow (line 18) | namespace oneflow { function GLOGCHECK (line 19) | void GLOGCHECK(bool value) { CHECK_OR_THROW(value); } function GLOGLOGFATAL (line 20) | void GLOGLOGFATAL(const char* error_msg) { LOG(FATAL) << error_msg; } FILE: oneflow/core/common/check.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/check_level.cpp type oneflow (line 22) | namespace oneflow { function IsEnvEnabled (line 24) | bool IsEnvEnabled(int32_t check_level) { FILE: oneflow/core/common/check_level.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/common/constant.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/container_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/common/container_util_test.cpp type oneflow (line 19) | namespace oneflow { type test (line 20) | namespace test { function TEST (line 22) | TEST(VectorAt, write_int_vector) { class A (line 36) | class A { method A (line 38) | explicit A(int a) : a(a) {} function TEST (line 43) | TEST(VectorAt, write_custom_class_vector) { FILE: oneflow/core/common/cost_util.h function std (line 33) | static const std::string& Repr() { function with_log_ (line 71) | bool with_log_{false}; FILE: oneflow/core/common/data_type.cpp type oneflow (line 19) | namespace oneflow { function IsBoolDataType (line 21) | bool IsBoolDataType(DataType data_type) { function IsIntegralDataType (line 31) | bool IsIntegralDataType(DataType data_type) { function IsFloatingDataType (line 40) | bool IsFloatingDataType(DataType data_type) { function IsHalfDataType (line 50) | bool IsHalfDataType(DataType data_type) { function IsComplexDataType (line 59) | bool IsComplexDataType(DataType data_type) { function IsTriviallyCopyableDataType (line 68) | bool IsTriviallyCopyableDataType(DataType data_type) { function IsIndexDataType (line 77) | bool IsIndexDataType(DataType data_type) { function IsSupportRequireGradDataType (line 86) | bool IsSupportRequireGradDataType(DataType data_type) { function NotSupportBoxingDataType (line 97) | bool NotSupportBoxingDataType(DataType data_type) { function GetSizeOfDataType (line 107) | size_t GetSizeOfDataType(DataType data_type) { function CheckDataType (line 147) | void CheckDataType() { FILE: oneflow/core/common/data_type.h function namespace (line 38) | namespace std { function namespace (line 60) | namespace detail { type GetDataType (line 160) | struct GetDataType type GetDataType (line 178) | struct GetDataType type GetDataType (line 180) | struct GetDataType type GetDataType (line 185) | struct GetDataType function T (line 200) | T GetZeroVal() { function T (line 206) | T GetOneVal() { function T (line 211) | T GetZeroVal() { function T (line 216) | T GetOneVal() { function T (line 284) | T* GetZeroPtr() { function T (line 290) | T* GetOnePtr() { function T (line 296) | T GetZeroVal() { function T (line 303) | T GetZeroVal() { function T (line 308) | T GetZeroVal() { function T (line 314) | T GetOneVal() { function T (line 321) | T GetOneVal() { function T (line 327) | T GetOneVal() { function T (line 333) | T GetMaxVal() { function T (line 339) | T GetMinVal() { type T (line 354) | typedef T type; type DevDType (line 359) | struct DevDType type half (line 361) | typedef half type; type DevDType (line 365) | struct DevDType type nv_bfloat16 (line 367) | typedef nv_bfloat16 type; FILE: oneflow/core/common/data_type_converter.h function namespace (line 27) | namespace oneflow { function OF_DEVICE_FUNC (line 74) | OF_DEVICE_FUNC static const T Call(U value) { function OF_DEVICE_FUNC (line 87) | OF_DEVICE_FUNC static const T Call(U value) { function OF_DEVICE_FUNC (line 100) | OF_DEVICE_FUNC static const T Call(U value) { function OF_DEVICE_FUNC (line 111) | OF_DEVICE_FUNC static const T Call(U value) { function OF_DEVICE_FUNC (line 119) | OF_DEVICE_FUNC static const T Call(U value) { return value; } function OF_DEVICE_FUNC (line 122) | OF_DEVICE_FUNC const int32_t Clamp(uint32_t value) { function OF_DEVICE_FUNC (line 126) | OF_DEVICE_FUNC const uint32_t Clamp(int32_t value) { return value < 0 ? ... function OF_DEVICE_FUNC (line 128) | OF_DEVICE_FUNC const int32_t Clamp(int64_t value) { type ClampHelper (line 135) | struct ClampHelper function OF_DEVICE_FUNC (line 136) | OF_DEVICE_FUNC static const int32_t Call(uint64_t value) { type ClampHelper (line 143) | struct ClampHelper function OF_DEVICE_FUNC (line 144) | OF_DEVICE_FUNC static const uint32_t Call(int64_t value) { type ClampHelper (line 152) | struct ClampHelper function OF_DEVICE_FUNC (line 153) | OF_DEVICE_FUNC static const uint32_t Call(uint64_t value) { function OF_DEVICE_FUNC (line 161) | OF_DEVICE_FUNC static const bool Call(T value) { return static_cast... function OF_DEVICE_FUNC (line 338) | OF_DEVICE_FUNC static const Out ConvertSatNorm(In value) { return Conver... function Out (line 344) | static const Out Convert(In value) { return value; } function OF_DEVICE_FUNC (line 345) | OF_DEVICE_FUNC static const Out ConvertNorm(In value) { function OF_DEVICE_FUNC (line 348) | OF_DEVICE_FUNC static const Out ConvertSat(In value) { return Clamp... function OF_DEVICE_FUNC (line 349) | OF_DEVICE_FUNC static const Out ConvertSatNorm(In value) { function OF_DEVICE_FUNC (line 368) | static OF_DEVICE_FUNC const T Convert(T value) { return value; } function OF_DEVICE_FUNC (line 369) | static OF_DEVICE_FUNC const T ConvertSat(T value) { return value; } function OF_DEVICE_FUNC (line 370) | static OF_DEVICE_FUNC const T ConvertNorm(T value) { return value; } function OF_DEVICE_FUNC (line 371) | static OF_DEVICE_FUNC const T ConvertSatNorm(T value) { return value; } function Out (line 381) | const Out Convert(In value) { function Out (line 386) | const Out ConvertNorm(In value) { function Out (line 391) | const Out ConvertSat(In value) { function Out (line 396) | const Out ConvertSatNorm(In value) { FILE: oneflow/core/common/data_type_converter_test.cpp type oneflow (line 26) | namespace oneflow { function T (line 32) | constexpr const T& clamp(const T& v, const T& lo, const T& hi) { function TEST (line 38) | TEST(ClampTest, Clamp) { function TEST (line 121) | TEST(ConvertSat, float2int) { function TEST (line 145) | TEST(ConvertNorm, int2int) { function TEST (line 150) | TEST(ConvertNorm, float2int) { function TEST (line 164) | TEST(ConvertSatNorm, float2int) { function TEST (line 178) | TEST(ConvertNorm, int2float) { function TEST (line 185) | TEST(Clamp1, int64_2_float16) { function TEST (line 192) | TEST(Clamp2, float16_2_int64) { FILE: oneflow/core/common/data_type_converter_test_static.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/decorator.h function namespace (line 26) | namespace oneflow { function RetT (line 59) | RetT Call(Arg0 arg0) { function RetT (line 76) | RetT Call(Arg0 arg0, Arg1 arg1) { function RetT (line 147) | RetT Call() { function RetT (line 156) | RetT Call(Arg0 arg0) { function RetT (line 178) | RetT Call(Arg0 arg0, Args... args) { FILE: oneflow/core/common/decorator_test.cpp type oneflow (line 20) | namespace oneflow { type test (line 21) | namespace test { function Inc (line 23) | Maybe Inc(int x) { return x + 1; } function IncByConstRef (line 25) | Maybe IncByConstRef(const int& x) { return x + 1; } function TEST (line 27) | TEST(ThreadLocal, scalar) { function TEST (line 34) | TEST(ThreadLocal, const_ref) { type Foo (line 43) | struct Foo { method New (line 44) | static Maybe New(int x) { return std::shared_ptr(new Foo... function TEST (line 51) | TEST(ThreadLocal, _class) { FILE: oneflow/core/common/device_type.cpp type oneflow (line 21) | namespace oneflow { function GetAllAvailableDeviceTypeNames (line 23) | std::vector GetAllAvailableDeviceTypeNames() { function PrintAvailableDevices (line 34) | std::string PrintAvailableDevices() { function PrintGeneratorAvailableDevices (line 39) | std::string PrintGeneratorAvailableDevices() { FILE: oneflow/core/common/device_type.h function namespace (line 21) | namespace std { function namespace (line 32) | namespace oneflow { FILE: oneflow/core/common/dtype_signature.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/eigen_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/either_ptr.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/env_var/bootstrap.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/env_var/debug_mode.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/env_var/eager.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/common/env_var/env_var.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/env_var/remat.h function namespace (line 20) | namespace oneflow { FILE: oneflow/core/common/env_var/stream.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/env_var/vm.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/error.cpp type oneflow (line 30) | namespace oneflow { function LogError (line 36) | void LogError(const Error& error) { function Error (line 48) | Error&& Error::AddStackFrame(Symbol error_stack_frame) { function Error (line 53) | Error&& Error::GetStackTrace(int64_t depth, int64_t skip_n_firsts) { function Error (line 104) | Error Error::Ok() { return std::make_shared(); } function Error (line 106) | Error Error::ProtoParseFailedError() { function Error (line 112) | Error Error::JobSetEmptyError() { function Error (line 118) | Error Error::DeviceTagNotFoundError() { function Error (line 124) | Error Error::InvalidValueError() { function Error (line 130) | Error Error::IndexError() { function Error (line 136) | Error Error::TypeError() { function Error (line 142) | Error Error::TimeoutError() { function Error (line 148) | Error Error::JobNameExistError() { function Error (line 154) | Error Error::JobNameEmptyError() { function Error (line 160) | Error Error::JobNameNotEqualError() { function Error (line 166) | Error Error::NoJobBuildAndInferCtxError() { function Error (line 172) | Error Error::JobConfFrozenError() { function Error (line 178) | Error Error::JobConfNotSetError() { function Error (line 184) | Error Error::JobConfRepeatedSetError() { function Error (line 190) | Error Error::JobTypeNotSetError() { function Error (line 196) | Error Error::LogicalBlobNameNotExistError() { function Error (line 202) | Error Error::LogicalBlobNameExistError() { function Error (line 208) | Error Error::LogicalBlobNameInvalidError() { function Error (line 214) | Error Error::OpNameExistError() { function Error (line 220) | Error Error::OpConfDeviceTagNoSetError() { function Error (line 226) | Error Error::PlacementError() { function Error (line 232) | Error Error::BlobSplitAxisInferError() { function Error (line 238) | Error Error::UnknownJobBuildAndInferError() { function Error (line 244) | Error Error::CheckFailedError() { function Error (line 250) | Error Error::ValueNotFoundError() { function Error (line 256) | Error Error::TodoError() { function Error (line 262) | Error Error::UnimplementedError() { function Error (line 268) | Error Error::RuntimeError() { function Error (line 274) | Error Error::OutOfMemoryError() { function Error (line 280) | Error Error::BoxingNotSupportedError() { function Error (line 286) | Error Error::OpKernelNotFoundError(const std::vector& err... function Error (line 295) | Error Error::MultipleOpKernelsMatchedError(const std::vector&... function GetErrorString (line 365) | std::string GetErrorString(const std::shared_ptr& error) { function ThrowError (line 376) | void ThrowError(const std::shared_ptr& error) { FILE: oneflow/core/common/error.h function namespace (line 47) | namespace oneflow { function namespace (line 84) | namespace std { function namespace (line 96) | namespace oneflow { FILE: oneflow/core/common/error_util.cpp type oneflow (line 21) | namespace oneflow { function StripSpace (line 25) | std::string StripSpace(std::string str) { function IsLetterNumberOrUnderline (line 34) | bool IsLetterNumberOrUnderline(char c) { function ShortenMsg (line 38) | Maybe ShortenMsg(std::string str) { function FormatFileOfStackFrame (line 79) | std::string FormatFileOfStackFrame(const std::string& file) { function FormatLineOfStackFrame (line 86) | std::string FormatLineOfStackFrame(const int64_t& line) { function FormatFunctionOfStackFrame (line 97) | std::string FormatFunctionOfStackFrame(const std::string& function) { function FormatMsgOfStackFrame (line 104) | Maybe FormatMsgOfStackFrame(std::string error_msg, bool i... function FormatMsgOfErrorType (line 116) | Maybe FormatMsgOfErrorType(const std::shared_ptr FormatErrorStr(const std::shared_ptr&... FILE: oneflow/core/common/error_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/common/exception.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/flat_shape.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/common/flat_shape.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/common/foreign_lock_helper.cpp type oneflow (line 19) | namespace oneflow { class NoForeignLockHelper (line 20) | class NoForeignLockHelper final : public ForeignLockHelper { method WithScopedRelease (line 21) | Maybe WithScopedRelease(const std::function()>& Ca... method WithScopedAcquire (line 25) | Maybe WithScopedAcquire(const std::function()>& Ca... FILE: oneflow/core/common/foreign_lock_helper.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/function_traits.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/hash.h function namespace (line 21) | namespace oneflow { function namespace (line 45) | namespace std { FILE: oneflow/core/common/hash_container.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/hash_eq_trait_ptr.h function namespace (line 19) | namespace oneflow { function namespace (line 40) | namespace std { FILE: oneflow/core/common/high_order_bool.h function namespace (line 29) | namespace oneflow { function expr_ (line 102) | expr_(expr) {} function ALWAYS_INLINE (line 104) | ALWAYS_INLINE bool get(const Context& context) const override { return !... FILE: oneflow/core/common/just.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/common/layout_standardize.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/common/math_util.cpp type oneflow (line 20) | namespace oneflow { function Gcd (line 22) | int64_t Gcd(int64_t m, int64_t n) { function Lcm (line 30) | int64_t Lcm(int64_t m, int64_t n) { return m * n / Gcd(m, n); } FILE: oneflow/core/common/math_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/maybe.h function namespace (line 28) | namespace oneflow { function data_or_error_ (line 47) | Maybe(const Error& error) : data_or_error_(error.stacked_error()) {} function Maybe (line 126) | static Maybe Ok() { return Maybe(); } function GetDataAndSerializedStackedError (line 138) | void GetDataAndSerializedStackedError(std::string* error_str) const { function std (line 168) | inline const std::shared_ptr& UninitializedValueError() { function error_or_scalar_ (line 180) | error_or_scalar_(UninitializedValueError()) {} function T (line 199) | T GetDataAndSerializedStackedError(std::string* error_str, const T& defa... function maybe_ptr_ (line 241) | Maybe(const Error& error) : maybe_ptr_(error) {} function T (line 259) | T GetDataAndSerializedStackedError(std::string* error_str) const { FILE: oneflow/core/common/maybe_test.cpp type oneflow (line 23) | namespace oneflow { type test (line 24) | namespace test { function TEST (line 26) | TEST(Maybe, JUST_MSG) { function TEST (line 61) | TEST(Maybe, CHECK_OR_RETURN) { function TEST (line 78) | TEST(Maybe, CHECK_OK) { function TEST (line 94) | TEST(Maybe, Noncopyable) { Maybe> a{std::make_u... FILE: oneflow/core/common/mem_util.cpp type oneflow (line 23) | namespace oneflow { type ProcStat (line 26) | struct ProcStat { function CPUSynchronize (line 35) | Maybe CPUSynchronize() { function ProcessMemUsage (line 43) | void ProcessMemUsage(double* vm_usage, double* resident_set) { function GetCPUMemoryUsed (line 69) | Maybe GetCPUMemoryUsed() { function FormatMemSize (line 76) | std::string FormatMemSize(uint64_t size) { FILE: oneflow/core/common/mem_util.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/common/meta_util.hpp type oneflow (line 22) | namespace oneflow { function for_each (line 25) | void for_each(const std::tuple& t, Func&& f, std::index_seque... function for_each_i (line 30) | void for_each_i(const std::tuple& t, Func&& f, std::index_seq... function make_tuple_from_sequence (line 38) | auto make_tuple_from_sequence(std::index_sequence) { function make_tuple_from_sequence (line 43) | constexpr auto make_tuple_from_sequence() { type detail (line 47) | namespace detail { function tuple_switch (line 49) | void tuple_switch(const std::size_t i, Tuple&& t, F&& f, std::index_... function tuple_switch (line 56) | inline void tuple_switch(const std::size_t i, Tuple&& t, F&& f) { FILE: oneflow/core/common/nd_index.cpp type oneflow (line 19) | namespace oneflow { function NdIndex (line 25) | NdIndex& NdIndex::operator=(const NdIndex& shape) { FILE: oneflow/core/common/nd_index.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/nd_index_offset_helper.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/nd_index_offset_helper_test.cpp type oneflow (line 25) | namespace oneflow { type test (line 27) | namespace test { function test_3d (line 30) | void test_3d() { function TEST (line 98) | TEST(NdIndexOffsetHelper, static_3d) { function TEST (line 103) | TEST(NdIndexOffsetHelper, dynamic_3d) { function test_constructor (line 111) | void test_constructor() { function TEST (line 141) | TEST(NdIndexOffsetHelper, constructor) { function test_stride_constructor (line 147) | void test_stride_constructor() { function TEST (line 170) | TEST(NdIndexStrideOffsetHelper, constructor) { FILE: oneflow/core/common/not_equal_to_previous_adjacent_iterator.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/notifier.cpp type oneflow (line 20) | namespace oneflow { function NotifierStatus (line 22) | NotifierStatus Notifier::Notify() { function NotifierStatus (line 34) | NotifierStatus Notifier::WaitAndClearNotifiedCnt() { FILE: oneflow/core/common/notifier.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/of_unused.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/common/op_args_reserved_size.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/common/op_args_vector.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/optional.h function namespace (line 27) | namespace oneflow { function T (line 89) | T value_or(const T& other) const { function reset (line 97) | void reset() { init_ = false; } function value_ (line 112) | value_(nullptr){} function explicit (line 115) | explicit OptionalBase(T value) : value_(&value) {} function value_ (line 116) | OptionalBase(const OptionalBase& base) : value_(base.value_) {} function T (line 133) | T value() { return *value_; } function value_type (line 137) | const value_type& value_or(const value_type& other) const { function reset (line 145) | void reset() { value_ = nullptr; } function explicit (line 165) | explicit OptionalBase(const T& value) : value_(std::make_shared(value... function explicit (line 166) | explicit OptionalBase(T&& value) : value_(std::make_shared(std::move(... function explicit (line 168) | explicit OptionalBase(const storage_type& value) : value_(value) {} function explicit (line 169) | explicit OptionalBase(storage_type&& value) : value_(std::move(value)) {} function storage_type (line 216) | const storage_type& value_or(const storage_type& other) const& { function storage_type (line 224) | storage_type value_or(const storage_type& other) && { function storage_type (line 232) | storage_type value_or(storage_type&& other) const& { function storage_type (line 240) | storage_type value_or(storage_type&& other) && { function type (line 251) | type value_or( function reset (line 297) | void reset() { value_.reset(); } type monadic_operations (line 309) | struct monadic_operations { function Optional (line 362) | Optional(NullOptType) // NOLINT(google-explicit-constructor) function has_value (line 396) | bool has_value() const { return base::has_value(); } function reset (line 453) | void reset() { base::reset(); } FILE: oneflow/core/common/optional_test.cpp type oneflow (line 22) | namespace oneflow { type test (line 23) | namespace test { function TEST (line 25) | TEST(Optional, copy_constructor) { function TEST (line 34) | TEST(Optional, move_constructor) { function TEST (line 43) | TEST(Optional, JUST) { function TEST (line 71) | TEST(Optional, reference) { function TEST (line 86) | TEST(Optional, non_scalar) { function TEST (line 109) | TEST(Optional, optional_just_error_throw) { function TEST (line 121) | TEST(Optional, monadic_operations) { FILE: oneflow/core/common/pcheck.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/permutation_iterator.h function class (line 26) | class PermutationIterator { FILE: oneflow/core/common/preprocessor_test.cpp type oneflow (line 20) | namespace oneflow { function TEST (line 22) | TEST(PP_SEQ, internal_seq_size) { function TEST (line 28) | TEST(PP_SEQ, internal_big_seq_size) { function TEST (line 37) | TEST(PP_SEQ, internal_for_each) { function TEST (line 46) | TEST(PP_TUPLE, internal_is_tuple_empty) { function TEST (line 79) | TEST(PP_SEQ, internal_seq_product_for_each) { function TEST (line 92) | TEST(PP, stringize) { function TEST (line 97) | TEST(PP, concate) { function TEST (line 102) | TEST(PP_SEQ, make_tuple_seq) { ASSERT_EQ(OF_PP_STRINGIZE(OF_PP_MAKE_TU... function TEST (line 104) | TEST(PP_SEQ, for_each_tuple) { function TEST (line 113) | TEST(PP_SEQ, outter_for_each_tuple) { function TEST (line 122) | TEST(PP_SEQ, nested_for_each_tuple) { function TEST (line 137) | TEST(PP_SEQ, seq_product_for_each) { FILE: oneflow/core/common/process_state.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/common/protobuf.cpp type oneflow (line 25) | namespace oneflow { function TryParseProtoFromTextFile (line 28) | bool TryParseProtoFromTextFile(const std::string& file_path, PbMessage... function ParseProtoFromTextFile (line 34) | void ParseProtoFromTextFile(const std::string& file_path, PbMessage* p... function TryParseProtoFromPbFile (line 39) | bool TryParseProtoFromPbFile(const std::string& file_path, PbMessage* ... function ParseProtoFromPbFile (line 44) | void ParseProtoFromPbFile(const std::string& file_path, PbMessage* pro... function PrintProtoToTextFile (line 48) | void PrintProtoToTextFile(const PbMessage& proto, const std::string& f... function PbMessage2TxtString (line 54) | std::string PbMessage2TxtString(const PbMessage& proto) { function PbMessage2TxtString (line 60) | void PbMessage2TxtString(const PbMessage& proto, std::string* str) { function TxtString2PbMessage (line 64) | bool TxtString2PbMessage(const std::string& proto_str, PbMessage* msg) { function FieldDefinedInPbMessage (line 68) | bool FieldDefinedInPbMessage(const PbMessage& msg, const std::string& ... function PbMessage (line 92) | PbMessage& GetMessageInPbMessage(const PbMessage& msg, const std::stri... function PbMessage (line 97) | PbMessage* MutableMessageInPbMessage(PbMessage* msg, const std::string... function PbMessage (line 102) | const PbMessage& GetMessageInPbMessage(const PbMessage& msg, int field... function PbMessage (line 110) | PbMessage* MutableMessageInPbMessage(PbMessage* msg, int field_index) { function GetFieldNameAndIndex4StrVal (line 189) | std::pair GetFieldNameAndIndex4StrVal(const std:... function GetStrValInPbFdOrPbRpf (line 197) | std::string GetStrValInPbFdOrPbRpf(const PbMessage& msg, const std::st... function HasStrFieldInPbFdOrPbRpf (line 208) | bool HasStrFieldInPbFdOrPbRpf(const PbMessage& msg, const std::string&... function ReplaceStrValInPbFdOrPbRpf (line 216) | std::string ReplaceStrValInPbFdOrPbRpf(PbMessage* msg, const std::stri... function PersistentOutStream (line 233) | PersistentOutStream& operator<<(PersistentOutStream& out_stream, const... FILE: oneflow/core/common/protobuf.h function namespace (line 36) | namespace oneflow { function const (line 209) | size_t operator()(const T& pb) const { function namespace (line 218) | namespace std { FILE: oneflow/core/common/range.cpp type oneflow (line 18) | namespace oneflow { function Range (line 41) | Range FindIntersectant(const Range& lhs, const Range& rhs) { FILE: oneflow/core/common/range.h function namespace (line 23) | namespace oneflow { function namespace (line 59) | namespace std { FILE: oneflow/core/common/registry_error.cpp type oneflow (line 19) | namespace oneflow { function CheckAndClearRegistryFlag (line 28) | Maybe CheckAndClearRegistryFlag() { function CatchRegistryError (line 35) | void CatchRegistryError(const std::function()>& handler) { FILE: oneflow/core/common/registry_error.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/scalar.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/common/scalar.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/common/shape.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 51) | FOR_RANGE(int, i, 0, small_shape.NumAxes()) { function FOR_RANGE (line 92) | FOR_RANGE(int, i, 0, this->NumAxes()) { type ConstShapeMixIn (line 98) | struct ConstShapeMixIn type MutShapeMixIn (line 99) | struct MutShapeMixIn type ConstShapeMixIn (line 100) | struct ConstShapeMixIn type ConstShapeMixIn (line 101) | struct ConstShapeMixIn type MutShapeMixIn (line 102) | struct MutShapeMixIn function Shape (line 104) | Shape CreateReducedShape(ShapeView shape, const AxisVector& axis_vec) { function Shape (line 113) | Shape CreateLeftExtendedShape(ShapeView shape, int ndims_left_extend_t... function Shape (line 123) | Shape ExpandDimIf0D(const Shape& shape) { function Shape (line 128) | Shape ExpandDimIf0D(ShapeView shape) { function Shape (line 133) | Shape CreateReducedShapeOrOnesShape(ShapeView shape, const AxisVector&... function ShiftNegativeAxis (line 138) | int64_t ShiftNegativeAxis(int64_t axis, const int64_t num_axes) { function Shape (line 152) | Shape& Shape::CheckNumAxesIdenticalAndAssign(ShapeView shape_view) { function Shape (line 158) | Shape& Shape::LeftOnesExtendedAssign(ShapeView shape_view) { function AxisVector (line 171) | AxisVector Shape::ShiftNegativeAxisVec(const AxisVector& axis_vec) con... function Shape (line 180) | Shape Shape::RemoveOnes(const AxisVector& axis_vec) const { function Shape (line 193) | Shape Shape::Ones(const int64_t num_axes) { function AxisVector (line 199) | AxisVector Shape::Axes4BroadcastTo(ShapeView broadcast_shape) const { function FOR_RANGE (line 229) | FOR_RANGE(int, i, 0, this->NumAxes()) { FILE: oneflow/core/common/shape.h function namespace (line 31) | namespace cfg { function elem_cnt (line 67) | int64_t elem_cnt() const; function T (line 89) | const T* tp() const { return static_cast(this); } function Set (line 94) | void Set(int64_t index, int64_t val) { function is_initialized_ (line 107) | is_initialized_(false) {} function NumAxes (line 139) | int64_t NumAxes() const { FILE: oneflow/core/common/shape_test.cpp type oneflow (line 21) | namespace oneflow { type test (line 23) | namespace test { function TEST (line 25) | TEST(Shape, constructor_0) { function TEST (line 30) | TEST(Shape, function_test_1) { FILE: oneflow/core/common/shape_vec.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/shape_view.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/common/shape_view.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/common/shared_or_scalar.h function namespace (line 25) | namespace oneflow { function ScalarT (line 99) | const ScalarT& scalar_value() const { FILE: oneflow/core/common/single_thread_obj_pool.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/common/single_thread_obj_pool_test.cpp type oneflow (line 20) | namespace oneflow { type obj_pool (line 21) | namespace obj_pool { type test (line 22) | namespace test { function TEST (line 24) | TEST(SingleThreadObjPool, naive) { type Int (line 30) | struct Int { // NOLINT method Int (line 31) | Int() : x(0) {} method Int (line 32) | explicit Int(int val) : x(val) {} function TEST (line 37) | TEST(SingleThreadObjPool, enable_reconstruct) { function TEST (line 43) | TEST(SingleThreadObjPool, disable_reconstruct) { FILE: oneflow/core/common/singleton.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/common/sized_buffer_view.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/common/small_vector.h function namespace (line 23) | namespace oneflow { function typename (line 53) | typename Base::const_iterator cend() { FILE: oneflow/core/common/spin_counter.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/common/spin_counter.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/static_check.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/common/static_global.h function namespace (line 22) | namespace oneflow { function RetT (line 39) | RetT Call(Arg0 arg0) { FILE: oneflow/core/common/steady_vector.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/common/steady_vector_test.cpp type oneflow (line 19) | namespace oneflow { type test (line 20) | namespace test { function TestSteadyVector (line 22) | void TestSteadyVector(int granularity) { function TEST (line 33) | TEST(SteadyVector, simple) { TestSteadyVector(6); } FILE: oneflow/core/common/str_util.cpp type oneflow (line 20) | namespace oneflow { type internal (line 22) | namespace internal { function JoinPathImpl (line 24) | std::string JoinPathImpl(std::initializer_list paths) { function GetHashKeyImpl (line 49) | std::string GetHashKeyImpl(std::initializer_list integers) { function Split (line 64) | void Split(const std::string& text, const std::string& delims, function Dirname (line 76) | std::string Dirname(const std::string& path) { function Basename (line 83) | std::string Basename(const std::string& path) { function CleanPath (line 89) | std::string CleanPath(const std::string& unclean_path) { function GetPrefixAndIndex (line 156) | void GetPrefixAndIndex(const std::string& prefix_and_idx, std::string*... function TryGetPrefixAndIndex (line 166) | bool TryGetPrefixAndIndex(const std::string& prefix_and_idx, std::stri... function ToLower (line 178) | std::string ToLower(const std::string& cap) { function GenAlphaNumericString (line 186) | std::string GenAlphaNumericString(size_t len) { FILE: oneflow/core/common/str_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/common/stream_type.h function namespace (line 24) | namespace oneflow { function namespace (line 62) | namespace std { FILE: oneflow/core/common/stride.cpp type oneflow (line 23) | namespace oneflow { function Stride (line 51) | Stride& Stride::CheckNumAxesIdenticalAndAssign(const Stride& stride) { FILE: oneflow/core/common/stride.h function namespace (line 24) | namespace oneflow { function namespace (line 47) | namespace std { FILE: oneflow/core/common/symbol.h function namespace (line 27) | namespace oneflow { function SymbolUtil (line 71) | SymbolUtil final { function namespace (line 153) | namespace std { FILE: oneflow/core/common/symbol_test.cpp type oneflow (line 20) | namespace oneflow { type test (line 21) | namespace test { type detail (line 23) | namespace detail { class SymObject (line 25) | class SymObject { method SymObject (line 27) | SymObject(const std::string& name) : name_(name) {} function TEST (line 39) | TEST(Symbol, shared_from_symbol) { type std (line 48) | namespace std { type hash (line 51) | struct hash final { FILE: oneflow/core/common/tensor_buffer.cpp type oneflow (line 19) | namespace oneflow { type detail (line 21) | namespace detail { function GetTensorBufferAlignedSize (line 27) | size_t GetTensorBufferAlignedSize(size_t origin_size, double factor) { function GetTensorBufferGrowthSize (line 33) | size_t GetTensorBufferGrowthSize(size_t origin_size) { function GetTensorBufferShrinkSize (line 39) | size_t GetTensorBufferShrinkSize(size_t origin_size) { function CheckTensorBufferDataType (line 45) | void CheckTensorBufferDataType(DataType val) { function TensorBuffer (line 128) | TensorBuffer& TensorBuffer::operator=(TensorBuffer&& other) noexcept { function Shape (line 164) | const Shape& TensorBuffer::shape() const { function DataType (line 169) | DataType TensorBuffer::data_type() const { function GetTensorBufferPoolSize (line 198) | size_t GetTensorBufferPoolSize(size_t base = kDefaultPoolSizeBase) { function GetTensorBufferPoolThreadLocalCacheSize (line 204) | size_t GetTensorBufferPoolThreadLocalCacheSize() { FILE: oneflow/core/common/tensor_buffer.h function namespace (line 24) | namespace oneflow { function class (line 71) | class TensorBuffer final { type GetDataType (line 129) | struct GetDataType function TensorBuffer (line 130) | inline TensorBuffer GetTypeByDataType(std::integral_constant(true, false); } FILE: oneflow/core/common/throw.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/common/to_string.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/common/tuple_hash.h function namespace (line 23) | namespace std { FILE: oneflow/core/common/type_traits.h function namespace (line 27) | namespace std { function namespace (line 107) | namespace oneflow { FILE: oneflow/core/common/util.cpp type oneflow (line 29) | namespace oneflow { function oneflow_cast (line 54) | float oneflow_cast(const std::string& s) { function oneflow_cast (line 62) | double oneflow_cast(const std::string& s) { function AbortSignalHandler (line 81) | void AbortSignalHandler(int signal) { std::abort(); } function GetAvailableCpuMemSize (line 84) | size_t GetAvailableCpuMemSize() { function IsKernelSafeInt32 (line 112) | bool IsKernelSafeInt32(int64_t n) { return n <= GetMaxVal() /... function CaseInsensitiveStringEquals (line 116) | bool CaseInsensitiveStringEquals(const std::string& lhs, const std::st... function StringToBool (line 122) | bool StringToBool(const std::string& str) { function StringToInteger (line 128) | bool StringToInteger(const std::string& str, int64_t* value) { function StringToFloat (line 139) | bool StringToFloat(const std::string& str, double* value) { function ParseBooleanFromEnv (line 152) | bool ParseBooleanFromEnv(const std::string& env_var, bool default_valu... function ParseIntegerFromEnv (line 161) | int64_t ParseIntegerFromEnv(const std::string& env_var, int64_t defaul... function ParseFloatFromEnv (line 172) | double ParseFloatFromEnv(const std::string& env_var, double default_va... function GetStringFromEnv (line 180) | std::string GetStringFromEnv(const std::string& env_var, const std::st... FILE: oneflow/core/common/util.h function namespace (line 49) | namespace fmt { function namespace (line 65) | namespace oneflow { function class (line 202) | class RoundModeGuard final { FILE: oneflow/core/common/wrap_dim_utils.h function namespace (line 19) | namespace oneflow { function Maybe (line 45) | static inline Maybe> dim_list_to_bitset( FILE: oneflow/core/common/zero_only_zip.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/control/bootstrap_client.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/control/bootstrap_server.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/control/ctrl_bootstrap.cpp type oneflow (line 25) | namespace oneflow { function BootstrapServer (line 116) | BootstrapServer* HostListCtrlBootstrap::mut_bootstrap_server() { retur... function BootstrapClient (line 117) | BootstrapClient* HostListCtrlBootstrap::mut_bootstrap_client() { retur... function BootstrapServer (line 176) | BootstrapServer* RankInfoCtrlBootstrap::mut_bootstrap_server() { retur... function BootstrapClient (line 177) | BootstrapClient* RankInfoCtrlBootstrap::mut_bootstrap_client() { retur... FILE: oneflow/core/control/ctrl_bootstrap.h function class (line 30) | class CtrlBootstrap { FILE: oneflow/core/control/ctrl_call.h function class (line 23) | class CtrlCallIf { FILE: oneflow/core/control/ctrl_client.cpp type oneflow (line 18) | namespace oneflow { function TryLockResult (line 67) | TryLockResult GrpcCtrlClient::TryLock(const std::string& name) { retur... FILE: oneflow/core/control/ctrl_server.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/control/ctrl_server.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/control/ctrl_service.cpp type oneflow (line 18) | namespace oneflow { function BuildOneRpcMethod (line 23) | const grpc::internal::RpcMethod BuildOneRpcMethod(std::shared_ptr BuildRpcMe... FILE: oneflow/core/control/ctrl_service.h function namespace (line 34) | namespace oneflow { FILE: oneflow/core/control/ctrl_test.cpp type oneflow (line 34) | namespace oneflow { function EnvProto (line 38) | EnvProto GetEnvProto(int port) { function Resource (line 47) | Resource GetResource() { function TEST (line 58) | TEST(CtrlServer, new_delete) { FILE: oneflow/core/control/ctrl_util.cpp type oneflow (line 30) | namespace oneflow { function sockaddr_in (line 36) | sockaddr_in GetSockAddr(const std::string& addr, uint16_t port) { FILE: oneflow/core/control/ctrl_util.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/control/host_list_bootstrap_client.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/control/host_list_bootstrap_client.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/control/host_list_bootstrap_server.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/control/host_list_bootstrap_server.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/control/rank_info_bootstrap_client.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/control/rank_info_bootstrap_client.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/control/rank_info_bootstrap_server.cpp type oneflow (line 23) | namespace oneflow { function GetHostFromUri (line 27) | std::string GetHostFromUri(const std::string& uri) { function rpc_bootstrap_server_sleep_seconds (line 36) | int64_t rpc_bootstrap_server_sleep_seconds() { function rpc_bootstrap_server_max_retry_times (line 42) | int64_t rpc_bootstrap_server_max_retry_times() { FILE: oneflow/core/control/rank_info_bootstrap_server.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/control/rpc_client.cpp type oneflow (line 21) | namespace oneflow { function rpc_client_max_retry_times (line 25) | int64_t rpc_client_max_retry_times() { function rpc_client_sleep_seconds (line 31) | int64_t rpc_client_sleep_seconds() { class ClientCall (line 39) | class ClientCall final { method ClientCall (line 42) | ClientCall() = default; function TryLockResult (line 70) | TryLockResult RpcClient::TryLock(const std::string& name) { FILE: oneflow/core/control/rpc_client.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/control/rpc_server.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/control/rpc_server.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/device/cuda_pseudo_bfloat16.h function __nv_bfloat16 (line 115) | __nv_bfloat16 __hneg(const __nv_bfloat16 a) { return -a; } function DEFINE_CUDA_PSEUDO_BFLOAT16_MATH_FUNC (line 161) | DEFINE_CUDA_PSEUDO_BFLOAT16_MATH_FUNC(cos) FILE: oneflow/core/device/cuda_pseudo_half.h function __half (line 109) | __half __hneg(const __half a) { return -a; } function DEFINE_CUDA_PSEUDO_HALF_MATH_FUNC (line 139) | DEFINE_CUDA_PSEUDO_HALF_MATH_FUNC(cos) FILE: oneflow/core/device/cuda_util.cpp type oneflow (line 34) | namespace oneflow { function GetAvailableGpuMemSize (line 137) | size_t GetAvailableGpuMemSize(int dev_id) { function GetCudaMallocHostFn (line 145) | std::function GetCudaMallocHostFn(int32_t... function cudaError_t (line 168) | cudaError_t NumaAwareCudaMallocHost(int32_t dev, void** ptr, size_t si... function CudaSynchronize (line 202) | void CudaSynchronize(int device_id) { function SetCudaDeviceIndex (line 207) | void SetCudaDeviceIndex(int device_id) { OF_CUDA_CHECK(cudaSetDevice(d... function GetCudaDeviceIndex (line 209) | int GetCudaDeviceIndex() { return GlobalProcessCtx::LocalRank(); } function GetCudaDeviceCount (line 211) | int GetCudaDeviceCount() { function GetCUDAMemoryUsed (line 218) | Maybe GetCUDAMemoryUsed() { function InitDevicePropVectorSize (line 246) | void InitDevicePropVectorSize() { function InitDeviceProperties (line 251) | void InitDeviceProperties(int device_id) { function cudaDeviceProp (line 258) | cudaDeviceProp* GetDeviceProperties(int device_id) { function InitCudaContextOnce (line 263) | void InitCudaContextOnce(int device_id) { function cudaError_t (line 275) | cudaError_t CudaDriverGetPrimaryCtxActive(int dev, int* active) { FILE: oneflow/core/device/cuda_util.h function namespace (line 49) | namespace oneflow { FILE: oneflow/core/device/cudnn_conv_util.cpp type oneflow (line 26) | namespace oneflow { function cudnnConvolutionFwdAlgo_t (line 34) | cudnnConvolutionFwdAlgo_t GetDefaultAlgo() { function cudnnConvolutionBwdDataAlgo_t (line 39) | cudnnConvolutionBwdDataAlgo_t GetDefaultAlgo (line 428) | struct CudnnConvAlgorithmSearch { method GetAlgoMaxCount (line 431) | static int GetAlgoMaxCount(CudnnConvResource* res) { method HeuristicSearch (line 437) | static void HeuristicSearch(const CudnnConvArgs& args, CudnnConvReso... method ExhaustiveSearch (line 448) | static void ExhaustiveSearch(const CudnnConvArgs& args, CudnnConvRes... type CudnnConvAlgorithmSearch (line 463) | struct CudnnConvAlgorithmSearch { method GetAlgoMaxCount (line 466) | static int GetAlgoMaxCount(CudnnConvResource* res) { method HeuristicSearch (line 473) | static void HeuristicSearch(const CudnnConvArgs& args, CudnnConvReso... method ExhaustiveSearch (line 484) | static void ExhaustiveSearch(const CudnnConvArgs& args, CudnnConvRes... type CudnnConvAlgorithmSearch (line 499) | struct CudnnConvAlgorithmSearch { method GetAlgoMaxCount (line 502) | static int GetAlgoMaxCount(CudnnConvResource* res) { method HeuristicSearch (line 509) | static void HeuristicSearch(const CudnnConvArgs& args, CudnnConvReso... method ExhaustiveSearch (line 520) | static void ExhaustiveSearch(const CudnnConvArgs& args, CudnnConvRes... function perf_t (line 535) | perf_t FindCudnnConvAlgorithm(CudnnConvArgs* args) { function perf_t (line 541) | perf_t FindCudnnConvAlgorithmWithResource(CudnnConvArgs* args, CudnnCo... function perf_t (line 555) | perf_t GetCudnnConvAlgorithmPerference(CudnnConvArgs* args, algo_t alg... function perf_t (line 561) | perf_t GetCudnnConvAlgorithmPerferenceWithResource(CudnnConvArgs* args... FILE: oneflow/core/device/cudnn_conv_util.h function namespace (line 26) | namespace user_op { function class (line 33) | class CudnnConvDesc final { type CudnnConvParams (line 51) | struct CudnnConvParams { function final (line 75) | struct CudnnConvArgs final { function class (line 96) | class CudnnConvResource { function cudnnHandle_t (line 116) | cudnnHandle_t cudnn_handle() override { return handle_; } function cudnnHandle_t (line 137) | cudnnHandle_t cudnn_handle() override; function namespace (line 209) | namespace oneflow { FILE: oneflow/core/device/cudnn_util.cpp type oneflow (line 19) | namespace oneflow { function cudnnDataType_t (line 23) | cudnnDataType_t GetCudnnDataType(DataType val) { function GetCudnnDataTypeByteSize (line 138) | size_t GetCudnnDataTypeByteSize(cudnnDataType_t data_type) { function cudnnHandle_t (line 194) | cudnnHandle_t CudnnHandlePool::Get() { FILE: oneflow/core/device/cudnn_util.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/device/device_id.cpp type oneflow (line 18) | namespace oneflow { function EncodeDeviceIdToInt64 (line 31) | int64_t EncodeDeviceIdToInt64(const DeviceId& device_id) { FILE: oneflow/core/device/device_id.h function hash (line 65) | size_t hash() const { function namespace (line 82) | namespace std { FILE: oneflow/core/device/ep_based_event_record.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/device/event_record.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/device/nccl_util.cpp type oneflow (line 18) | namespace oneflow { function NcclUniqueIdToString (line 22) | std::string NcclUniqueIdToString(const ncclUniqueId& unique_id) { function NcclUniqueIdFromString (line 26) | void NcclUniqueIdFromString(const std::string& str, ncclUniqueId* uniq... FILE: oneflow/core/device/nccl_util.h function namespace (line 33) | namespace oneflow { FILE: oneflow/core/eager/call_context.cpp type oneflow (line 19) | namespace oneflow { type eager (line 20) | namespace eager { function shared_to_weak (line 23) | vm::WeakEagerBlobObjectList shared_to_weak(const vm::EagerBlobObject... function CallContext (line 66) | CallContext DtrCallContext::ToCallContext() const { return CallConte... FILE: oneflow/core/eager/call_context.h function namespace (line 26) | namespace oneflow { function class (line 78) | class CallContext { function class (line 116) | class DtrCallContext { FILE: oneflow/core/eager/dev_vm_dep_object_consume_mode.h function namespace (line 20) | namespace oneflow { FILE: oneflow/core/eager/eager_blob_object.cpp type oneflow (line 24) | namespace oneflow { type vm (line 26) | namespace vm { function Shape (line 48) | const Shape& EagerBlobObject::shape() const { function Stride (line 55) | const Stride& EagerBlobObject::stride() const { function MutShapeView (line 72) | MutShapeView EagerBlobObject::mut_shape_view() { FILE: oneflow/core/eager/eager_blob_object.h function namespace (line 33) | namespace oneflow { FILE: oneflow/core/eager/local_dep_object.cpp type oneflow (line 20) | namespace oneflow { function NewLocalDepObject (line 22) | intrusive::shared_ptr NewLocalDepObject() { FILE: oneflow/core/eager/local_dep_object.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/eager/tensor_storage.cpp type oneflow (line 24) | namespace oneflow { type vm (line 25) | namespace vm { function unique_id (line 27) | int64_t unique_id() { function OpCallInstructionPolicy (line 117) | OpCallInstructionPolicy RematableTensorStorage::compute_op() const { FILE: oneflow/core/eager/tensor_storage.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/embedding/cache.cpp type oneflow (line 20) | namespace oneflow { type embedding (line 22) | namespace embedding { function NewCache (line 24) | std::unique_ptr NewCache(const CacheOptions& options) { FILE: oneflow/core/embedding/cache.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/embedding/cache_test.cpp type oneflow (line 21) | namespace oneflow { type embedding (line 23) | namespace embedding { function HasCudaDevice (line 29) | bool HasCudaDevice() { function TestCache (line 36) | void TestCache(Cache* cache, uint32_t line_size) { function TEST (line 214) | TEST(Cache, FullCache) { function TEST (line 229) | TEST(Cache, LruCache) { FILE: oneflow/core/embedding/cached_key_value_store.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/embedding/embedding_manager.cpp type oneflow (line 21) | namespace oneflow { type embedding (line 23) | namespace embedding { type IdStatistics (line 31) | struct IdStatistics { method IdStatistics (line 32) | IdStatistics() : final_num_unique(0), iter(-1) {} class DynamicTmpBufferAllocator (line 40) | class DynamicTmpBufferAllocator final : public TmpBufferAllocator { method DynamicTmpBufferAllocator (line 43) | DynamicTmpBufferAllocator(cudaStream_t stream, cudaMemPool_t pool) method Allocate (line 47) | void Allocate(void** ptr, size_t size) override { method Free (line 50) | void Free(void* ptr) override { OF_CUDA_CHECK(cudaFreeAsync(ptr, s... class DynamicAllocationEmbeddingState (line 57) | class DynamicAllocationEmbeddingState final : public EmbeddingState { method DynamicAllocationEmbeddingState (line 60) | DynamicAllocationEmbeddingState() method NewTmpBufferAllocator (line 87) | std::unique_ptr NewTmpBufferAllocator( method OnEmbeddingLookupStart (line 93) | void OnEmbeddingLookupStart(user_op::KernelComputeContext* ctx, in... method OnEmbeddingLookupEnd (line 139) | void OnEmbeddingLookupEnd(user_op::KernelComputeContext* ctx, int6... method OnEmbeddingGatherStart (line 143) | void OnEmbeddingGatherStart(user_op::KernelComputeContext* ctx, in... method OnEmbeddingGatherEnd (line 156) | void OnEmbeddingGatherEnd(user_op::KernelComputeContext* ctx, int6... method OnEmbeddingShuffleStart (line 160) | void OnEmbeddingShuffleStart(user_op::KernelComputeContext* ctx, i... method OnEmbeddingShuffleEnd (line 173) | void OnEmbeddingShuffleEnd(user_op::KernelComputeContext* ctx, int... method OnEmbeddingUpdateStart (line 177) | void OnEmbeddingUpdateStart(user_op::KernelComputeContext* ctx, in... method OnEmbeddingUpdateEnd (line 199) | void OnEmbeddingUpdateEnd(user_op::KernelComputeContext* ctx, int6... method OnEmbeddingPutStart (line 203) | void OnEmbeddingPutStart(user_op::KernelComputeContext* ctx, int64... method OnEmbeddingPutEnd (line 212) | void OnEmbeddingPutEnd(user_op::KernelComputeContext* ctx, int64_t... method OnEmbeddingFusedUpdatePutStart (line 217) | void OnEmbeddingFusedUpdatePutStart(user_op::KernelComputeContext*... method OnEmbeddingFusedUpdatePutEnd (line 227) | void OnEmbeddingFusedUpdatePutEnd(user_op::KernelComputeContext* c... method SetIdFinalNumUnique (line 231) | void SetIdFinalNumUnique(uint32_t final_num_unique, int64_t iter) ... method SetIdNumUniqueMatrix (line 238) | void SetIdNumUniqueMatrix(const std::vector& num_unique_... method GetIdNumUnique (line 245) | uint32_t GetIdNumUnique(int64_t iter) override { class StaticTmpBufferAllocator (line 280) | class StaticTmpBufferAllocator final : public TmpBufferAllocator { method StaticTmpBufferAllocator (line 283) | StaticTmpBufferAllocator(void* ptr, size_t size) : ptr_(ptr), offs... method Allocate (line 286) | void Allocate(void** ptr, size_t size) override { method Free (line 295) | void Free(void* ptr) override { class StaticAllocationEmbeddingState (line 305) | class StaticAllocationEmbeddingState final : public EmbeddingState { method StaticAllocationEmbeddingState (line 308) | StaticAllocationEmbeddingState() method NewTmpBufferAllocator (line 321) | std::unique_ptr NewTmpBufferAllocator( method OnEmbeddingLookupStart (line 328) | void OnEmbeddingLookupStart(user_op::KernelComputeContext* ctx, in... method OnEmbeddingLookupEnd (line 345) | void OnEmbeddingLookupEnd(user_op::KernelComputeContext* ctx, int6... method OnEmbeddingGatherStart (line 351) | void OnEmbeddingGatherStart(user_op::KernelComputeContext* ctx, in... method OnEmbeddingGatherEnd (line 358) | void OnEmbeddingGatherEnd(user_op::KernelComputeContext* ctx, int6... method OnEmbeddingShuffleStart (line 362) | void OnEmbeddingShuffleStart(user_op::KernelComputeContext* ctx, i... method OnEmbeddingShuffleEnd (line 372) | void OnEmbeddingShuffleEnd(user_op::KernelComputeContext* ctx, int... method OnEmbeddingUpdateStart (line 376) | void OnEmbeddingUpdateStart(user_op::KernelComputeContext* ctx, in... method OnEmbeddingUpdateEnd (line 392) | void OnEmbeddingUpdateEnd(user_op::KernelComputeContext* ctx, int6... method OnEmbeddingPutStart (line 397) | void OnEmbeddingPutStart(user_op::KernelComputeContext* ctx, int64... method OnEmbeddingPutEnd (line 406) | void OnEmbeddingPutEnd(user_op::KernelComputeContext* ctx, int64_t... method OnEmbeddingFusedUpdatePutStart (line 410) | void OnEmbeddingFusedUpdatePutStart(user_op::KernelComputeContext*... method OnEmbeddingFusedUpdatePutEnd (line 419) | void OnEmbeddingFusedUpdatePutEnd(user_op::KernelComputeContext* c... method SetIdFinalNumUnique (line 423) | void SetIdFinalNumUnique(uint32_t final_num_unique, int64_t iter) ... method SetIdNumUniqueMatrix (line 430) | void SetIdNumUniqueMatrix(const std::vector& num_unique_... method GetIdNumUnique (line 437) | uint32_t GetIdNumUnique(int64_t iter) override { function EmbeddingState (line 468) | EmbeddingState* EmbeddingManager::GetEmbeddingState(const std::strin... function KeyValueStore (line 492) | KeyValueStore* EmbeddingManager::GetKeyValueStore(const std::string&... FILE: oneflow/core/embedding/embedding_manager.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/embedding/full_cache.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/embedding/key_value_store.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/embedding/key_value_store_options.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/embedding/key_value_store_test.cpp type oneflow (line 25) | namespace oneflow { type embedding (line 27) | namespace embedding { function CreateTempDirectory (line 33) | std::string CreateTempDirectory() { function HasCudaDevice (line 42) | bool HasCudaDevice() { function TestKeyValueStore (line 49) | void TestKeyValueStore(KeyValueStore* store, size_t num_embeddings, ... function TEST (line 174) | TEST(PersistentTableKeyValueStore, PersistentTableKeyValueStore) { function TEST (line 194) | TEST(CachedKeyValueStore, LRU) { function TEST (line 221) | TEST(CachedKeyValueStore, Full) { function TEST (line 248) | TEST(MockKeyValueStore, Mock) { FILE: oneflow/core/embedding/kv_iterator.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/embedding/lru_cache.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/embedding/mock_key_value_store.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/embedding/persistent_table.cpp type oneflow (line 35) | namespace oneflow { type embedding (line 37) | namespace embedding { function T (line 63) | T* BytesOffset(T* ptr, size_t bytes) { function MemcpyOffset (line 68) | void MemcpyOffset(void* dst, size_t dst_off, const void* src, size_t... function InitOrCheckMetaValue (line 72) | void InitOrCheckMetaValue(const std::string& pathname, int64_t expec... function GetChunkName (line 87) | std::string GetChunkName(uint64_t chunk_id) { function GetChunkId (line 94) | uint64_t GetChunkId(const std::string& chunk_name) { function GetChunkId (line 101) | uint64_t GetChunkId(const std::string& filename, const std::string& ... function ListChunkFiles (line 106) | void ListChunkFiles(const std::string& base, const std::string& prefix, function GetLogicalBlockSize (line 120) | uint32_t GetLogicalBlockSize(uint32_t physical_block_size, uint32_t ... class AlignedBuffer (line 125) | class AlignedBuffer final { method AlignedBuffer (line 128) | explicit AlignedBuffer(size_t alignment) : alignment_(alignment), ... method Resize (line 131) | void Resize(size_t new_size) { class ChunkIteratorImpl (line 147) | class ChunkIteratorImpl : public PersistentTable::Iterator { method ChunkIteratorImpl (line 150) | ChunkIteratorImpl(uint32_t value_size, uint32_t logical_block_size... method Next (line 165) | void Next(uint32_t num_keys, uint32_t* return_keys, void* keys, vo... method Reset (line 182) | void Reset() override { pos_ = 0; } class AioEngine (line 197) | class AioEngine final { method AioEngine (line 200) | AioEngine() : ctx_{}, num_readings_(0) { method AsyncPread (line 212) | void AsyncPread(int fd, void* buf, size_t count, off_t offset) { method WaitUntilDone (line 226) | void WaitUntilDone() { type iocb (line 238) | struct iocb type iocb (line 239) | struct iocb type io_event (line 240) | struct io_event class Worker (line 252) | class Worker final { method Worker (line 255) | Worker() { thread_ = std::thread(&Worker::PullTask, this); } method Schedule (line 261) | void Schedule(IoTask task) { tasks_.Send(std::move(task)); } method Shutdown (line 263) | void Shutdown() { tasks_.Close(); } method PullTask (line 266) | void PullTask() { class SnapshotIteratorImpl (line 281) | class SnapshotIteratorImpl method SnapshotIteratorImpl (line 735) | SnapshotIteratorImpl(PersistentTableImpl* table, cons... method Next (line 752) | void Next(uint32_t num_keys, uint32_t* return_keys, void* keys, vo... method Reset (line 792) | void Reset() override { UNIMPLEMENTED(); } class PersistentTableImpl (line 284) | class PersistentTableImpl : public PersistentTable { method KeySize (line 290) | uint32_t KeySize() const override { return key_size_; } method ValueSize (line 292) | uint32_t ValueSize() const override { return value_size_; } class SnapshotIteratorImpl (line 732) | class SnapshotIteratorImpl : public PersistentTable::Iterator { method SnapshotIteratorImpl (line 735) | SnapshotIteratorImpl(PersistentTableImpl* table, cons... method Next (line 752) | void Next(uint32_t num_keys, uint32_t* return_keys, void* keys, vo... method Reset (line 792) | void Reset() override { UNIMPLEMENTED(); } function DispatchKeyType (line 810) | std::unique_ptr DispatchKeyType(const PersistentTab... function DispatchEngine (line 821) | std::unique_ptr DispatchEngine(const PersistentTabl... function NewPersistentTable (line 829) | std::unique_ptr NewPersistentTable(const Persistent... FILE: oneflow/core/embedding/persistent_table.h function namespace (line 23) | namespace embedding { FILE: oneflow/core/embedding/persistent_table_key_value_store.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/embedding/posix_file.h function namespace (line 32) | namespace oneflow { function class (line 141) | class PosixMappedFile final { function class (line 180) | class PosixFileLockGuard final { function f (line 199) | struct flock f {} function Unlock (line 207) | void Unlock() { FILE: oneflow/core/ep/common/active_device_guard.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { FILE: oneflow/core/ep/common/device.cpp type oneflow (line 18) | namespace oneflow { type ep (line 20) | namespace ep { function Event (line 22) | Event* Device::CreateEvent() { FILE: oneflow/core/ep/common/device_manager_registry.cpp type oneflow (line 20) | namespace oneflow { type ep (line 22) | namespace ep { class DeviceManagerRegistry::Impl (line 24) | class DeviceManagerRegistry::Impl { method Impl (line 27) | explicit Impl(DeviceManagerRegistry* registry) : registry_(registr... method DeviceManager (line 32) | DeviceManager* GetDeviceManagerOrNull(DeviceType device_type) { method DeviceManager (line 46) | DeviceManager* GetDeviceManager(DeviceType device_type) { method GetDevice (line 50) | std::shared_ptr GetDevice(DeviceType device_type, size_t d... method GetDeviceCount (line 54) | size_t GetDeviceCount(DeviceType device_type) { method GetDeviceCount (line 63) | size_t GetDeviceCount(const std::string& device_type_name) { method DumpVersionInfo (line 67) | static void DumpVersionInfo() { method GetDeviceTypeNameByDeviceType (line 74) | static std::string GetDeviceTypeNameByDeviceType(DeviceType device... method DeviceType (line 92) | static DeviceType GetDeviceTypeByDeviceTypeName(const std::string&... method RegisterDeviceManagerFactory (line 108) | static void RegisterDeviceManagerFactory(std::unique_ptr GetRegisteredDeviceTypes() { method IsDeviceTypeRegistered (line 129) | static bool IsDeviceTypeRegistered(DeviceType device_type) { function DeviceManager (line 159) | DeviceManager* DeviceManagerRegistry::GetDeviceManager(DeviceType de... function DeviceManager (line 163) | DeviceManager* DeviceManagerRegistry::GetDeviceManagerOrNull(DeviceT... function DeviceType (line 192) | DeviceType DeviceManagerRegistry::GetDeviceTypeByDeviceTypeName( FILE: oneflow/core/ep/common/onednn.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/add.cpp type oneflow (line 18) | namespace oneflow { type ep (line 20) | namespace ep { type primitive (line 21) | namespace primitive { FILE: oneflow/core/ep/common/primitive/batch_matmul.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { class BatchMatmulImpl (line 26) | class BatchMatmulImpl : public BatchMatmul { method BatchMatmulImpl (line 29) | BatchMatmulImpl(BlasTransposeType transpose_a, BlasTransposeType... method Launch (line 36) | void Launch(Stream* stream, size_t batch_size, size_t m, size_t ... class BatchMatmulFactoryImpl (line 74) | class BatchMatmulFactoryImpl : public BatchMatmulFactory { method BatchMatmulFactoryImpl (line 77) | BatchMatmulFactoryImpl() = default; method New (line 80) | std::unique_ptr New(DataType data_type, BlasTranspo... FILE: oneflow/core/ep/common/primitive/binary_functor.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/broadcast_elementwise_binary.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/broadcast_elementwise_unary.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/broadcast_matmul.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/broadcast_simplify_dims_test.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { function TestSimplifyBroadcastDims (line 27) | void TestSimplifyBroadcastDims(size_t num_src0_dims, const int64_t... function TEST (line 46) | TEST(Broadcast, SimplifyBroadcastDims) { FILE: oneflow/core/ep/common/primitive/constant_pad.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/copy_nd.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/elementwise_unary.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/matmul.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { class MatmulImpl (line 26) | class MatmulImpl : public Matmul { method MatmulImpl (line 29) | explicit MatmulImpl(std::unique_ptr&& batch_matmul) method Launch (line 33) | void Launch(Stream* stream, size_t m, size_t n, size_t k, Scalar... class MatmulFactoryImpl (line 43) | class MatmulFactoryImpl : public MatmulFactory { method MatmulFactoryImpl (line 46) | MatmulFactoryImpl() = default; method New (line 49) | std::unique_ptr New(DataType data_type, BlasTransposeTyp... FILE: oneflow/core/ep/common/primitive/permute.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/permute_impl.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/permute_test.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { type permute (line 24) | namespace permute { function TestSimplifyPermutation (line 29) | void TestSimplifyPermutation(size_t num_dims, const int64_t* src... function TEST (line 44) | TEST(Permute, SimplifyPermutation) { FILE: oneflow/core/ep/common/primitive/unary_functor.h function OF_DEVICE_FUNC (line 33) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 35) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 40) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) : alpha(attr0.Va... function OF_DEVICE_FUNC (line 42) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 64) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 66) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 79) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 81) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 94) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) : lambd(attr0.Va... function OF_DEVICE_FUNC (line 96) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 105) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) function OF_DEVICE_FUNC (line 108) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 124) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) : alpha(attr0.Va... function OF_DEVICE_FUNC (line 126) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 134) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 136) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 147) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 149) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 161) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 163) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 170) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 172) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 183) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 185) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 192) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) function OF_DEVICE_FUNC (line 195) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 206) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) : alpha(attr0.Va... function OF_DEVICE_FUNC (line 208) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 222) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) function OF_DEVICE_FUNC (line 225) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 234) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 236) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(!... function OF_DEVICE_FUNC (line 241) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 243) | OF_DEVICE_FUNC bool operator()(Src src) const { return false; } function OF_DEVICE_FUNC (line 248) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 250) | OF_DEVICE_FUNC bool operator()(Src src) const { return false; } function OF_DEVICE_FUNC (line 255) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 257) | OF_DEVICE_FUNC bool operator()(Src src) const { return true; } function OF_DEVICE_FUNC (line 268) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 270) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 275) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 277) | OF_DEVICE_FUNC uint8_t operator()(uint8_t src) const { return src; } function OF_DEVICE_FUNC (line 282) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 284) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(e... function OF_DEVICE_FUNC (line 289) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 291) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(e... function OF_DEVICE_FUNC (line 296) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 298) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 303) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 305) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 310) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 312) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 317) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 319) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 324) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 326) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 331) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 333) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(a... function OF_DEVICE_FUNC (line 338) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 340) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(c... function OF_DEVICE_FUNC (line 345) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 347) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(c... function OF_DEVICE_FUNC (line 352) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 354) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(c... function OF_DEVICE_FUNC (line 359) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 361) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(e... function OF_DEVICE_FUNC (line 366) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 368) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(e... function OF_DEVICE_FUNC (line 373) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 375) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(e... function OF_DEVICE_FUNC (line 380) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 382) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(f... function OF_DEVICE_FUNC (line 387) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 389) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(l... function OF_DEVICE_FUNC (line 394) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 396) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(l... function OF_DEVICE_FUNC (line 401) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 403) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(l... function OF_DEVICE_FUNC (line 408) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 410) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(l... function OF_DEVICE_FUNC (line 415) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 417) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(l... function OF_DEVICE_FUNC (line 422) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 424) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 431) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 433) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(-... function OF_DEVICE_FUNC (line 438) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 440) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 447) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 449) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 457) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 459) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(r... function OF_DEVICE_FUNC (line 464) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 466) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(n... function OF_DEVICE_FUNC (line 471) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 473) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(r... function OF_DEVICE_FUNC (line 478) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 480) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 487) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 489) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 503) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 505) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 510) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 512) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 517) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 519) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 524) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 526) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 531) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 533) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(t... function OF_DEVICE_FUNC (line 538) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 540) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 547) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 549) | OF_DEVICE_FUNC Dst operator()(Src src) const { function OF_DEVICE_FUNC (line 556) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 558) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 563) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 565) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(~... function OF_DEVICE_FUNC (line 570) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 572) | OF_DEVICE_FUNC Dst operator()(bool src) const { return static_cast(... function OF_DEVICE_FUNC (line 577) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 579) | OF_DEVICE_FUNC Dst operator()(Src src) const { return Dst{src.real(), -s... function OF_DEVICE_FUNC (line 584) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 586) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 591) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 593) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 598) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 599) | OF_DEVICE_FUNC Dst operator()(Src src) const { return Dst{src, 0.0}; } function OF_DEVICE_FUNC (line 604) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 606) | OF_DEVICE_FUNC Dst operator()(Src src) const { return Dst{0.0, src}; } FILE: oneflow/core/ep/common/primitive/util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/common/primitive/where.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ep/cpu/cpu_device.cpp type oneflow (line 22) | namespace oneflow { type ep (line 24) | namespace ep { function Stream (line 28) | Stream* CpuDevice::CreateStream() { return new CpuStream(this); } FILE: oneflow/core/ep/cpu/cpu_device.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/cpu/cpu_device_manager.cpp type oneflow (line 20) | namespace oneflow { type ep (line 22) | namespace ep { function DeviceManagerRegistry (line 29) | DeviceManagerRegistry* CpuDeviceManager::registry() const { return r... FILE: oneflow/core/ep/cpu/cpu_device_manager.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/cpu/cpu_device_manager_factory.cpp type oneflow (line 20) | namespace oneflow { type ep (line 22) | namespace ep { class CpuDeviceManagerFactory (line 26) | class CpuDeviceManagerFactory : public DeviceManagerFactory { method CpuDeviceManagerFactory (line 29) | CpuDeviceManagerFactory() = default; method NewDeviceManager (line 32) | std::unique_ptr NewDeviceManager(DeviceManagerRegis... method DeviceType (line 36) | DeviceType device_type() const override { return DeviceType::kCPU; } method device_type_name (line 38) | std::string device_type_name() const override { return "cpu"; } class MockDeviceManagerFactory (line 48) | class MockDeviceManagerFactory : public DeviceManagerFactory { method MockDeviceManagerFactory (line 51) | MockDeviceManagerFactory() = default; method NewDeviceManager (line 54) | std::unique_ptr NewDeviceManager(DeviceManagerRegis... method DeviceType (line 58) | DeviceType device_type() const override { return DeviceType::kMock... method device_type_name (line 60) | std::string device_type_name() const override { return "mock"; } class MetaDeviceManagerFactory (line 70) | class MetaDeviceManagerFactory : public DeviceManagerFactory { method MetaDeviceManagerFactory (line 73) | MetaDeviceManagerFactory() = default; method NewDeviceManager (line 76) | std::unique_ptr NewDeviceManager(DeviceManagerRegis... method DeviceType (line 80) | DeviceType device_type() const override { return DeviceType::kMeta; } method device_type_name (line 82) | std::string device_type_name() const override { return "meta"; } FILE: oneflow/core/ep/cpu/cpu_event.cpp type oneflow (line 18) | namespace oneflow { type ep (line 20) | namespace ep { FILE: oneflow/core/ep/cpu/cpu_event.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/cpu/cpu_random_generator.cpp type oneflow (line 21) | namespace oneflow { type ep (line 22) | namespace ep { type CPUGeneratorState (line 24) | struct CPUGeneratorState { FILE: oneflow/core/ep/cpu/cpu_random_generator.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/ep/cpu/cpu_stream.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { function DeviceType (line 23) | DeviceType CpuStream::device_type() const { return DeviceType::kCPU; } function CpuDevice (line 25) | CpuDevice* CpuStream::device() const { return device_; } FILE: oneflow/core/ep/cpu/cpu_stream.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/ep/cpu/primitive/add.cpp type primitive (line 25) | namespace primitive { function AddCpu (line 30) | void AddCpu(const T* const* srcs, T* dst, size_t count) { function AddCpu (line 39) | void AddCpu(const T* const* srcs, size_t arity, T* dst, size_t count) { class AddDefaultImpl (line 48) | class AddDefaultImpl : public Add { method AddDefaultImpl (line 51) | AddDefaultImpl() = default; method Launch (line 55) | void Launch(Stream* stream, const void* const* srcs, size_t arity, v... FILE: oneflow/core/ep/cpu/primitive/binary_functor.h function namespace (line 18) | namespace oneflow { FILE: oneflow/core/ep/cpu/primitive/broadcast_elementwise_binary.cpp type oneflow (line 30) | namespace oneflow { type ep (line 32) | namespace ep { type primitive (line 33) | namespace primitive { type broadcast_elementwise_binary (line 34) | namespace broadcast_elementwise_binary { function T (line 39) | T GetValue(Scalar value) { function float16 (line 44) | float16 GetValue(Scalar value) { function bfloat16 (line 49) | bfloat16 GetValue(Scalar value) { type BinaryLhsScalarFunctor (line 54) | struct BinaryLhsScalarFunctor { method BinaryLhsScalarFunctor (line 55) | BinaryLhsScalarFunctor(Src scalar, Scalar attr0, Scalar attr1) method Dst (line 57) | Dst operator()(Src src) const { return functor(scalar, src); } type BinaryRhsScalarFunctor (line 63) | struct BinaryRhsScalarFunctor { method BinaryRhsScalarFunctor (line 64) | BinaryRhsScalarFunctor(Src scalar, Scalar attr0, Scalar attr1) method Dst (line 66) | Dst operator()(Src src) const { return functor(src, scalar); } function LaunchElementwise (line 72) | void LaunchElementwise(CpuStream* cpu_stream, size_t simplified_... function LaunchBinaryLhsScalar (line 84) | void LaunchBinaryLhsScalar(CpuStream* cpu_stream, Src src0_value... function LaunchBinaryRhsScalar (line 93) | void LaunchBinaryRhsScalar(CpuStream* cpu_stream, Src src1_value... function LaunchRowWithMatrix (line 102) | void LaunchRowWithMatrix(CpuStream* cpu_stream, const int64_t* s... function LaunchMatrixWithRow (line 123) | void LaunchMatrixWithRow(CpuStream* cpu_stream, const int64_t* s... function LaunchColWithMatrix (line 144) | void LaunchColWithMatrix(CpuStream* cpu_stream, const int64_t* s... function LaunchMatrixWithCol (line 165) | void LaunchMatrixWithCol(CpuStream* cpu_stream, const int64_t* s... function LaunchGeneral (line 186) | void LaunchGeneral(CpuStream* cpu_stream, size_t simplified_num_... function LaunchGeneralDispatchIndexType (line 226) | void LaunchGeneralDispatchIndexType(CpuStream* cpu_stream, size_... function DispatchLaunch (line 244) | void DispatchLaunch(Stream* stream, size_t num_src0_dims, const ... class BroadcastElementwiseBinaryImpl (line 293) | class BroadcastElementwiseBinaryImpl : public BroadcastElementwi... method BroadcastElementwiseBinaryImpl (line 296) | BroadcastElementwiseBinaryImpl(Scalar attr0, Scalar attr1) : a... method Launch (line 299) | void Launch(Stream* stream, Scalar src0, size_t num_src1_dims,... method Launch (line 308) | void Launch(Stream* stream, size_t num_src0_dims, const int64_... method Launch (line 317) | void Launch(Stream* stream, size_t num_src0_dims, const int64_... function NewBroadcastElementwiseBinary (line 330) | std::unique_ptr NewBroadcastElementw... function OneDnnBroadcastDims (line 351) | inline void OneDnnBroadcastDims(dnnl::memory::dims* src0, size_t... class OneDnnBroadcastElementwiseBinaryImpl (line 370) | class OneDnnBroadcastElementwiseBinaryImpl : public BroadcastEle... method OneDnnBroadcastElementwiseBinaryImpl (line 373) | OneDnnBroadcastElementwiseBinaryImpl(Scalar attr0, Scalar attr... method Launch (line 376) | void Launch(Stream* stream, Scalar src0, size_t num_src1_dims,... method Launch (line 382) | void Launch(Stream* stream, size_t num_src0_dims, const int64_... method Launch (line 388) | void Launch(Stream* stream, size_t num_src0_dims, const int64_... function NewOneDnnBroadcastElementwiseBinary (line 500) | std::unique_ptr NewOneDnnBroadcastEl... class BroadcastElementwiseBinaryFactoryImpl (line 523) | class BroadcastElementwiseBinaryFactoryImpl : public BroadcastEl... method BroadcastElementwiseBinaryFactoryImpl (line 526) | BroadcastElementwiseBinaryFactoryImpl() = default; method New (line 529) | std::unique_ptr New(BinaryOp op, D... method New (line 534) | std::unique_ptr New(BinaryOp op, D... method New (line 539) | std::unique_ptr New(BinaryOp binar... FILE: oneflow/core/ep/cpu/primitive/broadcast_elementwise_unary.cpp type oneflow (line 25) | namespace oneflow { type ep (line 27) | namespace ep { type primitive (line 28) | namespace primitive { type broadcast_elementwise_unary (line 29) | namespace broadcast_elementwise_unary { function IsContiguous (line 39) | bool IsContiguous(size_t num_dims, const int64_t* dims, const in... function LaunchScalarFill (line 50) | void LaunchScalarFill(CpuStream* stream, Dst* dst, const Src* sr... function LaunchTensorFill (line 60) | void LaunchTensorFill(CpuStream* stream, Dst* dst, const Src* sr... function LaunchGeneral (line 72) | void LaunchGeneral(CpuStream* stream, Dst* dst, const Src* src, ... class BroadcastElementwiseUnaryImpl (line 113) | class BroadcastElementwiseUnaryImpl : public BroadcastElementwis... method BroadcastElementwiseUnaryImpl (line 116) | BroadcastElementwiseUnaryImpl(Scalar attr0, Scalar attr1) : at... method Launch (line 119) | void Launch(Stream* stream, size_t num_src_dims, const int64_t... method Launch (line 139) | void Launch(Stream* stream, size_t num_src_dims, const int64_t... function NewBroadcastElementwiseUnary (line 192) | std::unique_ptr NewBroadcastElementwi... class BroadcastElementwiseUnaryFactoryImpl (line 198) | class BroadcastElementwiseUnaryFactoryImpl : public BroadcastEle... method BroadcastElementwiseUnaryFactoryImpl (line 201) | BroadcastElementwiseUnaryFactoryImpl() = default; method New (line 204) | std::unique_ptr New(UnaryOp op, Dat... method New (line 209) | std::unique_ptr New(UnaryOp op, Dat... method New (line 214) | std::unique_ptr New(UnaryOp unary_o... FILE: oneflow/core/ep/cpu/primitive/broadcast_matmul.cpp type oneflow (line 21) | namespace oneflow { type ep (line 23) | namespace ep { type primitive (line 24) | namespace primitive { type broadcast_matmul (line 26) | namespace broadcast_matmul { type internal (line 28) | namespace internal { function CBLAS_TRANSPOSE (line 34) | CBLAS_TRANSPOSE GetCblasTranspose(BlasTransposeType transpose_... function CblasMatmul (line 48) | void CblasMatmul(CBLAS_TRANSPOSE trans_a, CBLAS_TRANSPOSE tran... function CblasMatmul (line 73) | void CblasMatmul(CBLAS_TRANSPOSE trans_a, CBLAS_TRANSPOSE tran... function LaunchCblasBroadcastMatmul (line 98) | void LaunchCblasBroadcastMatmul(Stream* /*stream*/, DataType d... function LaunchBroadcastMatmul (line 117) | void LaunchBroadcastMatmul(Stream* stream, DataType data_type,... class BroadcastMatmulFactoryImpl (line 144) | class BroadcastMatmulFactoryImpl : public BroadcastMatmulFacto... method BroadcastMatmulFactoryImpl (line 147) | BroadcastMatmulFactoryImpl() = default; method New (line 150) | std::unique_ptr New(DataType data_type, Bla... FILE: oneflow/core/ep/cpu/primitive/cast.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { type CpuCastFunctor (line 27) | struct CpuCastFunctor { method Call (line 28) | static void Call(const From* from, To* to, size_t count) { type CpuCastFunctor::value)>::type> (line 34) | struct CpuCastFunctor::value)>::type> (line 42) | struct CpuCastFunctor NewCast() { class CastFactoryImpl (line 81) | class CastFactoryImpl : public CastFactory { method CastFactoryImpl (line 84) | CastFactoryImpl() = default; method New (line 87) | std::unique_ptr New(DataType from, DataType to) override { FILE: oneflow/core/ep/cpu/primitive/constant_pad.cpp type oneflow (line 20) | namespace oneflow { type ep (line 22) | namespace ep { type primitive (line 24) | namespace primitive { function ConstantPadKernel (line 29) | void ConstantPadKernel(ConstantPadParams para... function float16 (line 55) | float16 GetValue(Scalar value) { function bfloat16 (line 60) | bfloat16 GetValue(Scalar value) { function LaunchKernel (line 65) | void LaunchKernel(ConstantPadParams params, S... function LaunchKernel (line 70) | void LaunchKernel(void* dst, const int64_t* dst_dims, const void* ... function DispatchIndexType (line 87) | void DispatchIndexType(void* dst, const int64_t* dst_dims, const v... function DispatchPackSize (line 100) | void DispatchPackSize(void* dst, int64_t* dst_dims, const void* sr... function LaunchWithSimplified (line 140) | void LaunchWithSimplified(size_t num_dims, void* dst, int64_t* dst... function SimplifyThenLaunch (line 168) | void SimplifyThenLaunch(size_t num_dims, const int64_t* src_dims, ... class ConstantPadImpl (line 186) | class ConstantPadImpl : public ConstantPad { method ConstantPadImpl (line 189) | ConstantPadImpl() = default; method Launch (line 192) | void Launch(Stream* stream, size_t num_dims, const int64_t* src_... function NewConstantPad (line 201) | std::unique_ptr NewConstantPad() { class ConstantPadFactoryImpl (line 205) | class ConstantPadFactoryImpl : public ConstantPadFactory { method ConstantPadFactoryImpl (line 208) | ConstantPadFactoryImpl() = default; method New (line 211) | std::unique_ptr New(DataType data_type) override { FILE: oneflow/core/ep/cpu/primitive/copy_nd.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { function CopyNdKernel (line 27) | void CopyNdKernel(CopyNdKernelParams params) { function LaunchKernel (line 47) | void LaunchKernel(Stream* stream, CopyNdKernelParams New(size_t max_num_dims) override { FILE: oneflow/core/ep/cpu/primitive/elementwise_unary.cpp type oneflow (line 22) | namespace oneflow { type ep (line 24) | namespace ep { type primitive (line 25) | namespace primitive { class ElementwiseUnaryImpl (line 30) | class ElementwiseUnaryImpl : public ElementwiseUnary { method ElementwiseUnaryImpl (line 33) | ElementwiseUnaryImpl(Scalar attr0, Scalar attr1) : attr0(attr0),... method Launch (line 36) | void Launch(Stream* stream, const void* src_ptr, void* dst_ptr, ... function NewElementwiseUnary (line 52) | std::unique_ptr NewElementwiseUnary(Scalar attr0... class ElementwiseUnaryFactoryImpl (line 57) | class ElementwiseUnaryFactoryImpl : public ElementwiseUnaryFactory { method ElementwiseUnaryFactoryImpl (line 60) | ElementwiseUnaryFactoryImpl() = default; method New (line 63) | std::unique_ptr New(UnaryOp unary_op, DataType... method New (line 68) | std::unique_ptr New(UnaryOp unary_op, DataType... method New (line 73) | std::unique_ptr New(UnaryOp unary_op, DataType... FILE: oneflow/core/ep/cpu/primitive/fill.cpp type oneflow (line 20) | namespace oneflow { type ep (line 22) | namespace ep { type primitive (line 23) | namespace primitive { function T (line 28) | T GetValue(Scalar value) { function float16 (line 33) | float16 GetValue(Scalar value) { function bfloat16 (line 38) | bfloat16 GetValue(Scalar value) { class FillImpl (line 43) | class FillImpl : public Fill { method FillImpl (line 46) | FillImpl() = default; method Launch (line 49) | void Launch(Stream* stream, void* dst, Scalar value, size_t coun... function NewFill (line 55) | std::unique_ptr NewFill() { class FillFactoryImpl (line 59) | class FillFactoryImpl : public FillFactory { method FillFactoryImpl (line 62) | FillFactoryImpl() = default; method New (line 65) | std::unique_ptr New(DataType data_type) override { FILE: oneflow/core/ep/cpu/primitive/memcpy.cpp type oneflow (line 18) | namespace oneflow { type ep (line 20) | namespace ep { type primitive (line 21) | namespace primitive { class MemcpyImpl (line 25) | class MemcpyImpl : public Memcpy { method MemcpyImpl (line 28) | MemcpyImpl() = default; method Launch (line 31) | void Launch(Stream* stream, void* dst, const void* src, size_t c... class MemcpyFactoryImpl (line 37) | class MemcpyFactoryImpl : public MemcpyFactory { method MemcpyFactoryImpl (line 40) | MemcpyFactoryImpl() = default; method New (line 43) | std::unique_ptr New(MemcpyKind kind) override { return s... FILE: oneflow/core/ep/cpu/primitive/memset.cpp type oneflow (line 18) | namespace oneflow { type ep (line 20) | namespace ep { type primitive (line 21) | namespace primitive { class MemsetImpl (line 25) | class MemsetImpl : public Memset { method MemsetImpl (line 28) | MemsetImpl() = default; method Launch (line 31) | void Launch(Stream* stream, void* ptr, int value, size_t count) { class MemsetFactoryImpl (line 36) | class MemsetFactoryImpl : public MemsetFactory { method MemsetFactoryImpl (line 39) | MemsetFactoryImpl() = default; method New (line 42) | std::unique_ptr New() override { return std::make_unique... FILE: oneflow/core/ep/cpu/primitive/permute.cpp type oneflow (line 22) | namespace oneflow { type ep (line 24) | namespace ep { type primitive (line 25) | namespace primitive { type permute (line 27) | namespace permute { type internal (line 29) | namespace internal { function PermuteKernel (line 34) | void PermuteKernel(PermuteKernelParams pa... function LaunchKernel (line 51) | void LaunchKernel(Stream* stream, const int64_t* src_dims, con... class PermuteImpl (line 57) | class PermuteImpl : public Permute { method PermuteImpl (line 60) | PermuteImpl() = default; method Launch (line 64) | void Launch(Stream* stream, DataType data_type, size_t num_d... class OneDnnPermuteImpl (line 74) | class OneDnnPermuteImpl : public Permute { method OneDnnPermuteImpl (line 77) | OneDnnPermuteImpl() = default; method Launch (line 81) | void Launch(Stream* stream, DataType data_type, size_t num_d... class PermuteFactoryImpl (line 133) | class PermuteFactoryImpl : public PermuteFactory { method PermuteFactoryImpl (line 136) | PermuteFactoryImpl() = default; method New (line 139) | std::unique_ptr New(size_t max_num_dims) override { FILE: oneflow/core/ep/cpu/primitive/softmax.cpp type oneflow (line 24) | namespace oneflow { type ep (line 26) | namespace ep { type primitive (line 27) | namespace primitive { type Algorithm (line 31) | enum class Algorithm { function SoftmaxCpu (line 37) | void SoftmaxCpu(size_t rows, size_t cols, const T* x, T* y) { class SoftmaxImpl (line 69) | class SoftmaxImpl : public SoftmaxBase { method SoftmaxImpl (line 72) | SoftmaxImpl() = default; method Launch (line 75) | void Launch(Stream* stream, size_t rows, size_t cols, const void... function SoftmaxOneDnn (line 83) | void SoftmaxOneDnn(Stream* stream, size_t rows, size_t cols, const... class OneDnnSoftmaxImpl (line 101) | class OneDnnSoftmaxImpl function NewOneDnnSoftmax (line 122) | std::unique_ptr NewOneDnnSoftmax() { function NewSoftmax (line 129) | std::unique_ptr NewSoftmax() { class GenericSoftmaxFactoryImpl (line 134) | class GenericSoftmaxFactoryImpl : public FactoryBase { method GenericSoftmaxFactoryImpl (line 137) | GenericSoftmaxFactoryImpl() = default; method New (line 140) | std::unique_ptr New(DataType data_type) override { FILE: oneflow/core/ep/cpu/primitive/softmax_backward.cpp type oneflow (line 24) | namespace oneflow { type ep (line 26) | namespace ep { type primitive (line 27) | namespace primitive { type Algorithm (line 31) | enum class Algorithm { function SoftmaxBackwardCpu (line 37) | void SoftmaxBackwardCpu(size_t rows, size_t cols, const T* y, cons... class SoftmaxBackwardImpl (line 66) | class SoftmaxBackwardImpl : public SoftmaxBackwardBase { method SoftmaxBackwardImpl (line 69) | SoftmaxBackwardImpl() = default; method Launch (line 72) | void Launch(Stream* stream, size_t rows, size_t cols, const void... function SoftmaxBackwardOneDnn (line 82) | void SoftmaxBackwardOneDnn(Stream* stream, size_t rows, size_t col... class OneDnnSoftmaxBackwardImpl (line 111) | class OneDnnSoftmaxBackwardImpl function NewOneDnnSoftmaxBackward (line 137) | std::unique_ptr NewOneDnnSoftmaxBackward() { function NewSoftmaxBackward (line 145) | std::unique_ptr NewSoftmaxBackward() { class GenericSoftmaxBackwardFactoryImpl (line 151) | class GenericSoftmaxBackwardFactoryImpl : public BackwardFactoryBa... method GenericSoftmaxBackwardFactoryImpl (line 154) | GenericSoftmaxBackwardFactoryImpl() = default; method New (line 157) | std::unique_ptr New(DataType data_type) ove... FILE: oneflow/core/ep/cpu/primitive/tensor_fill.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { type primitive (line 22) | namespace primitive { class TensorFillImpl (line 27) | class TensorFillImpl : public TensorFill { method TensorFillImpl (line 30) | TensorFillImpl() = default; method Launch (line 33) | void Launch(Stream* stream, const void* src, void* dst, size_t c... function NewTensorFill (line 40) | std::unique_ptr NewTensorFill() { class TensorFillFactoryImpl (line 44) | class TensorFillFactoryImpl : public TensorFillFactory { method TensorFillFactoryImpl (line 47) | TensorFillFactoryImpl() = default; method New (line 50) | std::unique_ptr New(DataType data_type) override { FILE: oneflow/core/ep/cpu/primitive/unary_functor.h function namespace (line 20) | namespace oneflow { type UnaryFunctor (line 98) | struct UnaryFunctor type UnaryFunctor (line 105) | struct UnaryFunctor function OF_DEVICE_FUNC (line 115) | OF_DEVICE_FUNC bool operator()(Src src) const { return std::isfinite(src... function OF_DEVICE_FUNC (line 120) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 121) | OF_DEVICE_FUNC Dst operator()(Src src) const { return static_cast(s... function OF_DEVICE_FUNC (line 126) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 128) | OF_DEVICE_FUNC Dst operator()(Src src) const { type UnaryFunctor (line 134) | struct UnaryFunctor function OF_DEVICE_FUNC (line 135) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 137) | OF_DEVICE_FUNC float operator()(float src) const { type UnaryFunctor (line 200) | struct UnaryFunctor function OF_DEVICE_FUNC (line 201) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 203) | OF_DEVICE_FUNC double operator()(double src) const { type UnaryFunctor (line 267) | struct UnaryFunctor function OF_DEVICE_FUNC (line 268) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 270) | OF_DEVICE_FUNC double operator()(double x) const { type UnaryFunctor (line 292) | struct UnaryFunctor function OF_DEVICE_FUNC (line 293) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 295) | OF_DEVICE_FUNC float operator()(float x) const { type UnaryFunctor (line 317) | struct UnaryFunctor function OF_DEVICE_FUNC (line 318) | OF_DEVICE_FUNC UnaryFunctor(Scalar attr0, Scalar attr1) {} function OF_DEVICE_FUNC (line 320) | OF_DEVICE_FUNC bfloat16 operator()(bfloat16 src) const { return std::abs... type UnaryFunctor (line 388) | struct UnaryFunctor type UnaryFunctor (line 395) | struct UnaryFunctor type UnaryFunctor (line 403) | struct UnaryFunctor type UnaryFunctor (line 411) | struct UnaryFunctor FILE: oneflow/core/ep/cpu/primitive/where.cpp type oneflow (line 21) | namespace oneflow { type ep (line 22) | namespace ep { type primitive (line 23) | namespace primitive { function BroadcastElementwiseWhereKernel (line 29) | void BroadcastElementwiseWhereKernel(CpuStream* cpu_stream, function ScalarWhereKernel (line 75) | void ScalarWhereKernel(const CondT* cond, const T* x, const T* y, ... function LaunchKernel (line 82) | void LaunchKernel(Stream* stream, const int64_t* cond_dims, const ... function LaunchScalarKernel (line 108) | void LaunchScalarKernel(Stream* stream, const CondT* cond, const T... class WhereImpl (line 113) | class WhereImpl : public Where { method WhereImpl (line 116) | explicit WhereImpl() = default; method Launch (line 119) | void Launch(Stream* stream, size_t num_cond_dims, const int64_t*... class WhereFactoryImpl (line 136) | class WhereFactoryImpl : public WhereFactory { method WhereFactoryImpl (line 139) | WhereFactoryImpl() = default; method New (line 142) | std::unique_ptr New(DataType cond_type, DataType data_typ... FILE: oneflow/core/ep/cuda/cuda_device.cpp type oneflow (line 30) | namespace oneflow { type ep (line 32) | namespace ep { function CreateConstBuffer (line 39) | void CreateConstBuffer(void** buf, T value, size_t n) { function Stream (line 129) | Stream* CudaDevice::CreateStream() { function cudaMemPool_t (line 204) | cudaMemPool_t CudaDevice::mem_pool() { return mem_pool_; } function cudaDeviceProp (line 207) | const cudaDeviceProp& CudaDevice::properties() const { return proper... FILE: oneflow/core/ep/cuda/cuda_device.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/ep/cuda/cuda_device_manager.cpp type oneflow (line 22) | namespace oneflow { type ep (line 24) | namespace ep { function DeviceManagerRegistry (line 29) | DeviceManagerRegistry* CudaDeviceManager::registry() const { return ... FILE: oneflow/core/ep/cuda/cuda_device_manager.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/cuda/cuda_device_manager_factory.cpp type oneflow (line 26) | namespace oneflow { type ep (line 28) | namespace ep { function GetCudaVersionString (line 32) | std::string GetCudaVersionString(int version) { function GetCudnnVersion (line 36) | bool GetCudnnVersion(libraryPropertyType type, int* version) { function GetCudnnVersionString (line 46) | bool GetCudnnVersionString(std::string* version) { function CudaDumpVersionInfo (line 58) | void CudaDumpVersionInfo() { class CudaDeviceManagerFactory (line 93) | class CudaDeviceManagerFactory : public DeviceManagerFactory { method CudaDeviceManagerFactory (line 96) | CudaDeviceManagerFactory() = default; method NewDeviceManager (line 99) | std::unique_ptr NewDeviceManager(DeviceManagerRegis... method DeviceType (line 103) | DeviceType device_type() const override { return DeviceType::kCUDA; } method device_type_name (line 105) | std::string device_type_name() const override { return "cuda"; } method DumpVersionInfo (line 107) | void DumpVersionInfo() const override { CudaDumpVersionInfo(); } FILE: oneflow/core/ep/cuda/cuda_event.cpp type oneflow (line 20) | namespace oneflow { type ep (line 22) | namespace ep { function cudaEvent_t (line 50) | cudaEvent_t CudaEvent::cuda_event() { return cuda_event_; } FILE: oneflow/core/ep/cuda/cuda_event.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/ep/cuda/cuda_matmul_mode.cpp type oneflow (line 19) | namespace oneflow { type ep (line 21) | namespace ep { FILE: oneflow/core/ep/cuda/cuda_matmul_mode.h function namespace (line 20) | namespace oneflow { FILE: oneflow/core/ep/cuda/cuda_random_generator.cpp type oneflow (line 26) | namespace oneflow { type ep (line 27) | namespace ep { function GetThreadNum (line 31) | int GetThreadNum(const cudaDeviceProp& prop) { FILE: oneflow/core/ep/cuda/cuda_random_generator.h function namespace (line 29) | namespace oneflow { FILE: oneflow/core/ep/cuda/cuda_stream.cpp type oneflow (line 26) | namespace oneflow { type ep (line 28) | namespace ep { function SetAffinityByDevice (line 34) | void SetAffinityByDevice(int dev_id) { function CheckVersionCompatibility (line 45) | void CheckVersionCompatibility(int compiletime_major, int compiletim... function CheckCudaRuntimeVersion (line 54) | void CheckCudaRuntimeVersion() { function CheckCublasVersion (line 68) | void CheckCublasVersion(cublasHandle_t handle) { function CheckCudnnVersion (line 89) | void CheckCudnnVersion() { function DeviceType (line 217) | DeviceType CudaStream::device_type() const { return DeviceType::kCUD... function CudaDevice (line 219) | CudaDevice* CudaStream::device() const { return device_; } function cudaStream_t (line 277) | cudaStream_t CudaStream::cuda_stream() const { return cuda_stream_; } function cublasHandle_t (line 279) | cublasHandle_t CudaStream::cublas_handle() const { return cublas_han... function cusolverDnHandle_t (line 282) | cusolverDnHandle_t CudaStream::cusolver_dn_handle() const { return c... function cublasLtHandle_t (line 286) | cublasLtHandle_t CudaStream::cublas_lt_handle() const { return cubla... function cudnnHandle_t (line 293) | cudnnHandle_t CudaStream::cudnn_handle() const { return cudnn_handle... function cudaDeviceProp (line 295) | const cudaDeviceProp& CudaStream::device_properties() const { return... FILE: oneflow/core/ep/cuda/cuda_stream.h function namespace (line 33) | namespace oneflow { FILE: oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp type oneflow (line 27) | namespace oneflow { type ep (line 29) | namespace ep { type primitive (line 30) | namespace primitive { type broadcast_matmul (line 32) | namespace broadcast_matmul { type internal (line 34) | namespace internal { function OptCudaDataType (line 40) | Optional OptCudaDataType(DataType data_type) { function cudaDataType_t (line 54) | cudaDataType_t GetCudaDataType(DataType data_type) { function CublasScalarParameter (line 68) | CublasScalarParameter GetCublasScalarParameter(Scalar scalar, ... function cudaDataType_t (line 84) | cudaDataType_t GetCublasScalarType(DataType data_type) { function cublasComputeType_t (line 94) | cublasComputeType_t GetComputeType(DataType data_type, CudaStr... function LaunchBroadcastMatmul (line 129) | void LaunchBroadcastMatmul(Stream* stream, DataType data_type,... class BroadcastMatmulFactoryImpl (line 228) | class BroadcastMatmulFactoryImpl : public BroadcastMatmulFacto... method BroadcastMatmulFactoryImpl (line 231) | BroadcastMatmulFactoryImpl() = default; method New (line 234) | std::unique_ptr New(DataType data_type, Bla... FILE: oneflow/core/ep/cuda/primitive/memcpy.cpp type oneflow (line 22) | namespace oneflow { type ep (line 24) | namespace ep { type primitive (line 25) | namespace primitive { class MemcpyImpl (line 29) | class MemcpyImpl : public Memcpy { method MemcpyImpl (line 32) | MemcpyImpl() = default; method Launch (line 35) | void Launch(Stream* stream, void* dst, const void* src, size_t c... class MemcpyFactoryImpl (line 42) | class MemcpyFactoryImpl : public MemcpyFactory { method MemcpyFactoryImpl (line 45) | MemcpyFactoryImpl() = default; method New (line 48) | std::unique_ptr New(MemcpyKind kind) override { FILE: oneflow/core/ep/cuda/primitive/memset.cpp type oneflow (line 22) | namespace oneflow { type ep (line 24) | namespace ep { type primitive (line 25) | namespace primitive { class MemsetImpl (line 29) | class MemsetImpl : public Memset { method MemsetImpl (line 32) | MemsetImpl() = default; method Launch (line 35) | void Launch(Stream* stream, void* ptr, int value, size_t count) ... class MemsetFactoryImpl (line 41) | class MemsetFactoryImpl : public MemsetFactory { method MemsetFactoryImpl (line 44) | MemsetFactoryImpl() = default; method New (line 47) | std::unique_ptr New() override { return std::unique_ptr<... FILE: oneflow/core/ep/include/active_device_guard.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/include/allocation_options.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/include/device.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/ep/include/device_manager.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/ep/include/device_manager_factory.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/device_manager_registry.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ep/include/event.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/add.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/batch_matmul.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/binary_op.h function namespace (line 23) | namespace ep { FILE: oneflow/core/ep/include/primitive/blas.h function namespace (line 21) | namespace ep { FILE: oneflow/core/ep/include/primitive/broadcast_elementwise_binary.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/broadcast_elementwise_unary.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/broadcast_matmul.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/cast.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/constant_pad.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/copy_nd.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/elementwise_unary.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/fast_integer_math.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/fill.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/log_softmax.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/log_softmax_backward.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/matmul.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/memcpy.h function namespace (line 23) | namespace ep { FILE: oneflow/core/ep/include/primitive/memset.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/one_hot.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/permute.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/primitive.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/softmax.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/softmax_backward.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/tensor_fill.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/primitive/unary_op.h function namespace (line 21) | namespace ep { FILE: oneflow/core/ep/include/primitive/where.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ep/include/random_generator.h function namespace (line 22) | namespace ep { FILE: oneflow/core/ep/include/stream.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ep/test/primitive/add_test.cpp type oneflow (line 23) | namespace oneflow { type ep (line 25) | namespace ep { type primitive (line 27) | namespace primitive { type test (line 29) | namespace test { function TestAdd (line 34) | void TestAdd(DeviceManagerRegistry* registry, const std::set(const Eigen::half& value) { function TestElementwiseBroadcastBinary (line 46) | void TestElementwiseBroadcastBinary(DeviceManagerRegistry* regis... function TestElementwiseBroadcastBinary (line 194) | void TestElementwiseBroadcastBinary(DeviceManagerRegistry* regis... function TestComputeBinary (line 207) | void TestComputeBinary(DeviceManagerRegistry* registry, const st... function TestLogicalBinary (line 225) | void TestLogicalBinary(DeviceManagerRegistry* registry, const st... function TEST_F (line 244) | TEST_F(PrimitiveTest, TestBinary) { FILE: oneflow/core/ep/test/primitive/broadcast_matmul_test.cpp type oneflow (line 23) | namespace oneflow { type ep (line 25) | namespace ep { type primitive (line 27) | namespace primitive { type test (line 29) | namespace test { function TestBroadcastMatmul (line 34) | void TestBroadcastMatmul(DeviceManagerRegistry* registry, const ... function TestBroadcastMatmul (line 143) | void TestBroadcastMatmul(DeviceManagerRegistry* registry, const ... function TestBroadcastMatmul (line 160) | void TestBroadcastMatmul(DeviceManagerRegistry* registry, const ... function TestBroadcastMatmul (line 169) | void TestBroadcastMatmul(DeviceManagerRegistry* registry, function TEST_F (line 177) | TEST_F(PrimitiveTest, TestBroadcastMatmul) { FILE: oneflow/core/ep/test/primitive/cast_test.cpp type oneflow (line 23) | namespace oneflow { type ep (line 25) | namespace ep { type primitive (line 27) | namespace primitive { type test (line 29) | namespace test { function TestCast (line 34) | void TestCast(DeviceManagerRegistry* registry, const std::set(std::tanh(sr... type LogicalNotFunctor (line 53) | struct LogicalNotFunctor { method Dst (line 54) | Dst operator()(Src src) { return static_cast(!src); } function EigenElementwise (line 58) | void EigenElementwise(FunctorT functor, Src* src, Dst* dst, cons... function TestElementwise (line 64) | void TestElementwise(DeviceManagerRegistry* registry, const std:... function TEST_F (line 108) | TEST_F(PrimitiveTest, TestElementwisePrimitive) { FILE: oneflow/core/ep/test/primitive/fill_test.cpp type oneflow (line 31) | namespace oneflow { type ep (line 33) | namespace ep { type primitive (line 35) | namespace primitive { type test (line 37) | namespace test { function TestFill (line 42) | void TestFill(DeviceManagerRegistry* registry, const std::set (line 29) | struct GetDataType : std::integral_constant& devices, size_t num_c... function TestWhere (line 157) | void TestWhere(DeviceManagerRegistry* registry, const std::set (line 174) | struct random { type random::value>> (line 183) | struct random::value>> { method T (line 184) | T operator()() { type random (line 192) | struct random { type random::value>> (line 201) | struct random::val... method T (line 202) | T operator()() { function TestScalarWhere (line 210) | void TestScalarWhere(DeviceManagerRegistry* registry, const std:... function TEST_F (line 257) | TEST_F(PrimitiveTest, TestWhere) { FILE: oneflow/core/ep/test/test_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/arg_tuple.cpp type oneflow (line 20) | namespace oneflow { function GetPair (line 24) | std::pair GetPair(const std::string& bn) { function InitArgName2BnIndex2TensorTupleIndex (line 31) | void InitArgName2BnIndex2TensorTupleIndex( FILE: oneflow/core/framework/arg_tuple.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/framework/attr_map.cpp type oneflow (line 25) | namespace oneflow { function AttrMap (line 77) | AttrMap& AttrMap::operator=(const AttrMap& other) { function AttrMap (line 152) | AttrMap MakeAttrMapFromUserOpConf(const UserOpConf& user_conf) { retur... FILE: oneflow/core/framework/attr_map.h function namespace (line 22) | namespace oneflow { function namespace (line 129) | namespace std { FILE: oneflow/core/framework/attr_map_test.cpp type oneflow (line 21) | namespace oneflow { type test (line 22) | namespace test { function TEST (line 24) | TEST(AttrMap, basic) { function TEST (line 55) | TEST(AttrMap, hash_value) { function TEST (line 64) | TEST(AttrMap, hash_map) { FILE: oneflow/core/framework/attr_value.cpp type oneflow (line 18) | namespace oneflow { function T (line 21) | const T& AttrValueCast(const user_op::AttrVal& attr_val) { function CastAttrValue (line 27) | std::shared_ptr CastAttrValue(const T& attr_val) { function CastAttrValue (line 32) | std::shared_ptr CastAttrValue(const T* attr_val) { function HashTypedAttrVal (line 37) | size_t HashTypedAttrVal(const T& val) { FILE: oneflow/core/framework/attr_value.h function namespace (line 30) | namespace oneflow { function T (line 161) | const T& val() const override { return *val_; } FILE: oneflow/core/framework/attr_value_accessor.cpp type oneflow (line 26) | namespace oneflow { type user_op (line 28) | namespace user_op { function Shape (line 53) | Shape AttrValueAccessor::Attr(const AttrValue& val) { function Stride (line 63) | Stride AttrValueAccessor::Attr(const AttrValue& val) { function FOR_RANGE (line 148) | FOR_RANGE(int32_t, i, 0, cpp_val.size()) { function FOR_RANGE (line 163) | FOR_RANGE(int32_t, i, 0, cpp_val.size()) { function MakeCppAttrValueFromProtoAttrValue (line 191) | Maybe MakeCppAttrValueFromProtoAttrValue(const ProtoT& attr... FILE: oneflow/core/framework/attr_value_accessor.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/auto_random_generator.cpp type oneflow (line 25) | namespace oneflow { type one (line 26) | namespace one { type AutoGeneratorState (line 28) | struct AutoGeneratorState { FILE: oneflow/core/framework/auto_random_generator.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/framework/autocast.cpp type oneflow (line 22) | namespace oneflow { type autocast (line 23) | namespace autocast { function DeviceType (line 31) | DeviceType* autocast_device_type() { function get_lower_precision_fp_from_device_type (line 52) | inline Symbol get_lower_precision_fp_from_device_type(DeviceT... function is_enabled (line 74) | bool is_enabled() { return *autocast_enabled(); } function set_enabled (line 75) | void set_enabled(bool enabled) { *autocast_enabled() = enabled; } function DeviceType (line 77) | DeviceType get_autocast_device_type() { return *autocast_device_type... function set_autocast_device_type (line 78) | void set_autocast_device_type(DeviceType device_type) { *autocast_de... function get_autocast_dtype (line 80) | Symbol get_autocast_dtype() { return *autocast_dtype(); } function get_autocast_cpu_dtype (line 81) | Symbol get_autocast_cpu_dtype() { return *autocast_cpu_dtype(... function get_autocast_gpu_dtype (line 82) | Symbol get_autocast_gpu_dtype() { return *autocast_gpu_dtype(... function set_autocast_dtype (line 84) | void set_autocast_dtype(Symbol dtype) { *autocast_dtype() = d... function set_autocast_cpu_dtype (line 85) | void set_autocast_cpu_dtype(Symbol dtype) { *autocast_cpu_dty... function set_autocast_gpu_dtype (line 86) | void set_autocast_gpu_dtype(Symbol dtype) { *autocast_gpu_dty... function is_autocast_cache_enabled (line 88) | bool is_autocast_cache_enabled() { return *cache_enabled(); } function set_autocast_cache_enabled (line 89) | void set_autocast_cache_enabled(bool enabled) { *cache_enabled() = e... function cached_cast (line 91) | Maybe cached_cast(const std::shared_ptr& t... function clear_cache (line 120) | void clear_cache() { cached_casts()->clear(); } function AutoCastColor (line 122) | AutoCastColor AutoCastMeta::autocast_color() const { return autocast... function MakeAutoCastMeta (line 163) | std::shared_ptr MakeAutoCastMeta( FILE: oneflow/core/framework/autocast.h function namespace (line 24) | namespace autocast { FILE: oneflow/core/framework/compute_complexity_fn_context.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/config_def.cpp type oneflow (line 20) | namespace oneflow { function ConfigDef (line 25) | ConfigDef* MutGlobalConfigDef() { function AttrValue (line 31) | AttrValue* AddAttrDef(const std::string& name, const std::string& desc... function ConfigDef (line 42) | const ConfigDef& GlobalEnvConfigDef() { return *MutGlobalConfigDef (line 84) | struct ConfigDefBuidler type ConfigDefBuidler (line 85) | struct ConfigDefBuidler type ConfigDefBuidler (line 86) | struct ConfigDefBuidler type ConfigDefBuidler (line 87) | struct ConfigDefBuidler FILE: oneflow/core/framework/config_def.h function ConfigDefBuidler (line 26) | ConfigDefBuidler final { FILE: oneflow/core/framework/consistency_check.cpp type oneflow (line 29) | namespace oneflow { type FlatMetaInfoConsistency (line 33) | struct FlatMetaInfoConsistency class CheckMetaInfoConsistencyAsyncTransportCtx (line 35) | class CheckMetaInfoConsistencyAsyncTransportCtx : public AsyncTranspor... method CheckMetaInfoConsistencyAsyncTransportCtx (line 37) | CheckMetaInfoConsistencyAsyncTransportCtx(const TransportToken& tran... function New (line 71) | static Maybe New(const Symbol& ... function Check (line 79) | Maybe Check(const Symbol& placement, function DataConsistencyCheck (line 153) | Maybe DataConsistencyCheck(const void* buffer_ptr, size_t buffer_s... function MetaInfoConsistencyCheckUtil (line 190) | Maybe MetaInfoConsistencyCheckUtil(const Symbol& pla... function IsMetaInfoConsistencyCheckDisable (line 210) | inline bool IsMetaInfoConsistencyCheckDisable() { function MetaInfoConsistencyCheck (line 226) | Maybe MetaInfoConsistencyCheck(const Symbol& placement, function MetaInfoConsistencyCheck (line 236) | Maybe MetaInfoConsistencyCheck(const Symbol& placement, function MetaInfoConsistencyCheck (line 245) | Maybe MetaInfoConsistencyCheck(const Symbol& placement, function MetaInfoConsistencyCheck (line 257) | Maybe MetaInfoConsistencyCheck(const Symbol& placement, FILE: oneflow/core/framework/consistency_check.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/framework/device.cpp type oneflow (line 29) | namespace oneflow { function CheckDeviceType (line 33) | void CheckDeviceType(const std::string& type) { function RawGetPlacement (line 138) | Maybe> RawGetPlacement(const Device& device) { function RawPlacement4Device (line 147) | Maybe> RawPlacement4Device(Symbol device) { function ParseDeviceString (line 157) | Maybe> ParseDeviceString(std::strin... FILE: oneflow/core/framework/device.h function namespace (line 29) | namespace oneflow { function ostream_formatter (line 87) | struct fmt::formatter> : ostream_format... function namespace (line 89) | namespace std { FILE: oneflow/core/framework/dtype.cpp type oneflow (line 25) | namespace oneflow { function GetDataTypeBytes (line 30) | std::size_t GetDataTypeBytes() { class DTypeMeta (line 39) | class DTypeMeta final { method DTypeMeta (line 41) | DTypeMeta(const std::string& name, bool is_signed, bool is_integer, ... method DTypeMeta (line 48) | DTypeMeta(const DTypeMeta&) = default; method DTypeMeta (line 49) | DTypeMeta(DTypeMeta&) = default; method is_signed (line 53) | bool is_signed() const { return is_signed_; } method is_integer (line 54) | bool is_integer() const { return is_integer_; } method is_floating_point (line 55) | bool is_floating_point() const { return is_floating_point_; } method is_complex (line 56) | bool is_complex() const { return is_complex_; } function DTypeMeta4DataType (line 66) | Maybe DTypeMeta4DataType(DataType data_type) { function data_type2dtype (line 98) | static HashMap> data_type2dtype{ function promoteTypes (line 166) | Symbol promoteTypes(const Symbol a, const Symbol ... function SetDefaultDType (line 264) | Maybe SetDefaultDType(const Symbol& dtype) { function GetDefaultDType (line 272) | Symbol GetDefaultDType() { FILE: oneflow/core/framework/dtype.h function explicit (line 54) | explicit DType(DataType data_type) : data_type_(data_type) {} function operator (line 57) | bool operator==(const DType& other) const { return this->data_type() == ... function is_signed (line 60) | bool is_signed() const; FILE: oneflow/core/framework/get_nd_sbp_signature_list_context.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/framework/global_param_grad_sync_mode.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/framework/global_param_grad_sync_mode.h function namespace (line 20) | namespace oneflow { FILE: oneflow/core/framework/global_tensor_infer_cache.cpp type oneflow (line 26) | namespace oneflow { type one (line 27) | namespace one { function OptionalEqual (line 31) | bool OptionalEqual(const Optional>& lhs, const Optiona... function MakeOp (line 155) | Maybe MakeOp(const UserOpExpr& user_op_expr, const AttrMap... function CheckInputParallelDescIdentical (line 163) | Maybe CheckInputParallelDescIdentical(const GlobalTensorMetaIn... function CheckIsDeviceSupportedByOp (line 191) | Maybe CheckIsDeviceSupportedByOp(const ParallelDesc& parallel_... class UserOpExprDeviceAndStreamInferContext (line 197) | class UserOpExprDeviceAndStreamInferContext final : public user_op::... method UserOpExprDeviceAndStreamInferContext (line 199) | UserOpExprDeviceAndStreamInferContext(const UserOpExpr* user_op_expr, method InputTensorDevice4ArgNameAndIndex (line 229) | Symbol InputTensorDevice4ArgNameAndIndex(const std::string... FILE: oneflow/core/framework/global_tensor_infer_cache.h function class (line 39) | class InputGlobalTensorMeta final { function class (line 67) | class GlobalTensorMetaInferArgs final { function namespace (line 150) | namespace std { function namespace (line 175) | namespace oneflow { FILE: oneflow/core/framework/id_util.cpp type oneflow (line 19) | namespace oneflow { function UniqueStr (line 21) | Maybe UniqueStr(const std::string& prefix) { return prefi... FILE: oneflow/core/framework/id_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/infer_nd_sbp_fn_context.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/infer_output_blob_time_shape_fn_context.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/infer_util.cpp type oneflow (line 25) | namespace oneflow { type user_op (line 27) | namespace user_op { FILE: oneflow/core/framework/infer_util.h function namespace (line 32) | namespace user_op { function namespace (line 39) | namespace user_op { FILE: oneflow/core/framework/instructions_builder.cpp type oneflow (line 64) | namespace oneflow { function RawGetCriticalSectionStream (line 68) | Maybe> RawGetCriticalSectionStream() { function RawGetLazyJobLauncherStream (line 75) | Maybe> RawGetLazyJobLauncherStream() { function NewSymbolId (line 208) | int64_t NewSymbolId() { type EnableStreamWaitOnReleaseTensor (line 429) | struct EnableStreamWaitOnReleaseTensor final method VisitCompute (line 431) | static bool VisitCompute() { return true; } method VisitHost2Device (line 432) | static bool VisitHost2Device() { return true; } method VisitDevice2Host (line 433) | static bool VisitDevice2Host() { return true; } method VisitCcl (line 434) | static bool VisitCcl() { return false; } method VisitBarrier (line 435) | static bool VisitBarrier() { return false; } method VisitCriticalSection (line 436) | static bool VisitCriticalSection() { return false; } method VisitLazyJobLauncher (line 437) | static bool VisitLazyJobLauncher() { return false; } method VisitPinnedCompute (line 438) | static bool VisitPinnedCompute() { return VisitCompute(); } function SmallSetInsert (line 491) | std::pair::iterator, bool> SmallSetInsert(SmallSe... function ForEachEagerBlobObjectsNeedingSoftSync (line 500) | Maybe ForEachEagerBlobObjectsNeedingSoftSync( function SupportingStreamWait (line 555) | bool SupportingStreamWait(Symbol from_stream, Symbol t... function GetDevice (line 687) | Maybe> GetDevice(const std::shared_ptr> GetDevice(const one::EagerLocalTensorImpl* tenso... function GetAccessStream (line 696) | Maybe> GetAccessStream(const T tensor) { function GetBarrierStream (line 743) | Maybe> GetBarrierStream() { function MutThreadLocalInstruction (line 771) | Maybe MutThreadLocalInstruction(Symbol strea... function SyncAccessSmallMem (line 792) | Maybe SyncAccessSmallMem(char* mem_ptr, size_t bytes, const T te... function SyncReadSmallMem (line 831) | Maybe SyncReadSmallMem(char* mem_ptr, size_t bytes, const T tens... FILE: oneflow/core/framework/instructions_builder.h function namespace (line 34) | namespace oneflow { FILE: oneflow/core/framework/layout.cpp type oneflow (line 21) | namespace oneflow { function layout_type2layout (line 24) | static const HashMap> layout_type2layout{ FILE: oneflow/core/framework/layout.h function LayoutType (line 24) | enum class LayoutType { function explicit (line 34) | explicit Layout(LayoutType layout_type) : layout_type_(layout_type) {} function operator (line 37) | bool operator==(const Layout& other) const { return this->layout_type() ... function std (line 39) | const std::string& name() const; FILE: oneflow/core/framework/load_library.cpp type oneflow (line 20) | namespace oneflow { function LoadLibrary (line 22) | Maybe LoadLibrary(const std::string& lib_path) { FILE: oneflow/core/framework/load_library.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/local_tensor_infer_cache.cpp type oneflow (line 25) | namespace oneflow { type one (line 26) | namespace one { function CheckIsDeviceSupportedByOp (line 30) | Maybe CheckIsDeviceSupportedByOp(const Device& device, const s... function CheckInputDeviceIdentical (line 35) | Maybe CheckInputDeviceIdentical(const LocalTensorMetaInferArgs... class UserOpExprDeviceAndStreamInferContext (line 52) | class UserOpExprDeviceAndStreamInferContext final : public user_op::... method UserOpExprDeviceAndStreamInferContext (line 54) | UserOpExprDeviceAndStreamInferContext(const UserOpExpr* user_op_expr, method InputTensorDevice4ArgNameAndIndex (line 81) | Symbol InputTensorDevice4ArgNameAndIndex(const std::string... function InferDeviceAndStream (line 103) | Maybe> InferDeviceAndStream(const UserOpExpr& user_op... FILE: oneflow/core/framework/local_tensor_infer_cache.h function hash_value (line 50) | size_t hash_value() const; function namespace (line 79) | namespace oneflow { FILE: oneflow/core/framework/multi_client_session_context.cpp type oneflow (line 46) | namespace oneflow { function GetCpuDeviceNum (line 50) | int32_t GetCpuDeviceNum() { return std::thread::hardware_concurrency(); } function IdState (line 209) | IdState MultiClientSessionContext::GetIdState() { FILE: oneflow/core/framework/multi_client_session_context.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/framework/multi_thread.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 21) | namespace user_op { function MultiThreadLoopInOpKernel (line 23) | void MultiThreadLoopInOpKernel(size_t num, std::function> FindOrCreateNdSbpString(Symbol ... function GetDualSbpParallel (line 39) | Maybe GetDualSbpParallel(const SbpParallel& sbp_parallel, SbpPar... function GetDualNdSbp (line 54) | Maybe> GetDualNdSbp(Symbol nd_sbp) { function GetNdSbpStrList (line 68) | Maybe> GetNdSbpStrList(const std::vector> GetNdSbpStrList(Symbol nd_sbp) { function GetDualNdSbpStrList (line 76) | Maybe> GetDualNdSbpStrList(Symbol nd_s... type private_details (line 80) | namespace private_details { function RawGetNdSbp (line 82) | Maybe> RawGetNdSbp(const std::vector>> RawGetSbpList(Symbol ... function RawContainSplitSbp (line 100) | bool RawContainSplitSbp(Symbol nd_sbp) { function RawNdSbpReplacePartialByBroadcast (line 107) | Maybe>> RawNdSbpReplacePartialByBroa... function SbpToString (line 128) | std::string SbpToString(Symbol sbp_sym) { return SbpToStr... function NdSbpToString (line 130) | std::string NdSbpToString(Symbol nd_sbp_sym) { return NdSbpToSt... function SbpToString (line 132) | std::string SbpToString(const SbpParallel& sbp) { function NdSbpToString (line 146) | std::string NdSbpToString(const NdSbp& nd_sbp) { function SetSbpAtAxis (line 157) | Maybe> SetSbpAtAxis(Symbol nd_sbp, Symbol> SetSbpAtAxis(const NdSbp& nd_sbp, const SbpParall... function SbpToNdSbp (line 170) | Maybe> SbpToNdSbp(Symbol sbp) { return SbpT... function SbpToNdSbp (line 172) | Maybe> SbpToNdSbp(const SbpParallel& sbp) { function Is1dSbp (line 179) | bool Is1dSbp(const NdSbp& nd_sbp) { FILE: oneflow/core/framework/nd_sbp.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/framework/nn_graph.cpp type oneflow (line 58) | namespace oneflow { function GetTensorValidInCurRank (line 62) | Maybe GetTensorValidInCurRank(const std::shared_ptr... function GetTensorMetaString (line 75) | Maybe GetTensorMetaString(const std::shared_ptr MakeEagerBlobObjectList(vm::EagerBlobObjectList* blob_list... function FOR_RANGE (line 156) | FOR_RANGE(size_t, i, 0, additional_var_names.size()) { function MultiThreadBroadcastFromMasterToWorkers (line 343) | std::set MultiThreadBroadcastFromMasterToWorkers(size_t w... function MultiThreadPushFromMasterToWorkers (line 377) | std::set MultiThreadPushFromMasterToWorkers(const std::st... function DumpCalculationPassName (line 402) | void DumpCalculationPassName(Job* job) { type GetCompileMethod (line 547) | struct GetCompileMethod final : public CompileModeVisitor RunLazyNNGraph(const one::TensorTuple& inputs, const one::... function SoftSyncNNGraphBuffers (line 915) | Maybe SoftSyncNNGraphBuffers(const one::TensorTuple& buffers, FILE: oneflow/core/framework/nn_graph.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/framework/nn_graph_if.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/op_builder.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { function OpBuilder (line 54) | OpBuilder& OpBuilder::Input(const std::string& input_name) { function OpBuilder (line 57) | OpBuilder& OpBuilder::Input(const std::string& input_name, const int... function OpBuilder (line 76) | OpBuilder& OpBuilder::Output(const std::string& output_name) { function OpBuilder (line 80) | OpBuilder& OpBuilder::Output(const std::string& output_name, const i... function OpBuilder (line 91) | OpBuilder& OpBuilder::Attr(const std::string& attr_name, const AttrV... FILE: oneflow/core/framework/op_builder.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/framework/op_definition.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/framework/op_expr.cpp type oneflow (line 31) | namespace oneflow { type one (line 32) | namespace one { class UserOpExprInferContext (line 169) | class UserOpExprInferContext : public user_op::InferContext { method UserOpExprInferContext (line 171) | UserOpExprInferContext(const UserOpExpr* user_op_expr, const AttrM... method Shape (line 241) | const Shape& InputShape(const std::string& name, int32_t index) co... method Shape (line 248) | const Shape& OutputShape(const std::string& name, int32_t index) c... method SetOutputShape (line 255) | void SetOutputShape(const std::string& name, int32_t index, const ... method Shape (line 264) | const Shape& Shape4ArgNameAndIndex(const std::string& arg_name, in... method SetShape4ArgNameAndIndex (line 268) | void SetShape4ArgNameAndIndex(const std::string& arg_name, int32_t... method Stride (line 273) | const Stride& InputStride(const std::string& name, int32_t index) ... method Stride (line 280) | const Stride& OutputStride(const std::string& name, int32_t index)... method SetOutputStride (line 287) | void SetOutputStride(const std::string& name, int32_t index, const... method Stride (line 296) | const Stride& Stride4ArgNameAndIndex(const std::string& arg_name, ... method SetStride4ArgNameAndIndex (line 300) | void SetStride4ArgNameAndIndex(const std::string& arg_name, int32_... method DataType (line 305) | DataType InputDType(const std::string& arg_name, int32_t index) co... method DataType (line 308) | DataType OutputDType(const std::string& arg_name, int32_t index) c... method SetOutputDType (line 311) | void SetOutputDType(const std::string& arg_name, int32_t index, Da... method DataType (line 314) | DataType Dtype4ArgNameAndIndex(const std::string& arg_name, int32_... method SetDtype4ArgNameAndIndex (line 317) | void SetDtype4ArgNameAndIndex(const std::string& arg_name, int32_t... method MemoryFormat (line 322) | MemoryFormat InputMemoryFormat(const std::string& arg_name, int32_... method MemoryFormat (line 325) | MemoryFormat OutputMemoryFormat(const std::string& arg_name, int32... method SetOutputMemoryFormat (line 328) | void SetOutputMemoryFormat(const std::string& arg_name, int32_t in... method MemoryFormat (line 332) | MemoryFormat MemoryFormat4ArgNameAndIndex(const std::string& arg_n... method SetMemoryFormat4ArgNameAndIndex (line 336) | void SetMemoryFormat4ArgNameAndIndex(const std::string& arg_name, ... method InputIsDynamic (line 341) | bool InputIsDynamic(const std::string& arg_name, int32_t index) co... method OutputIsDynamic (line 344) | bool OutputIsDynamic(const std::string& arg_name, int32_t index) c... method SetOutputIsDynamic (line 347) | void SetOutputIsDynamic(const std::string& arg_name, int32_t index... method IsDynamic4ArgNameAndIndex (line 350) | bool IsDynamic4ArgNameAndIndex(const std::string& arg_name, int32_... method SetIsDynamic4ArgNameAndIndex (line 353) | void SetIsDynamic4ArgNameAndIndex(const std::string& arg_name, int... method has_input (line 369) | bool has_input(const std::string& arg_name, int32_t index) const o... method has_output (line 374) | bool has_output(const std::string& arg_name, int32_t index) const ... method input_size (line 379) | int32_t input_size(const std::string& arg_name) const override { method output_size (line 383) | int32_t output_size(const std::string& arg_name) const override { function Get1DBroadcastNdSbp (line 405) | Symbol Get1DBroadcastNdSbp() { class UserOpExprPhysicalInferContext (line 415) | class UserOpExprPhysicalInferContext final : public UserOpExprInferC... method UserOpExprPhysicalInferContext (line 417) | UserOpExprPhysicalInferContext( method ParallelContext (line 435) | const ParallelContext& parallel_ctx() const override { return para... method ParallelDesc (line 436) | const ParallelDesc& parallel_desc() const override { return *paral... method SbpParallel (line 437) | const SbpParallel& SbpParallel4ArgNameAndIndex(const std::string& ... method NdSbp (line 442) | const NdSbp& NdSbp4ArgNameAndIndex(const std::string& name, int32_... method parallel_num (line 446) | int64_t parallel_num() const override { return 1; } class UserOpExprLogicalInferContext (line 454) | class UserOpExprLogicalInferContext final : public UserOpExprInferCo... method UserOpExprLogicalInferContext (line 456) | UserOpExprLogicalInferContext( method ParallelContext (line 479) | const ParallelContext& parallel_ctx() const override { return para... method ParallelDesc (line 480) | const ParallelDesc& parallel_desc() const override { return *paral... method SbpParallel (line 481) | const SbpParallel& SbpParallel4ArgNameAndIndex(const std::string& ... method NdSbp (line 489) | const NdSbp& NdSbp4ArgNameAndIndex(const std::string& name, int32_... method parallel_num (line 494) | int64_t parallel_num() const override { return parallel_desc_->par... class UserOpExprDeviceAndStreamInferContext (line 501) | class UserOpExprDeviceAndStreamInferContext final : public user_op::... method UserOpExprDeviceAndStreamInferContext (line 503) | UserOpExprDeviceAndStreamInferContext(const UserOpExpr* user_op_ex... method InputTensorDevice4ArgNameAndIndex (line 527) | Symbol InputTensorDevice4ArgNameAndIndex(const std::string... FILE: oneflow/core/framework/op_expr.h function namespace (line 37) | namespace oneflow { FILE: oneflow/core/framework/op_expr_grad_function.cpp type oneflow (line 22) | namespace oneflow { type one (line 23) | namespace one { FILE: oneflow/core/framework/op_expr_grad_function.h function namespace (line 27) | namespace oneflow { function override (line 163) | const override { function override (line 176) | const override { function override (line 186) | const override { function class (line 217) | class OpExprGradClosure { FILE: oneflow/core/framework/op_interpreter.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/framework/op_interpreter/dispatch_frame.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/framework/op_interpreter/dispatch_frame.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/op_interpreter/eager_global_op_interpreter.cpp type oneflow (line 39) | namespace oneflow { type one (line 40) | namespace one { function IsEnvEnableGlobalInputsWithInConsistentPlacement (line 44) | bool IsEnvEnableGlobalInputsWithInConsistentPlacement() { function IsInputsParallelDescIdentical (line 50) | Maybe IsInputsParallelDescIdentical( function MaxRankNumber (line 69) | Maybe MaxRankNumber(Symbol placement) { function MaxRankTensorPlacement (line 78) | Maybe> MaxRankTensorPlacement( function GetParallelDesc (line 105) | Maybe> GetParallelDesc(const TensorTuple& inputs, function GetDynamicOpGlobalFailedDebugString (line 116) | std::string GetDynamicOpGlobalFailedDebugString(const UserOpExpr& us... function IsAllZeroSizeTensorMeta (line 133) | Maybe IsAllZeroSizeTensorMeta(const std::vector CalcBoxingOutput(const std::shared_ptr& input,... function Interpret (line 174) | Maybe Interpret(const UserOpExpr& user_op_expr, const TensorTu... function RawGlobalToGlobal (line 303) | Maybe RawGlobalToGlobal(const GlobalToGlobalOpExpr& op_expr, c... FILE: oneflow/core/framework/op_interpreter/eager_local_op_interpreter.cpp type oneflow (line 48) | namespace oneflow { type one (line 49) | namespace one { function RawGetDefaultCpuDevice (line 53) | Maybe> RawGetDefaultCpuDevice() { return Device::New(... function GetDefaultDevice (line 57) | Maybe> GetDefaultDevice(const TensorTuple& inputs, co... function TensorImpl4Tensor (line 71) | Maybe TensorImpl4Tensor(const std::shared_ptr... function NaiveInterpret (line 78) | Maybe NaiveInterpret(const UserOpExpr& user_op_expr, const Ten... function BuildAndRunLocalCastInstruction (line 196) | static Maybe BuildAndRunLocalCastInstruction(const BuiltinOpEx... function EagerCclBroadcast (line 205) | Maybe EagerCclBroadcast(Symbol parall... function Broadcast (line 220) | Maybe Broadcast(const std::shared_ptr& tensor, int64... function Broadcast (line 239) | Maybe Broadcast(const TensorTuple& inputs, int64_t src_... function GetSyncedTensorIfBroadcast (line 263) | Maybe GetSyncedTensorIfBroadcast(const std::shared_ptr CalcPhysicalShape(Symbol global_tenso... function TryReshapeTensor (line 290) | Maybe TryReshapeTensor(const std::shared_ptr& tensor, function RawLocalToGlobal (line 310) | Maybe RawLocalToGlobal(const LocalToGlobalOpExpr& op_expr, con... function BuildAndRunDistributeSplitOrCloneInstruction (line 404) | static Maybe BuildAndRunDistributeSplitOrCloneInstruction(cons... function BuildAndRunDistributeConcatAndAddInstruction (line 423) | static Maybe BuildAndRunDistributeConcatAndAddInstruction(cons... FILE: oneflow/core/framework/op_interpreter/eager_local_op_interpreter.h function namespace (line 18) | namespace oneflow { FILE: oneflow/core/framework/op_interpreter/lazy_op_interpreter.cpp type oneflow (line 47) | namespace oneflow { type one (line 48) | namespace one { function BuildTensor (line 52) | Maybe BuildTensor(const OpAttribute& op_attribute, const std... function CheckTensorMatchAttr (line 84) | Maybe CheckTensorMatchAttr(const std::shared_ptr& tensor, function GetDeviceTagOfTensor (line 124) | Maybe GetDeviceTagOfTensor(const std::shared_ptr... function GetIsDynamicOfTensor (line 129) | bool GetIsDynamicOfTensor(const std::shared_ptr& tensor) { function GenNdSbpByTensor (line 137) | Maybe GenNdSbpByTensor(NdSbp* nd_sbp, const std::shared_ptr GenVariableOpConfNdSbpStringByTensor(VariableOpConf* var... function GetParallelDescOfTensor (line 166) | Maybe GetParallelDescOfTensor(const std::shared_... function NewScopeWithParallelConfAndCurScope (line 176) | Maybe NewScopeWithParallelConfAndCurScope(const ParallelConf&... function NewScopeWithParallelDescByTensor (line 189) | Maybe NewScopeWithParallelDescByTensor(const std::shared_ptr<... function GetGradAccStep (line 194) | Maybe GetGradAccStep() { function AddFreeEagerTensorToVariableOp (line 205) | Maybe AddFreeEagerTensorToVariableOp(const std::shared_ptr GradAccTryInsertUnpackAfterInput(const std::shared_ptr... function GradAccTryInsertRepeatAfterVar (line 291) | Maybe GradAccTryInsertRepeatAfterVar(const std::shared_ptr GradAccTryInsertPackBeforeOutput(const std::shared_ptr... function GradAccTryInsertRepeatTickBeforeSource (line 332) | Maybe GradAccTryInsertRepeatTickBeforeSource( function LazyInterpreterApplyImplForSourceUserOpExpr (line 621) | Maybe LazyInterpreterApplyImplForSourceUserOpExpr(const UserOp... function LazyInterpreterApplyImplForCopyUserOpExpr (line 705) | Maybe LazyInterpreterApplyImplForCopyUserOpExpr(const UserOpEx... FILE: oneflow/core/framework/op_interpreter/lazy_op_interpreter.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/op_interpreter/op_interpreter.cpp type oneflow (line 28) | namespace oneflow { type one (line 29) | namespace one { FILE: oneflow/core/framework/op_interpreter/op_interpreter_util.cpp type oneflow (line 30) | namespace oneflow { type one (line 31) | namespace one { function BuildEagerInterpreter (line 35) | std::shared_ptr BuildEagerInterpreter(const boo... function BuildLazyInterpreter (line 45) | std::shared_ptr BuildLazyInterpreter() { function ErrorString4Inputs (line 50) | std::string ErrorString4Inputs(const TensorTuple& inputs, const OpEx... function GetInterpreter (line 67) | Maybe GetInterpreter(const TensorTuple& inputs,... FILE: oneflow/core/framework/op_interpreter/op_interpreter_util.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/framework/op_kernel.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 21) | namespace user_op { FILE: oneflow/core/framework/op_kernel.h function virtual (line 102) | virtual DeviceType device_type() const = 0; FILE: oneflow/core/framework/op_kernel_infer_cache.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 22) | namespace user_op { function FOR_RANGE (line 53) | FOR_RANGE(int, i, 0, inputs.size()) { function FOR_RANGE (line 64) | FOR_RANGE(int, i, 0, ctx->outputs().size()) { FILE: oneflow/core/framework/op_kernel_infer_cache.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/framework/ordered_string_list.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/parallel_conf_util.cpp type oneflow (line 20) | namespace oneflow { function MakeParallelConf (line 34) | Maybe MakeParallelConf(const std::string& device_tag, FILE: oneflow/core/framework/parallel_conf_util.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/framework/parallel_conf_util_test.cpp type oneflow (line 21) | namespace oneflow { type test (line 22) | namespace test { function TEST (line 24) | TEST(ParallelConfUtil, MakeParallelConfSuccess) { function TEST (line 35) | TEST(ParallelConfUtil, MakeParallelConfError) { function TEST (line 44) | TEST(ParallelConfUtil, GetDeviceTagAndMachineDeviceIdsAndHierarchy) { FILE: oneflow/core/framework/placed_nd_sbp.cpp type oneflow (line 21) | namespace oneflow { function RawNew (line 25) | Maybe> RawNew(const Symbol& nd_sbp, FILE: oneflow/core/framework/placed_nd_sbp.h function namespace (line 24) | namespace oneflow { function namespace (line 51) | namespace std { FILE: oneflow/core/framework/placement_sbp_util.cpp type oneflow (line 34) | namespace oneflow { type private_details (line 36) | namespace private_details { function GetIndexesFromOffset (line 42) | Maybe GetIndexesFromOffset(const Stride& strides, int64_t offs... function GetOffsetFromIndexes (line 52) | Maybe GetOffsetFromIndexes(const Stride& strides, const IndexV... function GetSelectedIndex2OriginIndex (line 61) | Maybe GetSelectedIndex2OriginIndex( function GetSelectedShape (line 74) | Maybe GetSelectedShape(const Shape& hierarchy_shape, function CalcAxis2IsBroadcast (line 84) | Maybe>> CalcAxis2IsBroadcast(Symbol n... function CalcSelectedSubParallelDesc (line 94) | Maybe> CalcSelectedSubParallelDesc(Symbol> CalcSubParallelDesc4Axis(Symbol> GetSelectedParallelIds(const Shape& hier... function GetBroadcastSubParallelDesc (line 166) | Maybe> GetBroadcastSubParallelDesc(Symbol> MakeNdSbp(const SbpParallel& sbp) { function InitShapeAxis2NdSbpIndexes (line 180) | Maybe InitShapeAxis2NdSbpIndexes( function CheckSplitAxisExpandable (line 194) | Maybe CheckSplitAxisExpandable( function InitShapAxis2ExpandedDim (line 216) | Maybe InitShapAxis2ExpandedDim( function Flatten (line 258) | Maybe Flatten(const std::vector& shape_axis2... function InitOldAxis2NewAxisOffset (line 267) | Maybe InitOldAxis2NewAxisOffset(std::vector* old_axis... function ShiftSplitAxis (line 276) | Maybe> ShiftSplitAxis( function CalcDecomposableEquivalentShapeAndNdSbpPair (line 298) | Maybe, Symbol, Symbol... function CalcDecomposableEquivalent (line 332) | Maybe, Symbol>> CalcD... function InitDstNdSbpAxis2ExclusiveSrcNdSbpAxis (line 349) | Maybe InitDstNdSbpAxis2ExclusiveSrcNdSbpAxis( function MakeExclusiveSrcNdSbpAxis4DstNdSbpAxis (line 372) | Maybe MakeExclusiveSrcNdSbpAxis4DstNdSbpAxis( function IsNdSbpBoxingAcyclic (line 416) | Maybe IsNdSbpBoxingAcyclic( function InitNdSbpValidTransformationAxisSequence (line 432) | Maybe InitNdSbpValidTransformationAxisSequence( function IsNdSbpBoxingAcyclic (line 460) | Maybe IsNdSbpBoxingAcyclic(Symbol src_nd_sbp, Symbol> GetNdSbpValidTransformationAxisSequence(... function GetCyclicBoxingDebugString (line 483) | std::string GetCyclicBoxingDebugString( function GetPhysicalShape (line 506) | Maybe GetPhysicalShape(const Shape& shape, Symbol nd_sbp, function GetSubLogicalShape (line 512) | Maybe GetSubLogicalShape(Symbol tensor... function CalcSubGlobalTensorMeta (line 527) | Maybe> CalcSubGlobalTensorMeta( function ReplaceNdSbpComponent (line 540) | Maybe> ReplaceNdSbpComponent(Symbol nd_sbp, int... function ReplaceNdSbp (line 550) | Maybe> ReplaceNdSbp(Symbol> DecomposeIntoNaiveTran... function CalcBroadcastGroup (line 605) | Maybe>> CalcBroadcast... function RawCheckIsNdSbpBoxingAcyclic (line 672) | Maybe RawCheckIsNdSbpBoxingAcyclic(Symbol in, Symbo... function RawCheckIsNdSbpBoxingAcyclicWithDecompose (line 687) | Maybe RawCheckIsNdSbpBoxingAcyclicWithDecompose(Symbol>> GetBroadcastG... function GetBroadcastGroupWithoutAcrossNode (line 733) | Maybe>> GetBroadcastG... FILE: oneflow/core/framework/placement_sbp_util.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/framework/placement_sbp_util_test.cpp type oneflow (line 25) | namespace oneflow { type test (line 26) | namespace test { type GlobaProcessCtxScope (line 30) | struct GlobaProcessCtxScope final { method GlobaProcessCtxScope (line 31) | GlobaProcessCtxScope(GlobaProcessCtxScope&) = default; method GlobaProcessCtxScope (line 32) | GlobaProcessCtxScope(GlobaProcessCtxScope&&) = default; method GlobaProcessCtxScope (line 33) | GlobaProcessCtxScope& operator=(GlobaProcessCtxScope&) = default; method GlobaProcessCtxScope (line 34) | GlobaProcessCtxScope& operator=(GlobaProcessCtxScope&&) = default; method GlobaProcessCtxScope (line 35) | GlobaProcessCtxScope(int64_t node_size, int64_t world_size) { function InitSbpParallel (line 130) | void InitSbpParallel(SbpParallel* sbp_parallel, const std::string& s... function GetNdSbp (line 146) | Symbol GetNdSbp(Args... sbps) { function MakeGlobalTensorMeta (line 154) | Symbol MakeGlobalTensorMeta(Symbol> ReplacePlacementDeviceTag(Symbol TouchGlobalTensor(const std::shared_ptr& tens... FILE: oneflow/core/framework/placement_utils.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/random_generator.cpp type oneflow (line 33) | namespace oneflow { type one (line 34) | namespace one { function GetNonDeterministicRandom (line 38) | uint64_t GetNonDeterministicRandom() { function CPUSynchronize (line 45) | Maybe CPUSynchronize() { function DefaultGenerator (line 128) | Maybe DefaultGenerator(const std::string& device, int dev... function DefaultAutoGenerator (line 136) | Maybe DefaultAutoGenerator() { function DefaultCPUGenerator (line 144) | Maybe DefaultCPUGenerator() { function DefaultCUDAGenerator (line 149) | Maybe DefaultCUDAGenerator(int device_index) { function MakeAutoGenerator (line 167) | Maybe MakeAutoGenerator() { function MakeCPUGenerator (line 171) | Maybe MakeCPUGenerator() { function MakeCUDAGenerator (line 177) | Maybe MakeCUDAGenerator(int device_index) { function ManualSeedAllCudaGenerator (line 185) | Maybe ManualSeedAllCudaGenerator(uint64_t seed) { function MakeGenerator (line 196) | Maybe MakeGenerator(const std::string& device, int device... function DefaultGenerator (line 211) | Maybe DefaultGenerator(DeviceType device, int device_inde... function MakeGenerator (line 215) | Maybe MakeGenerator(DeviceType device, int device_index) { function ManualSeed (line 219) | Maybe ManualSeed(uint64_t seed) { function ManualSeed (line 225) | Maybe ManualSeed(uint64_t seed, const std::string& device, int... function ManualSeed (line 230) | Maybe ManualSeed(uint64_t seed, DeviceType device, int device_... FILE: oneflow/core/framework/random_generator.h function namespace (line 29) | namespace oneflow { FILE: oneflow/core/framework/rank_group_rpc_util.cpp type oneflow (line 28) | namespace oneflow { function CheckTransportToken (line 30) | Maybe CheckTransportToken(Symbol ra... FILE: oneflow/core/framework/rank_group_rpc_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/saved_tensor_hooks.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/sbp_context.cpp type oneflow (line 21) | namespace oneflow { function SplitImpl (line 25) | inline void SplitImpl(SbpSignature* sbp_sign, const std::string& bn, i... function BroadcastImpl (line 29) | inline void BroadcastImpl(SbpSignature* sbp_sign, const std::string& b... function PartialSumImpl (line 33) | inline void PartialSumImpl(SbpSignature* sbp_sign, const std::string& ... type user_op (line 39) | namespace user_op { function UserOpSbpSignatureBuilder (line 41) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::Split(const Op... function UserOpSbpSignatureBuilder (line 46) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::Split(const st... function UserOpSbpSignatureBuilder (line 52) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::Split( function UserOpSbpSignatureBuilder (line 60) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::Broadcast(cons... function UserOpSbpSignatureBuilder (line 65) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::Broadcast(cons... function UserOpSbpSignatureBuilder (line 70) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::Broadcast( function UserOpSbpSignatureBuilder (line 78) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::PartialSum(con... function UserOpSbpSignatureBuilder (line 83) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::PartialSum( function UserOpSbpSignatureBuilder (line 89) | UserOpSbpSignatureBuilder& UserOpSbpSignatureBuilder::PartialSum( function InferNdSbp4SrcOp (line 118) | Maybe InferNdSbp4SrcOp(user_op::InferNdSbpFnContext* ctx, cons... function SetSrcOpNdSbp (line 152) | Maybe SetSrcOpNdSbp(const NdSbpSignature& nd_sbp_sig, const st... FILE: oneflow/core/framework/sbp_context.h function class (line 29) | class UserOpSbpSignatureBuilder final { function virtual (line 64) | virtual DeviceType device_type() const = 0; FILE: oneflow/core/framework/sbp_infer_util.cpp type oneflow (line 32) | namespace oneflow { function CheckSbpParallel (line 39) | bool CheckSbpParallel(const SbpParallel& sbp_parallel) { function CheckNdSbp (line 45) | bool CheckNdSbp(const NdSbp& nd_sbp) { function Penalty4PartialInConsumer (line 53) | double Penalty4PartialInConsumer(double logical_blob_size, int32_t pro... function Ratio4Sbp (line 66) | int32_t Ratio4Sbp(const NdSbp& nd_sbp, const ParallelDesc& parallel_desc, function ComputCopyCostBetweenTwoSbpParallel (line 75) | Maybe ComputCopyCostBetweenTwoSbpParallel(const SbpParallel& p... function ComputCopyCostBetweenTwoDiffSbpParallel (line 154) | double ComputCopyCostBetweenTwoDiffSbpParallel(const SbpParallel& prod... function ComputCopyCostBetweenTwoNdSbp (line 202) | Maybe ComputCopyCostBetweenTwoNdSbp(const NdSbp& producer_nd_sbp, function ComputeEagerCopyCostBetweenNdSbp (line 248) | Maybe ComputeEagerCopyCostBetweenNdSbp(const NdSbp& producer_s... function GetComputeCopyCostFunc (line 368) | Maybe GetComputeCopyCostFunc() { function ReplaceHierarchy4ParallelDesc (line 377) | void ReplaceHierarchy4ParallelDesc(const ParallelDesc& old_parallel_desc, function CanMergeSplit (line 397) | bool CanMergeSplit(int32_t shape_value, int32_t merged_split_hierarchy... function PartialRatio4Producer (line 408) | int32_t PartialRatio4Producer(const NdSbp& sbp_producer, function BroadcastRatio4Consumer (line 413) | int32_t BroadcastRatio4Consumer(const NdSbp& sbp_consumer, function NdSbpDimReduce (line 418) | void NdSbpDimReduce(const Shape& hierarchy, const NdSbp& nd_sbp, Shape... function NdSbpsDimReduce (line 423) | void NdSbpsDimReduce(const Shape& hierarchy, const std::vector ComputeLazyCopyCostBetweenNdSbp(const NdSbp& producer_sb... function GetValidMaxCopyCost (line 746) | double GetValidMaxCopyCost() { function GetTransferCost (line 752) | double GetTransferCost() { function ResizeNdSbpSignature (line 759) | void ResizeNdSbpSignature(NdSbpSignature& nd_sbp_sig, int32_t size) { function SetNdSbpSignature (line 766) | void SetNdSbpSignature(NdSbpSignature* nd_sbp_signature, const SbpSign... function DfsGetNdSbpSignature (line 774) | void DfsGetNdSbpSignature(NdSbpSignature& nd_sbp_sig, int32_t depth, i... function MesureNdSbp (line 793) | size_t MesureNdSbp(const NdSbp& nd_sbp) { function MesureNdSbpSignature (line 817) | size_t MesureNdSbpSignature(const NdSbpSignature& nd_sbp_sig, const st... function DeduplicateNdSbpSignatureList (line 835) | void DeduplicateNdSbpSignatureList(std::vector* nd_sbp... function Storage4NdSbp (line 852) | double Storage4NdSbp(const NdSbp& nd_sbp, Shape& logical_shape, const ... function FilterNdSbpByLogicalShape (line 882) | Maybe FilterNdSbpByLogicalShape(const NdSbp& nd_sbp, Shape& logi... function ComputeCopyCostBetweenNdSbp (line 887) | Maybe ComputeCopyCostBetweenNdSbp(const NdSbp& producer_sbp_pa... function ComputeCopyCostWithMiddleNodes (line 898) | Maybe ComputeCopyCostWithMiddleNodes(const NdSbp& producer_sbp... function ComputeSbpInferPriority (line 988) | double ComputeSbpInferPriority(const NdSbp& producer_nd_sbp, const NdS... function Cost4GeneralBasicCommunication (line 1046) | double Cost4GeneralBasicCommunication(const NdSbp& producer_sbp_parallel, FILE: oneflow/core/framework/sbp_infer_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/sbp_infer_util_test.cpp type oneflow (line 21) | namespace oneflow { type test (line 22) | namespace test { function ParseNdSbpSignatureFromString (line 26) | bool ParseNdSbpSignatureFromString(const std::string& nd_sbp_signatu... function NdSbpSignature2String (line 90) | std::string NdSbpSignature2String(const NdSbpSignature& nd_sbp_signa... function TestDeduplicateNdSbpSignature (line 111) | void TestDeduplicateNdSbpSignature(const std::vector& n... function TEST (line 147) | TEST(SbpInferUtil, DeduplicateNdSbpSignatureList) { FILE: oneflow/core/framework/scope_util.cpp type oneflow (line 26) | namespace oneflow { function MakeDefaultScope (line 30) | Maybe MakeDefaultScope() { function MakeScope (line 44) | Maybe MakeScope(const JobConfigProto& config_proto, const Devic... function MakeInitialScope (line 62) | Maybe MakeInitialScope(const JobConfigProto& job_conf, Symbol GetCurrentScope() { function InitThreadLocalScopeStack (line 81) | Maybe InitThreadLocalScopeStack(const std::shared_ptr& sc... function ThreadLocalScopeStackPush (line 88) | Maybe ThreadLocalScopeStackPush(const std::shared_ptr& sc... function ThreadLocalScopeStackPop (line 94) | Maybe ThreadLocalScopeStackPop() { class BackwardPassScopeStorage (line 121) | class BackwardPassScopeStorage { method BackwardPassScopeStorage (line 125) | static BackwardPassScopeStorage* Global() { function FindOrCreateBackwardPassScope (line 136) | Maybe FindOrCreateBackwardPassScope(const std::shared_ptr SetDefaultSessionId(int64_t val) { function GetDefaultSessionId (line 40) | Maybe GetDefaultSessionId() { function RegsterSessionId (line 47) | bool RegsterSessionId(int64_t session_id) { function ClearSessionId (line 56) | bool ClearSessionId(int64_t session_id) { FILE: oneflow/core/framework/session_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/shut_down_util.cpp type oneflow (line 18) | namespace oneflow { function IsShuttingDown (line 29) | bool IsShuttingDown() { function SetShuttingDown (line 35) | void SetShuttingDown(bool arg_shutting_down) { FILE: oneflow/core/framework/shut_down_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/stream.cpp type oneflow (line 28) | namespace oneflow { function RawGetDefaultStreamByDevice (line 65) | Maybe> RawGetDefaultStreamByDevice(Symbol devic... function RawGetDefaultStreamByPlacement (line 69) | Maybe> RawGetDefaultStreamByPlacement(Symbol> RawGetAllocatorStream(Symbol stream) { FILE: oneflow/core/framework/stream.h function namespace (line 26) | namespace oneflow { function namespace (line 78) | namespace std { FILE: oneflow/core/framework/stream_allocator_is_pinned.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/stream_get_stream_type_name.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/stream_guard.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/framework/stream_guard.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/framework/stream_is_comm_net_stream.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/stream_mgr.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/framework/stream_mgr.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/framework/stream_need_soft_sync.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/stream_on_independent_thread.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/stream_set.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/core/framework/stream_set.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/stream_support_stream_wait.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/symbol_storage_util.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/core/framework/symbol_storage_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/sync_symbol_global_tensor_meta.cpp type oneflow (line 24) | namespace oneflow { type FlatGlobalTensorMeta (line 26) | struct FlatGlobalTensorMeta final { method New (line 27) | static Maybe New(uint64_t symbol_id, method Init (line 34) | Maybe Init(uint64_t symbol_id, Symbol g... method Check (line 46) | Maybe Check(uint64_t symbol_id, Symbol ... function SyncSymbolGlobalTensorMeta (line 67) | Maybe SyncSymbolGlobalTensorMeta(uint64_t symbol_id, FILE: oneflow/core/framework/sync_symbol_global_tensor_meta.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/sync_symbol_nd_sbp.cpp type oneflow (line 24) | namespace oneflow { function Check (line 54) | Maybe Check(const SbpParallel& sbp_parallel) const { function Check (line 86) | Maybe Check(uint64_t symbol_id, Symbol nd_sbp) const { class FlatNdSbpAsyncTransportCtx (line 102) | class FlatNdSbpAsyncTransportCtx : public AsyncTransportCtx { method FlatNdSbpAsyncTransportCtx (line 104) | FlatNdSbpAsyncTransportCtx(const TransportToken& transport_token, uint... method PrepareSendBufferAndCallback (line 110) | Maybe PrepareSendBufferAndCallback(int64_t rank, void** buffer, ... method PrepareRecvBufferAndCallback (line 120) | Maybe PrepareRecvBufferAndCallback(int64_t rank, void** buffer, ... method Check (line 130) | Maybe Check() const { function SyncSymbolNdSbp (line 146) | Maybe SyncSymbolNdSbp(uint64_t symbol_id, Symbol symbol) { FILE: oneflow/core/framework/sync_symbol_nd_sbp.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/sync_symbol_parallel_desc.cpp type oneflow (line 22) | namespace oneflow { type FlatParallelConf (line 27) | struct FlatParallelConf { method available_size (line 28) | size_t available_size() const { method capacity (line 35) | size_t capacity() const { return sizeof(FlatParallelConf); } method New (line 37) | static Maybe New(uint64_t symbol_id, Symbol Init(uint64_t symbol_id, Symbol parallel_d... method Check (line 57) | Maybe Check(uint64_t symbol_id, Symbol parallel_... function SyncSymbolParallelDesc (line 83) | Maybe SyncSymbolParallelDesc(uint64_t symbol_id, Symbol& grad_fn_node)... function set_grad_fn_output_index (line 280) | void set_grad_fn_output_index(int32_t idx) override { PRINT_BUG_PROMPT_A... function set_is_leaf (line 289) | void set_is_leaf(bool is_leaf) override { PRINT_BUG_PROMPT_AND_ABORT(); } function override (line 290) | const override { function set_autograd_meta (line 297) | void set_autograd_meta(const std::shared_ptr& autograd_met... function override (line 311) | const override { RETURN_ERROR_WITH_BUG_PROMPT(); } function override (line 334) | const override { return grad_fn_node_; } function set_grad_fn_node (line 338) | void set_grad_fn_node(const std::shared_ptr& grad_fn_node)... function set_grad_fn_output_index (line 342) | void set_grad_fn_output_index(int32_t idx) override { grad_fn_output_ind... function virtual (line 365) | virtual Maybe*> mut_device() override { return tensor_->m... function virtual (line 379) | virtual Maybe data() override { return tensor_->detach(); } function virtual (line 387) | virtual void set_grad_fn_node(const std::shared_ptr& grad_... function virtual (line 390) | virtual std::shared_ptr mut_grad_fn_node() override { function virtual (line 394) | virtual Maybe mut_eager_local_tensor_impl() overr... function virtual (line 420) | virtual Maybe set_consumer_nd_sbp_constraint(const Optional set_requires_grad(bool requires_grad) override { function virtual (line 438) | virtual Maybe set_retain_grad(bool retain_grad) override { function virtual (line 441) | virtual Maybe set_acc_grad(const std::shared_ptr& grad) ov... function virtual (line 444) | virtual Maybe mut_acc_grad() override { return tensor_->mut_acc_... function virtual (line 445) | virtual void set_is_leaf(bool is_leaf) override { return tensor_->set_is... function virtual (line 449) | virtual std::shared_ptr mut_autograd_meta() override { function virtual (line 452) | virtual void set_autograd_meta(const std::shared_ptr& auto... function virtual (line 456) | virtual user_op::TensorDesc* mut_tensor_meta() override { return tensor_... function virtual (line 457) | virtual Maybe set_data(const std::shared_ptr& other) overr... function virtual (line 465) | virtual Maybe offload() override { function virtual (line 469) | virtual Maybe load() override { function override (line 473) | const override { return JUST(tensor_->is_offloaded()); } function virtual (line 475) | virtual Maybe AsLocalTensor() override { function virtual (line 482) | virtual Maybe AsGlobalTensor() override { function explicit (line 511) | explicit LocalTensor(const std::shared_ptr& impl) { imp... function override (line 515) | const override { return impl_->shape(); } function override (line 516) | const override { return CHECK_JUST(DType::Get(impl_->dtype())); } function override (line 517) | const override { function override (line 521) | const override { function override (line 530) | const override { function override (line 535) | const override { return impl_->device(); } function override (line 548) | const override { function override (line 551) | const override { function override (line 554) | const override { return impl_->tensor_storage(); } function override (line 555) | const override { return impl_->has_eager_blob_object(); } function override (line 556) | const override { return impl_->stride(); } function override (line 557) | const override { return impl_->storage_offset(); } function override (line 561) | const override { return impl_->acc_grad(); } function override (line 562) | const override { return impl_->current_grad(); } function override (line 568) | const override { return impl_->is_pinned(); } function override (line 570) | const override { return impl_->tensor_meta(); } function set_is_leaf (line 589) | void set_is_leaf(bool is_leaf) override { impl_->set_is_leaf(is_leaf); } function override (line 590) | const override { function set_autograd_meta (line 594) | void set_autograd_meta(const std::shared_ptr& autograd_met... function LocalTensorImpl (line 606) | LocalTensorImpl* mut_impl() { return impl_.get(); } function override (line 617) | const override { return is_offloaded_; } function is_offloaded_ (line 636) | bool is_offloaded_{false}; function explicit (line 643) | explicit GlobalTensor(const std::shared_ptr& impl) { i... function override (line 647) | const override { return impl_->shape(); } function override (line 648) | const override { return CHECK_JUST(DType::Get(impl_->dtype())); } function override (line 649) | const override { return impl_->transport_token(); } function override (line 650) | const override { return impl_->nd_sbp(); } function override (line 651) | const override { return impl_->parallel_desc(); } function override (line 652) | const override { function override (line 666) | const override { function override (line 669) | const override { return impl_->cur_rank_phy_tensor(); } function override (line 674) | const override { return impl_->stride(); } function override (line 679) | const override { function override (line 682) | const override { function override (line 686) | const override { return impl_->tensor_storage(); } function override (line 687) | const override { return impl_->has_eager_blob_object(); } function override (line 696) | const override { return impl_->acc_grad(); } function override (line 697) | const override { return impl_->current_grad(); } function override (line 703) | const override { function set_is_leaf (line 724) | void set_is_leaf(bool is_leaf) override { impl_->set_is_leaf(is_leaf); } function override (line 725) | const override { function set_autograd_meta (line 729) | void set_autograd_meta(const std::shared_ptr& autograd_met... function GlobalTensorImpl (line 742) | GlobalTensorImpl* mut_impl() { return impl_.get(); } function override (line 744) | const override { function override (line 753) | const override { return is_offloaded_; } function is_offloaded_ (line 763) | bool is_offloaded_{false}; FILE: oneflow/core/framework/tensor_arg.cpp type oneflow (line 21) | namespace oneflow { type one (line 22) | namespace one { FILE: oneflow/core/framework/tensor_arg.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/framework/tensor_global_id.cpp type oneflow (line 22) | namespace oneflow { function RawGetMetaTransportToken (line 26) | Maybe> RawGetMetaTransportToken() { function NewTensorGlobalId (line 34) | Maybe NewTensorGlobalId() { return ++**JUST(GetMetaTra... type one (line 36) | namespace one { function InitGlobalId (line 43) | Maybe InitGlobalId(TensorTuple* outputs) { FILE: oneflow/core/framework/tensor_global_id.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/framework/tensor_impl.cpp type oneflow (line 39) | namespace oneflow { type one (line 40) | namespace one { function MemoryFormat (line 170) | MemoryFormat EagerLocalTensorImpl::memory_format() const { function GetPhysicalShape (line 210) | Maybe GetPhysicalShape(const Shape& logical_shape, const NdSb... function MemoryFormat (line 269) | MemoryFormat EagerGlobalTensorImpl::memory_format() const { FILE: oneflow/core/framework/tensor_impl.h function namespace (line 32) | namespace oneflow { function set_consumer_nd_sbp_constraint (line 159) | void set_consumer_nd_sbp_constraint(const Optional>& val) { function GlobalTensorMeta (line 163) | GlobalTensorMeta* mut_tensor_meta() { function override (line 198) | const override { return tensor_meta()->shape_ptr(); } function override (line 199) | const override { return tensor_meta()->stride_ptr(); } function override (line 209) | const override { return false; } function std (line 211) | const std::shared_ptr& mut_tensor_meta() overr... function override (line 216) | const override { RETURN_ERROR_WITH_BUG_PROMPT(); } function override (line 217) | const override { function override (line 220) | const override { RETURN_ERROR_WITH_BUG_PROMPT(); } function override (line 221) | const override { RETURN_ERROR_WITH_BUG_PROMPT(); } function std (line 243) | const std::shared_ptr& mut_tensor_meta() overr... function override (line 262) | const override { function override (line 266) | const override { return eager_blob_object_.get(); } function override (line 267) | const override { return storage_offset_; } function TensorStorage (line 269) | TensorStorage* mut_tensor_storage() { return tensor_storage_.get(); } function set_storage_offset (line 270) | void set_storage_offset(int64_t offset) { storage_offset_ = offset; } function MemoryFormat (line 325) | MemoryFormat memory_format() const override; function override (line 336) | const override { return cur_rank_phy_tensor_; } function reset_cur_rank_phy_tensor (line 337) | void reset_cur_rank_phy_tensor(const std::shared_ptr& val) { FILE: oneflow/core/framework/tensor_methods.cpp type oneflow (line 31) | namespace oneflow { type one (line 32) | namespace one { type view (line 33) | namespace view { function IsEnvViewDisabled (line 37) | bool IsEnvViewDisabled() { function IsViewApplicable (line 42) | bool IsViewApplicable(const std::shared_ptr& input) { function IsOverlappingMemorys (line 52) | static bool IsOverlappingMemorys(const std::vector& sizes, function MinStorageSize (line 70) | static int64_t MinStorageSize(const std::vector& sizes, function BasicView (line 82) | Maybe BasicView(const std::shared_ptr& input, cons... function BasicView (line 95) | Maybe BasicView(const std::shared_ptr& input, cons... function InplaceView (line 121) | Maybe InplaceView(const std::shared_ptr& input, cons... function Reshape (line 137) | Maybe Reshape(const std::shared_ptr& input, const ... function Reshape (line 142) | Maybe Reshape(const std::shared_ptr& input, const ... function Slice (line 168) | Maybe Slice(const std::shared_ptr& input, const st... function Unsqueeze (line 220) | Maybe Unsqueeze(const std::shared_ptr& input, cons... function InplaceUnsqueeze (line 264) | Maybe InplaceUnsqueeze(const std::shared_ptr& input,... function Squeeze (line 307) | Maybe Squeeze(const std::shared_ptr& input, function InplaceSqueeze (line 351) | Maybe InplaceSqueeze(const std::shared_ptr& input, function Expand (line 394) | Maybe Expand(const std::shared_ptr& input, const S... function InplaceExpand (line 455) | Maybe InplaceExpand(const std::shared_ptr& input, co... function Narrow (line 515) | Maybe Narrow(const std::shared_ptr& input, const i... function AsStridedGrad (line 558) | Maybe AsStridedGrad(const std::shared_ptr& dy, function AsStrided (line 681) | Maybe AsStrided(const std::shared_ptr& input, function InplaceAsStrided (line 708) | Maybe InplaceAsStrided(const std::shared_ptr& i... function Transpose (line 735) | Maybe Transpose(const std::shared_ptr& input, cons... function UnfoldTensor (line 778) | Maybe UnfoldTensor(const std::shared_ptr& input, c... function Diagonal (line 832) | Maybe Diagonal(const std::shared_ptr& input, const... function Touch (line 896) | Maybe Touch(std::shared_ptr input, Symbol stre... FILE: oneflow/core/framework/tensor_methods.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/tensor_name_scope.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { function TensorNameScope (line 22) | TensorNameScope* TensorNameScope::Global() { FILE: oneflow/core/framework/tensor_name_scope.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/framework/tensor_rpc_util.cpp type oneflow (line 30) | namespace oneflow { type private_details (line 31) | namespace private_details { type FlatTensorConsistency (line 33) | struct FlatTensorConsistency class CheckConsistencyAsyncTransportCtx (line 35) | class CheckConsistencyAsyncTransportCtx : public AsyncTransportCtx { method CheckConsistencyAsyncTransportCtx (line 37) | CheckConsistencyAsyncTransportCtx(const TransportToken& transport_... function New (line 71) | static Maybe New( function Check (line 81) | Maybe Check(Symbol tensor_meta, function LaunchTensorMetaConsistencyCheck (line 159) | Maybe LaunchTensorMetaConsistencyCheck( function BusyWaitAndCheck (line 174) | Maybe BusyWaitAndCheck(std::shared_ptr RunCallback(const std::shared_ptr& tensor, FILE: oneflow/core/framework/tensor_rpc_util.h function namespace (line 26) | namespace oneflow { function final (line 71) | struct DisableCheckGlobalTensorMetaScope final { FILE: oneflow/core/framework/tensor_storage.cpp type oneflow (line 20) | namespace oneflow { type one (line 21) | namespace one { FILE: oneflow/core/framework/tensor_storage.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/tensor_tuple.cpp type oneflow (line 19) | namespace oneflow { type one (line 20) | namespace one { FILE: oneflow/core/framework/tensor_tuple.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/framework/tensor_util.cpp type oneflow (line 26) | namespace oneflow { type one (line 27) | namespace one { function SyncAccessTensorWithTimeOut (line 29) | Maybe SyncAccessTensorWithTimeOut( function CopyLocalTensorDataTo (line 42) | Maybe CopyLocalTensorDataTo(const std::shared_ptr& inp... function GetTensorScope (line 63) | Maybe GetTensorScope(const std::shared_ptr& tensor) { function GetItemInScalarTensor (line 75) | Maybe GetItemInScalarTensor(const std::shared_ptr& sca... FILE: oneflow/core/framework/tensor_util.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/framework/to_string.cpp type oneflow (line 21) | namespace oneflow { function DeviceTag4DeviceType (line 23) | Maybe DeviceTag4DeviceType(DeviceType device_type) { function DeviceType4DeviceTag (line 32) | Maybe DeviceType4DeviceTag(const std::string& device_tag) { FILE: oneflow/core/framework/to_string.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/framework/transport_token.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/core/framework/transport_token.h function namespace (line 24) | namespace oneflow { function namespace (line 105) | namespace std { FILE: oneflow/core/framework/transport_util.cpp type oneflow (line 28) | namespace oneflow { function AccessToOtherRanks (line 37) | Maybe AccessToOtherRanks(const ForEachRankT& ForEachRank, const ... function AccessToAllOtherRanks (line 60) | Maybe AccessToAllOtherRanks(Symbol rank_group, const ... function AccessToNearbyRank (line 73) | Maybe AccessToNearbyRank(Symbol rank_group, const Tra... function RawGetTransportToken (line 84) | Maybe> RawGetTransportToken(const Tran... function GetAutoIncrementalTransportToken (line 93) | Maybe GetAutoIncrementalTransportToken(int64_t src_ran... function Send (line 103) | Maybe Send(const TransportToken& token, int64_t rank, void* buff... function Recv (line 118) | Maybe Recv(const TransportToken& token, int64_t rank, void* buff... function GetCurrentRankIndex (line 195) | Maybe GetCurrentRankIndex(const std::vector& rank_he... FILE: oneflow/core/framework/transport_util.h function namespace (line 25) | namespace oneflow { function final (line 115) | struct TransportUtil final { FILE: oneflow/core/framework/user_op_conf.cpp type oneflow (line 24) | namespace oneflow { type user_op (line 26) | namespace user_op { function OperatorConf (line 38) | const OperatorConf& UserOpConfWrapper::op_conf() const { return *op_... function UserOpConf (line 40) | const UserOpConf& UserOpConfWrapper::user_op_conf() const { return o... function TensorDesc (line 127) | const TensorDesc& UserOpWrapper::arg_tensor_desc(const std::string& ... function TensorDesc (line 133) | const TensorDesc& UserOpWrapper::TensorDesc4ArgNameAndIndex(const st... function UserOpConfWrapperBuilder (line 138) | UserOpConfWrapperBuilder& UserOpConfWrapperBuilder::InputBind( function UserOpConfWrapperBuilder (line 146) | UserOpConfWrapperBuilder& UserOpConfWrapperBuilder::Input(const std:... function UserOpConfWrapperBuilder (line 151) | UserOpConfWrapperBuilder& UserOpConfWrapperBuilder::Output(const std... function UserOpConfWrapperBuilder (line 155) | UserOpConfWrapperBuilder& UserOpConfWrapperBuilder::Output(const std... function UserOpConfWrapperBuilder (line 168) | UserOpConfWrapperBuilder& UserOpConfWrapperBuilder::ScopeSymbolId(in... function UserOpConfWrapperBuilder (line 173) | UserOpConfWrapperBuilder& UserOpConfWrapperBuilder::DeviceTag(const ... function UserOpConfWrapper (line 178) | UserOpConfWrapper UserOpConfWrapperBuilder::Build() { function CheckArgDefIsValidInUserOpConf (line 202) | Maybe CheckArgDefIsValidInUserOpConf( function CheckUserOpConfArgOrderValid (line 229) | Maybe CheckUserOpConfArgOrderValid( function AddAttrDefaultValueAndCheckValid (line 244) | Maybe AddAttrDefaultValueAndCheckValid(const UserOpDef& op_def, ... function AddAttrDefaultValueAndCheckValid (line 273) | Maybe AddAttrDefaultValueAndCheckValid(UserOpConf* user_conf) { function AddAttrDefaultValueAndCheckValid (line 281) | Maybe AddAttrDefaultValueAndCheckValid(const UserOpDef& op_def, ... function GetAttrTypeImpl (line 287) | Maybe GetAttrTypeImpl(const std::string& op_type_name, cons... function CheckAndCompleteUserOpConfImpl (line 298) | Maybe CheckAndCompleteUserOpConfImpl(const OperatorConf&... FILE: oneflow/core/framework/user_op_conf.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/framework/user_op_def.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 22) | namespace user_op { function AttrType (line 63) | AttrType UserOpDefWrapper::GetAttrType(const std::string& name) const { FILE: oneflow/core/framework/user_op_def.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/framework/user_op_hob.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/framework/user_op_kernel_registry.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 21) | namespace user_op { function OpKernelRegistry (line 23) | OpKernelRegistry& OpKernelRegistry::Name(const std::string& op_type_... function OpKernelRegistry (line 28) | OpKernelRegistry& OpKernelRegistry::SetCreateFn(OpKernelCreateFn fn) { function OpKernelRegistry (line 33) | OpKernelRegistry& OpKernelRegistry::SetInferTmpSizeFn(InferTmpSizeFn... function OpKernelRegistry (line 38) | OpKernelRegistry& OpKernelRegistry::SetInplaceProposalFn(InplaceProp... function OpKernelRegistry (line 43) | OpKernelRegistry& OpKernelRegistry::SetPriority(int32_t priority) { FILE: oneflow/core/framework/user_op_kernel_registry.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/framework/user_op_registry.cpp type oneflow (line 26) | namespace oneflow { type user_op (line 28) | namespace user_op { function InsertIfNotExists (line 32) | bool InsertIfNotExists(const std::string& name, HashSet... function OpRegistry (line 40) | OpRegistry& OpRegistry::Name(const std::string& op_type_name) { function OpRegistry (line 46) | OpRegistry& OpRegistry::ArgImpl(bool is_input, const std::string& na... function OpRegistry (line 74) | OpRegistry& OpRegistry::SupportCpuOnly() { function OpRegistry (line 79) | OpRegistry& OpRegistry::SupportNonContiguous() { function OpRegistry (line 84) | OpRegistry& OpRegistry::NoGrad() { function OpRegistry (line 89) | OpRegistry& OpRegistry::SetOutputBufferNum(int32_t num) { function OpRegistry (line 94) | OpRegistry& OpRegistry::Attr(const std::string& name, AttrType type) { function AddAttrWithDefault (line 105) | void AddAttrWithDefault(OpRegistryResult* result, const std::string&... function OpRegistry (line 141) | OpRegistry& OpRegistry::DefaultedAttr(const std::string& name, AttrT... function OpRegistry (line 148) | OpRegistry& OpRegistry::SetTensorDescInferFn(TensorDescInferFn tenso... function OpRegistry (line 154) | OpRegistry& OpRegistry::SetLogicalTensorDescInferFn(TensorDescInferF... function OpRegistry (line 159) | OpRegistry& OpRegistry::SetPhysicalTensorDescInferFn(TensorDescInfer... function OpRegistry (line 164) | OpRegistry& OpRegistry::SetCheckAttrFn(CheckAttrFn fn) { function OpRegistry (line 169) | OpRegistry& OpRegistry::SetGetSbpFn(GetSbpFn get_sbp_fn) { function OpRegistry (line 174) | OpRegistry& OpRegistry::SetSbpSignatureInferFn(SbpSignatureInferFn s... function OpRegistry (line 179) | OpRegistry& OpRegistry::SetInputArgModifyFn(InputArgModifyFn input_a... function OpRegistry (line 184) | OpRegistry& OpRegistry::SetOutputArgModifyFn(OutputArgModifyFn outpu... function OpRegistry (line 189) | OpRegistry& OpRegistry::SetOutputBlobTimeShapeInferFn( function OpRegistry (line 195) | OpRegistry& OpRegistry::SetNdSbpInferFn(NdSbpInferFn nd_sbp_infer_fn) { function OpRegistry (line 200) | OpRegistry& OpRegistry::SetDataTypeInferFn(DataTypeInferFn data_type... function OpRegistry (line 205) | OpRegistry& OpRegistry::SetDeviceAndStreamInferFn( function OpRegistry (line 211) | OpRegistry& OpRegistry::SetComputeComplexityFn(ComputeComplexityFn c... function OpRegistry (line 216) | OpRegistry& OpRegistry::SetGetNdSbpSignatureListFn(GetNdSbpSignature... function OpRegistry (line 221) | OpRegistry& OpRegistry::SetEnumerateNdSbpSignaturesFn(EnumerateNdSbp... function OpRegistry (line 226) | OpRegistry& OpRegistry::SetDumpNdSbpSignatureForOpConfFn( FILE: oneflow/core/framework/user_op_registry.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/framework/user_op_registry_manager.cpp type oneflow (line 25) | namespace oneflow { type user_op (line 29) | namespace user_op { function UserOpRegistryMgr (line 31) | UserOpRegistryMgr& UserOpRegistryMgr::Get() { function OpRegistry (line 36) | OpRegistry UserOpRegistryMgr::CheckAndGetOpRegistry(const std::strin... function OpRegistryResult (line 49) | const OpRegistryResult* UserOpRegistryMgr::GetOpRegistryResult(const... function OpKernelRegistry (line 55) | OpKernelRegistry UserOpRegistryMgr::CheckAndGetOpKernelRegistry(cons... function GetErrorMsgOfSearchedOp (line 67) | std::string GetErrorMsgOfSearchedOp(const KernelRegContext& ctx) { function UserOpHostMemoryInputRegistry (line 143) | UserOpHostMemoryInputRegistry& UserOpHostMemoryInputRegistry::Get() { FILE: oneflow/core/framework/user_op_registry_manager.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/framework/user_op_tensor.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/framework/util.h function namespace (line 21) | namespace std { function namespace (line 32) | namespace oneflow { FILE: oneflow/core/framework/variable_tensor_mgr.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/core/framework/variable_tensor_mgr.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/functional/function_library.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/functional/impl/activation_functor.cpp type oneflow (line 38) | namespace oneflow { type one (line 39) | namespace one { type functional (line 40) | namespace functional { type impl (line 42) | namespace impl { class ReluFunctor (line 44) | class ReluFunctor { method ReluFunctor (line 46) | ReluFunctor() { op_ = CHECK_JUST(one::OpBuilder("relu").Input(... class ReluGradFunctor (line 63) | class ReluGradFunctor : public BinaryFunctor { method ReluGradFunctor (line 65) | ReluGradFunctor() { class PReluFunctor (line 70) | class PReluFunctor { method PReluFunctor (line 72) | PReluFunctor() { class PReluGradFunctor (line 88) | class PReluGradFunctor { method PReluGradFunctor (line 90) | PReluGradFunctor() { class HardTanhFunctor (line 110) | class HardTanhFunctor { method HardTanhFunctor (line 112) | HardTanhFunctor() { class HardTanhGradFunctor (line 126) | class HardTanhGradFunctor { method HardTanhGradFunctor (line 128) | HardTanhGradFunctor() { class EluFunctor (line 143) | class EluFunctor { method EluFunctor (line 145) | EluFunctor() { op_ = CHECK_JUST(one::OpBuilder("elu").Input("i... class EluGradFunctor (line 156) | class EluGradFunctor { method EluGradFunctor (line 158) | EluGradFunctor() { class CeluFunctor (line 172) | class CeluFunctor { method CeluFunctor (line 174) | CeluFunctor() { op_ = CHECK_JUST(one::OpBuilder("celu").Input(... class CeluGradFunctor (line 194) | class CeluGradFunctor { method CeluGradFunctor (line 196) | CeluGradFunctor() { class GeluFunctor (line 210) | class GeluFunctor : public UnaryFunctor { method GeluFunctor (line 212) | GeluFunctor() { op_ = CHECK_JUST(one::OpBuilder("gelu").Input(... class GeluGradFunctor (line 215) | class GeluGradFunctor : public BinaryFunctor { method GeluGradFunctor (line 217) | GeluGradFunctor() { class FastGeluFunctor (line 222) | class FastGeluFunctor : public UnaryFunctor { method FastGeluFunctor (line 224) | FastGeluFunctor() { class FastGeluGradFunctor (line 229) | class FastGeluGradFunctor : public BinaryFunctor { method FastGeluGradFunctor (line 231) | FastGeluGradFunctor() { class QuickGeluFunctor (line 236) | class QuickGeluFunctor : public UnaryFunctor { method QuickGeluFunctor (line 238) | QuickGeluFunctor() { class QuickGeluGradFunctor (line 243) | class QuickGeluGradFunctor : public BinaryFunctor { method QuickGeluGradFunctor (line 245) | QuickGeluGradFunctor() { class SquareReLUFunctor (line 250) | class SquareReLUFunctor : public UnaryFunctor { method SquareReLUFunctor (line 252) | SquareReLUFunctor() { class SquareReLUGradFunctor (line 257) | class SquareReLUGradFunctor : public BinaryFunctor { method SquareReLUGradFunctor (line 259) | SquareReLUGradFunctor() { class GluFunctor (line 265) | class GluFunctor { method GluFunctor (line 267) | GluFunctor() {} class HardSigmoidFunctor (line 287) | class HardSigmoidFunctor { method HardSigmoidFunctor (line 289) | HardSigmoidFunctor() { class HardSigmoidGradFunctor (line 308) | class HardSigmoidGradFunctor : public BinaryFunctor { method HardSigmoidGradFunctor (line 310) | HardSigmoidGradFunctor() { class HardShrinkFunctor (line 316) | class HardShrinkFunctor { method HardShrinkFunctor (line 318) | HardShrinkFunctor() { class HardShrinkGradFunctor (line 341) | class HardShrinkGradFunctor { method HardShrinkGradFunctor (line 343) | HardShrinkGradFunctor() { class SoftmaxFunctorBase (line 357) | class SoftmaxFunctorBase { method SoftmaxFunctorBase (line 393) | SoftmaxFunctorBase() = default; class SoftmaxFunctor (line 399) | class SoftmaxFunctor : public SoftmaxFunctorBase { method SoftmaxFunctor (line 401) | SoftmaxFunctor() { class SoftmaxGradFunctor (line 406) | class SoftmaxGradFunctor { method SoftmaxGradFunctor (line 408) | SoftmaxGradFunctor() { class LogSoftmaxFunctor (line 421) | class LogSoftmaxFunctor : public SoftmaxFunctorBase { method LogSoftmaxFunctor (line 423) | LogSoftmaxFunctor() { class LogSoftmaxGradFunctor (line 428) | class LogSoftmaxGradFunctor { method LogSoftmaxGradFunctor (line 430) | LogSoftmaxGradFunctor() { class GumbelSoftmaxFunctor (line 444) | class GumbelSoftmaxFunctor { class HardSwishFunctor (line 497) | class HardSwishFunctor : public UnaryFunctor { method HardSwishFunctor (line 499) | HardSwishFunctor() { class HardSwishGradFunctor (line 504) | class HardSwishGradFunctor : public BinaryFunctor { method HardSwishGradFunctor (line 506) | HardSwishGradFunctor() { class LeakyReluFunctor (line 511) | class LeakyReluFunctor { method LeakyReluFunctor (line 513) | LeakyReluFunctor() { class LeakyReluGradFunctor (line 535) | class LeakyReluGradFunctor { method LeakyReluGradFunctor (line 537) | LeakyReluGradFunctor() { class RReluFunctor (line 551) | class RReluFunctor { method RReluFunctor (line 553) | RReluFunctor() { class RReluInplaceFunctor (line 581) | class RReluInplaceFunctor { class SoftplusFunctor (line 589) | class SoftplusFunctor { method SoftplusFunctor (line 591) | SoftplusFunctor() { class SoftplusGradFunctor (line 606) | class SoftplusGradFunctor { method SoftplusGradFunctor (line 608) | SoftplusGradFunctor() { class SiluFunctor (line 622) | class SiluFunctor : public UnaryFunctor { method SiluFunctor (line 624) | SiluFunctor() { op_ = CHECK_JUST(one::OpBuilder("silu").Input(... class SiluGradFunctor (line 627) | class SiluGradFunctor : public BinaryFunctor { method SiluGradFunctor (line 629) | SiluGradFunctor() { class MishFunctor (line 634) | class MishFunctor : public UnaryFunctor { method MishFunctor (line 636) | MishFunctor() { op_ = CHECK_JUST(one::OpBuilder("mish").Input(... class MishGradFunctor (line 639) | class MishGradFunctor : public BinaryFunctor { method MishGradFunctor (line 641) | MishGradFunctor() { class SeluFunctor (line 646) | class SeluFunctor : public UnaryFunctor { method SeluFunctor (line 648) | SeluFunctor() { op_ = CHECK_JUST(one::OpBuilder("selu").Input(... class SeluGradFunctor (line 651) | class SeluGradFunctor : public BinaryFunctor { method SeluGradFunctor (line 653) | SeluGradFunctor() { class SoftSignFunctor (line 658) | class SoftSignFunctor : public UnaryFunctor { method SoftSignFunctor (line 660) | SoftSignFunctor() { class SoftSignGradFunctor (line 665) | class SoftSignGradFunctor : public BinaryFunctor { method SoftSignGradFunctor (line 667) | SoftSignGradFunctor() { class SoftShrinkFunctor (line 672) | class SoftShrinkFunctor { method SoftShrinkFunctor (line 674) | SoftShrinkFunctor() { class ThresholdFunctor (line 700) | class ThresholdFunctor { method ThresholdFunctor (line 702) | ThresholdFunctor() { class ThresholdGradFunctor (line 717) | class ThresholdGradFunctor { method ThresholdGradFunctor (line 719) | ThresholdGradFunctor() { class SoftShrinkGradFunctor (line 734) | class SoftShrinkGradFunctor { method SoftShrinkGradFunctor (line 736) | SoftShrinkGradFunctor() { class FracFunctor (line 750) | class FracFunctor { method FracFunctor (line 752) | FracFunctor() { op_ = CHECK_JUST(one::OpBuilder("frac").Input(... class FracInplaceFunctor (line 761) | class FracInplaceFunctor { method FracInplaceFunctor (line 763) | FracInplaceFunctor() { op_ = CHECK_JUST(one::OpBuilder("frac")... function ONEFLOW_FUNCTION_LIBRARY (line 778) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/array_functor.cpp type oneflow (line 35) | namespace oneflow { type one (line 36) | namespace one { type functional (line 37) | namespace functional { type impl (line 38) | namespace impl { class ArgMaxFunctor (line 40) | class ArgMaxFunctor { method ArgMaxFunctor (line 42) | ArgMaxFunctor() { op_ = CHECK_JUST(one::OpBuilder("argmax").In... class ArgMinFunctor (line 98) | class ArgMinFunctor { method ArgMinFunctor (line 100) | ArgMinFunctor() {} class GlobalTensorConstantFunctor (line 113) | class GlobalTensorConstantFunctor { method GlobalTensorConstantFunctor (line 115) | GlobalTensorConstantFunctor() { class TensorConstantFunctor (line 159) | class TensorConstantFunctor { method TensorConstantFunctor (line 161) | TensorConstantFunctor() { class GlobalConstantFunctor (line 192) | class GlobalConstantFunctor { method GlobalConstantFunctor (line 194) | GlobalConstantFunctor() { op_ = CHECK_JUST(one::OpBuilder("con... class ConstantFunctor (line 250) | class ConstantFunctor { method ConstantFunctor (line 252) | ConstantFunctor() { op_ = CHECK_JUST(one::OpBuilder("constant"... class EmptyFunctor (line 286) | class EmptyFunctor { method EmptyFunctor (line 288) | EmptyFunctor() { op_ = CHECK_JUST(one::OpBuilder("empty").Outp... class EmptyStridedFunctor (line 321) | class EmptyStridedFunctor { class GlobalEmptyFunctor (line 337) | class GlobalEmptyFunctor { method GlobalEmptyFunctor (line 339) | GlobalEmptyFunctor() { op_ = CHECK_JUST(one::OpBuilder("empty"... class ZerosLikeFunctor (line 364) | class ZerosLikeFunctor : public UnaryFunctor { method ZerosLikeFunctor (line 366) | ZerosLikeFunctor() { class OnesLikeFunctor (line 371) | class OnesLikeFunctor : public UnaryFunctor { method OnesLikeFunctor (line 373) | OnesLikeFunctor() { class FullLikeFunctor (line 378) | class FullLikeFunctor { method FullLikeFunctor (line 380) | FullLikeFunctor() {} class FlattenFunctor (line 395) | class FlattenFunctor { method FlattenFunctor (line 397) | FlattenFunctor() = default; class WhereFunctor (line 438) | class WhereFunctor { method WhereFunctor (line 440) | WhereFunctor() { class WhereScalarXFunctor (line 454) | class WhereScalarXFunctor { method WhereScalarXFunctor (line 456) | WhereScalarXFunctor() = default; class WhereScalarYFunctor (line 480) | class WhereScalarYFunctor { method WhereScalarYFunctor (line 482) | WhereScalarYFunctor() = default; class WhereScalarXYFunctor (line 506) | class WhereScalarXYFunctor { method WhereScalarXYFunctor (line 508) | WhereScalarXYFunctor() = default; class ArgWhereFunctor (line 581) | class ArgWhereFunctor { method ArgWhereFunctor (line 583) | ArgWhereFunctor() { class NonZeroFunctor (line 598) | class NonZeroFunctor { method NonZeroFunctor (line 600) | NonZeroFunctor() {} class BroadcastLikeFunctor (line 640) | class BroadcastLikeFunctor { method BroadcastLikeFunctor (line 642) | BroadcastLikeFunctor() { class ConcatFunctor (line 690) | class ConcatFunctor { method ConcatFunctor (line 692) | ConcatFunctor() { class StackFunctor (line 758) | class StackFunctor { method StackFunctor (line 760) | StackFunctor() { class StackGradFunctor (line 804) | class StackGradFunctor { method StackGradFunctor (line 806) | StackGradFunctor() { class AtLeast1DFunctor (line 835) | class AtLeast1DFunctor { class AtLeast1DListFunctor (line 845) | class AtLeast1DListFunctor { class AtLeast2DFunctor (line 856) | class AtLeast2DFunctor { class AtLeast2DListFunctor (line 868) | class AtLeast2DListFunctor { class AtLeast3DFunctor (line 879) | class AtLeast3DFunctor { class AtLeast3DListFunctor (line 893) | class AtLeast3DListFunctor { class ColumnStackFunctor (line 904) | class ColumnStackFunctor { class HStackFunctor (line 919) | class HStackFunctor { class VStackFunctor (line 930) | class VStackFunctor { class RowStackFunctor (line 938) | class RowStackFunctor { class DStackFunctor (line 943) | class DStackFunctor { class ExpandFunctor (line 951) | class ExpandFunctor { method ExpandFunctor (line 953) | ExpandFunctor() { op_ = CHECK_JUST(one::OpBuilder("expand").In... class ExpandDimsFunctor (line 1003) | class ExpandDimsFunctor { method ExpandDimsFunctor (line 1005) | ExpandDimsFunctor() { class UnsqueezeMultipleFunctor (line 1024) | class UnsqueezeMultipleFunctor { method UnsqueezeMultipleFunctor (line 1026) | UnsqueezeMultipleFunctor() {} class InplaceUnsqueezeFunctor (line 1063) | class InplaceUnsqueezeFunctor { class SqueezeFunctor (line 1076) | class SqueezeFunctor { method SqueezeFunctor (line 1078) | SqueezeFunctor() { class InplaceSqueezeFunctor (line 1109) | class InplaceSqueezeFunctor { class RollFunctor (line 1140) | class RollFunctor { method RollFunctor (line 1142) | RollFunctor() { op_ = CHECK_JUST(one::OpBuilder("roll").Input(... class GatherFunctor (line 1165) | class GatherFunctor { method GatherFunctor (line 1167) | GatherFunctor() { class DimGatherFunctor (line 1181) | class DimGatherFunctor { method DimGatherFunctor (line 1183) | DimGatherFunctor() { method FOR_RANGE (line 1211) | FOR_RANGE(int32_t, i, 0, input->ndim()) { type DimScatterType (line 1230) | enum class DimScatterType { kUpdate, kAdd, kMultiply } function DimScatterTypeToString (line 1233) | std::string DimScatterTypeToString() { class DimScatterFunctorImpl (line 1243) | class DimScatterFunctorImpl { method DimScatterFunctorImpl (line 1245) | DimScatterFunctorImpl() class DimScatterFunctor (line 1272) | class DimScatterFunctor { class DimScatterScalarFunctorImpl (line 1293) | class DimScatterScalarFunctorImpl { method DimScatterScalarFunctorImpl (line 1295) | DimScatterScalarFunctorImpl() class DimScatterScalarFunctor (line 1321) | class DimScatterScalarFunctor { class DimScatterAddLikeFunctor (line 1340) | class DimScatterAddLikeFunctor { method DimScatterAddLikeFunctor (line 1342) | DimScatterAddLikeFunctor() { class ArgSortFunctor (line 1362) | class ArgSortFunctor { method ArgSortFunctor (line 1364) | ArgSortFunctor() { class SearchSortedFunctor (line 1383) | class SearchSortedFunctor { method SearchSortedFunctor (line 1385) | SearchSortedFunctor() { class SearchSortedScalarFunctor (line 1420) | class SearchSortedScalarFunctor { method SearchSortedScalarFunctor (line 1422) | SearchSortedScalarFunctor() { class GatherNdFunctor (line 1448) | class GatherNdFunctor { method GatherNdFunctor (line 1450) | GatherNdFunctor() { class ScatterNdFunctor (line 1463) | class ScatterNdFunctor { method ScatterNdFunctor (line 1465) | ScatterNdFunctor() { class TensorScatterNdUpdateFunctor (line 1480) | class TensorScatterNdUpdateFunctor { method TensorScatterNdUpdateFunctor (line 1482) | TensorScatterNdUpdateFunctor() { class ScatterNdLikeFunctor (line 1525) | class ScatterNdLikeFunctor { method ScatterNdLikeFunctor (line 1527) | ScatterNdLikeFunctor() { class ReshapeFunctor (line 1545) | class ReshapeFunctor { method ReshapeFunctor (line 1547) | ReshapeFunctor() { class ViewFunctor (line 1569) | class ViewFunctor { method ViewFunctor (line 1571) | ViewFunctor() { op_ = CHECK_JUST(one::OpBuilder("reshape").Inp... class ToContiguousFunctor (line 1592) | class ToContiguousFunctor { method ToContiguousFunctor (line 1594) | ToContiguousFunctor() { class InplaceToContiguousFunctor (line 1606) | class InplaceToContiguousFunctor { method InplaceToContiguousFunctor (line 1608) | InplaceToContiguousFunctor() { class NarrowFunctor (line 1645) | class NarrowFunctor { method NarrowFunctor (line 1647) | NarrowFunctor() { op_ = CHECK_JUST(one::OpBuilder("narrow").In... class NarrowGradFunctor (line 1677) | class NarrowGradFunctor { method NarrowGradFunctor (line 1679) | NarrowGradFunctor() { class SliceFunctor (line 1694) | class SliceFunctor { method SliceFunctor (line 1696) | SliceFunctor() { op_ = CHECK_JUST(one::OpBuilder("slice").Inpu... class SliceUpdateFunctor (line 1713) | class SliceUpdateFunctor { method SliceUpdateFunctor (line 1715) | SliceUpdateFunctor() { class SliceGradFunctor (line 1746) | class SliceGradFunctor { method SliceGradFunctor (line 1748) | SliceGradFunctor() { class UpsampleGradFunctor (line 1763) | class UpsampleGradFunctor { method UpsampleGradFunctor (line 1765) | UpsampleGradFunctor() { class CopyToDeviceFunctor (line 1782) | class CopyToDeviceFunctor { method CopyToDeviceFunctor (line 1784) | CopyToDeviceFunctor() { method TouchEpDevice (line 1805) | void TouchEpDevice(Symbol device) const { class CopyFunctor (line 1815) | class CopyFunctor { class FlipFunctor (line 1823) | class FlipFunctor { method FlipFunctor (line 1825) | FlipFunctor() { op_ = CHECK_JUST(one::OpBuilder("flip").Input(... class UnfoldTensorFunctor (line 1842) | class UnfoldTensorFunctor { method UnfoldTensorFunctor (line 1844) | UnfoldTensorFunctor() { class UnfoldTensorGradFunctor (line 1861) | class UnfoldTensorGradFunctor { method UnfoldTensorGradFunctor (line 1863) | UnfoldTensorGradFunctor() { class UpsampleLinear1DFunctor (line 1879) | class UpsampleLinear1DFunctor { method UpsampleLinear1DFunctor (line 1881) | UpsampleLinear1DFunctor() { class UpsampleLinear1DGradFunctor (line 1902) | class UpsampleLinear1DGradFunctor { method UpsampleLinear1DGradFunctor (line 1904) | UpsampleLinear1DGradFunctor() { class UpsampleNearest1DFunctor (line 1927) | class UpsampleNearest1DFunctor { method UpsampleNearest1DFunctor (line 1929) | UpsampleNearest1DFunctor() { class UpsampleNearest1DGradFunctor (line 1948) | class UpsampleNearest1DGradFunctor { method UpsampleNearest1DGradFunctor (line 1950) | UpsampleNearest1DGradFunctor() { class UpsampleNearest2DFunctor (line 1971) | class UpsampleNearest2DFunctor { method UpsampleNearest2DFunctor (line 1973) | UpsampleNearest2DFunctor() { class UpsampleNearest2DGradFunctor (line 1994) | class UpsampleNearest2DGradFunctor { method UpsampleNearest2DGradFunctor (line 1996) | UpsampleNearest2DGradFunctor() { class UpsampleBilinear2DFunctor (line 2019) | class UpsampleBilinear2DFunctor { method UpsampleBilinear2DFunctor (line 2021) | UpsampleBilinear2DFunctor() { class UpsampleBilinear2DGradFunctor (line 2042) | class UpsampleBilinear2DGradFunctor { method UpsampleBilinear2DGradFunctor (line 2044) | UpsampleBilinear2DGradFunctor() { class UpsampleBicubic2DFunctor (line 2067) | class UpsampleBicubic2DFunctor { method UpsampleBicubic2DFunctor (line 2069) | UpsampleBicubic2DFunctor() { class UpsampleBicubic2DGradFunctor (line 2090) | class UpsampleBicubic2DGradFunctor { method UpsampleBicubic2DGradFunctor (line 2092) | UpsampleBicubic2DGradFunctor() { class UpsampleNearest3DFunctor (line 2115) | class UpsampleNearest3DFunctor { method UpsampleNearest3DFunctor (line 2117) | UpsampleNearest3DFunctor() { class UpsampleNearest3DGradFunctor (line 2138) | class UpsampleNearest3DGradFunctor { method UpsampleNearest3DGradFunctor (line 2140) | UpsampleNearest3DGradFunctor() { class UpsampleTrilinear3DFunctor (line 2163) | class UpsampleTrilinear3DFunctor { method UpsampleTrilinear3DFunctor (line 2165) | UpsampleTrilinear3DFunctor() { class UpsampleTrilinear3DGradFunctor (line 2189) | class UpsampleTrilinear3DGradFunctor { method UpsampleTrilinear3DGradFunctor (line 2191) | UpsampleTrilinear3DGradFunctor() { class UnsortedSegmentSumLikeFunctor (line 2217) | class UnsortedSegmentSumLikeFunctor { method UnsortedSegmentSumLikeFunctor (line 2219) | UnsortedSegmentSumLikeFunctor() { class UnsortedSegmentSumFunctor (line 2239) | class UnsortedSegmentSumFunctor { method UnsortedSegmentSumFunctor (line 2241) | UnsortedSegmentSumFunctor() { class TrilFunctor (line 2260) | class TrilFunctor { method TrilFunctor (line 2262) | TrilFunctor() { op_ = CHECK_JUST(one::OpBuilder("tril").Input(... class InplaceTrilFunctor (line 2274) | class InplaceTrilFunctor { method InplaceTrilFunctor (line 2276) | InplaceTrilFunctor() { class TriuFunctor (line 2293) | class TriuFunctor { method TriuFunctor (line 2295) | TriuFunctor() { op_ = CHECK_JUST(one::OpBuilder("triu").Input(... class InplaceTriuFunctor (line 2306) | class InplaceTriuFunctor { method InplaceTriuFunctor (line 2308) | InplaceTriuFunctor() { class DiagFunctor (line 2325) | class DiagFunctor { method DiagFunctor (line 2327) | DiagFunctor() { op_ = CHECK_JUST(one::OpBuilder("diag").Input(... class DiagGradFunctor (line 2338) | class DiagGradFunctor { method DiagGradFunctor (line 2340) | DiagGradFunctor() { class DiagonalFunctor (line 2354) | class DiagonalFunctor { method DiagonalFunctor (line 2356) | DiagonalFunctor() { class DiagonalGradFunctor (line 2387) | class DiagonalGradFunctor { method DiagonalGradFunctor (line 2389) | DiagonalGradFunctor() { class SliceView1dContiguousFunctor (line 2404) | class SliceView1dContiguousFunctor { method SliceView1dContiguousFunctor (line 2406) | SliceView1dContiguousFunctor() = default; class TensorGetItemFunctor (line 2414) | class TensorGetItemFunctor { method TensorGetItemFunctor (line 2416) | TensorGetItemFunctor() {} class TensorSetItemFunctor (line 2476) | class TensorSetItemFunctor { method TensorSetItemFunctor (line 2478) | TensorSetItemFunctor() {} class CastLikeFunctor (line 2600) | class CastLikeFunctor { method CastLikeFunctor (line 2602) | CastLikeFunctor() { class ElementwiseMinimumGradFunctor (line 2615) | class ElementwiseMinimumGradFunctor { method ElementwiseMinimumGradFunctor (line 2617) | ElementwiseMinimumGradFunctor() { class ElementwiseMaximumGradFunctor (line 2636) | class ElementwiseMaximumGradFunctor { method ElementwiseMaximumGradFunctor (line 2638) | ElementwiseMaximumGradFunctor() { class DivGradFunctor (line 2657) | class DivGradFunctor { method DivGradFunctor (line 2659) | DivGradFunctor() { class BroadcastPowXGradFunctor (line 2677) | class BroadcastPowXGradFunctor { class BroadcastPowYGradFunctor (line 2692) | class BroadcastPowYGradFunctor { class IdentityFunctor (line 2707) | class IdentityFunctor { method IdentityFunctor (line 2709) | IdentityFunctor() { class AmpWhiteIdentityFunctor (line 2720) | class AmpWhiteIdentityFunctor { method AmpWhiteIdentityFunctor (line 2722) | AmpWhiteIdentityFunctor() { class AmpBlackIdentityFunctor (line 2733) | class AmpBlackIdentityFunctor { method AmpBlackIdentityFunctor (line 2735) | AmpBlackIdentityFunctor() { class ReduceSumLikeFunctor (line 2746) | class ReduceSumLikeFunctor { method ReduceSumLikeFunctor (line 2748) | ReduceSumLikeFunctor() { class BroadcastReduceSumLikeFunctor (line 2764) | class BroadcastReduceSumLikeFunctor { method BroadcastReduceSumLikeFunctor (line 2766) | BroadcastReduceSumLikeFunctor() {} class SplitFunctor (line 2787) | class SplitFunctor { method SplitFunctor (line 2789) | SplitFunctor() {} class UnbindFunctor (line 2811) | class UnbindFunctor { method UnbindFunctor (line 2813) | UnbindFunctor() {} class ChunkFunctor (line 2829) | class ChunkFunctor { method ChunkFunctor (line 2831) | ChunkFunctor() {} class SplitLikeFunctor (line 2854) | class SplitLikeFunctor { method SplitLikeFunctor (line 2856) | SplitLikeFunctor() { class SplitWithSizeFunctor (line 2885) | class SplitWithSizeFunctor { method SplitWithSizeFunctor (line 2887) | SplitWithSizeFunctor() {} class BatchGatherFunctor (line 2914) | class BatchGatherFunctor { method BatchGatherFunctor (line 2916) | BatchGatherFunctor() { class UnsortedBatchSegmentSumFunctor (line 2929) | class UnsortedBatchSegmentSumFunctor { method UnsortedBatchSegmentSumFunctor (line 2931) | UnsortedBatchSegmentSumFunctor() { class MaskedFillFunctor (line 2951) | class MaskedFillFunctor { method MaskedFillFunctor (line 2953) | MaskedFillFunctor() { method FOR_RANGE (line 2985) | FOR_RANGE(int64_t, i, 0, max_shape.NumAxes()) { class MeshgridFunctor (line 3001) | class MeshgridFunctor { class IndexSelectFunctor (line 3067) | class IndexSelectFunctor { function LocalTensorTo (line 3087) | Maybe LocalTensorTo(const std::shared_ptr& x, Sy... function GlobalTensorTo (line 3096) | Maybe GlobalTensorTo(const std::shared_ptr& x, c... class ToFunctor (line 3132) | class ToFunctor { class To2Functor (line 3158) | class To2Functor { class To3Functor (line 3185) | class To3Functor { class To4Functor (line 3199) | class To4Functor { class ToDeviceFunctor (line 3212) | class ToDeviceFunctor { class ToMemoryFormatFunctor (line 3238) | class ToMemoryFormatFunctor { method ToMemoryFormatFunctor (line 3240) | ToMemoryFormatFunctor() { class TopKFunctor (line 3255) | class TopKFunctor { method TopKFunctor (line 3257) | TopKFunctor() { op_ = CHECK_JUST(one::OpBuilder("top_k").Input... class InTopKFunctor (line 3302) | class InTopKFunctor { method InTopKFunctor (line 3304) | InTopKFunctor() { class TensorBufferToTensorFunctor (line 3325) | class TensorBufferToTensorFunctor { method TensorBufferToTensorFunctor (line 3327) | TensorBufferToTensorFunctor() { class TensorToTensorBufferFunctor (line 3341) | class TensorToTensorBufferFunctor { method TensorToTensorBufferFunctor (line 3343) | TensorToTensorBufferFunctor() { class GenTensorBufferFunctor (line 3356) | class GenTensorBufferFunctor { method GenTensorBufferFunctor (line 3358) | GenTensorBufferFunctor() { class RepeatFunctor (line 3374) | class RepeatFunctor { method RepeatFunctor (line 3376) | RepeatFunctor() {} class RepeatInterLeaveIndexFunctor (line 3432) | class RepeatInterLeaveIndexFunctor { method RepeatInterLeaveIndexFunctor (line 3434) | RepeatInterLeaveIndexFunctor() { class RepeatInterLeaveIntFunctor (line 3450) | class RepeatInterLeaveIntFunctor { method RepeatInterLeaveIntFunctor (line 3452) | RepeatInterLeaveIntFunctor() {} class RepeatInterLeaveTensorFunctor (line 3486) | class RepeatInterLeaveTensorFunctor { method RepeatInterLeaveTensorFunctor (line 3488) | RepeatInterLeaveTensorFunctor() {} class TileFunctor (line 3530) | class TileFunctor { method TileFunctor (line 3532) | TileFunctor() {} class TransposeAllDimPropertyFunctor (line 3548) | class TransposeAllDimPropertyFunctor { method TransposeAllDimPropertyFunctor (line 3550) | TransposeAllDimPropertyFunctor() {} class TransposeAllDimFunctionFunctor (line 3561) | class TransposeAllDimFunctionFunctor { method TransposeAllDimFunctionFunctor (line 3563) | TransposeAllDimFunctionFunctor() {} class ReshapeLikeFunctor (line 3574) | class ReshapeLikeFunctor { method ReshapeLikeFunctor (line 3576) | ReshapeLikeFunctor() { class PinMemoryFunctor (line 3589) | class PinMemoryFunctor { method PinMemoryFunctor (line 3591) | PinMemoryFunctor() { class FillFunctor (line 3638) | class FillFunctor { method FillFunctor (line 3640) | FillFunctor() { op_ = CHECK_JUST(one::OpBuilder("fill_").Input... class FillTensorFunctor (line 3662) | class FillTensorFunctor { method FillTensorFunctor (line 3664) | FillTensorFunctor() { class IndexAddFunctor (line 3691) | class IndexAddFunctor { method IndexAddFunctor (line 3693) | IndexAddFunctor() { class IndexAddInplaceFunctor (line 3730) | class IndexAddInplaceFunctor { method IndexAddInplaceFunctor (line 3732) | IndexAddInplaceFunctor() { class BroadcastShapesFunctor (line 3773) | class BroadcastShapesFunctor { class BroadcastTensorsFunctor (line 3780) | class BroadcastTensorsFunctor { class BinCountFunctor (line 3804) | class BinCountFunctor { method BinCountFunctor (line 3806) | BinCountFunctor() { class UniqueFunctor (line 3863) | class UniqueFunctor { method UniqueFunctor (line 3865) | UniqueFunctor() { class UniqueWithCountsFunctor (line 3898) | class UniqueWithCountsFunctor { method UniqueWithCountsFunctor (line 3900) | UniqueWithCountsFunctor() { class BaddBmmFunctor (line 3959) | class BaddBmmFunctor { class SortFunctor (line 3990) | class SortFunctor { class CloneFunctor (line 4019) | class CloneFunctor { class FusedCodegeexQkvReshapeFunctor (line 4024) | class FusedCodegeexQkvReshapeFunctor { method FusedCodegeexQkvReshapeFunctor (line 4026) | FusedCodegeexQkvReshapeFunctor() { function ONEFLOW_FUNCTION_LIBRARY (line 4052) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/binary_functor.cpp type oneflow (line 35) | namespace oneflow { type one (line 36) | namespace one { type functional (line 37) | namespace functional { type impl (line 39) | namespace impl { function IsCPUScalarTensor (line 43) | bool IsCPUScalarTensor(const std::shared_ptr& tensor) { function TensorDeviceToString (line 50) | std::string TensorDeviceToString(const std::shared_ptr& ... function CastDeviceForCPUScalarTensor (line 55) | Maybe CastDeviceForCPUScalarTensor(std::shared_ptr... class AddFunctor (line 67) | class AddFunctor { method AddFunctor (line 69) | AddFunctor() { class BroadcastPowFunctor (line 151) | class BroadcastPowFunctor : public BinaryFloatFunctor { method BroadcastPowFunctor (line 153) | BroadcastPowFunctor() { class SubFunctor (line 158) | class SubFunctor : public InplaceableBinaryFunctor { method SubFunctor (line 160) | SubFunctor() { class MulFunctor (line 182) | class MulFunctor { method MulFunctor (line 184) | MulFunctor() { class InplaceMulFunctor (line 204) | class InplaceMulFunctor { method InplaceMulFunctor (line 206) | InplaceMulFunctor() { class AddcmulBaseFunctor (line 236) | class AddcmulBaseFunctor { method AddcmulBaseFunctor (line 238) | AddcmulBaseFunctor() = default; class AddcmulFunctor (line 250) | class AddcmulFunctor : public AddcmulBaseFunctor { method AddcmulFunctor (line 252) | AddcmulFunctor() = default; class InplaceAddcmulFunctor (line 260) | class InplaceAddcmulFunctor : public AddcmulBaseFunctor { method InplaceAddcmulFunctor (line 262) | InplaceAddcmulFunctor() = default; class DivFunctor (line 270) | class DivFunctor : public BinaryFloatFunctor { method DivFunctor (line 272) | DivFunctor() { class DivFunctorMode (line 277) | class DivFunctorMode { method DivFunctorMode (line 279) | DivFunctorMode() {} class InplaceDivFunctor (line 301) | class InplaceDivFunctor { method InplaceDivFunctor (line 303) | InplaceDivFunctor() { class Atan2Functor (line 348) | class Atan2Functor : public BinaryFloatFunctor { method Atan2Functor (line 350) | Atan2Functor() { class PowFunctor (line 385) | class PowFunctor : public BinaryFloatFunctor { method PowFunctor (line 387) | PowFunctor() { class FloorDivFunctor (line 397) | class FloorDivFunctor : public BinaryFunctor { method FloorDivFunctor (line 399) | FloorDivFunctor() { class TruncDivFunctor (line 408) | class TruncDivFunctor : public BinaryFunctor { method TruncDivFunctor (line 410) | TruncDivFunctor() { class LerpFunctor (line 419) | class LerpFunctor { method LerpFunctor (line 421) | LerpFunctor() { class InplaceLerpFunctor (line 469) | class InplaceLerpFunctor { method InplaceLerpFunctor (line 471) | InplaceLerpFunctor() { class LerpGradFunctor (line 540) | class LerpGradFunctor { method LerpGradFunctor (line 542) | LerpGradFunctor() { class BroadcastFModFunctor (line 565) | class BroadcastFModFunctor : public BinaryFunctor { method BroadcastFModFunctor (line 567) | BroadcastFModFunctor() { class BroadcastEqualFunctor (line 572) | class BroadcastEqualFunctor : public BinaryFunctor { method BroadcastEqualFunctor (line 574) | BroadcastEqualFunctor() { class EqualFunctor (line 579) | class EqualFunctor { method EqualFunctor (line 581) | EqualFunctor() { class BroadcastNotEqualFunctor (line 600) | class BroadcastNotEqualFunctor : public BinaryFunctor { method BroadcastNotEqualFunctor (line 602) | BroadcastNotEqualFunctor() { class BroadcastGreaterFunctor (line 608) | class BroadcastGreaterFunctor : public BinaryFunctor { method BroadcastGreaterFunctor (line 610) | BroadcastGreaterFunctor() { class InplaceBroadcastGreaterFunctor (line 615) | class InplaceBroadcastGreaterFunctor { method InplaceBroadcastGreaterFunctor (line 617) | InplaceBroadcastGreaterFunctor() { class BroadcastGreaterEqualFunctor (line 641) | class BroadcastGreaterEqualFunctor : public BinaryFunctor { method BroadcastGreaterEqualFunctor (line 643) | BroadcastGreaterEqualFunctor() { class BroadcastLogicalAndFunctor (line 649) | class BroadcastLogicalAndFunctor : public BinaryFunctor { method BroadcastLogicalAndFunctor (line 651) | BroadcastLogicalAndFunctor() { class BroadcastLogicalOrFunctor (line 657) | class BroadcastLogicalOrFunctor : public BinaryFunctor { method BroadcastLogicalOrFunctor (line 659) | BroadcastLogicalOrFunctor() { class BroadcastLogicalXorFunctor (line 665) | class BroadcastLogicalXorFunctor : public BinaryFunctor { method BroadcastLogicalXorFunctor (line 667) | BroadcastLogicalXorFunctor() { class BroadcastBitwiseAndFunctor (line 673) | class BroadcastBitwiseAndFunctor : public BinaryFunctor { method BroadcastBitwiseAndFunctor (line 675) | BroadcastBitwiseAndFunctor() { class BroadcastBitwiseOrFunctor (line 681) | class BroadcastBitwiseOrFunctor : public BinaryFunctor { method BroadcastBitwiseOrFunctor (line 683) | BroadcastBitwiseOrFunctor() { class BroadcastBitwiseXorFunctor (line 689) | class BroadcastBitwiseXorFunctor : public BinaryFunctor { method BroadcastBitwiseXorFunctor (line 691) | BroadcastBitwiseXorFunctor() { class BroadcastLessFunctor (line 697) | class BroadcastLessFunctor : public BinaryFunctor { method BroadcastLessFunctor (line 699) | BroadcastLessFunctor() { class BroadcastLessEqualFunctor (line 704) | class BroadcastLessEqualFunctor : public BinaryFunctor { method BroadcastLessEqualFunctor (line 706) | BroadcastLessEqualFunctor() { class BroadcastIsCloseFunctor (line 712) | class BroadcastIsCloseFunctor { method BroadcastIsCloseFunctor (line 714) | BroadcastIsCloseFunctor() { class ScalarAddByTensorFunctor (line 737) | class ScalarAddByTensorFunctor : public InplaceableBinaryFunctor { method ScalarAddByTensorFunctor (line 739) | ScalarAddByTensorFunctor() { class HostScalarAddByTensorFunctor (line 746) | class HostScalarAddByTensorFunctor { method HostScalarAddByTensorFunctor (line 748) | HostScalarAddByTensorFunctor() { class ScalarSubByTensorFunctor (line 762) | class ScalarSubByTensorFunctor : public BinaryFunctor { method ScalarSubByTensorFunctor (line 764) | ScalarSubByTensorFunctor() { class ScalarMulByTensorFunctor (line 770) | class ScalarMulByTensorFunctor : public BinaryFunctor { method ScalarMulByTensorFunctor (line 772) | ScalarMulByTensorFunctor() { class ScalarDivByTensorFunctor (line 778) | class ScalarDivByTensorFunctor : public BinaryFunctor { method ScalarDivByTensorFunctor (line 780) | ScalarDivByTensorFunctor() { class BroadcastZetaFunctor (line 786) | class BroadcastZetaFunctor : public BinaryFloatFunctor { method BroadcastZetaFunctor (line 788) | BroadcastZetaFunctor() { class ZetaScalarTensorFunctor (line 793) | class ZetaScalarTensorFunctor { class ZetaTensorScalarFunctor (line 801) | class ZetaTensorScalarFunctor { function ONEFLOW_FUNCTION_LIBRARY (line 811) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/binary_functor.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/functional/impl/binary_grad_functor.cpp type oneflow (line 23) | namespace oneflow { type one (line 24) | namespace one { type functional (line 25) | namespace functional { type impl (line 27) | namespace impl { function ONEFLOW_FUNCTION_LIBRARY (line 65) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/comm_functor.cpp type oneflow (line 39) | namespace oneflow { type one (line 40) | namespace one { type functional (line 41) | namespace functional { type impl (line 43) | namespace impl { class EagerCclKernelRegContext (line 52) | class EagerCclKernelRegContext final : public user_op::KernelReg... method EagerCclKernelRegContext (line 54) | explicit EagerCclKernelRegContext(DeviceType device_type) : de... method DeviceType (line 57) | DeviceType device_type() const override { return device_type_; } method ParallelContext (line 58) | const ParallelContext& parallel_ctx() const override { PRINT_B... function RawCheckCclKernelRegistered (line 81) | Maybe RawCheckCclKernelRegistered(const std::string& op_ty... function IsSplitSbp (line 89) | bool IsSplitSbp(Symbol sbp_parallel) { return sbp_p... function EagerCclAllReduce (line 91) | Maybe EagerCclAllReduce(Symbol pa... function EagerCclReduceScatter (line 104) | Maybe EagerCclReduceScatter(Symbol EagerCclAllGather(Symbol pa... function EagerCclS2S (line 132) | Maybe EagerCclS2S(Symbol parallel... function EagerCclReduce (line 145) | Maybe EagerCclReduce(Symbol paral... function RankGroupAndDeviceType2AllReduceOpExpr (line 158) | Maybe RankGroupAndDeviceType2AllReduceOpExpr(Sy... function RankGroupAndDeviceType2AllGatherOpExpr (line 173) | Maybe RankGroupAndDeviceType2AllGatherOpExpr(Sy... function RankGroupAndDeviceType2ReduceScatterOpExpr (line 188) | Maybe RankGroupAndDeviceType2ReduceScatterOpExp... class CommBroadcastFunctor (line 207) | class CommBroadcastFunctor { method CommBroadcastFunctor (line 209) | CommBroadcastFunctor() = default; class CommBroadcastTensorsFunctor (line 219) | class CommBroadcastTensorsFunctor { method CommBroadcastTensorsFunctor (line 221) | CommBroadcastTensorsFunctor() = default; function RawStreamTouchFunctorOpExpr (line 235) | Maybe RawStreamTouchFunctorOpExpr(size_t input_... class StreamTouchFunctor (line 246) | class StreamTouchFunctor { method StreamTouchFunctor (line 248) | StreamTouchFunctor() = default; class LocalAllReduceFunctor (line 258) | class LocalAllReduceFunctor { method LocalAllReduceFunctor (line 260) | LocalAllReduceFunctor() = default; class GlobalAllReduceFunctor (line 283) | class GlobalAllReduceFunctor { method GlobalAllReduceFunctor (line 285) | GlobalAllReduceFunctor() = default; class GlobalReduceScatterFunctor (line 297) | class GlobalReduceScatterFunctor { method GlobalReduceScatterFunctor (line 299) | GlobalReduceScatterFunctor() = default; class LocalReduceScatterFunctor (line 322) | class LocalReduceScatterFunctor { method LocalReduceScatterFunctor (line 324) | LocalReduceScatterFunctor() = default; class GlobalAllGatherFunctor (line 354) | class GlobalAllGatherFunctor { method GlobalAllGatherFunctor (line 356) | GlobalAllGatherFunctor() = default; class LocalAllGatherFunctor (line 368) | class LocalAllGatherFunctor { method LocalAllGatherFunctor (line 370) | LocalAllGatherFunctor() = default; class GlobalS2SFunctor (line 400) | class GlobalS2SFunctor { method GlobalS2SFunctor (line 402) | GlobalS2SFunctor() = default; class SendFunctor (line 423) | class SendFunctor { method SendFunctor (line 425) | SendFunctor() { op_expr_ = CHECK_JUST(one::OpBuilder("send").I... class RecvFunctor (line 447) | class RecvFunctor { method RecvFunctor (line 449) | RecvFunctor() { op_expr_ = CHECK_JUST(one::OpBuilder("recv").O... class LocalReduceFunctor (line 497) | class LocalReduceFunctor { method LocalReduceFunctor (line 499) | LocalReduceFunctor() = default; function ONEFLOW_FUNCTION_LIBRARY (line 535) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/common.cpp type oneflow (line 31) | namespace oneflow { type one (line 32) | namespace one { type functional (line 33) | namespace functional { function InferUnifiedShapeForBroadcasting (line 36) | Maybe InferUnifiedShapeForBroadcasting(const Shape& input_s... function IsStaticZerosTensor (line 99) | bool IsStaticZerosTensor(const std::shared_ptr& x) { function IsInplaceValid (line 103) | bool IsInplaceValid(const std::shared_ptr& x) { function IsScalarTensor (line 107) | bool IsScalarTensor(const std::shared_ptr& x) { function ComputeNonOverlappingAndDense (line 111) | Maybe ComputeNonOverlappingAndDense(const std::shared_ptr IsNonOverlappingAndDense(const std::shared_ptr... function CheckAxis (line 149) | Maybe> CheckAxis(const std::vector& ... function CheckInplaceValid (line 165) | Maybe CheckInplaceValid(const std::shared_ptr& x) { function CheckInplaceCastValid (line 172) | Maybe CheckInplaceCastValid(const std::shared_ptr& x, function CheckInplaceShapeCanExpandTo (line 180) | Maybe CheckInplaceShapeCanExpandTo(const Shape& shape, const... function ComputeStride (line 207) | Optional ComputeStride(const Shape& shape, const Stride& s... function InferShapeUnspecifiedDim (line 254) | Maybe InferShapeUnspecifiedDim(const int64_t& elem_count, c... function InferUnifiedShapeForBroadcasting (line 285) | Maybe InferUnifiedShapeForBroadcasting(const std::vector>> InferUnifiedShapeForBro... function BroadcastSeedToAllRanks (line 321) | Maybe BroadcastSeedToAllRanks(uint64_t* seed, int64_t root) { function GetPermWhenTransposeAxisToLastDim (line 331) | Maybe> GetPermWhenTransposeAxisToLastDim(cons... function GetInversedPerm (line 346) | Maybe> GetInversedPerm(const std::vector, bool>> batchify(const st... function T (line 365) | T GetTensorItemValue(const std::shared_ptr& input) { function CheckNormalTensorStd (line 377) | Maybe CheckNormalTensorStd(const std::shared_ptr CheckNormalTensorStd(const float std) { FILE: oneflow/core/functional/impl/common.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/functional/impl/dataset_functor.cpp type oneflow (line 26) | namespace oneflow { type one (line 27) | namespace one { type functional (line 28) | namespace functional { type impl (line 30) | namespace impl { class ImageFlipFuntor (line 32) | class ImageFlipFuntor { method ImageFlipFuntor (line 34) | ImageFlipFuntor() { function ONEFLOW_FUNCTION_LIBRARY (line 49) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor... FILE: oneflow/core/functional/impl/eye_functor.cpp type oneflow (line 38) | namespace oneflow { type one (line 39) | namespace one { type functional (line 40) | namespace functional { type impl (line 42) | namespace impl { class EyeDevcieFunctor (line 44) | class EyeDevcieFunctor { method EyeDevcieFunctor (line 46) | EyeDevcieFunctor() { op_ = CHECK_JUST(one::OpBuilder("eye").Ou... class EyeDeviceStrFunctor (line 63) | class EyeDeviceStrFunctor { class GlobalEyeSbpListFunctor (line 73) | class GlobalEyeSbpListFunctor { method GlobalEyeSbpListFunctor (line 75) | GlobalEyeSbpListFunctor() { op_ = CHECK_JUST(one::OpBuilder("e... method FOR_RANGE (line 85) | FOR_RANGE(int32_t, i, 0, sbp_tuple.size()) { class GlobalEyeSbpFunctor (line 115) | class GlobalEyeSbpFunctor { class EyeInplaceFunctor (line 128) | class EyeInplaceFunctor { method EyeInplaceFunctor (line 130) | EyeInplaceFunctor() { op_ = CHECK_JUST(one::OpBuilder("eye").Out... function ONEFLOW_FUNCTION_LIBRARY (line 149) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/fused_attention_functor.cpp type oneflow (line 31) | namespace oneflow { type one (line 32) | namespace one { type functional (line 33) | namespace functional { type impl (line 35) | namespace impl { function ParseDims (line 39) | Maybe ParseDims(const std::string& name, const Shape& shap... function ParseDims (line 205) | Maybe ParseDims(const std::string& name, const Shape& shap... class FusedMultiHeadAttentionInferenceFunctor (line 215) | class FusedMultiHeadAttentionInferenceFunctor { method FusedMultiHeadAttentionInferenceFunctor (line 217) | FusedMultiHeadAttentionInferenceFunctor() = default; class FusedMultiHeadAttentionInferenceV2Functor (line 244) | class FusedMultiHeadAttentionInferenceV2Functor { type OpExprCacheKey (line 246) | struct OpExprCacheKey { type OpExprCacheKeyHash (line 255) | struct OpExprCacheKeyHash { method FusedMultiHeadAttentionInferenceV2Functor (line 262) | FusedMultiHeadAttentionInferenceV2Functor() { class FusedAttentionConcatPastKeyValueFunctor (line 530) | class FusedAttentionConcatPastKeyValueFunctor { method FusedAttentionConcatPastKeyValueFunctor (line 532) | FusedAttentionConcatPastKeyValueFunctor() { class FusedApplyRotaryEmbFunctor (line 616) | class FusedApplyRotaryEmbFunctor { method FusedApplyRotaryEmbFunctor (line 618) | FusedApplyRotaryEmbFunctor() { function ONEFLOW_FUNCTION_LIBRARY (line 738) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/global_cast.cpp type oneflow (line 51) | namespace oneflow { type one (line 52) | namespace one { type functional (line 53) | namespace functional { type impl (line 55) | namespace impl { function ParseEagerLocalToGlobalBalancedOverride (line 62) | Optional ParseEagerLocalToGlobalBalancedOverride() { function NeedSyncAndCheckShapeAndDtype (line 71) | bool NeedSyncAndCheckShapeAndDtype(bool check_meta_hint) { function New (line 84) | static Maybe New() { function New (line 89) | static Maybe New(const Shape& shape, DataT... function Check (line 95) | Maybe Check(const Shape& shape, DataType dtype) const { function Check (line 103) | Maybe Check(const FlatShapeAndDataType& flat_shape_dtype) ... function ToShape (line 111) | Maybe ToShape(Shape* shape) const { return this->shape().T... function ToShape (line 112) | Maybe ToShape() const { return shape().ToShape(); } function At (line 113) | int64_t At(int i) const { return shape().At(i); } function NumAxes (line 114) | int64_t NumAxes() const { return shape().NumAxes(); } function ShapeAndDataTypeConsistencyCheck (line 123) | Maybe ShapeAndDataTypeConsistencyCheck(const Symbol>> B... function FindRoot (line 190) | Maybe FindRoot(Symbol broadcast_parallel_... function BroadcastShapeAndDtype (line 200) | Maybe BroadcastShapeAndDtype(const Shape& ... function GetConcatenatedShapeAndCheckDtype (line 221) | Maybe GetConcatenatedShapeAndCheckDtype( function GetLogicalShapeAndDataType (line 325) | Maybe GetLogicalShapeAndDataType(Shape* logical_shape, Dat... function CheckNdSbpValid (line 354) | Maybe CheckNdSbpValid(Symbol nd_sbp, const Shape& l... function RawGetGlobalToGlobalOpExpr (line 369) | Maybe RawGetGlobalToGlobalOpExpr( function GlobalToGlobal (line 382) | Maybe GlobalToGlobal(const std::shared_ptr& x, S... function LocalToGlobal (line 415) | Maybe LocalToGlobal(const std::shared_ptr& x, Sy... class LocalToGlobalFunctor (line 471) | class LocalToGlobalFunctor { method LocalToGlobalFunctor (line 473) | LocalToGlobalFunctor() { class ToGlobalFunctor (line 512) | class ToGlobalFunctor { method ToGlobalFunctor (line 514) | ToGlobalFunctor() { class GlobalToLocalFunctor (line 558) | class GlobalToLocalFunctor { method GlobalToLocalFunctor (line 560) | GlobalToLocalFunctor() { function ONEFLOW_FUNCTION_LIBRARY (line 581) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/gradient_accumulation_functor.cpp type oneflow (line 25) | namespace oneflow { type one (line 26) | namespace one { type functional (line 27) | namespace functional { type impl (line 29) | namespace impl { class GradAccRepeatFunctor (line 31) | class GradAccRepeatFunctor { method GradAccRepeatFunctor (line 33) | GradAccRepeatFunctor() { class GradAccCollectFunctor (line 46) | class GradAccCollectFunctor { method GradAccCollectFunctor (line 48) | GradAccCollectFunctor() { class GradAccPackFunctor (line 61) | class GradAccPackFunctor { method GradAccPackFunctor (line 63) | GradAccPackFunctor() { class GradAccUnpackFunctor (line 76) | class GradAccUnpackFunctor { method GradAccUnpackFunctor (line 78) | GradAccUnpackFunctor() { function ONEFLOW_FUNCTION_LIBRARY (line 93) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/higher_derivative_functor.cpp type oneflow (line 32) | namespace oneflow { type one (line 33) | namespace one { type functional (line 34) | namespace functional { type impl (line 36) | namespace impl { class SinGradGradFunctor (line 38) | class SinGradGradFunctor { class CosGradGradFunctor (line 50) | class CosGradGradFunctor { class TanGradGradFunctor (line 62) | class TanGradGradFunctor { class SinhGradGradFunctor (line 76) | class SinhGradGradFunctor { class CoshGradGradFunctor (line 85) | class CoshGradGradFunctor { class TanhGradGradFunctor (line 94) | class TanhGradGradFunctor { class AsinGradGradFunctor (line 108) | class AsinGradGradFunctor { class AcosGradGradFunctor (line 126) | class AcosGradGradFunctor { class AtanGradGradFunctor (line 145) | class AtanGradGradFunctor { class AsinhGradGradFunctor (line 164) | class AsinhGradGradFunctor { class AcoshGradGradFunctor (line 183) | class AcoshGradGradFunctor { class AtanhGradGradFunctor (line 204) | class AtanhGradGradFunctor { class ErfGradGradFunctor (line 223) | class ErfGradGradFunctor { class ErfcGradGradFunctor (line 232) | class ErfcGradGradFunctor { class ExpGradGradFunctor (line 241) | class ExpGradGradFunctor { class Exp2GradGradFunctor (line 249) | class Exp2GradGradFunctor { class Expm1GradGradFunctor (line 257) | class Expm1GradGradFunctor { class LogGradGradFunctor (line 265) | class LogGradGradFunctor { class Log2GradGradFunctor (line 273) | class Log2GradGradFunctor { class Log10GradGradFunctor (line 283) | class Log10GradGradFunctor { class Log1pGradGradFunctor (line 293) | class Log1pGradGradFunctor { class LogSigmoidGradGradFunctor (line 302) | class LogSigmoidGradGradFunctor { class ReciprocalGradGradFunctor (line 311) | class ReciprocalGradGradFunctor { class ReciprocalNoNanGradGradFunctor (line 319) | class ReciprocalNoNanGradGradFunctor { class RsqrtGradGradFunctor (line 328) | class RsqrtGradGradFunctor { class SqrtGradGradFunctor (line 338) | class SqrtGradGradFunctor { class SquareGradGradFunctor (line 347) | class SquareGradGradFunctor { class SigmoidGradGradFunctor (line 355) | class SigmoidGradGradFunctor { class SiluGradGradFunctor (line 364) | class SiluGradGradFunctor { class SeluGradGradFunctor (line 389) | class SeluGradGradFunctor { class SoftSignGradGradFunctor (line 401) | class SoftSignGradGradFunctor { class HardSigmoidGradGradFunctor (line 423) | class HardSigmoidGradGradFunctor { class HardSwishGradGradFunctor (line 431) | class HardSwishGradGradFunctor { class SoftplusGradGradFunctor (line 442) | class SoftplusGradGradFunctor { class EluGradGradFunctor (line 465) | class EluGradGradFunctor { class CeluGradGradFunctor (line 476) | class CeluGradGradFunctor { class MaxPoolNdGradGradFunctor (line 488) | class MaxPoolNdGradGradFunctor { class MishGradGradFunctor (line 507) | class MishGradGradFunctor { class GeluGradGradFunctor (line 537) | class GeluGradGradFunctor { function ONEFLOW_FUNCTION_LIBRARY (line 560) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/linalg_functor.cpp type oneflow (line 33) | namespace oneflow { type one (line 34) | namespace one { type functional (line 35) | namespace functional { type impl (line 37) | namespace impl { type linalg (line 38) | namespace linalg { class CrossFunctor (line 40) | class CrossFunctor { method CrossFunctor (line 42) | CrossFunctor() { method FindValidDim (line 111) | Maybe FindValidDim(const Shape& shape) const { function ONEFLOW_FUNCTION_LIBRARY (line 132) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("LinalgC... FILE: oneflow/core/functional/impl/math_functor.cpp type oneflow (line 34) | namespace oneflow { type one (line 35) | namespace one { type functional (line 36) | namespace functional { type impl (line 37) | namespace impl { class AddNFunctor (line 39) | class AddNFunctor { method AddNFunctor (line 41) | AddNFunctor() { class ScalarMathBaseFunctor (line 73) | class ScalarMathBaseFunctor { method ScalarMathBaseFunctor (line 75) | explicit ScalarMathBaseFunctor(std::string op_name) { class ScalarAddFunctor (line 132) | class ScalarAddFunctor : public ScalarMathBaseFunctor { method ScalarAddFunctor (line 134) | ScalarAddFunctor() : ScalarMathBaseFunctor(/*op_name=*/"scalar... class ScalarAdd2Functor (line 153) | class ScalarAdd2Functor { class ScalarSubFunctor (line 174) | class ScalarSubFunctor { class ScalarSub2Functor (line 182) | class ScalarSub2Functor { class ScalarMulFunctor (line 190) | class ScalarMulFunctor : public ScalarMathBaseFunctor { method ScalarMulFunctor (line 192) | ScalarMulFunctor() : ScalarMathBaseFunctor(/*op_name=*/"scalar... class ScalarMul2Functor (line 195) | class ScalarMul2Functor { class InplaceScalarMulFunctor (line 202) | class InplaceScalarMulFunctor : public ScalarMathBaseFunctor { method InplaceScalarMulFunctor (line 204) | InplaceScalarMulFunctor() : ScalarMathBaseFunctor(/*op_name=*/... class ScalarDivFunctor (line 210) | class ScalarDivFunctor : public ScalarMathBaseFunctor { method ScalarDivFunctor (line 212) | ScalarDivFunctor() : ScalarMathBaseFunctor(/*op_name=*/"scalar... class ScalarDivModeFunctor (line 218) | class ScalarDivModeFunctor { class ScalarDiv2Functor (line 239) | class ScalarDiv2Functor { class ScalarDivMode2Functor (line 246) | class ScalarDivMode2Functor { class InplaceScalarDivFunctor (line 267) | class InplaceScalarDivFunctor : public ScalarMathBaseFunctor { method InplaceScalarDivFunctor (line 269) | InplaceScalarDivFunctor() : ScalarMathBaseFunctor(/*op_name=*/... class ScalarPowFunctor (line 275) | class ScalarPowFunctor : public ScalarMathBaseFunctor { method ScalarPowFunctor (line 277) | ScalarPowFunctor() : ScalarMathBaseFunctor(/*op_name=*/"scalar... class ScalarPowGradFunctor (line 280) | class ScalarPowGradFunctor { method ScalarPowGradFunctor (line 282) | ScalarPowGradFunctor() { class ScalarReversePowFunctor (line 303) | class ScalarReversePowFunctor : public ScalarMathBaseFunctor { method ScalarReversePowFunctor (line 305) | ScalarReversePowFunctor() : ScalarMathBaseFunctor(/*op_name=*/... class ScalarReversePowGradFunctor (line 311) | class ScalarReversePowGradFunctor { method ScalarReversePowGradFunctor (line 313) | ScalarReversePowGradFunctor() { class ScalarFloorDivFunctor (line 335) | class ScalarFloorDivFunctor : public ScalarMathBaseFunctor { method ScalarFloorDivFunctor (line 337) | ScalarFloorDivFunctor() : ScalarMathBaseFunctor(/*op_name=*/"s... class ScalarTruncDivFunctor (line 340) | class ScalarTruncDivFunctor : public ScalarMathBaseFunctor { method ScalarTruncDivFunctor (line 342) | ScalarTruncDivFunctor() : ScalarMathBaseFunctor(/*op_name=*/"s... class ScalarFModFunctor (line 345) | class ScalarFModFunctor : public ScalarMathBaseFunctor { method ScalarFModFunctor (line 347) | ScalarFModFunctor() : ScalarMathBaseFunctor(/*op_name=*/"scala... class ReduceMaxFunctor (line 350) | class ReduceMaxFunctor { method ReduceMaxFunctor (line 352) | ReduceMaxFunctor() { class ReduceMinFunctor (line 373) | class ReduceMinFunctor { method ReduceMinFunctor (line 375) | ReduceMinFunctor() { class MaxFunctor (line 396) | class MaxFunctor { class Max2Functor (line 405) | class Max2Functor { class MinFunctor (line 418) | class MinFunctor { class Min2Functor (line 427) | class Min2Functor { class AminFunctor (line 440) | class AminFunctor { class AmaxFunctor (line 453) | class AmaxFunctor { class ReduceSumWholeFunctor (line 466) | class ReduceSumWholeFunctor { method ReduceSumWholeFunctor (line 468) | ReduceSumWholeFunctor() { class ReduceSumFunctor (line 495) | class ReduceSumFunctor { method ReduceSumFunctor (line 497) | ReduceSumFunctor() { class ReduceNanSumFunctor (line 522) | class ReduceNanSumFunctor { method ReduceNanSumFunctor (line 524) | ReduceNanSumFunctor() { class ReduceNanSumWholeFunctor (line 550) | class ReduceNanSumWholeFunctor { method ReduceNanSumWholeFunctor (line 552) | ReduceNanSumWholeFunctor() { class ReduceAllWholeFunctor (line 580) | class ReduceAllWholeFunctor { method ReduceAllWholeFunctor (line 582) | ReduceAllWholeFunctor() { class ReduceAllFunctor (line 598) | class ReduceAllFunctor { method ReduceAllFunctor (line 600) | ReduceAllFunctor() { class ReduceAnyWholeFunctor (line 617) | class ReduceAnyWholeFunctor { method ReduceAnyWholeFunctor (line 619) | ReduceAnyWholeFunctor() { class ReduceAnyFunctor (line 635) | class ReduceAnyFunctor { method ReduceAnyFunctor (line 637) | ReduceAnyFunctor() { class ReduceDeviceStageBaseFunctor (line 655) | class ReduceDeviceStageBaseFunctor { method ReduceDeviceStageBaseFunctor (line 657) | ReduceDeviceStageBaseFunctor() class ReduceDeviceStageGradBaseFunctor (line 677) | class ReduceDeviceStageGradBaseFunctor { method ReduceDeviceStageGradBaseFunctor (line 679) | ReduceDeviceStageGradBaseFunctor() class ReduceMinDeviceStageFunctor (line 700) | class ReduceMinDeviceStageFunctor method GetOpName (line 703) | static std::string GetOpName() { return "reduce_min_device_sta... class ReduceMaxDeviceStageFunctor (line 706) | class ReduceMaxDeviceStageFunctor method GetOpName (line 709) | static std::string GetOpName() { return "reduce_max_device_sta... class ReduceMinDeviceStageGradFunctor (line 712) | class ReduceMinDeviceStageGradFunctor method GetOpName (line 715) | static std::string GetOpName() { return "reduce_min_device_sta... class ReduceMaxDeviceStageGradFunctor (line 718) | class ReduceMaxDeviceStageGradFunctor method GetOpName (line 721) | static std::string GetOpName() { return "reduce_max_device_sta... class ReduceGlobalStageBaseFunctor (line 725) | class ReduceGlobalStageBaseFunctor { method ReduceGlobalStageBaseFunctor (line 727) | ReduceGlobalStageBaseFunctor() class ReduceGlobalStageGradBaseFunctor (line 748) | class ReduceGlobalStageGradBaseFunctor { method ReduceGlobalStageGradBaseFunctor (line 750) | ReduceGlobalStageGradBaseFunctor() class ReduceMinGlobalStageFunctor (line 771) | class ReduceMinGlobalStageFunctor method GetOpName (line 774) | static std::string GetOpName() { return "reduce_min_global_sta... class ReduceMinGlobalStageGradFunctor (line 777) | class ReduceMinGlobalStageGradFunctor method GetOpName (line 780) | static std::string GetOpName() { return "reduce_min_global_sta... class ReduceMaxGlobalStageFunctor (line 783) | class ReduceMaxGlobalStageFunctor method GetOpName (line 786) | static std::string GetOpName() { return "reduce_max_global_sta... class ReduceMaxGlobalStageGradFunctor (line 789) | class ReduceMaxGlobalStageGradFunctor method GetOpName (line 792) | static std::string GetOpName() { return "reduce_max_global_sta... class ReduceMeanWholeFunctor (line 795) | class ReduceMeanWholeFunctor { method ReduceMeanWholeFunctor (line 797) | ReduceMeanWholeFunctor() {} class ReduceMeanFunctor (line 811) | class ReduceMeanFunctor { method ReduceMeanFunctor (line 813) | ReduceMeanFunctor() {} class ReduceProdWholeFunctor (line 835) | class ReduceProdWholeFunctor { method ReduceProdWholeFunctor (line 837) | ReduceProdWholeFunctor() { class MedianFunctor (line 869) | class MedianFunctor { method MedianFunctor (line 871) | MedianFunctor() { class MedianWithIndicesFunctor (line 889) | class MedianWithIndicesFunctor { method MedianWithIndicesFunctor (line 891) | MedianWithIndicesFunctor() { class ModeFunctor (line 926) | class ModeFunctor { method ModeFunctor (line 928) | ModeFunctor() { class ReduceProdFunctor (line 960) | class ReduceProdFunctor { method ReduceProdFunctor (line 962) | ReduceProdFunctor() { class LogSumExpFunctor (line 994) | class LogSumExpFunctor { method LogSumExpFunctor (line 996) | LogSumExpFunctor() {} method SqueezeMultiple (line 1019) | Maybe SqueezeMultiple(const std::shared_ptr quantile_output_shape(const... class ScalarQuantileFunctor (line 1198) | class ScalarQuantileFunctor { method ScalarQuantileFunctor (line 1200) | ScalarQuantileFunctor() {} method quantile_output_shape (line 1313) | static inline std::vector quantile_output_shape(const... class TransposeFunctor (line 1335) | class TransposeFunctor { method TransposeFunctor (line 1337) | TransposeFunctor() { class Transpose2dimFunctor (line 1363) | class Transpose2dimFunctor { method Transpose2dimFunctor (line 1365) | Transpose2dimFunctor() { class AsStridedFunctor (line 1392) | class AsStridedFunctor { method AsStridedFunctor (line 1394) | AsStridedFunctor() { class AsStridedGradFunctor (line 1419) | class AsStridedGradFunctor { method AsStridedGradFunctor (line 1421) | AsStridedGradFunctor() { class InplaceAsStridedFunctor (line 1441) | class InplaceAsStridedFunctor { class ArangeFunctor (line 1460) | class ArangeFunctor { method ArangeFunctor (line 1462) | ArangeFunctor() { op_ = CHECK_JUST(one::OpBuilder("arange").Ou... class Arange2Functor (line 1502) | class Arange2Functor { class GlobalArangeFunctor (line 1510) | class GlobalArangeFunctor { method GlobalArangeFunctor (line 1512) | GlobalArangeFunctor() { op_ = CHECK_JUST(one::OpBuilder("arang... class GlobalArange2Functor (line 1556) | class GlobalArange2Functor { class HannWindowFunctor (line 1566) | class HannWindowFunctor { class GlobalHannWindowFunctor (line 1602) | class GlobalHannWindowFunctor { class CastFunctor (line 1634) | class CastFunctor { method CastFunctor (line 1636) | CastFunctor() { op_ = CHECK_JUST(one::OpBuilder("cast").Input(... class ClampBaseFunctor (line 1655) | class ClampBaseFunctor { method ClampBaseFunctor (line 1657) | ClampBaseFunctor() { class ClampFunctor (line 1722) | class ClampFunctor : public ClampBaseFunctor { class ClampMinFunctor (line 1730) | class ClampMinFunctor : public ClampBaseFunctor { class ClampMaxFunctor (line 1737) | class ClampMaxFunctor : public ClampBaseFunctor { class ClampInplaceFunctor (line 1744) | class ClampInplaceFunctor : public ClampBaseFunctor { class ClampMinInplaceFunctor (line 1752) | class ClampMinInplaceFunctor : public ClampBaseFunctor { class ClampMaxInplaceFunctor (line 1759) | class ClampMaxInplaceFunctor : public ClampBaseFunctor { class ClipFunctor (line 1766) | class ClipFunctor { class ClipInplaceFunctor (line 1774) | class ClipInplaceFunctor { class SqrtSquareSumFunctor (line 1781) | class SqrtSquareSumFunctor { method SqrtSquareSumFunctor (line 1783) | SqrtSquareSumFunctor() { class VectorNormFunctor (line 1794) | class VectorNormFunctor { method VectorNormFunctor (line 1796) | VectorNormFunctor() {} class ScalarVectorNormFunctor (line 1862) | class ScalarVectorNormFunctor { method ScalarVectorNormFunctor (line 1864) | ScalarVectorNormFunctor() {} class ScalarMatrixNormFunctor (line 1892) | class ScalarMatrixNormFunctor { method ScalarMatrixNormFunctor (line 1894) | ScalarMatrixNormFunctor() {} class MatrixNormFunctor (line 1961) | class MatrixNormFunctor { method MatrixNormFunctor (line 1963) | MatrixNormFunctor() {} class NormFunctor (line 2008) | class NormFunctor { method NormFunctor (line 2010) | NormFunctor() {} class Norm2Functor (line 2089) | class Norm2Functor { method Norm2Functor (line 2091) | Norm2Functor() {} class ScalarNormFunctor (line 2122) | class ScalarNormFunctor { method ScalarNormFunctor (line 2124) | ScalarNormFunctor() {} class ScalarNorm2Functor (line 2152) | class ScalarNorm2Functor { method ScalarNorm2Functor (line 2154) | ScalarNorm2Functor() {} class InvFunctor (line 2182) | class InvFunctor { method InvFunctor (line 2184) | InvFunctor() { op_ = CHECK_JUST(one::OpBuilder("inv").Input("x... class DetFunctor (line 2201) | class DetFunctor { method DetFunctor (line 2203) | DetFunctor() { method GetPivotDet (line 2208) | Maybe GetPivotDet(const std::shared_ptr& pivot... class ClampGradFunctor (line 2267) | class ClampGradFunctor { method ClampGradFunctor (line 2269) | ClampGradFunctor() { class SelectFunctor (line 2326) | class SelectFunctor { method SelectFunctor (line 2328) | SelectFunctor() = default; class SelectTopNFunctor (line 2353) | class SelectTopNFunctor { method SelectTopNFunctor (line 2355) | SelectTopNFunctor() { op_ = CHECK_JUST(one::SelectTopNOpExpr::... class MinimumFunctor (line 2374) | class MinimumFunctor { method MinimumFunctor (line 2376) | MinimumFunctor() { class MaximumFunctor (line 2405) | class MaximumFunctor { method MaximumFunctor (line 2407) | MaximumFunctor() { class ScalarLogicalBaseFunctor (line 2436) | class ScalarLogicalBaseFunctor { method ScalarLogicalBaseFunctor (line 2438) | explicit ScalarLogicalBaseFunctor(std::string op_name) { class ScalarLogicalEqualFunctor (line 2480) | class ScalarLogicalEqualFunctor : public ScalarLogicalBaseFunctor { method ScalarLogicalEqualFunctor (line 2482) | ScalarLogicalEqualFunctor() : ScalarLogicalBaseFunctor(/*op_na... class ScalarLogicalEqual2Functor (line 2486) | class ScalarLogicalEqual2Functor { class ScalarLogicalNotEqualFunctor (line 2493) | class ScalarLogicalNotEqualFunctor : public ScalarLogicalBaseFun... method ScalarLogicalNotEqualFunctor (line 2495) | ScalarLogicalNotEqualFunctor() class ScalarLogicalNotEqual2Functor (line 2500) | class ScalarLogicalNotEqual2Functor { class ScalarLogicalGreaterFunctor (line 2507) | class ScalarLogicalGreaterFunctor : public ScalarLogicalBaseFunc... method ScalarLogicalGreaterFunctor (line 2509) | ScalarLogicalGreaterFunctor() : ScalarLogicalBaseFunctor(/*op_... class ScalarLogicalGreater2Functor (line 2513) | class ScalarLogicalGreater2Functor { class InplaceScalarLogicalGreaterFunctor (line 2520) | class InplaceScalarLogicalGreaterFunctor { method InplaceScalarLogicalGreaterFunctor (line 2522) | InplaceScalarLogicalGreaterFunctor() { class ScalarLogicalGreaterEqualFunctor (line 2568) | class ScalarLogicalGreaterEqualFunctor : public ScalarLogicalBas... method ScalarLogicalGreaterEqualFunctor (line 2570) | ScalarLogicalGreaterEqualFunctor() class ScalarLogicalGreaterEqual2Functor (line 2575) | class ScalarLogicalGreaterEqual2Functor { class ScalarLogicalLessFunctor (line 2582) | class ScalarLogicalLessFunctor : public ScalarLogicalBaseFunctor { method ScalarLogicalLessFunctor (line 2584) | ScalarLogicalLessFunctor() : ScalarLogicalBaseFunctor(/*op_nam... class ScalarLogicalLess2Functor (line 2588) | class ScalarLogicalLess2Functor { class ScalarLogicalLessEqualFunctor (line 2595) | class ScalarLogicalLessEqualFunctor : public ScalarLogicalBaseFu... method ScalarLogicalLessEqualFunctor (line 2597) | ScalarLogicalLessEqualFunctor() class ScalarLogicalLessEqual2Functor (line 2602) | class ScalarLogicalLessEqual2Functor { class ScalarLogicalAndFunctor (line 2609) | class ScalarLogicalAndFunctor : public ScalarLogicalBaseFunctor { method ScalarLogicalAndFunctor (line 2611) | ScalarLogicalAndFunctor() : ScalarLogicalBaseFunctor(/*op_name... class ScalarLogicalAnd2Functor (line 2615) | class ScalarLogicalAnd2Functor { class ScalarLogicalOrFunctor (line 2622) | class ScalarLogicalOrFunctor : public ScalarLogicalBaseFunctor { method ScalarLogicalOrFunctor (line 2624) | ScalarLogicalOrFunctor() : ScalarLogicalBaseFunctor(/*op_name=... class ScalarLogicalOr2Functor (line 2628) | class ScalarLogicalOr2Functor { class ScalarLogicalXorFunctor (line 2635) | class ScalarLogicalXorFunctor : public ScalarLogicalBaseFunctor { method ScalarLogicalXorFunctor (line 2637) | ScalarLogicalXorFunctor() : ScalarLogicalBaseFunctor(/*op_name... class ScalarLogicalXor2Functor (line 2641) | class ScalarLogicalXor2Functor { class ScalarBitwiseBaseFunctor (line 2648) | class ScalarBitwiseBaseFunctor { method ScalarBitwiseBaseFunctor (line 2650) | explicit ScalarBitwiseBaseFunctor(std::string op_name) { class ScalarLerpFunctor (line 2679) | class ScalarLerpFunctor { method ScalarLerpFunctor (line 2681) | ScalarLerpFunctor() { class ScalarInplaceLerpFunctor (line 2723) | class ScalarInplaceLerpFunctor { method ScalarInplaceLerpFunctor (line 2725) | ScalarInplaceLerpFunctor() { class ScalarLerpGradFunctor (line 2786) | class ScalarLerpGradFunctor { method ScalarLerpGradFunctor (line 2788) | ScalarLerpGradFunctor() { class ScalarBitwiseAndFunctor (line 2817) | class ScalarBitwiseAndFunctor : public ScalarBitwiseBaseFunctor { method ScalarBitwiseAndFunctor (line 2819) | ScalarBitwiseAndFunctor() : ScalarBitwiseBaseFunctor(/*op_name... class ScalarBitwiseAnd2Functor (line 2822) | class ScalarBitwiseAnd2Functor { class ScalarBitwiseOrFunctor (line 2829) | class ScalarBitwiseOrFunctor : public ScalarBitwiseBaseFunctor { method ScalarBitwiseOrFunctor (line 2831) | ScalarBitwiseOrFunctor() : ScalarBitwiseBaseFunctor(/*op_name=... class ScalarBitwiseOr2Functor (line 2834) | class ScalarBitwiseOr2Functor { class ScalarBitwiseXorFunctor (line 2841) | class ScalarBitwiseXorFunctor : public ScalarBitwiseBaseFunctor { method ScalarBitwiseXorFunctor (line 2843) | ScalarBitwiseXorFunctor() : ScalarBitwiseBaseFunctor(/*op_name... class ScalarBitwiseXor2Functor (line 2846) | class ScalarBitwiseXor2Functor { class StandardDeviationFunctor (line 2853) | class StandardDeviationFunctor { class VarianceFunctor (line 2935) | class VarianceFunctor { method VarianceFunctor (line 2937) | VarianceFunctor() { class RMSLayerNormalizationFunctor (line 2970) | class RMSLayerNormalizationFunctor { class DotFunctor (line 2993) | class DotFunctor { method DotFunctor (line 2995) | DotFunctor() { class MovedimVecFunctor (line 3006) | class MovedimVecFunctor { method MovedimVecFunctor (line 3008) | MovedimVecFunctor() = default; method CheckNoRepeat (line 3009) | static Maybe CheckNoRepeat(const std::vector& p... method FOR_RANGE (line 3047) | FOR_RANGE(size_t, i, 0, dim) { method FOR_RANGE (line 3057) | FOR_RANGE(size_t, i, 0, rest_dim) { order[destination_dims[i]]... class MovedimIntFunctor (line 3063) | class MovedimIntFunctor { method MovedimIntFunctor (line 3065) | MovedimIntFunctor() = default; class TensorSplitVecFunctor (line 3074) | class TensorSplitVecFunctor { method TensorSplitVecFunctor (line 3076) | TensorSplitVecFunctor() = default; class TensorSplitIntFunctor (line 3103) | class TensorSplitIntFunctor { method TensorSplitIntFunctor (line 3105) | TensorSplitIntFunctor() = default; class HsplitIntFunctor (line 3135) | class HsplitIntFunctor { method HsplitIntFunctor (line 3137) | HsplitIntFunctor() = default; class HsplitVecFunctor (line 3154) | class HsplitVecFunctor { method HsplitVecFunctor (line 3156) | HsplitVecFunctor() = default; class VsplitIntFunctor (line 3168) | class VsplitIntFunctor { method VsplitIntFunctor (line 3170) | VsplitIntFunctor() = default; class VsplitVecFunctor (line 3186) | class VsplitVecFunctor { method VsplitVecFunctor (line 3188) | VsplitVecFunctor() = default; class ErfinvFunctor (line 3199) | class ErfinvFunctor { method ErfinvFunctor (line 3201) | ErfinvFunctor() { op_ = CHECK_JUST(one::OpBuilder("erfinv").In... class ErfinvInplaceFunctor (line 3210) | class ErfinvInplaceFunctor { method ErfinvInplaceFunctor (line 3212) | ErfinvInplaceFunctor() { class GeluWithApproximateFunctor (line 3227) | class GeluWithApproximateFunctor { class CumBaseFunctor (line 3239) | class CumBaseFunctor { method CumBaseFunctor (line 3241) | explicit CumBaseFunctor(std::string op_name) { class CumsumFunctor (line 3265) | class CumsumFunctor : public CumBaseFunctor { method CumsumFunctor (line 3267) | CumsumFunctor() : CumBaseFunctor("cumsum") {} class CumProdFunctor (line 3270) | class CumProdFunctor : public CumBaseFunctor { method CumProdFunctor (line 3272) | CumProdFunctor() : CumBaseFunctor("cumprod") {} class CumGradBaseFunctor (line 3275) | class CumGradBaseFunctor { class CumProdGradFunctor (line 3280) | class CumProdGradFunctor : public CumGradBaseFunctor { method CumProdGradFunctor (line 3282) | CumProdGradFunctor() { function sumproduct_pair (line 3307) | static Maybe sumproduct_pair(const std::shared_ptr<... function einsum_check_label (line 3439) | bool einsum_check_label(unsigned char label) { return std::isalp... function einsum_label_to_index (line 3441) | uint8_t einsum_label_to_index(unsigned char label) { function einsum_index_to_label (line 3446) | unsigned char einsum_index_to_label(uint8_t index) { class EinSumFunctor (line 3462) | class EinSumFunctor { method EinSumFunctor (line 3464) | EinSumFunctor() {} class TruncFunctor (line 3870) | class TruncFunctor { method TruncFunctor (line 3872) | TruncFunctor() { op_ = CHECK_JUST(one::OpBuilder("trunc").Inpu... class AddCDivFunctor (line 3880) | class AddCDivFunctor { method AddCDivFunctor (line 3882) | AddCDivFunctor() {} class InplaceAddCDivFunctor (line 3890) | class InplaceAddCDivFunctor { method InplaceAddCDivFunctor (line 3892) | InplaceAddCDivFunctor() {} type fft_norm_mode (line 3907) | enum class fft_norm_mode { function use_optimized_cufft_path (line 3913) | bool use_optimized_cufft_path(const std::vector& fft_di... function fft_norm_mode (line 3925) | static fft_norm_mode fft_norm_from_string(const Optional resize_fft_input(const std::shared_ptr> promote_type_fft(Symbol type, bool... method promote_tensor_fft (line 4021) | Maybe promote_tensor_fft(const std::shared_ptr... method maybe_wrap_dims (line 4034) | Maybe maybe_wrap_dims(std::vector& dims, int64_... method calculate_fftn_shape_and_dims (line 4057) | Maybe calculate_fftn_shape_and_dims(const std::shared_pt... method parse_input_n_and_dims (line 4094) | Maybe parse_input_n_and_dims(const std::shared_ptr permute_and_reshape(const std::shared_ptr parse_c2r_input_n_and_dims(const std::shared_ptr apply_reduction(const Maybe& x, const st... method LossFunctorBase (line 1609) | LossFunctorBase() = default; class MseLossFunctor (line 1613) | class MseLossFunctor : public LossFunctorBase { method MseLossFunctor (line 1615) | MseLossFunctor() {} class L1LossFunctor (line 1626) | class L1LossFunctor : public LossFunctorBase { method L1LossFunctor (line 1628) | L1LossFunctor() {} class SmoothL1LossFunctor (line 1639) | class SmoothL1LossFunctor : LossFunctorBase { method SmoothL1LossFunctor (line 1641) | SmoothL1LossFunctor() { class KLDivLossFunctor (line 1657) | class KLDivLossFunctor : public LossFunctorBase { method KLDivLossFunctor (line 1659) | KLDivLossFunctor() { class MarginRankingLossFunctor (line 1682) | class MarginRankingLossFunctor : public LossFunctorBase { class BinaryCrossEntropyLossFunctor (line 1701) | class BinaryCrossEntropyLossFunctor : public LossFunctorBase { method BinaryCrossEntropyLossFunctor (line 1703) | BinaryCrossEntropyLossFunctor() { class BinaryCrossEntropyWithLogitsLossFunctor (line 1730) | class BinaryCrossEntropyWithLogitsLossFunctor : public LossFunct... method BinaryCrossEntropyWithLogitsLossFunctor (line 1732) | BinaryCrossEntropyWithLogitsLossFunctor() { class NLLLossFunctor (line 1814) | class NLLLossFunctor { method NLLLossFunctor (line 1816) | NLLLossFunctor() { class CrossEntropyFunctor (line 1908) | class CrossEntropyFunctor { method CrossEntropyFunctor (line 1910) | CrossEntropyFunctor() { class CrossEntropyLabelSmoothingFunctor (line 1991) | class CrossEntropyLabelSmoothingFunctor { method CrossEntropyLabelSmoothingFunctor (line 1993) | CrossEntropyLabelSmoothingFunctor() { class CrossEntropyProbFunctor (line 2085) | class CrossEntropyProbFunctor : public LossFunctorBase { method CrossEntropyProbFunctor (line 2087) | CrossEntropyProbFunctor() { class SparseCrossEntropyFunctor (line 2138) | class SparseCrossEntropyFunctor { method SparseCrossEntropyFunctor (line 2140) | SparseCrossEntropyFunctor() { class SparseCrossEntropyMsFunctor (line 2159) | class SparseCrossEntropyMsFunctor { method SparseCrossEntropyMsFunctor (line 2161) | SparseCrossEntropyMsFunctor() { class SparseSoftmaxCrossEntropyFunctor (line 2180) | class SparseSoftmaxCrossEntropyFunctor { method SparseSoftmaxCrossEntropyFunctor (line 2182) | SparseSoftmaxCrossEntropyFunctor() { method RunWithMsVersion (line 2237) | Maybe RunWithMsVersion(const std::shared_ptr SparseSoftmaxCrossEntropyOperator(const std::sha... method LazySparseSoftmaxCrossEntropyMsOperator (line 2275) | Maybe LazySparseSoftmaxCrossEntropyMsOperator( method EagerSparseSoftmaxCrossEntropyMsOperator (line 2285) | Maybe EagerSparseSoftmaxCrossEntropyMsOperator( class SoftmaxCrossEntropyFunctor (line 2385) | class SoftmaxCrossEntropyFunctor { method SoftmaxCrossEntropyFunctor (line 2387) | SoftmaxCrossEntropyFunctor() { class SoftmaxCrossEntropyGradFunctor (line 2405) | class SoftmaxCrossEntropyGradFunctor { method SoftmaxCrossEntropyGradFunctor (line 2407) | SoftmaxCrossEntropyGradFunctor() { class CombinedMarginLossFunctor (line 2425) | class CombinedMarginLossFunctor { method CombinedMarginLossFunctor (line 2427) | CombinedMarginLossFunctor() { class CtcLossFunctor (line 2447) | class CtcLossFunctor { method CtcLossFunctor (line 2449) | CtcLossFunctor() { class TripletMarginLossFunctor (line 2553) | class TripletMarginLossFunctor { method TripletMarginLossFunctor (line 2555) | TripletMarginLossFunctor() {} class AffineGridFunctor (line 2603) | class AffineGridFunctor { method AffineGridFunctor (line 2605) | AffineGridFunctor() { class GridSampleFunctor (line 2619) | class GridSampleFunctor { method GridSampleFunctor (line 2621) | GridSampleFunctor() { class NormalizationFunctor (line 2639) | class NormalizationFunctor { method NormalizationFunctor (line 2641) | NormalizationFunctor() { class NormalizationAddReluFunctor (line 2770) | class NormalizationAddReluFunctor { method NormalizationAddReluFunctor (line 2772) | NormalizationAddReluFunctor() { method BuildFusedNormalizationOp (line 2787) | Maybe BuildFusedNormalizationOp(bool stats, b... class ConstantPadFunctor (line 2857) | class ConstantPadFunctor { method ConstantPadFunctor (line 2859) | ConstantPadFunctor() { class ReflectionPadFunctor (line 2901) | class ReflectionPadFunctor { method ReflectionPadFunctor (line 2903) | ReflectionPadFunctor() { class ReplicationPadFunctor (line 3002) | class ReplicationPadFunctor { method ReplicationPadFunctor (line 3004) | ReplicationPadFunctor() { class PadFunctor (line 3089) | class PadFunctor { class DropoutFunctor (line 3107) | class DropoutFunctor { method DropoutFunctor (line 3109) | DropoutFunctor() { function MakeFeatureNoise (line 3162) | Maybe MakeFeatureNoise(const std::shared_ptr DropoutImpl(const std::shared_ptr& in... class Dropout1dFunctor (line 3194) | class Dropout1dFunctor { class Dropout2dFunctor (line 3216) | class Dropout2dFunctor { class Dropout3dFunctor (line 3244) | class Dropout3dFunctor { class DropoutGradFunctor (line 3269) | class DropoutGradFunctor { method DropoutGradFunctor (line 3271) | DropoutGradFunctor() { class UnfoldFunctor (line 3286) | class UnfoldFunctor { method UnfoldFunctor (line 3288) | UnfoldFunctor() { class FoldFunctor (line 3310) | class FoldFunctor { method FoldFunctor (line 3312) | FoldFunctor() { fold_op_ = CHECK_JUST(one::OpBuilder("fold").I... class OneHotFunctor (line 3333) | class OneHotFunctor { method OneHotFunctor (line 3335) | OneHotFunctor() { class PairwiseDistanceFunctor (line 3377) | class PairwiseDistanceFunctor { class CosineSimilarityFunctor (line 3389) | class CosineSimilarityFunctor { class L2NormalizeFunctor (line 3439) | class L2NormalizeFunctor { method L2NormalizeFunctor (line 3441) | L2NormalizeFunctor() { class NormalizeFunctor (line 3474) | class NormalizeFunctor { class FusedSelfAttentionFunctor (line 3493) | class FusedSelfAttentionFunctor { method FusedSelfAttentionFunctor (line 3495) | FusedSelfAttentionFunctor() { class FusedSelfAttentionGradFunctor (line 3513) | class FusedSelfAttentionGradFunctor { method FusedSelfAttentionGradFunctor (line 3515) | FusedSelfAttentionGradFunctor() { class FusedScaleTrilSoftmaxMaskScaleFunctor (line 3537) | class FusedScaleTrilSoftmaxMaskScaleFunctor { method FusedScaleTrilSoftmaxMaskScaleFunctor (line 3539) | FusedScaleTrilSoftmaxMaskScaleFunctor() { class L2NormalizeGradFunctor (line 3575) | class L2NormalizeGradFunctor { method L2NormalizeGradFunctor (line 3577) | L2NormalizeGradFunctor() { class FusedBiasAddGeluFunctor (line 3598) | class FusedBiasAddGeluFunctor { method FusedBiasAddGeluFunctor (line 3600) | FusedBiasAddGeluFunctor() { class FusedBiasAddGeluGradFunctor (line 3615) | class FusedBiasAddGeluGradFunctor { method FusedBiasAddGeluGradFunctor (line 3617) | FusedBiasAddGeluGradFunctor() { class FusedBiasAddDropoutFunctor (line 3637) | class FusedBiasAddDropoutFunctor { method FusedBiasAddDropoutFunctor (line 3639) | FusedBiasAddDropoutFunctor() { class FusedGluFunctor (line 3689) | class FusedGluFunctor { method FusedGluFunctor (line 3691) | FusedGluFunctor() { class FusedScaleTrilFunctor (line 3825) | class FusedScaleTrilFunctor { method FusedScaleTrilFunctor (line 3827) | FusedScaleTrilFunctor() { class FusedScaleMaskSoftmaxFunctor (line 3862) | class FusedScaleMaskSoftmaxFunctor { method FusedScaleMaskSoftmaxFunctor (line 3864) | FusedScaleMaskSoftmaxFunctor() { class FusedScaleMaskSoftmaxDropoutFunctor (line 3880) | class FusedScaleMaskSoftmaxDropoutFunctor { method FusedScaleMaskSoftmaxDropoutFunctor (line 3882) | FusedScaleMaskSoftmaxDropoutFunctor() { class FusedBiasAddScaleMaskSoftmaxDropoutFunctor (line 3930) | class FusedBiasAddScaleMaskSoftmaxDropoutFunctor { method FusedBiasAddScaleMaskSoftmaxDropoutFunctor (line 3932) | FusedBiasAddScaleMaskSoftmaxDropoutFunctor() { class CtcGreedyDecoderFunctor (line 3974) | class CtcGreedyDecoderFunctor { method CtcGreedyDecoderFunctor (line 3976) | CtcGreedyDecoderFunctor() { class PariticalFCSampleDisableBoxing (line 3996) | class PariticalFCSampleDisableBoxing { method PariticalFCSampleDisableBoxing (line 3998) | PariticalFCSampleDisableBoxing() { class NmsFunctor (line 4015) | class NmsFunctor { method NmsFunctor (line 4017) | NmsFunctor() { op_ = CHECK_JUST(one::OpBuilder("nms").Input("i... class RoiAlignFunctor (line 4030) | class RoiAlignFunctor { method RoiAlignFunctor (line 4032) | RoiAlignFunctor() { class RoiAlignGradFunctor (line 4050) | class RoiAlignGradFunctor { method RoiAlignGradFunctor (line 4052) | RoiAlignGradFunctor() { class FusedDotFeatureInteractionFunctor (line 4076) | class FusedDotFeatureInteractionFunctor { method FusedDotFeatureInteractionFunctor (line 4078) | FusedDotFeatureInteractionFunctor() { class FusedCrossFeatureInteractionFunctor (line 4138) | class FusedCrossFeatureInteractionFunctor { method FusedCrossFeatureInteractionFunctor (line 4140) | FusedCrossFeatureInteractionFunctor() { class OneEmbeddingIdShuffleFunctor (line 4169) | class OneEmbeddingIdShuffleFunctor { method OneEmbeddingIdShuffleFunctor (line 4171) | OneEmbeddingIdShuffleFunctor() { class OneEmbeddingEmbeddingShuffleFunctor (line 4211) | class OneEmbeddingEmbeddingShuffleFunctor { method OneEmbeddingEmbeddingShuffleFunctor (line 4213) | OneEmbeddingEmbeddingShuffleFunctor() { class OneEmbeddingEmbeddingGradientShuffleFunctor (line 4242) | class OneEmbeddingEmbeddingGradientShuffleFunctor { method OneEmbeddingEmbeddingGradientShuffleFunctor (line 4244) | OneEmbeddingEmbeddingGradientShuffleFunctor() { class OneEmbeddingLookupFunctor (line 4273) | class OneEmbeddingLookupFunctor { method OneEmbeddingLookupFunctor (line 4275) | OneEmbeddingLookupFunctor() { class OneEmbeddingFusedLookupFunctor (line 4303) | class OneEmbeddingFusedLookupFunctor { method OneEmbeddingFusedLookupFunctor (line 4305) | OneEmbeddingFusedLookupFunctor() { class OneEmbeddingFusedLookupGradFunctor (line 4369) | class OneEmbeddingFusedLookupGradFunctor { method OneEmbeddingFusedLookupGradFunctor (line 4371) | OneEmbeddingFusedLookupGradFunctor() { class OneEmbeddingEmbeddingPutFunctor (line 4392) | class OneEmbeddingEmbeddingPutFunctor { method OneEmbeddingEmbeddingPutFunctor (line 4394) | OneEmbeddingEmbeddingPutFunctor() { class OneEmbeddingUniqueKeyValuePairFunctor (line 4417) | class OneEmbeddingUniqueKeyValuePairFunctor { method OneEmbeddingUniqueKeyValuePairFunctor (line 4419) | OneEmbeddingUniqueKeyValuePairFunctor() { class OneEmbeddingSgdUpdateFunctor (line 4454) | class OneEmbeddingSgdUpdateFunctor { method OneEmbeddingSgdUpdateFunctor (line 4456) | OneEmbeddingSgdUpdateFunctor() { class OneEmbeddingAdamUpdateFunctor (line 4552) | class OneEmbeddingAdamUpdateFunctor { method OneEmbeddingAdamUpdateFunctor (line 4554) | OneEmbeddingAdamUpdateFunctor() { class OneEmbeddingAdagradUpdateFunctor (line 4641) | class OneEmbeddingAdagradUpdateFunctor { method OneEmbeddingAdagradUpdateFunctor (line 4643) | OneEmbeddingAdagradUpdateFunctor() { class OneEmbeddingFtrlUpdateFunctor (line 4706) | class OneEmbeddingFtrlUpdateFunctor { method OneEmbeddingFtrlUpdateFunctor (line 4708) | OneEmbeddingFtrlUpdateFunctor() { class DeformConv2dFunctor (line 4767) | class DeformConv2dFunctor { method DeformConv2dFunctor (line 4769) | DeformConv2dFunctor() { class RocAucScoreFunctor (line 4810) | class RocAucScoreFunctor { method RocAucScoreFunctor (line 4812) | RocAucScoreFunctor() { class MultiTensorSgdUpdateFunctor (line 4826) | class MultiTensorSgdUpdateFunctor { method MultiTensorSgdUpdateFunctor (line 4828) | MultiTensorSgdUpdateFunctor() { class MultiTensorMomentumUpdateFunctor (line 4858) | class MultiTensorMomentumUpdateFunctor { method MultiTensorMomentumUpdateFunctor (line 4860) | MultiTensorMomentumUpdateFunctor() { class MultiTensorAdamUpdateFunctor (line 4897) | class MultiTensorAdamUpdateFunctor { method MultiTensorAdamUpdateFunctor (line 4899) | MultiTensorAdamUpdateFunctor() { class MatrixVectorProductFunctor (line 4940) | class MatrixVectorProductFunctor { method MatrixVectorProductFunctor (line 4942) | MatrixVectorProductFunctor() { class BatchNormStatsFunctor (line 4965) | class BatchNormStatsFunctor { method BatchNormStatsFunctor (line 4967) | BatchNormStatsFunctor() { class BatchNormGatherStatsWithCountsFunctor (line 4983) | class BatchNormGatherStatsWithCountsFunctor { method BatchNormGatherStatsWithCountsFunctor (line 4985) | BatchNormGatherStatsWithCountsFunctor() { class BatchNormElemtFunctor (line 5035) | class BatchNormElemtFunctor { method BatchNormElemtFunctor (line 5037) | BatchNormElemtFunctor() { class BatchNormBackwardReduceFunctor (line 5063) | class BatchNormBackwardReduceFunctor { method BatchNormBackwardReduceFunctor (line 5065) | BatchNormBackwardReduceFunctor() { class BatchNormBackwardElemtFunctor (line 5091) | class BatchNormBackwardElemtFunctor { method BatchNormBackwardElemtFunctor (line 5093) | BatchNormBackwardElemtFunctor() { class FusedFastGeluMulFunctor (line 5125) | class FusedFastGeluMulFunctor { method FusedFastGeluMulFunctor (line 5127) | FusedFastGeluMulFunctor() { class FusedFastGeluMulGradFunctor (line 5143) | class FusedFastGeluMulGradFunctor { method FusedFastGeluMulGradFunctor (line 5145) | FusedFastGeluMulGradFunctor() { class GroupedMatmulBiasFunctor (line 5164) | class GroupedMatmulBiasFunctor { method GroupedMatmulBiasFunctor (line 5166) | GroupedMatmulBiasFunctor() { class GroupedMatmulFunctor (line 5216) | class GroupedMatmulFunctor { method GroupedMatmulFunctor (line 5218) | GroupedMatmulFunctor() { class MultiTensorYoloV5WeightUpdateFunctor (line 5258) | class MultiTensorYoloV5WeightUpdateFunctor { method MultiTensorYoloV5WeightUpdateFunctor (line 5260) | MultiTensorYoloV5WeightUpdateFunctor() { class FusedScaleMaskBiasSoftmaxFunctor (line 5290) | class FusedScaleMaskBiasSoftmaxFunctor { method FusedScaleMaskBiasSoftmaxFunctor (line 5292) | FusedScaleMaskBiasSoftmaxFunctor() { class FusedScaleMaskBiasSoftmaxGradFunctor (line 5335) | class FusedScaleMaskBiasSoftmaxGradFunctor { method FusedScaleMaskBiasSoftmaxGradFunctor (line 5337) | FusedScaleMaskBiasSoftmaxGradFunctor() { class FusedClipGradFunctor (line 5355) | class FusedClipGradFunctor { method FusedClipGradFunctor (line 5357) | FusedClipGradFunctor() { class NonContiguousBinaryOpFunctor (line 5379) | class NonContiguousBinaryOpFunctor { method NonContiguousBinaryOpFunctor (line 5381) | NonContiguousBinaryOpFunctor() { class NonContiguousBinaryOpGradFunctor (line 5403) | class NonContiguousBinaryOpGradFunctor { method NonContiguousBinaryOpGradFunctor (line 5405) | NonContiguousBinaryOpGradFunctor() { function pad_last_dim (line 5431) | Maybe pad_last_dim(const std::shared_ptr check_rnn_cell_forward_input(const std::shared_ptr check_rnn_cell_forward_hidden(const std::shared_ptr<... function check_attributes (line 81) | Maybe check_attributes(const std::shared_ptr&... function linear (line 120) | Maybe linear(const std::shared_ptr& input, type CellParams (line 134) | struct CellParams { method CellParams (line 135) | CellParams(const std::shared_ptr _w_ih, // NOLINT method matmul_ih (line 148) | Maybe matmul_ih(const std::shared_ptr& in... method matmul_hh (line 152) | Maybe matmul_hh(const std::shared_ptr& h)... method matmul_hr (line 156) | Maybe matmul_hr(const std::shared_ptr& h)... method linear_ih (line 161) | Maybe linear_ih(const std::shared_ptr& in... method linear_hh (line 165) | Maybe linear_hh(const std::shared_ptr& h)... function gather_params (line 174) | static Maybe> gather_params(const Tensor... type SimpleCell (line 206) | struct SimpleCell { type GRUCell (line 223) | struct GRUCell { type LSTMCell (line 273) | struct LSTMCell { class RnnTanhCellFunctor (line 326) | class RnnTanhCellFunctor { method RnnTanhCellFunctor (line 328) | RnnTanhCellFunctor() {} class RnnReluCellFunctor (line 349) | class RnnReluCellFunctor { method RnnReluCellFunctor (line 351) | RnnReluCellFunctor() {} class GruCellFunctor (line 372) | class GruCellFunctor { method GruCellFunctor (line 374) | GruCellFunctor() {} class LstmCellFunctor (line 394) | class LstmCellFunctor { method LstmCellFunctor (line 396) | LstmCellFunctor() {} class FusedGruCellFunctor (line 419) | class FusedGruCellFunctor { method FusedGruCellFunctor (line 421) | FusedGruCellFunctor() { class FusedGruCellGradFunctor (line 461) | class FusedGruCellGradFunctor { method FusedGruCellGradFunctor (line 463) | FusedGruCellGradFunctor() { class FusedLstmCellFunctor (line 527) | class FusedLstmCellFunctor { method FusedLstmCellFunctor (line 529) | FusedLstmCellFunctor() { class FusedLstmCellGradFunctor (line 571) | class FusedLstmCellGradFunctor { method FusedLstmCellGradFunctor (line 573) | FusedLstmCellGradFunctor() { function _rnn_impl (line 647) | Maybe _rnn_impl(const std::shared_ptr&... class RnnTanhInputFunctor (line 733) | class RnnTanhInputFunctor { method RnnTanhInputFunctor (line 735) | RnnTanhInputFunctor() {} class RnnReluInputFunctor (line 747) | class RnnReluInputFunctor { method RnnReluInputFunctor (line 749) | RnnReluInputFunctor() {} function _rnn_pack_sequence_impl (line 762) | Maybe _rnn_pack_sequence_impl(const std::shared_ptr... class RnnTanhDataFunctor (line 912) | class RnnTanhDataFunctor { method RnnTanhDataFunctor (line 914) | RnnTanhDataFunctor() {} class RnnReluDataFunctor (line 927) | class RnnReluDataFunctor { method RnnReluDataFunctor (line 929) | RnnReluDataFunctor() {} function _lstm_impl (line 942) | Maybe _lstm_impl(const std::shared_ptr... class LstmInputFunctor (line 1046) | class LstmInputFunctor { method LstmInputFunctor (line 1048) | LstmInputFunctor() {} function _lstm_pack_sequence_impl (line 1060) | Maybe _lstm_pack_sequence_impl(const std::shared_pt... class LstmDataFunctor (line 1245) | class LstmDataFunctor { method LstmDataFunctor (line 1247) | LstmDataFunctor() {} class GruInputFunctor (line 1260) | class GruInputFunctor { method GruInputFunctor (line 1262) | GruInputFunctor() {} class GruDataFunctor (line 1274) | class GruDataFunctor { method GruDataFunctor (line 1276) | GruDataFunctor() {} function checkLongTensor (line 1289) | Maybe checkLongTensor(const std::shared_ptr& ... class PackPaddedSequenceFunctor (line 1297) | class PackPaddedSequenceFunctor { method PackPaddedSequenceFunctor (line 1299) | PackPaddedSequenceFunctor() {} function ONEFLOW_FUNCTION_LIBRARY (line 1420) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/slice_boxing_functor.cpp type oneflow (line 27) | namespace oneflow { type one (line 28) | namespace one { type functional (line 29) | namespace functional { type impl (line 31) | namespace impl { function IsSplitSbp (line 35) | bool IsSplitSbp(Symbol sbp_parallel) { return sbp_p... function EagerSToB (line 37) | Maybe EagerSToB(Symbol in_paralle... function EagerPToB (line 53) | Maybe EagerPToB(Symbol in_paralle... function EagerNaiveSToS (line 67) | Maybe EagerNaiveSToS(Symbol in_pa... function EagerBToS (line 85) | Maybe EagerBToS(Symbol in_paralle... function EagerPToS (line 101) | Maybe EagerPToS(Symbol in_paralle... function EagerSToP (line 117) | Maybe EagerSToP(Symbol in_paralle... class EagerSToBFunctor (line 135) | class EagerSToBFunctor { method EagerSToBFunctor (line 137) | EagerSToBFunctor() = default; class EagerPToBFunctor (line 159) | class EagerPToBFunctor { method EagerPToBFunctor (line 161) | EagerPToBFunctor() = default; class EagerNaiveSToSFunctor (line 177) | class EagerNaiveSToSFunctor { method EagerNaiveSToSFunctor (line 179) | EagerNaiveSToSFunctor() = default; class EagerBToSFunctor (line 207) | class EagerBToSFunctor { method EagerBToSFunctor (line 209) | EagerBToSFunctor() = default; class EagerPToSFunctor (line 231) | class EagerPToSFunctor { method EagerPToSFunctor (line 233) | EagerPToSFunctor() = default; class EagerSToPFunctor (line 255) | class EagerSToPFunctor { method EagerSToPFunctor (line 257) | EagerSToPFunctor() = default; function ONEFLOW_FUNCTION_LIBRARY (line 281) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/test_functor.cpp type oneflow (line 30) | namespace oneflow { type one (line 31) | namespace one { type functional (line 32) | namespace functional { type impl (line 34) | namespace impl { class ThrowErrorFunctor (line 36) | class ThrowErrorFunctor final { method ThrowErrorFunctor (line 38) | ThrowErrorFunctor() { function ONEFLOW_FUNCTION_LIBRARY (line 53) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("Th... FILE: oneflow/core/functional/impl/unary_functor.cpp type oneflow (line 24) | namespace oneflow { type one (line 25) | namespace one { type functional (line 26) | namespace functional { type impl (line 28) | namespace impl { function ONEFLOW_FUNCTION_LIBRARY (line 174) | ONEFLOW_FUNCTION_LIBRARY(m) { FILE: oneflow/core/functional/impl/unary_functor.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/functional/impl/util_ops_functor.cpp type oneflow (line 30) | namespace oneflow { type one (line 31) | namespace one { type functional (line 32) | namespace functional { type impl (line 34) | namespace impl { class UtilOpsFunctor (line 36) | class UtilOpsFunctor { class IsNanFunctor (line 46) | class IsNanFunctor final : public UtilOpsFunctor { method IsNanFunctor (line 48) | IsNanFunctor() { op_ = CHECK_JUST(one::OpBuilder("isnan").Inpu... class IsInfFunctor (line 51) | class IsInfFunctor final : public UtilOpsFunctor { method IsInfFunctor (line 53) | IsInfFunctor() { op_ = CHECK_JUST(one::OpBuilder("isinf").Inpu... class IsFiniteFunctor (line 56) | class IsFiniteFunctor final : public UtilOpsFunctor { method IsFiniteFunctor (line 58) | IsFiniteFunctor() { class DependFunctor (line 63) | class DependFunctor { method DependFunctor (line 65) | DependFunctor() { class DependTupleFunctor (line 78) | class DependTupleFunctor { method DependTupleFunctor (line 80) | DependTupleFunctor() { method _dispatch (line 94) | Maybe _dispatch(const std::shared_ptr& in... function ONEFLOW_FUNCTION_LIBRARY (line 109) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("IsNan"); } function ONEFLOW_FUNCTION_LIBRARY (line 110) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("IsInf"); } function ONEFLOW_FUNCTION_LIBRARY (line 111) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("IsFi... function ONEFLOW_FUNCTION_LIBRARY (line 112) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("Depend... function ONEFLOW_FUNCTION_LIBRARY (line 113) | ONEFLOW_FUNCTION_LIBRARY(m) { m.add_functor("D... FILE: oneflow/core/functional/packed_functor.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/functional/sequence_function.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/functional/tensor_index.cpp type oneflow (line 31) | namespace oneflow { type one (line 32) | namespace one { type functional (line 33) | namespace functional { function CountSpecifiedDims (line 37) | int64_t CountSpecifiedDims(const TensorIndex& index) { function ExpandMaskIndex (line 55) | Maybe ExpandMaskIndex(const std::shared_ptr& ... function ExpandIndices (line 88) | Maybe ExpandIndices(const TensorTuple& indices) { function IsContinuousSubspace (line 139) | Maybe IsContinuousSubspace(const TensorTuple& indices) { function TransposeFront (line 156) | Maybe> TransposeFront(const std::shared_ptr AdjustSubspace(const std::shared_ptr& input,... function HasFalseIndex (line 213) | Maybe HasFalseIndex(const TensorIndex& index) { function IsValidScalarTensorIndex (line 219) | bool IsValidScalarTensorIndex(const std::shared_ptr& ... function PermuteBackForGlobalTensor (line 225) | Maybe PermuteBackForGlobalTensor(const std::shared_ptr& tensor) { function PrepareSliceIndices (line 255) | Maybe PrepareSliceIndices(const TensorIndex& index, const Sh... function RemoveExpandDimSlice (line 377) | Maybe> RemoveExpandDimSlice( function ApplyAdvancedIndexing (line 394) | Maybe ApplyAdvancedIndexing(const std::shared_ptr&... function ApplyAdvancedIndexingUpdate (line 437) | Maybe ApplyAdvancedIndexingUpdate(const std::shared_ptr ApplySelectIndexing(const std::shared_ptr UnifyInputAndIndicesOnDevice(const std::shared_ptr ComputeCommonDType(const TensorTuple& tensor_tuple) { function CheckHasDifferentInputDType (line 47) | bool CheckHasDifferentInputDType(const TensorTuple& tensor_tuple) { function CastToSameType (line 56) | Maybe CastToSameType(TensorTuple& tensor_tuple, const Symbol... function TensorProcessor (line 67) | TensorProcessor& TensorProcessor::AddInputs(const TensorTuple& ini... function TensorProcessor (line 75) | TensorProcessor& TensorProcessor::AddInputs(const TensorTuple& ini... function TensorProcessor (line 84) | TensorProcessor& TensorProcessor::PromoteInputsToCommonDtype(bool ... function TensorProcessor (line 89) | TensorProcessor& TensorProcessor::PromoteInputsToCommonDtype( function TensorProcessor (line 96) | TensorProcessor& TensorProcessor::PromoteIntegerInputsToFloatDtype... function IsAllContiguous (line 145) | static bool IsAllContiguous(const TensorTuple& tensors) { FILE: oneflow/core/functional/tensor_processor.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/graph/boxing/b21_sub_task_graph_builder.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { FILE: oneflow/core/graph/boxing/b21_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/boxing_logger.cpp type oneflow (line 20) | namespace oneflow { function ShapeToString (line 29) | std::string ShapeToString(const Shape& shape) { function ParallelDescToString (line 41) | std::string ParallelDescToString(const ParallelDesc& parallel_desc) { function NdSbpToCsvString (line 60) | std::string NdSbpToCsvString(const NdSbp& nd_sbp) { function MakeBoxingLoggerCsvRow (line 71) | std::string MakeBoxingLoggerCsvRow(const SubTskGphBuilderStatus& status, FILE: oneflow/core/graph/boxing/boxing_logger.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/boxing/ccl_sub_task_graph_builder.cpp type oneflow (line 25) | namespace oneflow { function CclInitCollectiveNode (line 31) | void CclInitCollectiveNode(CollectiveBoxingGenericTaskNode* node, cons... function FindRootParallelId (line 68) | int64_t FindRootParallelId(const ParallelDesc& multi_device, const Par... function IsSourceTimeShape (line 85) | bool IsSourceTimeShape(const Shape& shape) { return shape.elem_cnt() =... function FOR_RANGE (line 99) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 128) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 159) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 208) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 239) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 288) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 328) | FOR_RANGE(int64_t, out_id, 0, out_parallel_desc.parallel_num()) { function FOR_RANGE (line 389) | FOR_RANGE(int64_t, i, 0, out_parallel_desc.parallel_num()) { function FOR_RANGE (line 430) | FOR_RANGE(int64_t, i, 0, in_parallel_desc.parallel_num()) { FILE: oneflow/core/graph/boxing/ccl_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/chain_sub_task_graph_builder.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/graph/boxing/chain_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/collective_boxing_sub_task_graph_builder.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/boxing/collective_boxing_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/collective_boxing_util.cpp type oneflow (line 18) | namespace oneflow { type boxing (line 20) | namespace boxing { type collective (line 22) | namespace collective { function Shape (line 26) | Shape GetSplitShape(const RankDesc& rank_desc) { function Shape (line 34) | Shape GetFlattenSplitShape(const RankDesc& rank_desc) { function GenericOpHasInput (line 44) | bool GenericOpHasInput(const RankDesc& rank_desc) { function GenericOpHasOutput (line 59) | bool GenericOpHasOutput(const RankDesc& rank_desc) { function Shape (line 74) | Shape GenericOpGetInputShape(const RankDesc& rank_desc) { function Shape (line 90) | Shape GenericOpGetOutputShape(const RankDesc& rank_desc) { FILE: oneflow/core/graph/boxing/collective_boxing_util.h function namespace (line 26) | namespace oneflow { function namespace (line 58) | namespace std { FILE: oneflow/core/graph/boxing/fallback_to_cpu_slice_boxing_sub_task_graph_builder.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, in_id, 0, in_parallel_desc.parallel_num()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, out_id, 0, out_parallel_desc.parallel_num()) { function FOR_RANGE (line 65) | FOR_RANGE(size_t, i, 0, sorted_out_tasks->size()) { FILE: oneflow/core/graph/boxing/fallback_to_cpu_slice_boxing_sub_task_graph_builder.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/boxing/hierarchical_sub_task_graph_builder.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/graph/boxing/hierarchical_sub_task_graph_builder_impl.cpp type oneflow (line 36) | namespace oneflow { function Make1DSubTskGphBuilder (line 40) | std::shared_ptr Make1DSubTskGphBuilder() { function MergeParallelConf (line 54) | void MergeParallelConf(const ParallelDesc& parallel_desc_0, const Para... function NewUniqueIdGbc (line 75) | inline std::string NewUniqueIdGbc() { class NDNcclSendRecvBoxingSubTskGphBuilder (line 87) | class NDNcclSendRecvBoxingSubTskGphBuilder final : public Hierarchical... method NDNcclSendRecvBoxingSubTskGphBuilder (line 90) | NDNcclSendRecvBoxingSubTskGphBuilder() {} method Build (line 93) | Maybe Build(SubTskGphBuilderCtx* ctx, class Dim0NdSbpMismatchedSubTskGphBuilder (line 149) | class Dim0NdSbpMismatchedSubTskGphBuilder final : public HierarchicalS... method Dim0NdSbpMismatchedSubTskGphBuilder (line 152) | Dim0NdSbpMismatchedSubTskGphBuilder() { method Build (line 158) | Maybe Build(SubTskGphBuilderCtx* ctx, class Same2DHierarchySubTskGphBuilder (line 188) | class Same2DHierarchySubTskGphBuilder final : public HierarchicalSubTs... method Same2DHierarchySubTskGphBuilder (line 191) | Same2DHierarchySubTskGphBuilder() { method Build (line 198) | Maybe Build(SubTskGphBuilderCtx* ctx, type DispatchHierarchicalSubTskGphBuilder::Impl (line 230) | struct DispatchHierarchicalSubTskGphBuilder::Impl { FILE: oneflow/core/graph/boxing/hierarchical_sub_task_graph_builder_impl.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/hierarchical_sub_task_graph_builder_util.cpp type oneflow (line 18) | namespace oneflow { function FOR_RANGE (line 56) | FOR_RANGE(int64_t, i, 0, num_groups) { function FOR_RANGE (line 125) | FOR_RANGE(int64_t, i, 0, group_size) { FILE: oneflow/core/graph/boxing/hierarchical_sub_task_graph_builder_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/boxing/naive_b2b_sub_task_graph_builder.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 30) | FOR_RANGE(int64_t, out_id, 0, out_parallel_desc.parallel_num()) { FILE: oneflow/core/graph/boxing/naive_b2b_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/naive_b2p_sub_task_graph_builder.cpp type oneflow (line 21) | namespace oneflow { function FOR_RANGE (line 36) | FOR_RANGE(int64_t, out_id, 0, out_parallel_desc.parallel_num()) { function FOR_RANGE (line 47) | FOR_RANGE(int64_t, out_id, 0, out_parallel_desc.parallel_num()) { FILE: oneflow/core/graph/boxing/naive_b2p_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/one_to_one_sub_task_graph_builder.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/graph/boxing/one_to_one_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/slice_boxing_sub_task_graph_builder.cpp type oneflow (line 25) | namespace oneflow { function IsCopyNdPrimitiveSupported (line 29) | bool IsCopyNdPrimitiveSupported(DeviceType device_type, int64_t ndims) { function FOR_RANGE (line 106) | FOR_RANGE(int64_t, out_id, 0, out_pd.parallel_num()) { function FOR_RANGE (line 203) | FOR_RANGE(int64_t, out_id, 0, out_pd.parallel_num()) { FILE: oneflow/core/graph/boxing/slice_boxing_sub_task_graph_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/sub_task_graph_builder.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/graph/boxing/sub_task_graph_builder_context.cpp type oneflow (line 18) | namespace oneflow { function TaskGraph (line 22) | TaskGraph* SubTskGphBuilderCtx::task_graph() { return task_graph_; } FILE: oneflow/core/graph/boxing/sub_task_graph_builder_context.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/boxing/sub_task_graph_builder_status_util.cpp type oneflow (line 18) | namespace oneflow { function BuildSubTskGphBuilderStatus (line 20) | Maybe BuildSubTskGphBuilderStatus(const std::s... function MakeComposedSubTskGphBuilderStatus (line 26) | Maybe MakeComposedSubTskGphBuilderStatus( FILE: oneflow/core/graph/boxing/sub_task_graph_builder_status_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing/sub_task_graph_builder_util.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/boxing/sub_task_graph_builder_util.h function namespace (line 24) | namespace oneflow { function NodeType (line 69) | NodeType* FindNearestNode(const std::vector from_nodes, FILE: oneflow/core/graph/boxing_identity_task_node.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/boxing_identity_task_node.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/boxing_zeros_task_node.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/boxing_zeros_task_node.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/collective_boxing_pack_task_node.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/collective_boxing_pack_task_node.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/collective_boxing_task_node.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/collective_boxing_task_node.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/collective_boxing_unpack_task_node.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/collective_boxing_unpack_task_node.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/compute_task_node.cpp type oneflow (line 20) | namespace oneflow { function OpNode (line 24) | const OpNode* OpNodeOnEdge(TaskEdge* edge, TaskNode* (TaskEdge::*GetNo... function GetCompTaskNodesOnEdge (line 40) | std::vector GetCompTaskNodesOnEdge( function NewFakeDataRegstDesc (line 66) | std::shared_ptr NewFakeDataRegstDesc() { function OpNode (line 122) | const OpNode* CompTaskNode::GetOneSuccOpNodeOnEdge(TaskEdge* edge) { function OpNode (line 126) | const OpNode* CompTaskNode::GetOnePredOpNodeOnEdge(TaskEdge* edge) { function CompTaskNode (line 145) | CompTaskNode* NewCompTaskNode4OpNode(const OpNode* op_node) { FILE: oneflow/core/graph/compute_task_node.h function namespace (line 25) | namespace oneflow { function class (line 83) | class OpCompTaskNodeCreator { function class (line 101) | class FnOpCompTaskNodeCreator : public OpCompTaskNodeCreator { FILE: oneflow/core/graph/copy_task_node.cpp type oneflow (line 21) | namespace oneflow { function OperatorConf (line 88) | OperatorConf CopyHdTaskNode::NewCopyOpConf() { function OperatorConf (line 119) | OperatorConf CopyCommNetTaskNode::NewCopyOpConf() { FILE: oneflow/core/graph/copy_task_node.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/exec_graph.cpp type oneflow (line 21) | namespace oneflow { function CheckPhysicalBlobDesc (line 76) | Maybe CheckPhysicalBlobDesc(const BlobDesc& logical, const NdSbp... function CheckPhysicalBlobDesc (line 84) | Maybe CheckPhysicalBlobDesc( function InferPhysicalBlobDesc (line 107) | Maybe InferPhysicalBlobDesc( FILE: oneflow/core/graph/exec_graph.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/graph/fake_consumed_regst_provider.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/graph/graph.h function namespace (line 25) | namespace oneflow { function ForEachStart (line 502) | auto ForEachStart = [&](const std::function& Handler) { function class (line 757) | class BitSet { FILE: oneflow/core/graph/inplace_lbi_graph.cpp type oneflow (line 19) | namespace oneflow { function IsSourceNode (line 23) | bool IsSourceNode(const Operator& op) { function CheckSubGraph (line 42) | void CheckSubGraph(const HashSet& nodes) { function InplaceLbiNode (line 63) | const InplaceLbiNode* GetRoot(const HashSet& no... function InplaceLbiNode (line 75) | const InplaceLbiNode* FindSoleIsMutableIbnConsumer(const SourceOpInpla... function InplaceLbiNode (line 86) | InplaceLbiNode* CreateNode(const LogicalBlobId& lbi, function GetUnconnectedNodes (line 100) | void GetUnconnectedNodes(const HashSet& nodes, function InplaceLbiNode (line 111) | const InplaceLbiNode* GetFirstDiffNode(const std::vector& lbis, co... function RemoveUnconnectedNodes (line 149) | void RemoveUnconnectedNodes(HashSet* nodes, function InplaceLbiEdge (line 158) | const InplaceLbiEdge* InplaceLbiNode::GetValidInEdge( function InplaceLbiEdge (line 164) | const InplaceLbiEdge* InplaceLbiNode::GetSoleValidInEdge( function InplaceLbiEdge (line 385) | const InplaceLbiEdge* InplaceLbiGraph::FindFirstIntraOpRefConflictMutR... function InplaceLbiEdge (line 424) | const InplaceLbiEdge* InplaceLbiGraph::FindFirstConstRefConflictMutRef... function InplaceLbiEdge (line 444) | const InplaceLbiEdge* InplaceLbiGraph::FindFirstInterOpRefConflictMutR... FILE: oneflow/core/graph/inplace_lbi_graph.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/graph/inplace_regst_graph.cpp type oneflow (line 18) | namespace oneflow { function MakeGetterRegstDesc4RegstDescId (line 22) | std::function MakeGetterRegstDesc4Regs... FILE: oneflow/core/graph/inplace_regst_graph.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/graph/nccl_send_recv_boxing_task_node.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/graph/nccl_send_recv_boxing_task_node.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/graph/node.cpp type oneflow (line 18) | namespace oneflow { function NewNodeId (line 20) | int64_t NewNodeId() { function NewEdgeId (line 25) | int64_t NewEdgeId() { FILE: oneflow/core/graph/node.h function namespace (line 22) | namespace oneflow { function NodeType (line 58) | NodeType* src_node() const { return src_node_; } function NodeType (line 59) | NodeType* dst_node() const { return dst_node_; } function EdgeType (line 83) | EdgeType* SoleInEdge() const { function EdgeType (line 87) | EdgeType* SoleOutEdge() const { function ForEachNodeOnInEdge (line 95) | void ForEachNodeOnInEdge(std::function Handler) const { FILE: oneflow/core/graph/normal_forward_compute_task_node.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/graph/op_graph.cpp type oneflow (line 26) | namespace oneflow { function SbpParallel (line 62) | const SbpParallel& OpNode::SbpParallel4BnInOp(const std::string& bn_in... function SbpParallel (line 66) | const SbpParallel& OpNode::SbpParallel4Lbi(const LogicalBlobId& lbi) c... function NdSbp (line 73) | const NdSbp& OpNode::NdSbp4BnInOp(const std::string& bn_in_op) const { function NdSbp (line 77) | const NdSbp& OpNode::NdSbp4Lbi(const LogicalBlobId& lbi) const { function BlobDesc (line 119) | const BlobDesc& OpNode::LogicalBlobDesc4Lbi(const LogicalBlobId& lbi) ... function OpNode (line 126) | const OpNode& OpNode::SrcNode4Ibn(const std::string& bn_in_op) const { function OpNode (line 130) | OpNode* OpNode::MutSrcNode4Ibn(const std::string& bn_in_op) const { function OpNode (line 136) | const OpNode& OpNode::ProducerOpNode4Lbi(const LogicalBlobId& lbi) con... function OpNode (line 142) | OpNode* OpNode::MutSrcNode4InputLbi(const LogicalBlobId& lbi) const { function MakeGetterParallelDesc4OpName (line 222) | std::function(const std::string&)> MakeGetterPara... function OpNode (line 371) | const OpNode* OpGraph::OpNode4OpName(const std::string& op_name) const { function SbpParallel (line 425) | const SbpParallel& OpGraph::GetSbpParallel(const std::string& op_name, function NdSbp (line 431) | const NdSbp& OpGraph::GetNdSbp(const std::string& op_name, const Logic... function DataType (line 436) | DataType OpGraph::GetBlobDataType(const LogicalBlobId& lbi) const { function BlobDesc (line 442) | const BlobDesc& OpGraph::GetLogicalBlobDesc(const LogicalBlobId& lbi) ... function LogicalBlobId (line 455) | LogicalBlobId OpGraph::GetLogicalBlobIdKey(const std::string& op_name, FILE: oneflow/core/graph/op_graph.h function namespace (line 26) | namespace oneflow { function SbpParallel (line 48) | const SbpParallel& SbpParallel4Lbi(const LogicalBlobId& lbi) const; function explicit (line 81) | explicit OpEdge(std::shared_ptr> lbis, function explicit (line 104) | explicit OpGraph(const Job& job) { CHECK_JUST(Init(job)); } function CheckIsDAG (line 145) | void CheckIsDAG() const; FILE: oneflow/core/graph/plan_task_graph.cpp type oneflow (line 19) | namespace oneflow { function TaskProto (line 50) | const TaskProto* PlanTaskGraph::TaskProto4TaskId(int64_t task_id) const { FILE: oneflow/core/graph/plan_task_graph.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/slice_boxing_task_node.cpp type oneflow (line 21) | namespace oneflow { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, ordered_in_data_edges_.size()) { function FOR_RANGE (line 59) | FOR_RANGE(size_t, i, 0, op->input_bns().size()) { function OperatorConf (line 89) | OperatorConf SliceBoxingTaskNode::GetBoxingOpConf() { FILE: oneflow/core/graph/slice_boxing_task_node.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/graph/straighten_nodes.cpp type oneflow (line 34) | namespace oneflow { type TaskClassifier (line 38) | enum TaskClassifier : int { class TopoStruct (line 48) | class TopoStruct { function MoveFrontBetweenMaps (line 120) | void MoveFrontBetweenMaps(std::map& source, function ShouldRunASAP (line 129) | bool ShouldRunASAP(TaskType task_type) { function IsTransferNode (line 153) | bool IsTransferNode(TaskType task_type) { function TaskClassifier (line 176) | TaskClassifier GetTaskClassifier(const TaskNode* node, bool nccl_use_c... function MaxProducerMinLayer (line 204) | int32_t MaxProducerMinLayer(HashMap* task_node2... function FindTrunk (line 390) | void FindTrunk(HashMap* task_node2topo_struct) { function FindMinLifetime (line 422) | void FindMinLifetime(HashMap* task_node2topo_st... function ShortGpuTime (line 454) | bool ShortGpuTime(const OperatorConf& op_conf) { function InitDecideParameters (line 475) | void InitDecideParameters(StraightenAlgorithmTag sat, function MaximumOverlapNum (line 501) | int32_t MaximumOverlapNum(StraightenAlgorithmTag sat, bool nccl_use_co... function StraightenNodes (line 523) | void StraightenNodes(TaskGraph* task_graph, std::vector* or... FILE: oneflow/core/graph/straighten_nodes.h type StraightenOrder (line 29) | enum StraightenOrder : int { FILE: oneflow/core/graph/stream_id.cpp type oneflow (line 19) | namespace oneflow { function EncodeStreamIdToInt64 (line 48) | int64_t EncodeStreamIdToInt64(const StreamId& stream_id) { function StreamId (line 56) | StreamId DecodeStreamIdFromInt64(int64_t stream_id_val) { FILE: oneflow/core/graph/stream_id.h function class (line 23) | class StreamId { function namespace (line 69) | namespace std { FILE: oneflow/core/graph/stream_index_generator.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/stream_index_generator.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/task_graph.cpp type oneflow (line 48) | namespace oneflow { function IsMemcpyPrimitiveSupported (line 55) | bool IsMemcpyPrimitiveSupported(DeviceType device_type, ep::primitive:... function IsMemcpyHtoDSupported (line 60) | bool IsMemcpyHtoDSupported(DeviceType device_type) { function IsMemcpyDtoHSupported (line 64) | bool IsMemcpyDtoHSupported(DeviceType device_type) { function IsConnectToTickOp (line 68) | bool IsConnectToTickOp(const TaskNode* node) { function IsSubsetTickOpConf (line 76) | bool IsSubsetTickOpConf(const OperatorConf& op_conf) { function IsTickOpConf (line 80) | bool IsTickOpConf(const OperatorConf& conf) { function IsOptimizerPassOp (line 96) | bool IsOptimizerPassOp(const Operator* op) { function IsSpecialOpNotConsiderMergeInChain (line 107) | bool IsSpecialOpNotConsiderMergeInChain(const Operator* op) { function IsTaskNodeProducedRegstHasMultiRegstNum (line 130) | bool IsTaskNodeProducedRegstHasMultiRegstNum(const TaskNode* node) { function CanBeMergedInChain (line 137) | bool CanBeMergedInChain(const TaskNode* node) { function GetTaskNodeTimeShape (line 148) | std::shared_ptr GetTaskNodeTimeShape(const TaskNode* node) { function TraverseConnectedSubGraphMergeInThisChain (line 154) | void TraverseConnectedSubGraphMergeInThisChain(TaskNode* this_node, co... function MakeGetterTaskNode4SoleOpName (line 185) | std::function MakeGetterTaskNode4SoleOp... function IsLbiOnTaskEdge (line 202) | bool IsLbiOnTaskEdge(const TaskEdge* edge, const LogicalBlobId& lbi) { function MakePredicatorIsLbiAllConsumersReachable (line 209) | std::function function IsInplaceAllowed (line 244) | bool IsInplaceAllowed( function CreateBoxingLogger (line 270) | std::unique_ptr CreateBoxingLogger() { function MakeGetterTaskNode4MachineId7ThrdId (line 279) | Maybe MakeGetterTaskNode4MachineId7ThrdId( function StreamId (line 297) | StreamId GetStreamId(const OpNode* op_node, int64_t parallel_id, TaskT... function TaskType (line 320) | TaskType TaskType4OpNode(const OpNode* op_node) { function CompTaskNode (line 327) | CompTaskNode* GenCompTaskNode( function GenSortedCompTaskNodes (line 344) | void GenSortedCompTaskNodes(const OpNode* op_node, std::vector RegisterCreateSubTskGphBuilderFn(DeviceType device_type, function TaskEdge (line 533) | TaskEdge* TaskGraph::NewTaskEdgeWithLbi(const LogicalBlobId& lbi) { function TaskEdge (line 539) | TaskEdge* TaskGraph::NewTaskEdgeWithLbis(const std::vector New() { function Maybe (line 162) | static Maybe New( function Maybe (line 191) | static Maybe New( FILE: oneflow/core/graph/task_graph_rebuild_ctx.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/graph/task_graph_rebuild_ctx.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/task_id.cpp type oneflow (line 19) | namespace oneflow { function EncodeTaskIdToInt64 (line 52) | int64_t EncodeTaskIdToInt64(const TaskId& task_id) { function TaskId (line 61) | TaskId DecodeTaskIdFromInt64(int64_t task_id_val) { function MachineId4ActorId (line 73) | int64_t MachineId4ActorId(int64_t actor_id) { function ThrdId4ActorId (line 77) | int64_t ThrdId4ActorId(int64_t actor_id) { FILE: oneflow/core/graph/task_id.h function class (line 23) | class TaskId { function namespace (line 63) | namespace std { FILE: oneflow/core/graph/task_id_generator.cpp type oneflow (line 21) | namespace oneflow { function TaskId (line 52) | TaskId TaskIdGenerator::Generate(const StreamId& stream_id) { FILE: oneflow/core/graph/task_id_generator.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/task_node.cpp type oneflow (line 22) | namespace oneflow { function ForEachDataEdge (line 26) | void ForEachDataEdge(const std::unordered_set& edges, function StreamId (line 68) | const StreamId& TaskNode::stream_id() const { function DeviceType (line 73) | DeviceType TaskNode::device_type() const { return stream_id().device_i... function MemZoneId (line 271) | MemZoneId TaskNode::MemZoneId121() const { function RegstDesc (line 285) | RegstDesc* TaskNode::BuildCtrlRegstDesc(TaskNode* dst_node) { function RegstDesc (line 290) | RegstDesc* TaskNode::BuildCtrlRegstDesc(TaskNode* dst_node, std::strin... function RegstDescProto (line 466) | RegstDescProto* FindOrCreateProducedCtrlRegstDesc(TaskProto* task_proto, function RegstDescIdSet (line 477) | RegstDescIdSet* FindOrCreateConsumedCtrlRegstDescIdSet(TaskProto* task... function TaskEdge (line 507) | TaskEdge* TaskNode::GetSoleEdge(void (TaskNode::*ForEachEdge)(const st... function TaskEdge (line 525) | TaskEdge* TaskNode::SoleInDataEdge() const { return GetSoleEdge(&TaskN... function TaskEdge (line 527) | TaskEdge* TaskNode::SoleOutDataEdge() const { return GetSoleEdge(&Task... FILE: oneflow/core/graph/task_node.h function namespace (line 26) | namespace std { function namespace (line 37) | namespace oneflow { function DeviceType (line 71) | DeviceType device_type() const; function virtual (line 110) | virtual MemZoneId MemZoneId121() const; function AddLbi (line 194) | void AddLbi(const LogicalBlobId& lbi) { lbis_.insert(lbi); } function AddLbis (line 195) | void AddLbis(const std::vector& lbis) { lbis_.insert(lbis... FILE: oneflow/core/graph/task_stream_id.h function StreamId (line 30) | inline StreamId GenerateComputeTaskStreamId(int64_t rank, DeviceType dev... function StreamId (line 37) | inline StreamId GenerateNamedTaskStreamId(const DeviceId& device_id, con... FILE: oneflow/core/graph/task_stream_index_manager.cpp type oneflow (line 22) | namespace oneflow { function StreamIndexGenerator (line 24) | StreamIndexGenerator* TaskStreamIndexManager::GetGenerator(const Devic... function GenerateComputeTaskStreamIndex (line 108) | StreamId::stream_index_t GenerateComputeTaskStreamIndex(DeviceType dev... FILE: oneflow/core/graph/task_stream_index_manager.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph/task_type_visitor.h function namespace (line 27) | namespace oneflow { type CreateTransportTask (line 54) | struct CreateTransportTask function Maybe (line 56) | static Maybe VisitCopyCommNet() { return new CopyCom... function Maybe (line 57) | static Maybe VisitSliceBoxing() { return new SliceBo... function Maybe (line 58) | static Maybe VisitCollectiveBoxingGeneric() { function Maybe (line 61) | static Maybe VisitBoxingIdentity() { return new Boxi... function Maybe (line 62) | static Maybe VisitCollectiveBoxingPack() { function Maybe (line 65) | static Maybe VisitCollectiveBoxingUnpack() { function Maybe (line 68) | static Maybe VisitBoxingZeros() { return new BoxingZ... function Maybe (line 69) | static Maybe VisitNcclSendRecvBoxing() { FILE: oneflow/core/graph/transport_task_node.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/graph/transport_task_node.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/graph_impl/acc_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class AccCompTaskNode (line 21) | class AccCompTaskNode final : public CompTaskNode { method AccCompTaskNode (line 24) | AccCompTaskNode() = default; method TaskType (line 26) | TaskType GetTaskType() const override { return TaskType::kAcc; } FILE: oneflow/core/graph_impl/acc_ctrl_tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class AccCtrlTickCompTaskNode (line 21) | class AccCtrlTickCompTaskNode final : public CompTaskNode { method AccCtrlTickCompTaskNode (line 24) | AccCtrlTickCompTaskNode() = default; method TaskType (line 26) | TaskType GetTaskType() const override { return TaskType::kAccCtrlTic... FILE: oneflow/core/graph_impl/acc_tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class AccTickCompTaskNode (line 21) | class AccTickCompTaskNode final : public CompTaskNode { method AccTickCompTaskNode (line 24) | AccTickCompTaskNode() = default; method TaskType (line 26) | TaskType GetTaskType() const override { return TaskType::kAccTick; } FILE: oneflow/core/graph_impl/callback_notify_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class CallbackNotifyCompTaskNode (line 21) | class CallbackNotifyCompTaskNode final : public CompTaskNode { method CallbackNotifyCompTaskNode (line 24) | CallbackNotifyCompTaskNode() = default; method TaskType (line 27) | TaskType GetTaskType() const override { return TaskType::kCallbackNo... FILE: oneflow/core/graph_impl/case_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class CaseCompTaskNode (line 21) | class CaseCompTaskNode final : public CompTaskNode { method CaseCompTaskNode (line 24) | CaseCompTaskNode() = default; method TaskType (line 31) | TaskType GetTaskType() const override { return TaskType::kCase; } function FOR_RANGE (line 44) | FOR_RANGE(int64_t, obn_id, 0, op()->output_bns().size()) { function FOR_RANGE (line 69) | FOR_RANGE(int64_t, obn_id, 0, sole_op->output_bns().size()) { FILE: oneflow/core/graph_impl/critical_section_wait_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class CriticalSectionWaitTickCompTaskNode (line 21) | class CriticalSectionWaitTickCompTaskNode final : public CompTaskNode { method CriticalSectionWaitTickCompTaskNode (line 24) | CriticalSectionWaitTickCompTaskNode() = default; method IsMeaningLess (line 27) | bool IsMeaningLess() override { return false; } method TaskType (line 28) | TaskType GetTaskType() const override { return TaskType::kCriticalSe... FILE: oneflow/core/graph_impl/decode_h2d_compute_task_node.cpp type oneflow (line 20) | namespace oneflow { class DecodeH2DCompTaskNode (line 22) | class DecodeH2DCompTaskNode final : public CompTaskNode { method DecodeH2DCompTaskNode (line 25) | DecodeH2DCompTaskNode() = default; method TaskType (line 32) | TaskType GetTaskType() const override { return TaskType::kDecodeH2D; } function CompTaskNode (line 67) | CompTaskNode* CreateCompTaskNodeByOpDeviceType(const OperatorConf& op_... FILE: oneflow/core/graph_impl/device_tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class DeviceTickCompTaskNode (line 21) | class DeviceTickCompTaskNode final : public CompTaskNode { method DeviceTickCompTaskNode (line 24) | DeviceTickCompTaskNode() = default; method IsMeaningLess (line 27) | bool IsMeaningLess() override { return false; } method TaskType (line 28) | TaskType GetTaskType() const override { return TaskType::kDeviceTick; } FILE: oneflow/core/graph_impl/distribute_concat_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class DistributeConcatCompTaskNode (line 21) | class DistributeConcatCompTaskNode final : public CompTaskNode { method DistributeConcatCompTaskNode (line 24) | DistributeConcatCompTaskNode() = default; method TaskType (line 31) | TaskType GetTaskType() const override { return TaskType::kDistribute... FILE: oneflow/core/graph_impl/distribute_split_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class DistributeSplitCompTaskNode (line 21) | class DistributeSplitCompTaskNode final : public CompTaskNode { method DistributeSplitCompTaskNode (line 24) | DistributeSplitCompTaskNode() = default; method TaskType (line 31) | TaskType GetTaskType() const override { return TaskType::kDistribute... FILE: oneflow/core/graph_impl/dst_subset_tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class DstSubsetTickCompTaskNode (line 21) | class DstSubsetTickCompTaskNode final : public CompTaskNode { method DstSubsetTickCompTaskNode (line 24) | DstSubsetTickCompTaskNode() = default; method IsMeaningLess (line 27) | bool IsMeaningLess() override { return false; } method TaskType (line 28) | TaskType GetTaskType() const override { return TaskType::kDstSubsetT... FILE: oneflow/core/graph_impl/esac_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class EsacCompTaskNode (line 21) | class EsacCompTaskNode final : public CompTaskNode { method EsacCompTaskNode (line 24) | EsacCompTaskNode() = default; method ConsumeFakeRegsts (line 29) | void ConsumeFakeRegsts() override { UNIMPLEMENTED() << "EsacCompTask... method TaskType (line 31) | TaskType GetTaskType() const override { return TaskType::kEsac; } function FOR_RANGE (line 40) | FOR_RANGE(int64_t, ibn_id, 0, op()->input_bns().size()) { function FOR_RANGE (line 67) | FOR_RANGE(int64_t, ibn_id, 0, sole_op->input_bns().size()) { FILE: oneflow/core/graph_impl/normal_forward_compute_task_node.cpp type oneflow (line 20) | namespace oneflow { function RegstNum4OpSameOutputBlob (line 24) | size_t RegstNum4OpSameOutputBlob(OperatorConf::OpTypeCase op_type_case) { function GetOutRegstNameByObn (line 34) | std::string GetOutRegstNameByObn(const std::string& obn) { return "__"... function RegstNum4Op (line 48) | size_t RegstNum4Op(const Operator& sole_op) { FILE: oneflow/core/graph_impl/pack_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class PackCompTaskNode (line 21) | class PackCompTaskNode final : public CompTaskNode { method PackCompTaskNode (line 24) | PackCompTaskNode() = default; method TaskType (line 27) | TaskType GetTaskType() const override { return TaskType::kPack; } FILE: oneflow/core/graph_impl/reentrant_lock_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class ReentrantLockCompTaskNode (line 21) | class ReentrantLockCompTaskNode final : public CompTaskNode { method ReentrantLockCompTaskNode (line 24) | ReentrantLockCompTaskNode() = default; method IsMeaningLess (line 27) | bool IsMeaningLess() override { return false; } method TaskType (line 28) | TaskType GetTaskType() const override { return TaskType::kReentrantL... FILE: oneflow/core/graph_impl/repeat_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class RepeatCompTaskNode (line 21) | class RepeatCompTaskNode final : public CompTaskNode { method RepeatCompTaskNode (line 24) | RepeatCompTaskNode() = default; method TaskType (line 31) | TaskType GetTaskType() const override { return TaskType::kRepeat; } FILE: oneflow/core/graph_impl/source_tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class SourceTickCompTaskNode (line 21) | class SourceTickCompTaskNode final : public CompTaskNode { method SourceTickCompTaskNode (line 24) | SourceTickCompTaskNode() = default; method ConsumeAllRegsts (line 28) | void ConsumeAllRegsts() override {} method ConsumeFakeRegsts (line 29) | void ConsumeFakeRegsts() override {} method IsMeaningLess (line 31) | bool IsMeaningLess() override { return false; } method TaskType (line 33) | TaskType GetTaskType() const override { return TaskType::kSourceTick; } FILE: oneflow/core/graph_impl/src_subset_tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class SrcSubsetTickCompTaskNode (line 21) | class SrcSubsetTickCompTaskNode final : public CompTaskNode { method SrcSubsetTickCompTaskNode (line 24) | SrcSubsetTickCompTaskNode() = default; method IsMeaningLess (line 27) | bool IsMeaningLess() override { return false; } method TaskType (line 28) | TaskType GetTaskType() const override { return TaskType::kSrcSubsetT... FILE: oneflow/core/graph_impl/ssp_variable_proxy_task_node.cpp type oneflow (line 21) | namespace oneflow { class SspVariableProxyCompTaskNode (line 23) | class SspVariableProxyCompTaskNode final : public CompTaskNode { method SspVariableProxyCompTaskNode (line 26) | SspVariableProxyCompTaskNode() = default; method ProduceAllRegstsAndBindEdges (line 29) | void ProduceAllRegstsAndBindEdges() override { method ConsumeAllRegsts (line 65) | void ConsumeAllRegsts() override { method ConsumeFakeRegsts (line 70) | void ConsumeFakeRegsts() override { ConsumeFakeRegst("var"); } method TaskType (line 72) | TaskType GetTaskType() const override { return TaskType::kSspVariabl... method BuildExecGphAndRegst (line 75) | void BuildExecGphAndRegst() override { method BuildExecGphStructAndBindInRegst (line 82) | void BuildExecGphStructAndBindInRegst() { method BindInplacebetweenVarAndRef (line 89) | void BindInplacebetweenVarAndRef() { method BuildOutRegst (line 98) | void BuildOutRegst() { method InferProducedDataRegstTimeShape (line 112) | void InferProducedDataRegstTimeShape() override { NaiveInferProduced... FILE: oneflow/core/graph_impl/tick_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class TickCompTaskNode (line 21) | class TickCompTaskNode final : public CompTaskNode { method TickCompTaskNode (line 24) | TickCompTaskNode() = default; method IsMeaningLess (line 27) | bool IsMeaningLess() override { return false; } method TaskType (line 28) | TaskType GetTaskType() const override { return TaskType::kTick; } FILE: oneflow/core/graph_impl/unpack_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class UnpackCompTaskNode (line 21) | class UnpackCompTaskNode final : public CompTaskNode { method UnpackCompTaskNode (line 24) | UnpackCompTaskNode() = default; method TaskType (line 27) | TaskType GetTaskType() const override { return TaskType::kUnpack; } FILE: oneflow/core/graph_impl/wait_and_send_ids_compute_task_node.cpp type oneflow (line 19) | namespace oneflow { class WaitAndSendIdsCompTaskNode (line 21) | class WaitAndSendIdsCompTaskNode final : public CompTaskNode { method WaitAndSendIdsCompTaskNode (line 24) | WaitAndSendIdsCompTaskNode() = default; method ConsumeAllRegsts (line 28) | void ConsumeAllRegsts() override {} method ConsumeFakeRegsts (line 29) | void ConsumeFakeRegsts() override {} method IsMeaningLess (line 31) | bool IsMeaningLess() override { return false; } method TaskType (line 33) | TaskType GetTaskType() const override { return TaskType::kWaitAndSen... FILE: oneflow/core/hardware/basic_device_descriptor_list.cpp type oneflow (line 18) | namespace oneflow { type hardware (line 20) | namespace hardware { FILE: oneflow/core/hardware/basic_device_descriptor_list.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/hardware/cuda_device_descriptor.cpp type oneflow (line 24) | namespace oneflow { type hardware (line 26) | namespace hardware { type CudaDeviceDescriptor::Impl (line 42) | struct CudaDeviceDescriptor::Impl { FILE: oneflow/core/hardware/cuda_device_descriptor.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/hardware/cuda_device_descriptor_class.cpp type oneflow (line 29) | namespace oneflow { type hardware (line 31) | namespace hardware { class CudaDeviceDescriptorClass (line 39) | class CudaDeviceDescriptorClass : public DeviceDescriptorClass { method CudaDeviceDescriptorClass (line 41) | CudaDeviceDescriptorClass() = default; method QueryDeviceDescriptorList (line 44) | std::shared_ptr QueryDeviceDescriptorL... method Name (line 57) | std::string Name() const override { return kCudaDeviceDescriptorCl... method SerializeDeviceDescriptorList (line 59) | void SerializeDeviceDescriptorList(const std::shared_ptr DeserializeDeviceDescr... method DumpDeviceDescriptorListSummary (line 86) | void DumpDeviceDescriptorListSummary(const std::shared_ptr descrip... method RegisteredCount (line 39) | size_t RegisteredCount() { method GetRegistered (line 49) | std::shared_ptr GetRegistered(size_t ... method GetRegistered (line 54) | std::shared_ptr GetRegistered(const s... method DeviceClassRegistryStorage (line 61) | static DeviceClassRegistryStorage& Instance() { FILE: oneflow/core/hardware/device_descriptor_class.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/hardware/device_descriptor_list.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/hardware/net_ib_device_descriptor.cpp type oneflow (line 22) | namespace oneflow { type hardware (line 24) | namespace hardware { function GetPCIBusID (line 37) | void GetPCIBusID(const std::string& name, std::string* pci_bus_id) { type NetIBDeviceDescriptor::Impl (line 51) | struct NetIBDeviceDescriptor::Impl { function NetIBDeviceDescriptorLinkLayer (line 72) | NetIBDeviceDescriptorLinkLayer NetIBDeviceDescriptor::LinkLayer() co... FILE: oneflow/core/hardware/net_ib_device_descriptor.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/hardware/net_ib_device_descriptor_class.cpp type oneflow (line 27) | namespace oneflow { type hardware (line 29) | namespace hardware { class NetIBDeviceDescriptorClass (line 37) | class NetIBDeviceDescriptorClass : public DeviceDescriptorClass { method NetIBDeviceDescriptorClass (line 39) | NetIBDeviceDescriptorClass() = default; method QueryDeviceDescriptorList (line 42) | std::shared_ptr QueryDeviceDescriptorL... method Name (line 68) | std::string Name() const override { return kNetIBDeviceDescriptorC... method SerializeDeviceDescriptorList (line 70) | void SerializeDeviceDescriptorList(const std::shared_ptr DeserializeDeviceDescr... method DumpDeviceDescriptorListSummary (line 97) | void DumpDeviceDescriptorListSummary(const std::shared_ptr QueryDeviceDescriptorL... method Name (line 81) | std::string Name() const override { return kNetSocketDeviceDescrip... method SerializeDeviceDescriptorList (line 83) | void SerializeDeviceDescriptorList(const std::shared_ptr DeserializeDeviceDescr... method DumpDeviceDescriptorListSummary (line 111) | void DumpDeviceDescriptorListSummary(const std::shared_ptr GetCPUAffinit... method GetMemoryAffinity (line 58) | std::shared_ptr GetMemoryA... method GetCPUAffinityByPCIBusID (line 62) | std::shared_ptr GetCPUAffinit... method GetMemoryAffinityByPCIBusID (line 67) | std::shared_ptr GetMemoryA... method SetCPUAffinity (line 72) | void SetCPUAffinity( method SetMemoryAffinity (line 75) | void SetMemoryAffinity( class HWLocCPUAffinityDescriptor (line 81) | class HWLocCPUAffinityDescriptor : public TopologyCPUAffinityDescrip... method HWLocCPUAffinityDescriptor (line 84) | explicit HWLocCPUAffinityDescriptor(hwloc_cpuset_t hwloc_cpu_set) method hwloc_cpuset_t (line 88) | hwloc_cpuset_t HWLocCPUSet() const { return hwloc_cpu_set_; } class HWLocMemoryAffinityDescriptor (line 94) | class HWLocMemoryAffinityDescriptor : public TopologyMemoryAffinityD... method HWLocMemoryAffinityDescriptor (line 97) | explicit HWLocMemoryAffinityDescriptor(hwloc_bitmap_t hwloc_bitmap... method hwloc_bitmap_t (line 101) | hwloc_bitmap_t HWLocBitmap() const { return hwloc_bitmap_; } method hwloc_membind_policy_t (line 102) | hwloc_membind_policy_t HWLocPolicy() const { return policy_; } class HWLocTopologyDescriptor (line 109) | class HWLocTopologyDescriptor : public TopologyDescriptor { method GetCPUAffinity (line 114) | std::shared_ptr GetCPUAffinit... method GetMemoryAffinity (line 120) | std::shared_ptr GetMemoryA... method GetCPUAffinityByPCIBusID (line 127) | std::shared_ptr GetCPUAffinit... method GetMemoryAffinityByPCIBusID (line 137) | std::shared_ptr GetMemoryA... method SetCPUAffinity (line 147) | void SetCPUAffinity( method SetMemoryAffinity (line 154) | void SetMemoryAffinity( method Query (line 162) | static std::shared_ptr Query() { method Deserialize (line 175) | static std::shared_ptr Deserialize(... method Serialize (line 192) | void Serialize(std::string* serialized) const { method hwloc_obj_t (line 202) | hwloc_obj_t GetNonIOAncestorByPCIBusID(const std::string& pci_bus_... method HWLocTopologyDescriptor (line 209) | explicit HWLocTopologyDescriptor(hwloc_topology_t topology) : topo... function QueryTopologyDescriptor (line 215) | std::shared_ptr QueryTopologyDescriptor() { function DeserializeTopologyDescriptor (line 224) | std::shared_ptr DeserializeTopologyDescrip... function SerializeTopologyDescriptor (line 240) | void SerializeTopologyDescriptor(const std::shared_ptr (line 83) | struct PushBackPtrFieldName { method Call (line 85) | static void Call(WalkCtxType* ctx, const char* field_name) { type FilterPointerFieldName (line 91) | struct FilterPointerFieldName { method Call (line 92) | static void Call(WalkCtxType* ctx, FieldType* field, const char* fie... type FilterPointerFieldNameUntil (line 98) | struct FilterPointerFieldNameUntil { method Call (line 99) | static bool Call(WalkCtxType* ctx, FieldType* field) { function TEST (line 105) | TEST(DSS, filter_field) { function TEST (line 113) | TEST(DSS, filter_field_until) { type TestDssUnion (line 126) | struct TestDssUnion { type UnionField (line 130) | struct UnionField { type StaticDumpFieldName (line 144) | struct StaticDumpFieldName { method Call (line 145) | static void Call(WalkCtxType* ctx, const char* field_name, const cha... function TEST (line 151) | TEST(DSS, union_field) { function TEST (line 175) | TEST(DSS, static_verbose_field) { FILE: oneflow/core/intrusive/flat_msg.h type FlatMsgGetDefault (line 197) | struct FlatMsgGetDefault type _FLAT_MSG_ONEOF_ENUM_TYPE (line 224) | enum _FLAT_MSG_ONEOF_ENUM_TYPE function FlatMsgRepeatedField (line 440) | FlatMsgRepeatedField final { FILE: oneflow/core/intrusive/flat_msg_test.cpp type oneflow (line 20) | namespace oneflow { type DumpFieldName (line 25) | struct DumpFieldName { method Call (line 26) | static void Call(WalkCtxType* ctx, FieldType* field, const char* fie... function GetFieldNames (line 32) | std::vector GetFieldNames(T* flat_msg) { function CheckSoleFieldName (line 39) | void CheckSoleFieldName(T* flat_msg, const std::string& expected) { function TEST (line 50) | TEST(FlatMsg, optional) { function TEST (line 74) | TEST(FlatMsg, oneof) { FILE: oneflow/core/intrusive/flat_msg_view.h function namespace (line 26) | namespace oneflow { function Match (line 230) | static bool Match(FlatMsgViewT* flat_msg_view, const ValueType* data_ptr... function Match (line 249) | static bool Match(FlatMsgViewT* self, const ContainerT& container) { function Match (line 261) | static bool Match(FlatMsgViewT* self, const std::vector__Init__(); } class TestListHead (line 43) | class TestListHead : public intrusive::ListHead { method TestListHead (line 45) | TestListHead() { this->__Init__(); } function TEST (line 50) | TEST(TestListHook, init) { function TEST (line 56) | TEST(TestListHook, append_to) { function TEST (line 64) | TEST(TestListHook, clear) { function TEST (line 73) | TEST(ListHead, empty) { function TEST (line 78) | TEST(ListHead, push_front) { function TEST (line 97) | TEST(ListHead, end) { function TEST (line 105) | TEST(ListHead, begin) { function TEST (line 116) | TEST(ListHead, last) { function TEST (line 127) | TEST(ListHead, push_back) { function TEST (line 138) | TEST(ListHead, erase) { function TEST (line 154) | TEST(ListHead, pop_front) { function TEST (line 170) | TEST(ListHead, pop_back) { function TEST (line 186) | TEST(ListHead, Next) { function TEST (line 203) | TEST(ListHead, prev_item) { FILE: oneflow/core/intrusive/list_test.cpp type oneflow (line 25) | namespace oneflow { type test (line 27) | namespace test { class TestListItem (line 31) | class TestListItem : public intrusive::Base { method __Init__ (line 33) | void __Init__() { clear_cnt(); } method __Delete__ (line 34) | void __Delete__() { method has_cnt (line 39) | bool has_cnt() const { return cnt_ != nullptr; } method cnt (line 40) | int cnt() const { return *cnt_; } method is_foo_list_empty (line 41) | bool is_foo_list_empty() const { return foo_list_.empty(); } method set_cnt (line 44) | void set_cnt(int* val) { cnt_ = val; } method clear_cnt (line 45) | void clear_cnt() { cnt_ = nullptr; } method ref_cnt (line 48) | size_t ref_cnt() const { return intrusive_ref_.ref_cnt(); } method TestListItem (line 56) | TestListItem() : foo_list_(), intrusive_ref_(), cnt_() {} function TEST (line 63) | TEST(List, empty) { function TEST (line 69) | TEST(List, empty_Begin) { function TEST (line 80) | TEST(List, empty_Next) { function TEST (line 96) | TEST(List, PushFront) { function TEST (line 110) | TEST(List, destructor) { function TEST (line 135) | TEST(List, PushBack) { function TEST (line 149) | TEST(List, Erase) { function TEST (line 166) | TEST(List, PopBack) { function TEST (line 183) | TEST(List, PopFront) { function TEST (line 199) | TEST(List, Clear) { function TEST (line 213) | TEST(List, UNSAFE_FOR_EACH_PTR) { function TEST (line 231) | TEST(List, FOR_EACH) { class TestIntrusiveListHead (line 254) | class TestIntrusiveListHead final : public intrusive::Base { method FooList (line 259) | const FooList& foo_list() const { return foo_list_; } method FooList (line 261) | FooList* mut_foo_list() { return &foo_list_; } method TestIntrusiveListHead (line 267) | TestIntrusiveListHead() : intrusive_ref_(), foo_list_() {} function TEST (line 272) | TEST(List, intrusive_list_for_each) { class TestIntrusiveListHeadWrapper (line 298) | class TestIntrusiveListHeadWrapper final : public intrusive::Base { method TestIntrusiveListHead (line 301) | const TestIntrusiveListHead& head() const { method TestIntrusiveListHead (line 307) | TestIntrusiveListHead* mut_head() { method clear_head (line 311) | void clear_head() { method TestIntrusiveListHeadWrapper (line 319) | TestIntrusiveListHeadWrapper() : intrusive_ref_(), head_() {} function TEST (line 324) | TEST(List, nested_list_delete) { function TEST (line 348) | TEST(List, MoveTo) { FILE: oneflow/core/intrusive/mutexed_list.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/intrusive/object_pool.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/intrusive/object_pool_test.cpp type oneflow (line 23) | namespace oneflow { type intrusive (line 25) | namespace intrusive { type test (line 27) | namespace test { class IntrusiveFoo (line 31) | class IntrusiveFoo final // NOLINT method IntrusiveFoo (line 35) | IntrusiveFoo() = default; function TEST (line 43) | TEST(ObjectPool_kThreadUnsafeAndDisableDestruct, append_to_pool) { function TEST (line 50) | TEST(ObjectPool_kThreadUnsafeAndDisableDestruct, recycle) { FILE: oneflow/core/intrusive/ref.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/intrusive/reflective.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/intrusive/shared_ptr.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/intrusive/skiplist.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/intrusive/skiplist_hook.h function namespace (line 26) | namespace oneflow { function LevelZeroHookOffset (line 99) | int LevelZeroHookOffset() { function __Init__ (line 105) | void __Init__() { hook_.NullptrClear(); } function T (line 108) | T* mut_key() { return &key_; } function Clear (line 112) | void Clear() { function self_type (line 117) | static self_type* Find(const key_type& key, hook_type* head, int size_sh... function self_type (line 124) | static self_type* Erase(const key_type& key, hook_type* head, int size_s... function Erase (line 130) | static void Erase(self_type* elem) { elem->hook_.Erase(); } function SkipListHook (line 154) | static SkipListHook* ThisPtr4HookPtr(ListHook* list_hook_ptr, int level) { function RandomNumLevels (line 165) | static int32_t RandomNumLevels(int size_shift) { function ListHook (line 173) | static ListHook* SearchLastBottomHookLessThan(const key_type& key, hook_... function IteratorHookOffset (line 205) | int IteratorHookOffset() { function __Init__ (line 209) | void __Init__() { function value_type (line 217) | value_type* Begin() { function value_type (line 224) | value_type* Find(const key_type& key) { function value_type (line 229) | const value_type* Find(const key_type& key) const { function value_type (line 235) | value_type* Erase(const key_type& key) { function Erase (line 240) | void Erase(value_type* elem) { function Clear (line 267) | void Clear() { function empty_debug (line 271) | bool empty_debug() const { FILE: oneflow/core/intrusive/skiplist_hook_test.cpp type oneflow (line 20) | namespace oneflow { type intrusive (line 22) | namespace intrusive { type test (line 24) | namespace test { class TestSkipListHead (line 27) | class TestSkipListHead final : public SkipListHead {... method TestSkipListHead (line 29) | TestSkipListHead() { this->__Init__(); } method TestSkipListHead (line 30) | TestSkipListHead(const TestSkipListHead&) = delete; method TestSkipListHead (line 31) | TestSkipListHead(TestSkipListHead&&) = delete; method TestSkipListHead (line 32) | TestSkipListHead& operator==(const TestSkipListHead&) = delete; method TestSkipListHead (line 33) | TestSkipListHead& operator==(TestSkipListHead&&) = delete; type FooSkipListElem (line 37) | struct FooSkipListElem { method FooSkipListElem (line 38) | FooSkipListElem() : value() { key.__Init__(); } function TEST (line 46) | TEST(SkipListHook, empty) { function TEST (line 52) | TEST(SkipListHook, insert_naive) { function TEST (line 69) | TEST(SkipListHook, erase_by_key) { function TEST (line 82) | TEST(SkipListHook, erase_by_elem) { function TEST (line 95) | TEST(SkipListHook, insert_many) { function TEST (line 122) | TEST(SkipListHook, erase_many_by_key) { function TEST (line 145) | TEST(SkipListHook, erase_many_by_elem) { FILE: oneflow/core/intrusive/skiplist_test.cpp type oneflow (line 20) | namespace oneflow { type intrusive (line 22) | namespace intrusive { type test (line 24) | namespace test { class SkipListFoo (line 28) | class SkipListFoo final : public intrusive::Base { method __Init__ (line 30) | void __Init__() { clear_is_deleted(); } method __Delete__ (line 31) | void __Delete__() { method has_is_deleted (line 36) | bool has_is_deleted() const { return is_deleted_ != nullptr; } method is_deleted (line 37) | int is_deleted() const { return *is_deleted_; } method foo_map_key (line 38) | int32_t foo_map_key() const { return foo_map_key_.key(); } method set_is_deleted (line 40) | void set_is_deleted(int* val) { is_deleted_ = val; } method clear_is_deleted (line 41) | void clear_is_deleted() { is_deleted_ = nullptr; } method set_foo_map_key (line 43) | void set_foo_map_key(int32_t val) { *foo_map_key_.mut_key() = va... method ref_cnt (line 45) | size_t ref_cnt() const { return intrusive_ref_.ref_cnt(); } method SkipListFoo (line 51) | SkipListFoo() : intrusive_ref_(), is_deleted_(), foo_map_key_() {} class SkipListFooContainer (line 59) | class SkipListFooContainer final : public intrusive::Base { method Key2SkipListFoo (line 64) | const Key2SkipListFoo& foo_map() const { return foo_map_; } method Key2SkipListFoo (line 66) | Key2SkipListFoo* mut_foo_map() { return &foo_map_; } method SkipListFooContainer (line 72) | SkipListFooContainer() : intrusive_ref_(), foo_map_() {} function TEST (line 79) | TEST(SkipList, empty) { function TEST (line 85) | TEST(SkipList, insert_naive) { function TEST (line 101) | TEST(SkipList, insert_twice) { function TEST (line 111) | TEST(SkipList, erase_by_key) { function TEST (line 123) | TEST(SkipList, erase_by_elem) { function TEST (line 135) | TEST(SkipList, insert_many) { function TEST (line 162) | TEST(SkipList, erase_many_by_key) { function TEST (line 185) | TEST(SkipList, erase_many_by_elem) { function TEST (line 208) | TEST(SkipList, MAP_HEAD) { function TEST (line 241) | TEST(SkipList, FOR_EACH) { FILE: oneflow/core/intrusive/static_counter.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/intrusive/static_counter_test.cpp type oneflow (line 21) | namespace oneflow { type test (line 23) | namespace test { function TEST (line 31) | TEST(StaticCounter, eq0) { static_assert(STATIC_COUNTER(static_count... function TEST (line 37) | TEST(StaticCounter, eq1) { static_assert(STATIC_COUNTER(static_count... function TEST (line 41) | TEST(StaticCounter, eq1_again) { static_assert(STATIC_COUNTER(static... function TEST (line 47) | TEST(StaticCounter, eq2) { static_assert(STATIC_COUNTER(static_count... FILE: oneflow/core/intrusive/struct_traits.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/intrusive/struct_traits_test.cpp type oneflow (line 20) | namespace oneflow { type test (line 22) | namespace test { type OneflowTestNamespaceFoo (line 26) | struct OneflowTestNamespaceFoo { method OneflowTestNamespaceFoo (line 27) | OneflowTestNamespaceFoo() : x(0), bar(0), const_bar(0) {} function TEST (line 34) | TEST(StructField, mutable_struct_mutable_field) { function TEST (line 43) | TEST(StructField, mutable_struct_const_field) { function TEST (line 52) | TEST(StructField, const_struct_mutable_field) { function TEST (line 61) | TEST(StructField, const_struct_const_field) { type X (line 70) | struct X { type Y (line 75) | struct Y { function TEST (line 80) | TEST(StructField, compose) { FILE: oneflow/core/ipc/shared_memory.cpp type oneflow (line 32) | namespace oneflow { type ipc (line 33) | namespace ipc { function ShmOpen (line 40) | int ShmOpen(const std::string& shm_name, int* fd, bool create) { function ShmOpen (line 48) | int ShmOpen(std::string* shm_name, int* fd, bool create) { function ShmMap (line 59) | int ShmMap(int fd, const size_t shm_size, void** ptr) { function ShmSetUp (line 66) | Maybe ShmSetUp(std::string* shm_name, size_t shm_size, bool c... function ShmSetUp (line 81) | Maybe ShmSetUp(const std::string& shm_name, size_t* shm_size,... function GetContentsOfShmDirectory (line 97) | Maybe> GetContentsOfShmDirectory() { function SharedMemoryManager (line 115) | SharedMemoryManager& SharedMemoryManager::get() { FILE: oneflow/core/ipc/shared_memory.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job/checkpointing_config_def.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job/cluster_instruction.cpp type oneflow (line 24) | namespace oneflow { function GetHaltAckCtrlKey (line 28) | std::string GetHaltAckCtrlKey(int64_t machine_id) { function GetClusterInstructionKey (line 34) | std::string GetClusterInstructionKey() { class ObsoleteCtrlKeys (line 39) | class ObsoleteCtrlKeys { method ObsoleteCtrlKeys (line 41) | ObsoleteCtrlKeys() = default; method ForEach (line 45) | void ForEach(const CallbackT& Callback) const { method Clear (line 50) | void Clear() { method Add (line 54) | void Add(const std::string& key) { function OccasionallyClearCtrlKV (line 66) | void OccasionallyClearCtrlKV(const std::string& key) { function PushClusterInstruction (line 82) | void PushClusterInstruction(const ClusterInstructionProto& cluster_ins... function PullClusterInstruction (line 88) | void PullClusterInstruction(ClusterInstructionProto* cluster_instructi... FILE: oneflow/core/job/cluster_instruction.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/coordinator.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/executor.cpp type oneflow (line 18) | namespace oneflow { type boxing (line 20) | namespace boxing { type collective (line 22) | namespace collective { FILE: oneflow/core/job/collective_boxing/executor.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/executor_backend.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/executor_backend_manager.cpp type oneflow (line 19) | namespace oneflow { type boxing (line 21) | namespace boxing { type collective (line 23) | namespace collective { function ExecutorBackendMgr (line 25) | ExecutorBackendMgr& ExecutorBackendMgr::Get() { FILE: oneflow/core/job/collective_boxing/executor_backend_manager.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/request_store.cpp type oneflow (line 24) | namespace oneflow { type boxing (line 26) | namespace boxing { type collective (line 28) | namespace collective { type RequestEntryToken (line 97) | struct RequestEntryToken { function RequestEntry (line 112) | RequestEntry* RequestStore::GetRequestEntry(void* request_entry_to... FILE: oneflow/core/job/collective_boxing/request_store.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/runtime_request_info.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/scheduler.cpp type oneflow (line 28) | namespace oneflow { type boxing (line 30) | namespace boxing { type collective (line 32) | namespace collective { function CanMergeIntoCurGroup (line 36) | bool CanMergeIntoCurGroup(RequestStore* request_store, const Reque... function HasRankInteraction (line 48) | bool HasRankInteraction(const DeviceSet& a, const DeviceSet& b) { class RequestHandle (line 60) | class RequestHandle final { method RequestHandle (line 63) | RequestHandle(int32_t local_rank, void* request_entry_token, voi... method local_rank (line 69) | int32_t local_rank() const { return local_rank_; } class GroupToken (line 81) | class GroupToken final { method GroupToken (line 84) | GroupToken(DeviceType device_type, void* backend_group_token) method DeviceType (line 88) | DeviceType device_type() { return device_type_; } class ExecutorImpl (line 97) | class ExecutorImpl : public Executor { method ExecutorImpl (line 99) | ExecutorImpl() = default; function GroupToken (line 155) | GroupToken* ExecutorImpl::CreateGroupToken(const std::vector& job_ids)... function SchedulerPlanToken (line 248) | SchedulerPlanToken* Scheduler::AddPlan(const Plan& plan) { method SchedulerPlanToken (line 240) | explicit SchedulerPlanToken(const std::vector& job_ids)... function RequestHandle (line 274) | RequestHandle* Scheduler::CreateRequestHandle(const RankDesc& rank... method RequestHandle (line 63) | RequestHandle(int32_t local_rank, void* request_entry_token, voi... method local_rank (line 69) | int32_t local_rank() const { return local_rank_; } FILE: oneflow/core/job/collective_boxing/scheduler.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job/collective_boxing/static_group_coordinator.cpp type oneflow (line 25) | namespace oneflow { type boxing (line 27) | namespace boxing { type collective (line 29) | namespace collective { function SortRequestIdsByOrder (line 33) | void SortRequestIdsByOrder(RequestStore* request_store, std::vecto... function HasRankInteractionOnDeviceSet (line 41) | bool HasRankInteractionOnDeviceSet(const DeviceSet& a, const Devic... type GroupState (line 53) | struct GroupState { method GroupState (line 54) | explicit GroupState(int32_t group_size) : index2is_ready(group_s... type RequestGroupIndex (line 67) | struct RequestGroupIndex { class GroupToken (line 72) | class GroupToken type StaticGroupRequestsInfo (line 74) | struct StaticGroupRequestsInfo { type StaticGroupRequestsInfoToken (line 81) | struct StaticGroupRequestsInfoToken { type StaticGroupCoordinator::Impl (line 86) | struct StaticGroupCoordinator::Impl { FILE: oneflow/core/job/collective_boxing/static_group_coordinator.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/compile_mode.cpp type oneflow (line 21) | namespace oneflow { type CompileModeName (line 25) | struct CompileModeName final : public CompileModeVisitor Name2CompileMode() { function GetValidCompileModeNames (line 41) | std::string GetValidCompileModeNames() { function CurrentCompileMode (line 54) | Maybe CurrentCompileMode() { FILE: oneflow/core/job/compile_mode.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/compiler.cpp type oneflow (line 28) | namespace oneflow { FILE: oneflow/core/job/compiler.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job/critical_section_desc.cpp type oneflow (line 22) | namespace oneflow { function CriticalSection (line 24) | CriticalSection* CriticalSectionDesc::AddCriticalSection(int64_t job_i... function CriticalSection (line 54) | const CriticalSection& CriticalSectionDesc::GetCriticalSection(int64_t... function CriticalSection (line 59) | CriticalSection* CriticalSectionDesc::MutCriticalSection(int64_t criti... function FOR_RANGE (line 71) | FOR_RANGE(int64_t, i, 0, critical_sections_.size()) { function FOR_RANGE (line 81) | FOR_RANGE(int64_t, i, 0, critical_sections_.size()) { function FOR_RANGE (line 94) | FOR_RANGE(int64_t, i, 0, critical_sections_.size()) { function FOR_RANGE (line 109) | FOR_RANGE(int64_t, i, 0, critical_sections_.size()) { FILE: oneflow/core/job/critical_section_desc.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job/critical_section_instance.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/eager_ccl_comm_manager.cpp type oneflow (line 19) | namespace oneflow { function EagerCclCommMgrBuilder (line 23) | EagerCclCommMgrBuilder& EagerCclCommMgrBuilder::Get() { FILE: oneflow/core/job/eager_ccl_comm_manager.h function namespace (line 23) | namespace oneflow { function EagerCclCommMgr (line 66) | EagerCclCommMgr* NewCclCommMgr(DeviceType device_type) const { function class (line 86) | class UserKernelUnifiedCclCommInitRegistry final { FILE: oneflow/core/job/eager_nccl_comm_manager.cpp type oneflow (line 29) | namespace oneflow { function GetNcclUniqueIdRpcKey (line 33) | std::string GetNcclUniqueIdRpcKey(const std::vector& a, function CreateNcclComm (line 57) | void CreateNcclComm(ncclComm_t* comm, const int dev, const std::string... function NeedUnifiedNcclCommInit (line 83) | bool NeedUnifiedNcclCommInit(const OperatorConf& op_conf) { function ncclComm_t (line 111) | ncclComm_t EagerNcclCommMgr::GetCommForDevice( function ncclComm_t (line 134) | ncclComm_t EagerNcclCommMgr::GetCommForDeviceAndStreamName( function FOR_RANGE (line 161) | FOR_RANGE(int64_t, parallel_id, 0, parallel_desc.parallel_num()) { function FOR_RANGE (line 176) | FOR_RANGE(int64_t, parallel_id, 0, parallel_desc.parallel_num()) { FILE: oneflow/core/job/eager_nccl_comm_manager.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/job/env_desc.cpp type oneflow (line 19) | namespace oneflow { function BootstrapConf (line 21) | const BootstrapConf& EnvDesc::bootstrap_conf() const { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, machine_num) { FILE: oneflow/core/job/env_desc.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/env_global_objects_scope.cpp type oneflow (line 62) | namespace oneflow { function LogDir (line 66) | std::string LogDir(const std::string& log_dir) { function InitLogging (line 73) | void InitLogging(const CppLoggingConf& logging_conf) { function GetDefaultCpuDeviceNum (line 88) | int32_t GetDefaultCpuDeviceNum() { return std::thread::hardware_concur... function Resource (line 90) | Resource GetDefaultResource(const EnvProto& env_proto) { function SetCpuDeviceManagerNumThreads (line 101) | void SetCpuDeviceManagerNumThreads() { function ClearAllSymbol (line 112) | void ClearAllSymbol() { function CommNetIBEnabled (line 121) | bool CommNetIBEnabled() { function InitRDMA (line 281) | Maybe InitRDMA() { function RDMAIsInitialized (line 305) | Maybe RDMAIsInitialized() { function DestoryRDMA (line 313) | Maybe DestoryRDMA() { FILE: oneflow/core/job/env_global_objects_scope.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job/function_config_def.cpp type oneflow (line 18) | namespace oneflow {} FILE: oneflow/core/job/global_for.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/core/job/global_for.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/global_mode.cpp type oneflow (line 21) | namespace oneflow { function GetGlobalParallelDescFromDevice (line 23) | Symbol GetGlobalParallelDescFromDevice(const Optional> GetInterfaceOpName2JobIds( function InitJobId2MutualExclusionJobIds (line 73) | std::vector> InitJobId2MutualExclusionJobIds( function GetMutualExclusionJobGroups (line 104) | std::vector> GetMutualExclusionJobGroups( function MergeReusedChunk (line 159) | void MergeReusedChunk(HashMap* chunk_id2chunk, function MergeSharedMemBlockR2L (line 235) | void MergeSharedMemBlockR2L(RegstDescProto* lhs, RegstDescProto* rhs, function MergeSharedInterfaceMemBlock (line 288) | void MergeSharedInterfaceMemBlock(const std::vector (line 47) | struct hash<::oneflow::MemAllocAlgoType> { type oneflow (line 55) | namespace oneflow { type MemAllocAlgoType (line 35) | enum MemAllocAlgoType { function GenDeviceUniqueId (line 59) | int64_t GenDeviceUniqueId(int64_t machine_id, int64_t device_id) { function TryConnectWithMemSafeGuardCtrlRegstDesc (line 63) | void TryConnectWithMemSafeGuardCtrlRegstDesc(TaskProto* src_task_proto... type MemoryChain (line 77) | struct MemoryChain { function InitMemoryChains (line 84) | void InitMemoryChains(Plan* plan, function IsReachableToAnyOtherTask (line 142) | bool IsReachableToAnyOtherTask(const TaskProto* src_task, const HashSe... function IsTaskConnectedL2R (line 151) | bool IsTaskConnectedL2R(const TaskProto* src, const TaskProto* dst) { function GenMemChainTasksAndRegsts (line 160) | void GenMemChainTasksAndRegsts( function GenRegstAllocFreeTimeLineAndRegstLifetimes (line 238) | void GenRegstAllocFreeTimeLineAndRegstLifetimes( function MemReusedLifetimeFirstAlgo (line 338) | void MemReusedLifetimeFirstAlgo( function MemReusedTimeLineAlgo (line 356) | void MemReusedTimeLineAlgo( function MemReusedMemVolumeFirstAlgo (line 376) | void MemReusedMemVolumeFirstAlgo( function SelectAlgorithmGenMemBlockOffset4Regsts (line 400) | void SelectAlgorithmGenMemBlockOffset4Regsts( function CountMemAllocAlgoNum (line 427) | int64_t CountMemAllocAlgoNum() { function InitAlgo2Result (line 448) | void InitAlgo2Result( FILE: oneflow/core/job/intra_job_mem_sharing_util.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/job/job_build_and_infer_ctx.cpp type oneflow (line 34) | namespace oneflow { function ResetOpConfName (line 41) | void ResetOpConfName(OperatorConf* op_conf, const std::string& new_op_... function GetOpNames (line 57) | Maybe GetOpNames(const Job& job, HashSet* op_names) { function UpdateOpName2AncestorsNeedNoGrad (line 64) | void UpdateOpName2AncestorsNeedNoGrad( function FOR_RANGE (line 469) | FOR_RANGE(int32_t, i, 0, sub_op_list_size) { function FOR_RANGE (line 485) | FOR_RANGE(int32_t, i, 0, sub_op_list_size) { sub_lbis->at(i).set_op_na... function Job (line 848) | const Job& JobBuildAndInferCtx::job() const { return *job_; } function ParallelConf (line 855) | ParallelConf LazyJobBuildAndInferCtx::GetLocalOpParallelConf(const Par... function FOR_RANGE (line 886) | FOR_RANGE(int32_t, i, 0, parallel_desc.parallel_num()) { function FOR_RANGE (line 900) | FOR_RANGE(int32_t, i, 0, parallel_desc.parallel_num()) { function OpConf2ClassName (line 1050) | std::string OpConf2ClassName(const OperatorConf& op_conf) { function FormateUserConf (line 1064) | void FormateUserConf(nlohmann::json& json_conf) { function FormateVariableConf (line 1109) | void FormateVariableConf(nlohmann::json& json_conf) { FILE: oneflow/core/job/job_build_and_infer_ctx.h function namespace (line 29) | namespace oneflow { function class (line 159) | class LazyJobBuildAndInferCtx : public JobBuildAndInferCtx { FILE: oneflow/core/job/job_build_and_infer_ctx_mgr.cpp type oneflow (line 26) | namespace oneflow { function JobBuildAndInferCtx (line 44) | JobBuildAndInferCtx* LazyJobBuildAndInferCtxMgr::NewJobBuildAndInferCt... function GlobalJobBuildAndInferCtxMgr (line 96) | Maybe GlobalJobBuildAndInferCtxMgr() { function GetJobBuildAndInferCtx (line 100) | Maybe GetJobBuildAndInferCtx(const std::string& ... function GetCurInferCtx (line 105) | Maybe GetCurInferCtx() { FILE: oneflow/core/job/job_build_and_infer_ctx_mgr.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/job/job_builder.cpp type oneflow (line 27) | namespace oneflow { function GetParallelHierarchyNumAxes (line 31) | int64_t GetParallelHierarchyNumAxes( function SetNdSbpSignature4Oba (line 44) | void SetNdSbpSignature4Oba(Job* job, function SetSbpSignature4Oba (line 55) | void SetSbpSignature4Oba(Job* job, const OpBlobArg& oba, const SbpPara... function AddOrSetNdSbpSignature4OpName (line 61) | void AddOrSetNdSbpSignature4OpName( function AddOrSetSbpSignature4OpName (line 75) | void AddOrSetSbpSignature4OpName(Job* job, const std::string& op_name, function MakeGetterParallelConf4OpName (line 84) | std::function MakeGetterParal... function FOR_RANGE (line 100) | FOR_RANGE(int32_t, i, 0, job->net().op_size()) { function FOR_RANGE (line 105) | FOR_RANGE(int32_t, i, 0, job->placement().placement_group_size()) { function FOR_RANGE (line 126) | FOR_RANGE(int32_t, i, 0, job->placement().blob_placement_group_size()) { function SbpParallel (line 400) | SbpParallel* JobBuilder::MutSbpParallel4Oba(const OpBlobArg& oba) const { function SbpSignature (line 422) | const SbpSignature JobBuilder::SbpSignature4OpName(const std::string& ... function NdSbpSignature (line 441) | const NdSbpSignature& JobBuilder::NdSbpSignature4OpName(const std::str... FILE: oneflow/core/job/job_builder.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job/job_desc.cpp type oneflow (line 29) | namespace oneflow { function CheckFunctionConfig (line 33) | void CheckFunctionConfig(const JobConfigProto& job_conf) { function AttrValue (line 61) | const AttrValue& JobDesc::GetFunctionFlagVal(const std::string& field_... function IsInterfaceOpConf (line 70) | bool IsInterfaceOpConf(const OperatorConf& op_conf) { function JobDesc (line 81) | const JobDesc& GlobalJobDesc() { return *Singleton::Get(); } FILE: oneflow/core/job/job_desc.h function namespace (line 29) | namespace oneflow { FILE: oneflow/core/job/job_instance.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/job_interpreter.cpp type oneflow (line 30) | namespace oneflow { type one (line 31) | namespace one { function InitEnv (line 36) | Maybe InitEnv(const one::TensorTuple& graph_inputs, const std::... function OpConfToUserOpExpr (line 48) | Maybe OpConfToUserOpExpr(const OperatorConf& op_conf) { function GetInputTensors (line 63) | Maybe>> GetInputTen... function GetOutputNamesOfOp (line 78) | OpArgsVector GetOutputNamesOfOp(const UserOpConf& user_... function IsViewOp (line 87) | bool IsViewOp(const std::shared_ptr& op) { function RunViewOp (line 91) | Maybe RunViewOp(const std::shared_ptr& op, Env& en... function RawRunGlobalNormalOp (line 110) | Maybe RawRunGlobalNormalOp(const std::shared_ptr& ... function RunGlobalNormalOp (line 157) | Maybe RunGlobalNormalOp(const std::shared_ptr& op,... function RunNormalOp (line 167) | Maybe RunNormalOp(const std::shared_ptr& op, Env& ... function GetOutdatedTensorsAfterOp (line 181) | std::vector> GetOutdatedTensorsAfterOp(cons... function InitOpExprs (line 206) | Maybe InitOpExprs(const std::shared_ptr& graph) { function InterpretJob (line 222) | Maybe InterpretJob(const one::TensorTuple& graph_i... FILE: oneflow/core/job/job_interpreter.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/job/job_ir.cpp type oneflow (line 18) | namespace oneflow { function ConvertJobToTosaIR (line 22) | Maybe ConvertJobToTosaIR(Job* job) { function SaveJobToIR (line 26) | Maybe SaveJobToIR(Job* job, const std::string& path) { function ConvertJobToIR (line 30) | Maybe ConvertJobToIR(Job* job) { function LoadJobFromIR (line 34) | Maybe LoadJobFromIR(Job* job, const std::string& path) { FILE: oneflow/core/job/job_ir.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/job_set_compile_ctx.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/lazy_mode.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job/lazy_mode.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/local_sig_infer_hint.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/memory_share_strategy.cpp type oneflow (line 25) | namespace oneflow { function IsLifetimeExcluded (line 31) | bool IsLifetimeExcluded(const std::pair& a, FILE: oneflow/core/job/memory_share_strategy.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/job/nd_sbp_infer_hint.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job/nd_sbp_util.cpp type oneflow (line 21) | namespace oneflow { function GetTensorSliceView (line 23) | std::vector GetTensorSliceView(const int64_t parallel... function TensorSliceView (line 58) | TensorSliceView GetTensorSliceView4ParallelRank(const Shape& parallel_... function TensorSliceView (line 102) | TensorSliceView GetTensorSliceView4ParallelId(const Shape& parallel_hi... function GetTensorSliceView (line 111) | std::vector GetTensorSliceView(const Shape& parallel_... function TensorSliceView (line 121) | TensorSliceView GetBroadcastTensorSliceView(const BlobDesc& blob_desc) { function NdSbpHasPartialParallel (line 125) | bool NdSbpHasPartialParallel(const NdSbp& nd_sbp) { function NdSbpHasBroadcastParallel (line 133) | bool NdSbpHasBroadcastParallel(const NdSbp& nd_sbp) { function NdSbpIsAllBroadcast (line 141) | bool NdSbpIsAllBroadcast(const NdSbp& nd_sbp) { function NdSbpIsAllPartialSum (line 148) | bool NdSbpIsAllPartialSum(const NdSbp& nd_sbp) { function NdSbpIsAllSplit (line 155) | bool NdSbpIsAllSplit(const NdSbp& nd_sbp, int64_t axis) { FILE: oneflow/core/job/nd_sbp_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/oneflow.cpp type std (line 43) | namespace std { type hash (line 46) | struct hash { type oneflow (line 56) | namespace oneflow { type ReentrantLockBackEdge (line 74) | struct ReentrantLockBackEdge { function cluster_thrd_ids_key (line 79) | std::string cluster_thrd_ids_key(const std::string& plan_name) { function ctrl_regst_desc_info_key (line 83) | std::string ctrl_regst_desc_info_key(const std::string& plan_name) { function job_id2job_conf (line 87) | std::string job_id2job_conf(const std::string& plan_name) { return pla... function GetCollectiveBoxingPlanKey (line 89) | std::string GetCollectiveBoxingPlanKey(const std::string& plan_name) { function sub_plan_key (line 93) | std::string sub_plan_key(const std::string& plan_name, int64_t machine... function block7chunk_key (line 97) | std::string block7chunk_key(const std::string& plan_name, int64_t mach... function PushPlan (line 101) | void PushPlan(const std::string& plan_name, Plan&& plan) { function PullPlan (line 152) | void PullPlan(const std::string& plan_name, Plan* plan) { function CompileCurJobOnMaster (line 185) | Maybe CompileCurJobOnMaster(Job* job, Plan* plan, bool need_job_... function MergePlan (line 204) | void MergePlan(Plan* plan, Plan&& other) { function MergeSubPlan (line 226) | void MergeSubPlan(Plan* plan, std::vector&& sub_plans) { function RegstDescProto (line 232) | RegstDescProto* GetSoleDataRegstDescProto(TaskProto* task) { function OperatorConf (line 243) | const OperatorConf& GetSoleOpConf(Plan* plan, const TaskProto& task) { function UpdateSoleObnRegstDescId (line 250) | void UpdateSoleObnRegstDescId(Plan* plan, TaskProto* task) { function LinkTickTaskProto (line 269) | void LinkTickTaskProto(Plan* plan, TaskProto* identity_tick, TaskProto... function LinkMainPlan (line 288) | void LinkMainPlan(Plan* plan, Plan&& main_plan, function GetMemSharingOpBlobInfo (line 354) | void GetMemSharingOpBlobInfo(const JobBuilder& job_builder, const std:... function FilterOpName2ParallelBlobConf (line 381) | void FilterOpName2ParallelBlobConf( function CheckNonDistributeOptimizerAvailable (line 400) | void CheckNonDistributeOptimizerAvailable(const std::vector MakeMainJobComponent( function MakeCallbackNotifierSinkTick (line 547) | Maybe MakeCallbackNotifierSinkTick( function MakeMainJob (line 598) | Maybe MakeMainJob(Job* main_job, function ConnectCriticalSectionEndToReentrantLockEnd (line 671) | Maybe ConnectCriticalSectionEndToReentrantLockEnd( function CompileMainJob (line 717) | Maybe CompileMainJob(Job* main_job, const std::vector CompileJobsAndMergePlans(const PbRpf& job_confs, Plan... function CompileJobsAndPushMergedPlan (line 869) | Maybe CompileJobsAndPushMergedPlan(const PbRpf& job_confs) { FILE: oneflow/core/job/oneflow.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/job/parallel_desc.cpp type oneflow (line 31) | namespace oneflow { function GetDeviceCount (line 35) | int64_t GetDeviceCount(DeviceType device_type) { function GlobalDeviceIdsContaining (line 42) | bool GlobalDeviceIdsContaining(const MachineId2DeviceIdList& bigger, function ParseDeviceNameConf (line 58) | Maybe> ParseDeviceNameConf(const std::... function ParseMachineAndDeviceIdList (line 66) | Maybe ParseMachineAndDeviceIdList(const ParallelConf& parall... function GetTensorDevice4CurrentProcessCtx (line 170) | Maybe> GetTensorDevice4CurrentProcessCtx(Symbol GetPartIdAndPartNumFromParallelCtx( function ParallelConf (line 378) | ParallelConf GenParallelConfOfCpuZeroOnMaster() { function ParallelConf (line 385) | ParallelConf GenParallelConfOfCpuZeroOnAllMachines() { function ParallelConf (line 394) | ParallelConf GenParallelConfOfCpuOnAllRanks() { function CalcParallelId4CurrentProcessCtx (line 407) | Maybe> CalcParallelId4CurrentProcessCtx(Symbol CalcParallelContext4CurrentProcessCtx( function RawReplaceDeviceType (line 432) | Maybe> RawReplaceDeviceType(Symbol ... function RanksToString (line 439) | Maybe RanksToString(int64_t axis, const int64_t* ranks, c... function RawPlacementToString (line 452) | Maybe RawPlacementToString(Symbol placement) { function RawGetTensorDevice (line 478) | Maybe> RawGetTensorDevice(Symbol parallel... function RawTxtStringToPlacement (line 486) | Maybe> RawTxtStringToPlacement(const std::string&... function RawCheckDeviceIdsIsValid (line 492) | Maybe RawCheckDeviceIdsIsValid(Symbol placement) { function RawGetParallelDescOfThisRank (line 497) | Maybe> RawGetParallelDescOfThisRank(const std::st... FILE: oneflow/core/job/parallel_desc.h function namespace (line 30) | namespace oneflow { function namespace (line 167) | namespace private_details { function namespace (line 178) | namespace std { FILE: oneflow/core/job/parallel_desc_test.cpp type oneflow (line 25) | namespace oneflow { type test (line 27) | namespace test { type GlobaProcessCtxScope (line 31) | struct GlobaProcessCtxScope final { method GlobaProcessCtxScope (line 32) | GlobaProcessCtxScope(int64_t node_size, int64_t world_size) { function TEST (line 44) | TEST(ParallelDesc, continuous_1n4d) { function TEST (line 54) | TEST(ParallelDesc, continuous_1n4d_multi_process) { function TEST (line 69) | TEST(ParallelDesc, continuous_1n4d_multi_process_with_rank) { function TEST (line 82) | TEST(ParallelDesc, discrete_1n4d) { function TEST (line 93) | TEST(ParallelDesc, continuous_2n8d) { function TEST (line 104) | TEST(ParallelDesc, discrete_2n8d) { function TEST (line 117) | TEST(GetBroadcastGroup, naive_1n1d) { function TEST (line 129) | TEST(GetBroadcastGroup, naive_1n4d) { function TEST (line 148) | TEST(GetBroadcastGroup, naive_2n8d) { FILE: oneflow/core/job/pipeline_config_def.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job/placement_scope.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/job/placement_scope.h function namespace (line 23) | namespace oneflow { function namespace (line 57) | namespace std { FILE: oneflow/core/job/plan_util.cpp type oneflow (line 33) | namespace oneflow { function RegstDescProto (line 35) | RegstDescProto* PlanUtil::GetSoleProducedDataRegst(TaskProto* task_pro... function SetVariableOpNamesForVariableAndRepeatRegst (line 59) | void SetVariableOpNamesForVariableAndRepeatRegst(Plan* plan) { function GenChunkForMultiNNGraphMemoryReuseInMultiClient (line 126) | void GenChunkForMultiNNGraphMemoryReuseInMultiClient( function IsCollectiveBoxingTaskType (line 878) | bool IsCollectiveBoxingTaskType(TaskType task_type) { function IsCollectiveBoxingNode (line 882) | bool IsCollectiveBoxingNode(const PlanTaskNode* node) { type CollectiveBoxingRequestInfo (line 902) | struct CollectiveBoxingRequestInfo { function GetDeviceDesc (line 909) | void GetDeviceDesc(const TaskProto* task_proto, boxing::collective::De... type MemBlockMemoryInfo (line 1057) | struct MemBlockMemoryInfo { method MemBlockMemoryInfo (line 1062) | MemBlockMemoryInfo() : mem_block_id(-1), mem_block_mem_size(-1), reg... type ChunkMemoryInfo (line 1065) | struct ChunkMemoryInfo { method ChunkMemoryInfo (line 1069) | ChunkMemoryInfo() : chunk_id(-1), chunk_mem_size(-1) {} type RankDeviceMemoryInfo (line 1072) | struct RankDeviceMemoryInfo { method RankDeviceMemoryInfo (line 1081) | RankDeviceMemoryInfo() function StreamId (line 1401) | StreamId PlanUtil::GetStreamId(const TaskProto& task) { FILE: oneflow/core/job/plan_util.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/job/qat_config_def.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job/rank_compiler.cpp type oneflow (line 28) | namespace oneflow { FILE: oneflow/core/job/rank_compiler.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/job/rank_group.cpp type oneflow (line 25) | namespace oneflow { function CalcDefaultParallelDesc (line 50) | Maybe> CalcDefaultParallelDesc(DeviceType device_... function AllWorldRanks (line 74) | Maybe> AllWorldRanks() { FILE: oneflow/core/job/rank_group.h function namespace (line 29) | namespace oneflow { function namespace (line 70) | namespace std { FILE: oneflow/core/job/rank_group_scope.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job/rank_group_scope.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/rank_group_test.cpp type oneflow (line 23) | namespace oneflow { type test (line 24) | namespace test { function TEST (line 26) | TEST(RankGroup, two_rank) { function TEST (line 39) | TEST(RankGroup, nonconsecutive_rank) { FILE: oneflow/core/job/resource_desc.cpp type oneflow (line 26) | namespace oneflow { function Machine (line 37) | Machine ResourceDesc::machine(int32_t idx) const { function CollectiveBoxingConf (line 65) | CollectiveBoxingConf ResourceDesc::collective_boxing_conf() const { FILE: oneflow/core/job/resource_desc.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job/runtime.cpp type oneflow (line 34) | namespace oneflow { function SendCmdMsg (line 38) | void SendCmdMsg(const std::vector& tasks, ActorCmd c... function HandoutTasks (line 45) | void HandoutTasks(const std::vector& tasks) { function HasNonCtrlConsumedRegstDescId (line 52) | bool HasNonCtrlConsumedRegstDescId(const TaskProto& task) { FILE: oneflow/core/job/runtime.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job/runtime_buffer_managers_scope.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/job/runtime_buffer_managers_scope.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/runtime_buffers_scope.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/job/runtime_buffers_scope.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/runtime_context.cpp type oneflow (line 18) | namespace oneflow { function GetRunningActorCountKeyByJobId (line 39) | std::string GetRunningActorCountKeyByJobId(int64_t job_id) { FILE: oneflow/core/job/runtime_context.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job/runtime_job_descs.cpp type oneflow (line 18) | namespace oneflow { function JobDesc (line 27) | const JobDesc& RuntimeJobDescs::job_desc(int64_t job_id) const { FILE: oneflow/core/job/runtime_job_descs.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/sbp_infer_hint.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job/sbp_parallel.cpp type oneflow (line 21) | namespace oneflow { function MakeSplitSbpParallel (line 69) | Maybe> MakeSplitSbpParallel(int axis) { function MakeBroadcastSbpParallel (line 76) | Maybe> MakeBroadcastSbpParallel() { function MakePartialSumSbpParallel (line 82) | Maybe> MakePartialSumSbpParallel() { function SbpParallel (line 91) | SbpParallel GetDualSbpParallel(const SbpParallel& sbp_parallel) { function IsSbpSignatureContaining (line 105) | bool IsSbpSignatureContaining(const SbpSignature& bigger, const SbpSig... function FilterSbpSignatureList (line 115) | void FilterSbpSignatureList(const SbpSignatureList& sbp_sig_list, cons... function ComputCopyCostBetweenTwoSbpParallel (line 124) | double ComputCopyCostBetweenTwoSbpParallel(const SbpInferHint& produce... function ComputeIbnCopyCost4SbpSig (line 135) | double ComputeIbnCopyCost4SbpSig( function MakeGetterIbnCopyCost4SbpSig (line 148) | std::function MakeGetterIbnCopyCost4SbpSig( function MakeGetterOrderValue4SbpSig (line 162) | std::function MakeGetterOrderVal... function SortSbpSignatureListByCopyCost (line 174) | void SortSbpSignatureListByCopyCost( function IsValidSbpParallelString (line 192) | bool IsValidSbpParallelString(const std::string& sbp_str) { function ParseNdSbpFromLongString (line 197) | bool ParseNdSbpFromLongString(const std::string& nd_sbp_str, NdSbp* nd... function NdSbpToLongString (line 208) | std::string NdSbpToLongString(const NdSbp& nd_sbp) { function ParseSbpParallelFromString (line 217) | bool ParseSbpParallelFromString(const std::string& sbp_str, SbpParalle... function SbpParallelToString (line 248) | std::string SbpParallelToString(const SbpParallel& sbp_parallel) { function ParseNdSbpFromStringList (line 252) | bool ParseNdSbpFromStringList(const std::vector& sbp_str_... function NdSbpToStringList (line 259) | std::vector NdSbpToStringList(const NdSbp& nd_sbp) { function SbpSignatureToNdSbpSignature (line 267) | void SbpSignatureToNdSbpSignature(const SbpSignature& sbp_signature, function NdSbpSignatureToSbpSignature (line 274) | void NdSbpSignatureToSbpSignature(const NdSbpSignature& nd_sbp_signature, function CheckSbpSignatureAndNdSbpEquals (line 282) | void CheckSbpSignatureAndNdSbpEquals(const SbpSignature& sbp_sig, function NdSbpAllSameSplitParallel (line 294) | bool NdSbpAllSameSplitParallel(const NdSbp& nd_sbp) { function NdSbpSignatureToString (line 304) | Maybe NdSbpSignatureToString(const NdSbpSignature& nd_sbp... function NdSbpSignatureToString (line 339) | Maybe NdSbpSignatureToString(const NdSbpSignature& nd_sbp... function NdSbpSignatureListToString (line 347) | Maybe NdSbpSignatureListToString(const std::vector NdSbpSignatureListToString(const std::vector NewScopeSymbolId( FILE: oneflow/core/job/scope.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/job/session.cpp type oneflow (line 22) | namespace oneflow { function NewSessionId (line 24) | int64_t NewSessionId() { FILE: oneflow/core/job/session.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job/ssp_config_def.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job/utils/progress_bar.cpp type oneflow (line 22) | namespace oneflow { function LogProgress (line 24) | Maybe LogProgress(const std::string& task_name, bool is_end) { FILE: oneflow/core/job/utils/progress_bar.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job/version.cpp type oneflow (line 19) | namespace oneflow { function DumpVersionInfo (line 21) | void DumpVersionInfo() { FILE: oneflow/core/job/version.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/adadelta_optim.cpp type oneflow (line 27) | namespace oneflow { function GenVariableOutputLbn (line 31) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 36) | OperatorConf GenerateAdadeltaHelperVariableConf(const VariableOp& op, ... function GenerateAdadeltaOptimizerOpConf (line 47) | void GenerateAdadeltaOptimizerOpConf(JobPassCtx* ctx, const OpNode& va... FILE: oneflow/core/job_rewriter/adagrad_optm.cpp type oneflow (line 27) | namespace oneflow { function GenVariableOutputLbn (line 31) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 36) | OperatorConf GenerateAdagradHelperVariableConf(const VariableOp& op, c... function GenerateAdagradOptimizerOpConf (line 48) | void GenerateAdagradOptimizerOpConf(JobPassCtx* ctx, const OpNode& var... FILE: oneflow/core/job_rewriter/adam_optm.cpp type oneflow (line 19) | namespace oneflow { type BiasCorrectionFactorCacheKey (line 21) | struct BiasCorrectionFactorCacheKey { class BiasCorrectionFactorState (line 46) | class BiasCorrectionFactorState final : public JobPassState { method BiasCorrectionFactorState (line 48) | BiasCorrectionFactorState() {} method GetLbn (line 51) | std::string GetLbn(float beta, std::string bias_correction_name, Par... function GenVariableOutputLbn (line 73) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 78) | OperatorConf GenerateAdamHelperVariableOpConf(const VariableOp& op, co... function GenerateOptimizerOpConf (line 90) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, type std (line 32) | namespace std { type hash (line 35) | struct hash { type oneflow (line 44) | namespace oneflow { type BiasCorrectionFactorCacheKey (line 21) | struct BiasCorrectionFactorCacheKey { class BiasCorrectionFactorState (line 46) | class BiasCorrectionFactorState final : public JobPassState { method BiasCorrectionFactorState (line 48) | BiasCorrectionFactorState() {} method GetLbn (line 51) | std::string GetLbn(float beta, std::string bias_correction_name, Par... function GenVariableOutputLbn (line 73) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 78) | OperatorConf GenerateAdamHelperVariableOpConf(const VariableOp& op, co... function GenerateOptimizerOpConf (line 90) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, FILE: oneflow/core/job_rewriter/add_ssp_variable_proxy.cpp type oneflow (line 23) | namespace oneflow { class AddSspVariableProxyPass (line 27) | class AddSspVariableProxyPass final : public JobPass { method AddSspVariableProxyPass (line 29) | AddSspVariableProxyPass(const AddSspVariableProxyPass&) = delete; method AddSspVariableProxyPass (line 30) | AddSspVariableProxyPass(AddSspVariableProxyPass&&) = delete; method AddSspVariableProxyPass (line 31) | AddSspVariableProxyPass() = default; method Apply (line 34) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 41) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 45) | Maybe Apply(const OpGraph& op_graph, JobBuilder* job_builder) ... method ForEachTrainableVarOpNode (line 82) | Maybe ForEachTrainableVarOpNode( method AddSspVarProxyOp (line 97) | Maybe AddSspVarProxyOp(OpNode* op_node, JobBuilder* job_builde... method ReplaceVarWithSspVarProxyOp (line 105) | Maybe ReplaceVarWithSspVarProxyOp( method IsInOptimizerPass (line 126) | Maybe IsInOptimizerPass(int64_t scope_symbol_id) const { method AddSspVarProxyOp (line 131) | Maybe AddSspVarProxyOp(const LogicalBlobId& old_var_out_lbi, i... FILE: oneflow/core/job_rewriter/auto_learning_rate.cpp type oneflow (line 21) | namespace oneflow { class AutoLearningRate (line 25) | class AutoLearningRate final : public JobPass { method AutoLearningRate (line 28) | AutoLearningRate() = default; method IsEnabled (line 31) | bool IsEnabled(const JobPassCtx& ctx) const { return ctx.job_desc().... method Apply (line 35) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { class oneflow::OpNode (line 47) | class oneflow::OpNode function FOR_RANGE (line 64) | FOR_RANGE(int64_t, i, 0, train_conf.optimizer_conf_size()) { FILE: oneflow/core/job_rewriter/auto_mixed_precision.cpp type oneflow (line 27) | namespace oneflow { function VerifyAMPList (line 31) | void VerifyAMPList(const AMPList& amp_list) { function NoCastRegistry (line 40) | NoCastRegistry* GetNoCastRegistry() { function FindInNoCastRegisry (line 45) | bool FindInNoCastRegisry(const std::string& op_type, const OpArg& op_a... function MakePredicatorIsAllowedToRunWithHalf (line 53) | std::function MakePredicatorIsAllowedToRunWithHalf(cons... function InsertCastOpImpl (line 66) | void InsertCastOpImpl(bool f2h, const OpGraph& op_graph, const HashSet... class AutoMixedPrecision (line 156) | class AutoMixedPrecision final : public JobPass { method AutoMixedPrecision (line 159) | AutoMixedPrecision() method IsEnabled (line 166) | bool IsEnabled(const JobPassCtx& ctx) const { type NoCastRegistrar (line 325) | struct NoCastRegistrar final { method NoCastRegistrar (line 326) | NoCastRegistrar(const std::string& op_type, OpArg&& op_arg) { type amp (line 367) | namespace amp { function IsNoCast (line 369) | bool IsNoCast(const std::string& op_type, const OpArg& op_arg) { FILE: oneflow/core/job_rewriter/auto_mixed_precision.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/auto_mixed_precision_lists.cpp type oneflow (line 18) | namespace oneflow { function AMPList (line 20) | const AMPList& AutoMixedPrecisionLists::WhiteList() { function AMPList (line 58) | const AMPList& AutoMixedPrecisionLists::BlackList() { function AMPList (line 64) | const AMPList& AutoMixedPrecisionLists::GrayList() { function AMPList (line 145) | const AMPList& AutoMixedPrecisionLists::ClearList() { FILE: oneflow/core/job_rewriter/auto_mixed_precision_lists.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/job_rewriter/auto_parallel.cpp type oneflow (line 26) | namespace oneflow { class AutoParallelPass (line 30) | class AutoParallelPass final : public JobPass { method AutoParallelPass (line 33) | AutoParallelPass() = default; method Apply (line 38) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/auto_train_step.cpp type oneflow (line 22) | namespace oneflow { class AutoTrainStep (line 26) | class AutoTrainStep final : public JobPass { method AutoTrainStep (line 29) | AutoTrainStep() = default; FILE: oneflow/core/job_rewriter/autograd.cpp type oneflow (line 31) | namespace oneflow { function TrainConf (line 35) | const TrainConf& GetTrainConf() { return GlobalJobDesc().job_conf().tr... function ScopeSymbolId4Lbi (line 37) | int64_t ScopeSymbolId4Lbi(const OpGraph& op_graph, const LogicalBlobId... function AnyLbiWithDiffLbi (line 41) | bool AnyLbiWithDiffLbi(const OpEdge* op_edge) { function CheckNotReachableAmongOpNodes (line 64) | void CheckNotReachableAmongOpNodes(const OpGraph& op_graph, const std:... function GetLossOpNodes (line 74) | Maybe GetLossOpNodes(const OpGraph& op_graph, std::list... function GetLossOpNodesAndAscendants (line 89) | Maybe GetLossOpNodesAndAscendants(const OpGraph& op_graph, HashS... function ParallelConf (line 102) | const ParallelConf& ProducerParallelConf4Lbi(const OpGraph& op_graph, ... function ScaleModelDiffByConstantLossInstanceNum (line 106) | void ScaleModelDiffByConstantLossInstanceNum(const OpGraph& op_graph, ... function TryLocalCastTotalLossInstanceNum (line 131) | Maybe TryLocalCastTotalLossInstanceNum( function ScaleModelDiffByDynamicLossInstanceNum (line 171) | void ScaleModelDiffByDynamicLossInstanceNum( function AllSplitDistribution (line 247) | bool AllSplitDistribution(const NdSbp& nd_sbp) { function ForEachAggregatedParamGroup (line 254) | void ForEachAggregatedParamGroup( function MakeScopeSymbolId (line 270) | int64_t MakeScopeSymbolId(const JobConfigProto& job_conf, const Parall... function AddLbns (line 279) | std::string AddLbns(JobBuilder* job_builder, const std::vector MakeGetterLossOpNode4OpName( function ScaleModelDiffByLossInstanceNum (line 658) | Maybe ScaleModelDiffByLossInstanceNum(const OpGraph& op_graph, J... function ScaleInitialDiffByLossScale (line 716) | Maybe ScaleInitialDiffByLossScale( function ScaleModelDiffByLossScale (line 833) | void ScaleModelDiffByLossScale(JobPassCtx* ctx, const OpGraph& op_grap... function RegularizeGradient (line 926) | void RegularizeGradient(const OpGraph& op_graph, JobBuilder* job_builder, function ClipGradient (line 957) | void ClipGradient(JobPassCtx* ctx, const OpGraph& op_graph, JobBuilder... function AddDiffParallelCast (line 967) | void AddDiffParallelCast(const OpGraph& op_graph, JobBuilder* job_buil... function AddDiffHalf2FloatCast (line 997) | void AddDiffHalf2FloatCast(const OpGraph& op_graph, JobBuilder* job_bu... function AddDiffStaticShapeCast (line 1019) | void AddDiffStaticShapeCast(const OpGraph& op_graph, JobBuilder* job_b... function CountNotFiniteIfNeeded (line 1039) | Maybe CountNotFiniteIfNeeded(JobPassCtx* ctx, const OpGraph& op_... FILE: oneflow/core/job_rewriter/autograd.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job_rewriter/autotick.cpp type oneflow (line 30) | namespace oneflow { function NewMutOpConTickInputHelper (line 34) | std::unique_ptr NewMutOpConTickInputHelper(co... function PrependTickByParallelDesc (line 43) | void PrependTickByParallelDesc(const OpGraph& op_graph, JobBuilder* jo... function FindJobSoleSrcSubsetTickOpConf (line 66) | Maybe FindJobSoleSrcSubsetTickOpConf(const Job& j... function BuildDstSubsetTickOpAndParallelConf (line 77) | Maybe BuildDstSubsetTickOpAndParallelConf(const HashSet CreateDstSubsetTickAndSinkTicks( function CreateDstSubsetTickAndSinkTicks (line 149) | Maybe CreateDstSubsetTickAndSinkTicks(CriticalSection* critical_... function BuildSrcSubsetTickOpAndParallelConf (line 162) | Maybe BuildSrcSubsetTickOpAndParallelConf(OperatorConf* src_subs... function CreateSourceTicksAndSrcSubsetTick (line 175) | Maybe CreateSourceTicksAndSrcSubsetTick( function CreateSourceTicksAndSrcSubsetTick (line 202) | Maybe CreateSourceTicksAndSrcSubsetTick(CriticalSection* critica... function ConnectSrcSubsetTickAndOtherTick (line 214) | Maybe ConnectSrcSubsetTickAndOtherTick(const OperatorConf& src_s... function GetSrcSubsetTickOpNode (line 232) | Maybe GetSrcSubsetTickOpNode(const OpGraph& op_graph) { function OperatorConf (line 245) | OperatorConf MakeTickOpConf(const std::string& tick_name) { function OperatorConf (line 253) | OperatorConf MakeDeviceTickOpConf(const std::string& tick_name) { function OperatorConf (line 262) | OperatorConf AppendTick(const std::string tick_name, const std::vector... function OperatorConf (line 274) | OperatorConf AppendTick(const std::string tick_name, const std::list& op_nodes, JobBu... function OperatorConf (line 303) | OperatorConf AppendAccTick(const Shape& src_shape, const std::list GetOpNames(const HashSet& op_n... function InitOpTypeCase2OpNodes (line 336) | Maybe InitOpTypeCase2OpNodes( function ForEachInputCriticalSectionOpNodes (line 349) | Maybe ForEachInputCriticalSectionOpNodes( function ForEachOutputCriticalSectionOpNodes (line 365) | Maybe ForEachOutputCriticalSectionOpNodes( function AddTickForTimeShape (line 382) | Maybe> AddTickForTimeShape(const Shape& src_... function AddGlobalInputOutputCriticalSection (line 409) | Maybe AddGlobalInputOutputCriticalSection( function MultiClientAddOneWaitAndSendIdsOp (line 456) | Maybe MultiClientAddOneWaitAndSendIdsOp(JobBuilder* job_builder,... function MultiClientAddWaitAndSendIds (line 490) | Maybe MultiClientAddWaitAndSendIds( function MultiClientAddCallbackNotifier (line 530) | Maybe MultiClientAddCallbackNotifier(JobBuilder* job_builder, in... function AutoPrependTick (line 551) | Maybe AutoPrependTick(const OpGraph& op_graph, JobBuilder* job_b... function AddTickForTimeShape (line 559) | Maybe AddTickForTimeShape(const OpGraph& op_graph, JobBuilder* j... function AutoSourceAndSinkTick (line 574) | Maybe AutoSourceAndSinkTick( function MultiClientAutoSourceAndSinkTick (line 600) | Maybe MultiClientAutoSourceAndSinkTick(const OpGraph& op_graph, ... function InsertCriticalSectionSrcAndDstTicks (line 628) | Maybe InsertCriticalSectionSrcAndDstTicks( function InsertSrcSubsetTickAndDstSubsetTick (line 665) | Maybe InsertSrcSubsetTickAndDstSubsetTick( function InsertCriticalSectionWaitTicks (line 697) | Maybe InsertCriticalSectionWaitTicks(const OpGraph& op_graph, Jo... function InsertCriticalSectionCallbackTicks (line 729) | Maybe InsertCriticalSectionCallbackTicks(const OpGraph&... function MultiClientAutoCriticalSectionTick (line 748) | Maybe MultiClientAutoCriticalSectionTick( function ConnectCriticalSectionCallbackToJobSoleDstSubsetTick (line 768) | Maybe ConnectCriticalSectionCallbackToJobSoleDstSubsetTick( function MultiClientAutoInterfaceCriticalSectionTick (line 788) | Maybe MultiClientAutoInterfaceCriticalSectionTick(const OpGraph&... FILE: oneflow/core/job_rewriter/autotick.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/job_rewriter/boxing_with_middle_nodes.cpp type oneflow (line 26) | namespace oneflow { function BoxingWithMiddleNodes (line 28) | Maybe BoxingWithMiddleNodes(const OpGraph& op_graph, JobBuilder*... FILE: oneflow/core/job_rewriter/boxing_with_middle_nodes.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/calculation_pass.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/job_rewriter/calculation_pass.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/checkpointing_pass.cpp type oneflow (line 27) | namespace oneflow { class CheckpointingPass (line 32) | class CheckpointingPass final : public JobPass { method CheckpointingPass (line 35) | CheckpointingPass() = default; method Apply (line 38) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 45) | bool IsEnabled(const JobPassCtx& ctx) const { return ctx.job_desc().... function Scope (line 54) | const Scope& Scope4OpNode(const OpNode* op_node) { function IsForwardPassScope (line 62) | bool IsForwardPassScope(const Scope& scope) { function IsForwardPass7CheckpointingScope (line 66) | bool IsForwardPass7CheckpointingScope(const Scope& scope) { function CollectAllCheckpointingOpsInForwardPass (line 70) | void CollectAllCheckpointingOpsInForwardPass( function GenConnectedCheckpointingSubgraphs (line 92) | void GenConnectedCheckpointingSubgraphs( FILE: oneflow/core/job_rewriter/clip_by_global_norm_job_pass_state.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/clone_grad.cpp type oneflow (line 19) | namespace oneflow { function GenerateCloneGradOpIfNeed (line 21) | Maybe GenerateCloneGradOpIfNeed( FILE: oneflow/core/job_rewriter/clone_grad.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/cudnn_fused_normalization_add_relu_pass.cpp type oneflow (line 23) | namespace oneflow { function IsFusedBnAddReluSupported (line 27) | bool IsFusedBnAddReluSupported() { function IsNormalizationAddReluOp (line 35) | bool IsNormalizationAddReluOp(const OperatorConf& op) { function NeedDoPass (line 41) | bool NeedDoPass(const Job& job) { class CudnnFusedNormalizationAddReluPass (line 47) | class CudnnFusedNormalizationAddReluPass final : public JobPass { method CudnnFusedNormalizationAddReluPass (line 49) | CudnnFusedNormalizationAddReluPass() = default; method IsEnabled (line 52) | bool IsEnabled(const JobPassCtx& ctx) const { FILE: oneflow/core/job_rewriter/cutlass_conv_tuning_warmup_pass.cpp type oneflow (line 27) | namespace oneflow { class CutlassConvTuningWarmupPass (line 34) | class CutlassConvTuningWarmupPass final : public JobPass { method CutlassConvTuningWarmupPass (line 36) | CutlassConvTuningWarmupPass() = default; FILE: oneflow/core/job_rewriter/delay_variable_op_execution_pass.cpp type oneflow (line 20) | namespace oneflow { class DelayVariableOpExecutionPass (line 22) | class DelayVariableOpExecutionPass final : public JobPass { method DelayVariableOpExecutionPass (line 24) | DelayVariableOpExecutionPass() = default; FILE: oneflow/core/job_rewriter/device_tick_autotick.cpp type oneflow (line 18) | namespace oneflow { class MutDeviceTickOpConTickInputHelper (line 22) | class MutDeviceTickOpConTickInputHelper final : public MutOpConTickInp... method MutDeviceTickOpConTickInputHelper (line 24) | MutDeviceTickOpConTickInputHelper() : MutOpConTickInputHelper() {} method VirtualIsTickInputBound (line 26) | bool VirtualIsTickInputBound() const override { method OperatorConf (line 30) | OperatorConf NewTickInputBoundOpConf(const std::string& lbn) const o... FILE: oneflow/core/job_rewriter/do_parallel_cast_before_widening_type_cast_pass.cpp type oneflow (line 21) | namespace oneflow { class DoParallelCastBeforeWideningTypeCast (line 25) | class DoParallelCastBeforeWideningTypeCast final : public JobPass { method DoParallelCastBeforeWideningTypeCast (line 27) | DoParallelCastBeforeWideningTypeCast() = default; method IsEnabled (line 30) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 35) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/dump_blob_parallel_conf_pass.cpp type oneflow (line 20) | namespace oneflow { class DumpBlobParallelConfPass (line 24) | class DumpBlobParallelConfPass final : public JobPass { method DumpBlobParallelConfPass (line 27) | DumpBlobParallelConfPass() = default; method IsEnabled (line 30) | bool IsEnabled(const JobPassCtx& ctx) const { return true; } method Apply (line 32) | Maybe Apply(const OpGraph& op_graph, Job* job) const { method Apply (line 39) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/dump_variable_info_pass.cpp type oneflow (line 20) | namespace oneflow { function GetNdSbpString (line 24) | std::string GetNdSbpString(const VariableOpConf& conf, const ParallelD... class DumpVariableInfoPass (line 40) | class DumpVariableInfoPass final : public JobPass { method DumpVariableInfoPass (line 42) | DumpVariableInfoPass() = default; method IsEnabled (line 45) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 50) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/dynamic_loss_scale_job_pass_state.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/dynamic_loss_scale_schedule_pass.cpp type oneflow (line 22) | namespace oneflow { class DynamicLossScaleSchedulePass (line 26) | class DynamicLossScaleSchedulePass final : public JobPass { method DynamicLossScaleSchedulePass (line 29) | DynamicLossScaleSchedulePass() = default; FILE: oneflow/core/job_rewriter/eliminate_dead_nodes_pass.cpp type oneflow (line 19) | namespace oneflow { class EliminateDeadNodesPass (line 23) | class EliminateDeadNodesPass final : public JobPass { method EliminateDeadNodesPass (line 25) | EliminateDeadNodesPass() = default; method Apply (line 30) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { function IsNoSideEffect (line 37) | static bool IsNoSideEffect(const OpNode* op_node) { FILE: oneflow/core/job_rewriter/fix_pipeline_stage_id_pass.cpp type oneflow (line 27) | namespace oneflow { class FixPipelineStageIdPass (line 31) | class FixPipelineStageIdPass final : public JobPass { method FixPipelineStageIdPass (line 34) | FixPipelineStageIdPass() = default; method Apply (line 37) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 44) | bool IsEnabled(const JobPassCtx& ctx) const { function Scope (line 52) | const Scope& Scope4ScopeSymbolId(int64_t scope_symbol_id) { function Scope (line 57) | const Scope& Scope4OpNode(const OpNode* op_node) { function OpNodeHasScope (line 63) | bool OpNodeHasScope(const OpNode* node) { return node->op().op_conf().... function GetStageIdHint (line 65) | int64_t GetStageIdHint(const OpNode* node) { function ParallelDesc2HashString (line 69) | std::string ParallelDesc2HashString(const ParallelDesc& parallel_desc) { function NewScopeWithStageId (line 80) | Maybe NewScopeWithStageId(int64_t old_scope_symbol_id, int64_... FILE: oneflow/core/job_rewriter/ftrl_optm.cpp type oneflow (line 27) | namespace oneflow { function GenVariableOutputLbn (line 31) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 36) | OperatorConf GenerateFtrlHelperVariableConf(const VariableOp& op, cons... function GenerateFtrlOptimizerOpConf (line 48) | void GenerateFtrlOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op... FILE: oneflow/core/job_rewriter/fuse_add_to_output_pass.cpp type oneflow (line 21) | namespace oneflow { class FuseAddToOutputPass (line 25) | class FuseAddToOutputPass final : public JobPass { method FuseAddToOutputPass (line 27) | FuseAddToOutputPass() = default; method IsEnabled (line 30) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 35) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/fuse_bce_reduce_mean_fw_bw_pass.cpp type oneflow (line 19) | namespace oneflow { function UpdateConsumerOpConf (line 23) | void UpdateConsumerOpConf(const OpNode* consumer, const LogicalBlobId&... class FuseBCEReduceMeanFwBwPass (line 40) | class FuseBCEReduceMeanFwBwPass final : public JobPass { method FuseBCEReduceMeanFwBwPass (line 42) | FuseBCEReduceMeanFwBwPass() = default; method IsEnabled (line 45) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 50) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/fuse_cast_scale_pass.cpp type oneflow (line 19) | namespace oneflow { class FuseCastScalePass (line 23) | class FuseCastScalePass final : public JobPass { method FuseCastScalePass (line 25) | FuseCastScalePass() = default; method IsEnabled (line 28) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 33) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/fuse_consecutive_add_pass.cpp type oneflow (line 22) | namespace oneflow { class FuseConsecutiveAddPass (line 26) | class FuseConsecutiveAddPass final : public JobPass { method FuseConsecutiveAddPass (line 28) | FuseConsecutiveAddPass() = default; method Apply (line 33) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/fuse_embedding_interaction_pass.cpp type oneflow (line 19) | namespace oneflow { class FuseEmbeddingShuffleInteractionPass (line 23) | class FuseEmbeddingShuffleInteractionPass final : public JobPass { method FuseEmbeddingShuffleInteractionPass (line 25) | FuseEmbeddingShuffleInteractionPass() = default; method IsEnabled (line 28) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 38) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/fuse_model_update_cast_pass.cpp type oneflow (line 19) | namespace oneflow { class FuseModelUpdateCastOpsPass (line 23) | class FuseModelUpdateCastOpsPass final : public JobPass { method FuseModelUpdateCastOpsPass (line 25) | FuseModelUpdateCastOpsPass() = default; method IsEnabled (line 28) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 35) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/fuse_update_ops_pass.cpp type oneflow (line 19) | namespace oneflow { class FuseUpdateOpsPass (line 23) | class FuseUpdateOpsPass final : public JobPass { method FuseUpdateOpsPass (line 25) | FuseUpdateOpsPass() = default; method IsEnabled (line 28) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 33) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/generate_optimizer_op_confs.cpp type oneflow (line 25) | namespace oneflow { class GenerateOptimizerOpConfs (line 29) | class GenerateOptimizerOpConfs final : public JobPass { method GenerateOptimizerOpConfs (line 32) | GenerateOptimizerOpConfs() = default; method IsEnabled (line 35) | bool IsEnabled(const JobPassCtx& ctx) const { return ctx.job_desc().... function FilterCurModelLbi2ModelDiffLbiByName (line 40) | void FilterCurModelLbi2ModelDiffLbiByName( function WithCalculationPassScope (line 52) | Maybe WithCalculationPassScope(const std::string& pass_nam... FILE: oneflow/core/job_rewriter/group_boxing_by_dst_parallel.cpp type oneflow (line 23) | namespace oneflow { function Scope (line 25) | const Scope& Scope4ScopeSymbolId(int64_t scope_symbol_id) { function Scope (line 30) | const Scope& Scope4OpNode(const OpNode* op_node) { function OpNodeHasScope (line 36) | bool OpNodeHasScope(const OpNode* node) { return node->op().op_conf().... function GetStageIdHint (line 38) | int64_t GetStageIdHint(const OpNode* node) { function GroupBoxingByDstParallel (line 42) | Maybe GroupBoxingByDstParallel(const OpGraph& op_graph, JobBuild... FILE: oneflow/core/job_rewriter/group_boxing_by_dst_parallel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/job_rewriter/indexed_slices_optimizer_rewrite_pass.cpp type oneflow (line 19) | namespace oneflow { class IndexedSlicesOptimizerRewritePass (line 21) | class IndexedSlicesOptimizerRewritePass final : public JobPass { method IndexedSlicesOptimizerRewritePass (line 23) | IndexedSlicesOptimizerRewritePass() = default; method IsEnabled (line 26) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 33) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/input_autotick.cpp type oneflow (line 18) | namespace oneflow { class MutInputOpConTickInputHelper (line 22) | class MutInputOpConTickInputHelper final : public MutOpConTickInputHel... method MutInputOpConTickInputHelper (line 24) | MutInputOpConTickInputHelper() : MutOpConTickInputHelper() {} method VirtualIsTickInputBound (line 26) | bool VirtualIsTickInputBound() const override { return op_conf().inp... method OperatorConf (line 28) | OperatorConf NewTickInputBoundOpConf(const std::string& lbn) const o... FILE: oneflow/core/job_rewriter/insert_nccl_logical_op_pass.cpp type oneflow (line 35) | namespace oneflow { class InsertNcclLogicalOpPass (line 41) | class InsertNcclLogicalOpPass final : public JobPass { method InsertNcclLogicalOpPass (line 44) | InsertNcclLogicalOpPass() = default; method Apply (line 47) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 54) | bool IsEnabled(const JobPassCtx& ctx) const { function IsTickOpConf (line 63) | bool IsTickOpConf(const OperatorConf& op_conf) { function IsBreakpointOpNode (line 72) | bool IsBreakpointOpNode(const OpNode* node) { function IsAccOpNode (line 101) | bool IsAccOpNode(const OpNode* node) { function IsRepeatOpNode (line 106) | bool IsRepeatOpNode(const OpNode* node) { function GetOpNodeTimeShape (line 111) | std::shared_ptr GetOpNodeTimeShape(const OpNode* op_node) { function GetOpNodeInputTimeShape (line 115) | std::shared_ptr GetOpNodeInputTimeShape(const OpNode* op_... function GetOpNodeFastestTimeShape (line 119) | std::shared_ptr GetOpNodeFastestTimeShape(const OpNode* o... function SharedPtrShapeEqual (line 123) | bool SharedPtrShapeEqual(const std::shared_ptr& lhs, function FindAllConnectedSubgraphForGpuExecOrder (line 128) | void FindAllConnectedSubgraphForGpuExecOrder(std::vector BuildScopeWithReducedParallelDesc(int64_t old_scope_sym... function TryBuildNcclLogicalOpConf (line 454) | bool TryBuildNcclLogicalOpConf(OperatorConf* ret, const OpNode* src_no... function InsertNcclLogicalOpsAsCloseAsPossibleToDstNode (line 523) | void InsertNcclLogicalOpsAsCloseAsPossibleToDstNode( function GenAfterAccSubgraph (line 571) | void GenAfterAccSubgraph(std::vector* ordered_after_acc... type InsertNcclSubGraph (line 615) | struct InsertNcclSubGraph { type PlacementNcclSubGraghsInfo (line 623) | struct PlacementNcclSubGraghsInfo { function InitInsertNcclSubGraphInfoFromSet (line 629) | void InitInsertNcclSubGraphInfoFromSet( function GetStreamIndexName (line 644) | std::string GetStreamIndexName(uint32_t id) { return "NCCL_COMPUTE_" +... function InsertNcclLogicalOpsInSubGraph (line 646) | int64_t InsertNcclLogicalOpsInSubGraph(const OpGraph& op_graph, JobBui... function InsertNcclLogicalOpsAfterAcc (line 717) | void InsertNcclLogicalOpsAfterAcc(const OpGraph& op_graph, JobBuilder*... FILE: oneflow/core/job_rewriter/insert_pinned_identity_op_pass.cpp type oneflow (line 19) | namespace oneflow { class InsertPinnedIdentityOpPass (line 23) | class InsertPinnedIdentityOpPass final : public JobPass { method InsertPinnedIdentityOpPass (line 25) | InsertPinnedIdentityOpPass() = default; function InsertPinnedIdentityOp (line 31) | Maybe InsertPinnedIdentityOp(JobBuilder* job_builder, con... FILE: oneflow/core/job_rewriter/job_completer.cpp type oneflow (line 32) | namespace oneflow { function CheckOpGraph (line 36) | Maybe CheckOpGraph(const OpGraph& op_graph) { function CheckAndLogOpGraph (line 51) | Maybe CheckAndLogOpGraph(const Job& job) { function WithOpGraphAndMutJob (line 65) | Maybe WithOpGraphAndMutJob(Job* job, function WithOpGraphAndMutJobBuilder (line 72) | Maybe WithOpGraphAndMutJobBuilder( function SetCtrlInOpName4VariableOp (line 80) | Maybe SetCtrlInOpName4VariableOp(const OpGraph& op_graph, JobBui... FILE: oneflow/core/job_rewriter/job_completer.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job_rewriter/job_pass.cpp type oneflow (line 18) | namespace oneflow { function RegisterJobPass (line 29) | void RegisterJobPass(const std::string& pass_name, const JobPass* pass) { function HasJobPass (line 33) | bool HasJobPass(const std::string& pass_name) { function JobPass (line 37) | const JobPass& JobPass4Name(const std::string& pass_name) { FILE: oneflow/core/job_rewriter/job_pass.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job_rewriter/lamb_optm.cpp type oneflow (line 19) | namespace oneflow { type BiasCorrectionFactorCacheKey (line 21) | struct BiasCorrectionFactorCacheKey { class BiasCorrectionFactorState (line 44) | class BiasCorrectionFactorState final : public JobPassState { method BiasCorrectionFactorState (line 46) | BiasCorrectionFactorState() {} function GenVariableOutputLbn (line 59) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 64) | OperatorConf GenerateLAMBHelperVariableOpConf(const VariableOp& op, co... function SetScalarShapeAndNdSbpConf (line 76) | void SetScalarShapeAndNdSbpConf(const ParallelDesc& parallel_desc, Ope... function GenerateOptimizerOpConf (line 86) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, type std (line 30) | namespace std { type hash (line 32) | struct hash { type oneflow (line 41) | namespace oneflow { type BiasCorrectionFactorCacheKey (line 21) | struct BiasCorrectionFactorCacheKey { class BiasCorrectionFactorState (line 44) | class BiasCorrectionFactorState final : public JobPassState { method BiasCorrectionFactorState (line 46) | BiasCorrectionFactorState() {} function GenVariableOutputLbn (line 59) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 64) | OperatorConf GenerateLAMBHelperVariableOpConf(const VariableOp& op, co... function SetScalarShapeAndNdSbpConf (line 76) | void SetScalarShapeAndNdSbpConf(const ParallelDesc& parallel_desc, Ope... function GenerateOptimizerOpConf (line 86) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, FILE: oneflow/core/job_rewriter/lars_optm.cpp type oneflow (line 19) | namespace oneflow { function GenerateOptimizerOpConf (line 23) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, FILE: oneflow/core/job_rewriter/logical_chain_pass.cpp type oneflow (line 35) | namespace oneflow { class LogicalChainPass (line 41) | class LogicalChainPass final : public JobPass { method LogicalChainPass (line 44) | LogicalChainPass() = default; method Apply (line 47) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 54) | bool IsEnabled(const JobPassCtx& ctx) const { return EnableLogicalCh... function IsTickOpConf (line 59) | bool IsTickOpConf(const OperatorConf& op_conf) { function IsBreakpointOpNode (line 68) | bool IsBreakpointOpNode(const OpNode* node) { function IsAccOrPackOpNode (line 95) | bool IsAccOrPackOpNode(const OpNode* node) { function IsAccOpNode (line 102) | bool IsAccOpNode(const OpNode* node) { function IsRepeatOpNode (line 107) | bool IsRepeatOpNode(const OpNode* node) { function GetOpNodeFastestTimeShape (line 112) | std::shared_ptr GetOpNodeFastestTimeShape(const OpNode* o... function GetOpNodeInputTimeShape (line 116) | std::shared_ptr GetOpNodeInputTimeShape(const OpNode* op_... function SharedPtrShapeEqual (line 120) | bool SharedPtrShapeEqual(const std::shared_ptr& lhs, function IsOpEdge121Connected (line 125) | bool IsOpEdge121Connected(const OpNode* src_node, const OpNode* dst_no... function GetLogicalChainsWithTimeShape (line 154) | void GetLogicalChainsWithTimeShape(std::vector>... type LogicalChain (line 208) | struct LogicalChain { method LogicalChain (line 211) | explicit LogicalChain(int64_t val) : logical_chain_id(val) { CHECK_G... type PlacementLogicalChainsInfo (line 214) | struct PlacementLogicalChainsInfo { method PlacementLogicalChainsInfo (line 219) | PlacementLogicalChainsInfo() : seed_parallel_desc(nullptr) {} function InitPlacementLogicalChainsInfoFromSet (line 222) | void InitPlacementLogicalChainsInfoFromSet( function CreateAfterAccLogicalChain (line 240) | void CreateAfterAccLogicalChain(const std::shared_ptr& a... function TryMergeAfterAccLogicalChainToMaxLogicalChain (line 287) | void TryMergeAfterAccLogicalChainToMaxLogicalChain( FILE: oneflow/core/job_rewriter/momentum_optm.cpp type oneflow (line 20) | namespace oneflow { function GenerateOptimizerOpConf (line 24) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, FILE: oneflow/core/job_rewriter/multi_tensor_model_update.cpp type oneflow (line 20) | namespace oneflow { type SGDOptimizerKey (line 22) | struct SGDOptimizerKey { type AdamOptimizerKey (line 49) | struct AdamOptimizerKey { function AddScaleAndSkipLbn (line 141) | void AddScaleAndSkipLbn(user_op::UserOpConfWrapperBuilder& multi_tenso... function AddProcessedVariable (line 153) | void AddProcessedVariable(HashSet& processed_variable_list, function IfVariableProcessed (line 169) | bool IfVariableProcessed(const HashSet& processed_variabl... class MultiTensorModelUpdatePass (line 193) | class MultiTensorModelUpdatePass final : public JobPass { method MultiTensorModelUpdatePass (line 195) | MultiTensorModelUpdatePass() = default; method IsEnabled (line 198) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 204) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { type std (line 80) | namespace std { type hash (line 83) | struct hash { type hash (line 107) | struct hash { type oneflow (line 137) | namespace oneflow { type SGDOptimizerKey (line 22) | struct SGDOptimizerKey { type AdamOptimizerKey (line 49) | struct AdamOptimizerKey { function AddScaleAndSkipLbn (line 141) | void AddScaleAndSkipLbn(user_op::UserOpConfWrapperBuilder& multi_tenso... function AddProcessedVariable (line 153) | void AddProcessedVariable(HashSet& processed_variable_list, function IfVariableProcessed (line 169) | bool IfVariableProcessed(const HashSet& processed_variabl... class MultiTensorModelUpdatePass (line 193) | class MultiTensorModelUpdatePass final : public JobPass { method MultiTensorModelUpdatePass (line 195) | MultiTensorModelUpdatePass() = default; method IsEnabled (line 198) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 204) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/nccl_logical_chain_strict_order_pass.cpp type oneflow (line 30) | namespace oneflow { class NcclLogicalChainStrictOrderPass (line 34) | class NcclLogicalChainStrictOrderPass final : public JobPass { method NcclLogicalChainStrictOrderPass (line 37) | NcclLogicalChainStrictOrderPass() = default; method Apply (line 40) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 47) | bool IsEnabled(const JobPassCtx& ctx) const { function IsAccOrPackOpNode (line 54) | bool IsAccOrPackOpNode(const OpNode* node) { function InsertCtrlOpBetweenBwChainAndAccChain (line 61) | Maybe InsertCtrlOpBetweenBwChainAndAccChain( FILE: oneflow/core/job_rewriter/nccl_logical_op_fusion_pass.cpp type oneflow (line 32) | namespace oneflow { class NcclLogicalOpFusionPass (line 39) | class NcclLogicalOpFusionPass final : public JobPass { method NcclLogicalOpFusionPass (line 42) | NcclLogicalOpFusionPass() = default; method Apply (line 45) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 52) | bool IsEnabled(const JobPassCtx& ctx) const { function IsNcclLogicalOpNode (line 62) | bool IsNcclLogicalOpNode(const OpNode* node) { function ReplaceNcclOpsWithFusionOp (line 84) | Maybe ReplaceNcclOpsWithFusionOp(std::vector* nccl... type NcclFusionBucket (line 172) | struct NcclFusionBucket { method NcclFusionBucket (line 175) | NcclFusionBucket() : fusion_bucket_size(0) {} function GenNcclFusionKey (line 178) | std::string GenNcclFusionKey(const OpNode* nccl_op) { function GetNcclOpMemSize (line 190) | int64_t GetNcclOpMemSize(const OpNode* nccl_op) { function AppendOrCreatFusionBucket (line 203) | void AppendOrCreatFusionBucket(std::vector* buckets,... FILE: oneflow/core/job_rewriter/normalization_exponential_average_auto_tick_rewrite_pass.cpp type oneflow (line 19) | namespace oneflow { class NormalizationExponentialAverageAutoTickPass (line 21) | class NormalizationExponentialAverageAutoTickPass final : public JobPa... method NormalizationExponentialAverageAutoTickPass (line 23) | NormalizationExponentialAverageAutoTickPass() = default; FILE: oneflow/core/job_rewriter/optimizer.cpp type oneflow (line 20) | namespace oneflow { function AddOptimizerOp (line 29) | void AddOptimizerOp(JobPassCtx* ctx, const OpNode& var_op_node, const ... function GetOptimizerWeightDecayRate (line 36) | float GetOptimizerWeightDecayRate(const OptimizerConf& optimizer_conf,... function SetDynamicLossScaleSkipIf (line 67) | void SetDynamicLossScaleSkipIf(JobPassCtx* ctx, user_op::UserOpConfWra... FILE: oneflow/core/job_rewriter/optimizer.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job_rewriter/optimizer_placement_optimization_pass.cpp type oneflow (line 31) | namespace oneflow { function GetSoleOutBlobSize (line 35) | int64_t GetSoleOutBlobSize(const OpNode* node) { class DataParallelNodeSequence (line 41) | class DataParallelNodeSequence final { method DataParallelNodeSequence (line 43) | DataParallelNodeSequence(std::vector nodes, int64_t o... method OpNode (line 51) | const OpNode* GetVariableNode() const { return nodes_.front(); } method OpNode (line 53) | const OpNode* GetLastNode() const { return nodes_.back(); } method order (line 55) | int64_t order() const { return order_; } method ParallelDesc (line 59) | const ParallelDesc& parallel_desc() const { return nodes_.front()->p... method model_size (line 61) | int64_t model_size() const { return model_size_; } method len (line 63) | int64_t len() const { return len_; } method resize (line 65) | void resize(const int64_t size) { function ParallelConf (line 81) | ParallelConf NonDistributedParallelConf4ParallelId(const ParallelDesc&... function GetDataParallelVariableAndNaiveSuccNode (line 93) | Maybe GetDataParallelVariableAndNaiveSuccNode( function SetBroadcastParallel4OpNodeIbn (line 132) | void SetBroadcastParallel4OpNodeIbn(JobBuilder* builder, const OpNode*... function SetBroadcastParallel4Consumers (line 142) | void SetBroadcastParallel4Consumers(JobBuilder* builder, const Sequenc... function SetNdSbp4OpNodeIbn (line 154) | void SetNdSbp4OpNodeIbn(JobBuilder* builder, const OpNode* node, const... function SetNdSbp4Consumers (line 162) | void SetNdSbp4Consumers(JobBuilder* builder, const SequencePtr& sequen... function MakeGetterOpNode2TopoOrder (line 212) | std::function MakeGetterOpNode2TopoOrder(const... function GetMinConsumerOrder (line 222) | int64_t GetMinConsumerOrder(const OpGraph& op_graph, const OpNode* node, function ForEachDataParallelNodeSequence (line 231) | void ForEachDataParallelNodeSequence(const OpGraph& op_graph, function SequenceCompSortedByOrderAsc (line 245) | bool SequenceCompSortedByOrderAsc(const SequencePtr& lhs, const Sequen... function SequenceCompSortedByModelSizeDesc (line 249) | bool SequenceCompSortedByModelSizeDesc(const SequencePtr& lhs, const S... function ForEachParallelSortedNodeSequence (line 253) | void ForEachParallelSortedNodeSequence( function IsS0Parallel (line 269) | bool IsS0Parallel(const SbpParallel& sbp_parallel) { function IsS0Parallel (line 273) | bool IsS0Parallel(const SbpSignature& signature, const std::string& bn) { function IsNdSbpMatch (line 277) | bool IsNdSbpMatch(const NdSbpSignature& signature, const std::string& ... function IsNdSbpSupported4Op (line 281) | bool IsNdSbpSupported4Op(const OpNode* node, const NdSbp& nd_sbp) { function IsS0SignatureSupported (line 295) | bool IsS0SignatureSupported(const OpNode* node) { function ForEachModelSizeBalancedPartition (line 310) | void ForEachModelSizeBalancedPartition( function IsSplitValid (line 331) | bool IsSplitValid(const Shape& shape, const NdSbp& nd_sbp, const Shape... function GenerateSplitSignature (line 351) | void GenerateSplitSignature(const NdSbp& var_nd_sbp, const OperatorCon... function ShardSequence (line 381) | void ShardSequence(JobBuilder* builder, const int64_t threshold, const... function RewriteDistributedSplit (line 453) | Maybe RewriteDistributedSplit(const OpGraph& op_graph, JobBuilde... function RewriteNonDistributed (line 469) | Maybe RewriteNonDistributed(const OpGraph& op_graph, JobBuilder*... class OptimizerPlacementOptimizationPass (line 513) | class OptimizerPlacementOptimizationPass final : public JobPass { method OptimizerPlacementOptimizationPass (line 516) | OptimizerPlacementOptimizationPass() = default; method Apply (line 519) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/pass_util.cpp type oneflow (line 18) | namespace oneflow { function IsNodeInList (line 20) | bool IsNodeInList(const HashSet& op_list, OpNode* node) { function ReplaceSlashToDash4Lbn (line 26) | std::string ReplaceSlashToDash4Lbn(std::string lbn) { function DfsTopoGraphTraversal (line 31) | void DfsTopoGraphTraversal(const OpGraph& graph, bool reversed, function MakePredicatorIsSafeToDelete (line 57) | std::function MakePredicatorIsSafeToDelet... function IsUserOpWithTypeName (line 74) | bool IsUserOpWithTypeName(const OperatorConf& op_conf, const std::stri... function GenParallelConfKey (line 78) | std::string GenParallelConfKey(const ParallelConf& conf) { FILE: oneflow/core/job_rewriter/pass_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/job_rewriter/pipeline_buffer_pass.cpp type oneflow (line 25) | namespace oneflow { class PipelineBufferPass (line 29) | class PipelineBufferPass final : public JobPass { method PipelineBufferPass (line 32) | PipelineBufferPass() = default; method Apply (line 35) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { method IsEnabled (line 42) | bool IsEnabled(const JobPassCtx& ctx) const { function Scope (line 53) | const Scope& Scope4ScopeSymbolId(int64_t scope_symbol_id) { function Scope (line 58) | const Scope& Scope4OpNode(const OpNode* op_node) { function IsForwardPass (line 64) | bool IsForwardPass(const OpNode* node) { function IsBackwardPass (line 68) | bool IsBackwardPass(const OpNode* node) { function OpNodeHasScope (line 72) | bool OpNodeHasScope(const OpNode* node) { return node->op().op_conf().... function IsIdentityBufferOrRepeatOpNode (line 74) | bool IsIdentityBufferOrRepeatOpNode(const OpNode* node) { function GetStageIdHint (line 83) | int64_t GetStageIdHint(const OpNode* node) { function TryInsertOrUseBufferOpToDstNode (line 87) | void TryInsertOrUseBufferOpToDstNode( function TryInsertOrUseBufferOpBothSrcDst (line 141) | void TryInsertOrUseBufferOpBothSrcDst( FILE: oneflow/core/job_rewriter/prune_amp_white_identity_op_pass.cpp type oneflow (line 19) | namespace oneflow { function IsAmpIdentityOp (line 23) | bool IsAmpIdentityOp(const OperatorConf& op) { function NeedDoPass (line 29) | bool NeedDoPass(const Job& job) { class PruneAmpWhiteIdentityOpPass (line 33) | class PruneAmpWhiteIdentityOpPass final : public JobPass { method PruneAmpWhiteIdentityOpPass (line 35) | PruneAmpWhiteIdentityOpPass() = default; FILE: oneflow/core/job_rewriter/prune_cast_to_static_shape_op_pass.cpp type oneflow (line 19) | namespace oneflow { function IsRelatedOp (line 23) | bool IsRelatedOp(const OperatorConf& op) { function NeedDoPass (line 27) | bool NeedDoPass(const Job& job) { class PruneCastToStaticShapeOpsPass (line 31) | class PruneCastToStaticShapeOpsPass final : public JobPass { method PruneCastToStaticShapeOpsPass (line 33) | PruneCastToStaticShapeOpsPass() = default; method IsEnabled (line 36) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 41) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/prune_depend_op_pass.cpp type oneflow (line 26) | namespace oneflow { type UpdatedNodeInfo (line 30) | struct UpdatedNodeInfo { function IsDependyOp (line 39) | bool IsDependyOp(const OperatorConf& op) { function NeedDoPass (line 43) | bool NeedDoPass(const Job& job) { function OpNode (line 47) | const OpNode* GetNodeFromEdgeByTensorName(const OpNode* op_node, function OpNode (line 67) | const OpNode* GetNodeFromInputEdge(const OpNode* op_node) { function OpNode (line 71) | const OpNode* GetNodeFromInCtrlEdge(const OpNode* op_node) { function LogicalBlobId (line 75) | LogicalBlobId GetNewLbi(const OpNode* src_node, const OpNode* depend_n... class PruneDependOpPass (line 89) | class PruneDependOpPass final : public JobPass { method PruneDependOpPass (line 91) | PruneDependOpPass() = default; FILE: oneflow/core/job_rewriter/prune_parallel_cast_op_pass.cpp type oneflow (line 19) | namespace oneflow { function IsParallelCastOp (line 23) | bool IsParallelCastOp(const OperatorConf& op_conf) { function NeedDoPass (line 30) | bool NeedDoPass(const Job& job) { class PruneParallelCastOpsPass (line 34) | class PruneParallelCastOpsPass final : public JobPass { method PruneParallelCastOpsPass (line 36) | PruneParallelCastOpsPass() = default; method IsEnabled (line 39) | bool IsEnabled(const JobPassCtx& ctx) const { return ctx.job_desc().... method Apply (line 42) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/prune_pinned_identity_op_pass.cpp type oneflow (line 19) | namespace oneflow { class PrunePinnedIdentityOpPass (line 23) | class PrunePinnedIdentityOpPass final : public JobPass { method PrunePinnedIdentityOpPass (line 25) | PrunePinnedIdentityOpPass() = default; function PrunePinnedIdentityOp (line 31) | Maybe PrunePinnedIdentityOp(JobBuilder* job_builder, FILE: oneflow/core/job_rewriter/quantization_aware_training.cpp type oneflow (line 29) | namespace oneflow { function VerifyQATList (line 43) | Maybe VerifyQATList(const OpTypeSet& op_list) { function GetScaleLbn (line 53) | Maybe GetScaleLbn(const std::string& lbn) { function IsConvBiasEdge (line 58) | Maybe IsConvBiasEdge(const QatConfig& qat_config, const OpEdge* ... function IsWeightEdge (line 112) | bool IsWeightEdge(const OpEdge* edge) { function IsBnInputEdge (line 116) | bool IsBnInputEdge(const OpEdge* edge) { function OpTypeName4OpNode (line 122) | std::string OpTypeName4OpNode(const OpNode* node) { function OperatorConf (line 129) | OperatorConf Get1DZeroVariableOpConf(std::string name, const int64_t s... function GetInferenceOutputNode (line 143) | Maybe GetInferenceOutputNode(const OpGraph& op_graph, OpNode*... function PerLayerQuantizationAttr4Config (line 163) | bool PerLayerQuantizationAttr4Config(const QatConfig& qat_config) { function QuantizationSchemeAttr4QatConfig (line 167) | std::string QuantizationSchemeAttr4QatConfig(const QatConfig& qat_conf... function QuantizationFormulaAttr4QatConfig (line 172) | Maybe QuantizationFormulaAttr4QatConfig(const QatConfig& ... function Int8List4QatConfig (line 183) | Maybe Int8List4QatConfig(const QatConfig& qat_config) { function TransparentList4QatConfig (line 195) | Maybe TransparentList4QatConfig(const QatConfig& qat_config) { function InsertQuantOpAfterInt8Ops4QatConfig (line 206) | Maybe InsertQuantOpAfterInt8Ops4QatConfig(const QatConfig& qat_c... function MultiplyOp (line 217) | user_op::UserOpConfWrapper MultiplyOp(const std::string& name, const s... function MinMaxObserver (line 231) | Maybe MinMaxObserver(const std::string& na... function MovingMinMaxObserver (line 251) | Maybe MovingMinMaxObserver( function FakeQuantOp (line 291) | Maybe FakeQuantOp(const std::string& name,... function GetScaleAndZeroPointLbn4Edge (line 313) | Maybe GetScaleAndZeroPointLbn4Edge(OpEdge* edge, const std::stri... function ReplaceInputLbn4DstNodeOfEdge (line 352) | Maybe ReplaceInputLbn4DstNodeOfEdge(OpEdge* edge, const std::str... class QuantAwareTraining (line 365) | class QuantAwareTraining final : public JobPass { method QuantAwareTraining (line 368) | QuantAwareTraining() = default; method IsEnabled (line 370) | bool IsEnabled(const JobPassCtx& ctx) const { function IsNodeQuantizationEnabled (line 383) | Maybe IsNodeQuantizationEnabled(const OpNode& node) { FILE: oneflow/core/job_rewriter/replace_embedding_ops_pass.cpp type oneflow (line 24) | namespace oneflow { function BuildIdentityOp (line 28) | std::string BuildIdentityOp(JobBuilder* job_builder, const std::string... function DynamicLossScaleAddGradient (line 43) | Maybe DynamicLossScaleAddGradient( function BuildEmbeddingLookup (line 106) | void BuildEmbeddingLookup( function BuildEmbeddingShuffle (line 179) | void BuildEmbeddingShuffle(JobBuilder* job_builder, const std::string&... function BuildEmbeddingGradientShuffle (line 211) | void BuildEmbeddingGradientShuffle( function GetLossInstanceNumScaleFactor (line 295) | double GetLossInstanceNumScaleFactor(const OpGraph& op_graph, JobBuild... function BuildIdShuffle (line 350) | void BuildIdShuffle(bool use_system_gather, const std::string& embeddi... function MakeConstantInitializerAttr (line 470) | void MakeConstantInitializerAttr(const int64_t embedding_size, const i... function ScaleGrad (line 488) | void ScaleGrad(JobPassCtx* ctx, const OpGraph& op_graph, JobBuilder* j... function IsSupportFusedUpdatePut (line 551) | bool IsSupportFusedUpdatePut(const bool is_full_cache, const bool enab... function BuildEmbeddingUpdate (line 570) | void BuildEmbeddingUpdate( function UpdateConsumerOpConf (line 731) | void UpdateConsumerOpConf(const OpNode* consumer, const LogicalBlobId&... function GlobalAbsMaxMin (line 748) | std::string GlobalAbsMaxMin(JobBuilder* job_builder, function GlobalNorm (line 816) | std::string GlobalNorm(JobBuilder* job_builder, function GetClampCoeff (line 874) | std::string GetClampCoeff(JobBuilder* job_builder, const std::string& ... function ClipGradByGlobalNorm (line 920) | void ClipGradByGlobalNorm(JobPassCtx* ctx, const OpGraph& op_graph, Jo... function FilterCurGradLbnAndUpdateOpConfPairs (line 998) | void FilterCurGradLbnAndUpdateOpConfPairs( function FilterEmbeddingGradients (line 1010) | void FilterEmbeddingGradients(JobPassCtx* ctx, const OpGraph& op_graph... function IsRelatedOp (line 1034) | bool IsRelatedOp(const OperatorConf& op) { function NeedDoPass (line 1038) | bool NeedDoPass(const Job& job) { class ReplaceEmbeddingOps (line 1044) | class ReplaceEmbeddingOps final : public JobPass { method ReplaceEmbeddingOps (line 1046) | ReplaceEmbeddingOps() = default; method IsEnabled (line 1049) | bool IsEnabled(const JobPassCtx& ctx) const { return ctx.job_desc().... method Apply (line 1052) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/rmsprop_optm.cpp type oneflow (line 19) | namespace oneflow { function GenVariableOutputLbn (line 23) | std::string GenVariableOutputLbn(const OperatorConf& op_conf) { function OperatorConf (line 28) | OperatorConf GenerateRmspropHelperVariableOpConf(const VariableOp& op,... function GenerateOptimizerOpConf (line 40) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, FILE: oneflow/core/job_rewriter/sequential_one_embedding_shuffle_ops_pass.cpp type oneflow (line 20) | namespace oneflow { class SequentialOneEmbeddingOpsPass (line 24) | class SequentialOneEmbeddingOpsPass final : public JobPass { method SequentialOneEmbeddingOpsPass (line 26) | SequentialOneEmbeddingOpsPass() = default; method IsEnabled (line 29) | bool IsEnabled(const JobPassCtx& ctx) const { method Apply (line 34) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/sgd_optm.cpp type oneflow (line 19) | namespace oneflow { function GenerateOptimizerOpConf (line 23) | void GenerateOptimizerOpConf(JobPassCtx* ctx, const OpNode& var_op_node, FILE: oneflow/core/job_rewriter/source_user_op_auto_tick.cpp type oneflow (line 19) | namespace oneflow { class MutUserOpConTickInputHelper (line 23) | class MutUserOpConTickInputHelper final : public MutOpConTickInputHelp... method MutUserOpConTickInputHelper (line 25) | MutUserOpConTickInputHelper() : MutOpConTickInputHelper() {} method VirtualIsTickInputBound (line 27) | bool VirtualIsTickInputBound() const override { return !op_conf().us... method OperatorConf (line 29) | OperatorConf NewTickInputBoundOpConf(const std::string& lbn) const o... FILE: oneflow/core/job_rewriter/split_sparse_softmax_cross_entropy_op_pass.cpp type oneflow (line 19) | namespace oneflow { function NeedDoPass (line 23) | bool NeedDoPass(const Job& job) { function UpdateProbConsumerOpConf (line 29) | void UpdateProbConsumerOpConf(const std::string& new_prob_lbn, const O... class SplitSparseSoftmaxCrossEntropyOpPass (line 43) | class SplitSparseSoftmaxCrossEntropyOpPass final : public JobPass { method SplitSparseSoftmaxCrossEntropyOpPass (line 45) | SplitSparseSoftmaxCrossEntropyOpPass() = default; method Apply (line 50) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/system_op_fill_job_name_pass.cpp type oneflow (line 20) | namespace oneflow { class SystemOpFillJobNamePass (line 24) | class SystemOpFillJobNamePass final : public JobPass { method SystemOpFillJobNamePass (line 27) | SystemOpFillJobNamePass() = default; method IsEnabled (line 30) | bool IsEnabled(const JobPassCtx& ctx) const { return true; } method Apply (line 32) | Maybe Apply(Job* job, JobPassCtx* ctx) const override { FILE: oneflow/core/job_rewriter/tick_autotick.cpp type oneflow (line 18) | namespace oneflow { class MutTickOpConTickInputHelper (line 22) | class MutTickOpConTickInputHelper final : public MutOpConTickInputHelp... method MutTickOpConTickInputHelper (line 24) | MutTickOpConTickInputHelper() : MutOpConTickInputHelper() {} method VirtualIsTickInputBound (line 26) | bool VirtualIsTickInputBound() const override { return op_conf().tic... method OperatorConf (line 28) | OperatorConf NewTickInputBoundOpConf(const std::string& lbn) const o... FILE: oneflow/core/job_rewriter/variable_autotick.cpp type oneflow (line 18) | namespace oneflow { class MutVariableOpConTickInputHelper (line 22) | class MutVariableOpConTickInputHelper final : public MutOpConTickInput... method MutVariableOpConTickInputHelper (line 24) | MutVariableOpConTickInputHelper() : MutOpConTickInputHelper() {} method VirtualIsTickInputBound (line 26) | bool VirtualIsTickInputBound() const override { return op_conf().var... method OperatorConf (line 28) | OperatorConf NewTickInputBoundOpConf(const std::string& lbn) const o... FILE: oneflow/core/kernel/assign_kernel.cpp type oneflow (line 18) | namespace oneflow { class AssignKernel (line 20) | class AssignKernel final : public Kernel { method AssignKernel (line 23) | AssignKernel() = default; method IsStateless (line 27) | bool IsStateless() const override { return false; } FILE: oneflow/core/kernel/blob_access_checker_kernel_observer.cpp type oneflow (line 19) | namespace oneflow { function ForEachObnAndIsHeaderInferedBeforeCompute (line 24) | void ForEachObnAndIsHeaderInferedBeforeCompute(KernelContext* kernel_c... function ForEachObnAndIsMutableByConsumer (line 38) | void ForEachObnAndIsMutableByConsumer(KernelContext* kernel_ctx, const... function SetOutputBlobProducerInferAccessChecker (line 51) | void SetOutputBlobProducerInferAccessChecker(KernelContext* kernel_ctx... function SetOutputBlobProducerComputeAccessChecker (line 59) | void SetOutputBlobProducerComputeAccessChecker(KernelContext* kernel_c... function SetOutputBlobConsumerAccessChecker (line 72) | void SetOutputBlobConsumerAccessChecker(KernelContext* kernel_ctx, con... FILE: oneflow/core/kernel/blob_access_checker_kernel_observer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/blob_tensor_view.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 21) | namespace user_op { function ShapeView (line 25) | ShapeView BlobTensorView::shape_view() const { return blob_->shape(); } function MutShapeView (line 27) | MutShapeView BlobTensorView::mut_shape_view() { return *blob_->mut_s... function Stride (line 29) | const Stride& BlobTensorView::stride() const { return blob_->stride(... function DataType (line 31) | DataType BlobTensorView::data_type() const { return blob_->data_type... function MemoryFormat (line 33) | MemoryFormat BlobTensorView::memory_format() const { return blob_->m... function MemoryCase (line 35) | const MemoryCase& BlobTensorView::mem_case() const { return blob_->m... function Blob (line 43) | Blob* BlobTensorView::blob() const { return blob_; } FILE: oneflow/core/kernel/blob_tensor_view.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/boxing_kernel.cpp type oneflow (line 24) | namespace oneflow { class BoxingKernel (line 27) | class BoxingKernel final : public Kernel { method BoxingKernel (line 30) | BoxingKernel() = default; function ConstructPbRpf (line 43) | PbRpf ConstructPbRpf(const std::string& s) { function CalcSumOfBlobs (line 51) | void CalcSumOfBlobs(KernelContext* ctx, const std::function BnInOp2Blob, method OneElemSize (line 95) | size_t OneElemSize() const { return GetSizeOfDataType(BnInOp2Blob_(b... method TotalElemNum (line 97) | int64_t TotalElemNum() const { return seg_num_ * elem_sum_.back(); } method CalcContinuousElemNumStartFrom (line 100) | std::tuple CalcContinuousElemNumStartFrom(int64_t i... function ConcatSplitPartDataContent (line 128) | void ConcatSplitPartDataContent(ep::Stream* stream, const DataContentD... function ConcatSplitDataContent (line 170) | void ConcatSplitDataContent(ep::Stream* stream, FILE: oneflow/core/kernel/boxing_zeros_kernel.cpp type oneflow (line 20) | namespace oneflow { class BoxingZerosKernel (line 22) | class BoxingZerosKernel final : public Kernel { method BoxingZerosKernel (line 25) | BoxingZerosKernel() = default; FILE: oneflow/core/kernel/broadcast_to_compatible_with_kernel.cpp type oneflow (line 19) | namespace oneflow { class BroadcastToCompatibleWithKernel (line 22) | class BroadcastToCompatibleWithKernel final : public Kernel { method BroadcastToCompatibleWithKernel (line 25) | BroadcastToCompatibleWithKernel() = default; function FOR_RANGE (line 40) | FOR_RANGE(int64_t, i, 0, num_axes) { FILE: oneflow/core/kernel/callback_notify_kernel.cpp type oneflow (line 22) | namespace oneflow { class CallbackNotifyKernel (line 25) | class CallbackNotifyKernel final : public Kernel { method CallbackNotifyKernel (line 28) | CallbackNotifyKernel() = default; method IsStateless (line 32) | bool IsStateless() const override { return false; } FILE: oneflow/core/kernel/case_kernel.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/kernel/case_kernel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/chain_kernel_observer.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/kernel/chain_kernel_observer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/collective_boxing_kernels.cpp type oneflow (line 22) | namespace oneflow { function CollectiveBoxingActorContext (line 28) | CollectiveBoxingActorContext* GetCollectiveBoxingActorContext(KernelCo... class CollectiveBoxingKernelState (line 34) | class CollectiveBoxingKernelState final : public KernelState { method CollectiveBoxingKernelState (line 37) | explicit CollectiveBoxingKernelState(const RankDesc& rank_desc) method RequestHandle (line 42) | RequestHandle* request_handle() { return request_handle_; } class CollectiveBoxingGenericKernel (line 48) | class CollectiveBoxingGenericKernel final : public Kernel { method CollectiveBoxingGenericKernel (line 51) | CollectiveBoxingGenericKernel() = default; method IsKernelLaunchSynchronized (line 56) | bool IsKernelLaunchSynchronized() const override { return false; } FILE: oneflow/core/kernel/collective_boxing_pack_kernel.cpp type oneflow (line 20) | namespace oneflow { class CollectiveBoxingPackKernel (line 22) | class CollectiveBoxingPackKernel final : public Kernel { method CollectiveBoxingPackKernel (line 25) | CollectiveBoxingPackKernel() = default; method IsStateless (line 29) | bool IsStateless() const override { return false; } function FOR_RANGE (line 55) | FOR_RANGE(int64_t, i, 0, transpose_in_dim_vec.size()) { FILE: oneflow/core/kernel/collective_boxing_unpack_kernel.cpp type oneflow (line 20) | namespace oneflow { class CollectiveBoxingUnpackKernel (line 22) | class CollectiveBoxingUnpackKernel final : public Kernel { method CollectiveBoxingUnpackKernel (line 25) | CollectiveBoxingUnpackKernel() = default; method IsStateless (line 29) | bool IsStateless() const override { return false; } function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 1, transpose_in_dim_vec.size()) { perm.emplace_b... FILE: oneflow/core/kernel/constant_like_kernel.cpp type oneflow (line 20) | namespace oneflow { class ConstantLikeKernel (line 22) | class ConstantLikeKernel final : public Kernel { method ConstantLikeKernel (line 25) | ConstantLikeKernel() : is_init_(false) {} method ForwardDataContent (line 31) | void ForwardDataContent(KernelContext* ctx) const override { FILE: oneflow/core/kernel/cpu_check_numerics_kernel_observer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/cpu_numerics_kernel_observer.cpp type oneflow (line 19) | namespace oneflow { function HasNotFinite (line 24) | bool HasNotFinite(const int64_t elem_cnt, const T* data_ptr) { function HasNotFiniteCpu (line 31) | bool HasNotFiniteCpu(ep::Stream* stream, const Blob* blob) { function DumpBlob (line 43) | void DumpBlob(KernelContext* ctx, const std::string& bn) { function DumpBlobs (line 51) | void DumpBlobs(KernelContext* ctx, const Kernel* kernel) { FILE: oneflow/core/kernel/critical_section_callback_tick_kernel.cpp type oneflow (line 22) | namespace oneflow { class CriticalSectionCallbackTickKernel (line 24) | class CriticalSectionCallbackTickKernel final : public Kernel { method CriticalSectionCallbackTickKernel (line 27) | CriticalSectionCallbackTickKernel() = default; method IsStateless (line 31) | bool IsStateless() const override { return false; } FILE: oneflow/core/kernel/critical_section_wait_tick_kernel.cpp type oneflow (line 22) | namespace oneflow { class CriticalSectionWaitTickKernel (line 24) | class CriticalSectionWaitTickKernel final : public Kernel { method CriticalSectionWaitTickKernel (line 27) | CriticalSectionWaitTickKernel() = default; method IsStateless (line 31) | bool IsStateless() const override { return false; } FILE: oneflow/core/kernel/cuda_check_numerics_kernel_observer.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/kernel/cuda_graph_support.h function namespace (line 20) | namespace oneflow { FILE: oneflow/core/kernel/distribute_kernels.cpp type oneflow (line 19) | namespace oneflow { class DistributeAddKernel (line 21) | class DistributeAddKernel final : public Kernel { method DistributeAddKernel (line 24) | DistributeAddKernel() = default; function Blob (line 36) | const Blob* DistributeAddKernel::GetInBlob(KernelContext* ctx) const { class DistributeCloneKernel (line 50) | class DistributeCloneKernel final : public Kernel { method DistributeCloneKernel (line 53) | DistributeCloneKernel() = default; function Blob (line 65) | Blob* DistributeCloneKernel::GetOutBlob(KernelContext* ctx) const { class DistributeConcatKernel (line 79) | class DistributeConcatKernel final : public Kernel { method DistributeConcatKernel (line 82) | DistributeConcatKernel() = default; function Blob (line 94) | const Blob* DistributeConcatKernel::GetInBlob(KernelContext* ctx) const { class DistributeSplitKernel (line 108) | class DistributeSplitKernel final : public Kernel { method DistributeSplitKernel (line 111) | DistributeSplitKernel() = default; function Blob (line 129) | Blob* DistributeSplitKernel::GetOutBlob(KernelContext* ctx) const { FILE: oneflow/core/kernel/dynamic_reshape_kernel.cpp type oneflow (line 18) | namespace oneflow { class DynamicReshapeKernel (line 20) | class DynamicReshapeKernel final : public Kernel { method DynamicReshapeKernel (line 23) | DynamicReshapeKernel() = default; FILE: oneflow/core/kernel/dynamic_reshape_like_kernel.cpp type oneflow (line 18) | namespace oneflow { class DynamicReshapeLikeKernel (line 20) | class DynamicReshapeLikeKernel final : public Kernel { method DynamicReshapeLikeKernel (line 23) | DynamicReshapeLikeKernel() = default; FILE: oneflow/core/kernel/esac_kernel.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/kernel/esac_kernel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/identity_kernel.cpp type oneflow (line 20) | namespace oneflow { class IdentityKernel (line 22) | class IdentityKernel final : public Kernel { method IdentityKernel (line 25) | IdentityKernel() = default; FILE: oneflow/core/kernel/image_decoder_random_crop_resize_kernel.cpp type oneflow (line 42) | namespace oneflow { type Task (line 48) | struct Task { type Work (line 55) | struct Work { type ROI (line 63) | struct ROI { class ROIGenerator (line 70) | class ROIGenerator { class RandomCropROIGenerator (line 76) | class RandomCropROIGenerator : public ROIGenerator { method RandomCropROIGenerator (line 78) | explicit RandomCropROIGenerator(RandomCropGenerator* crop_generator) method Generate (line 82) | void Generate(int width, int height, ROI* roi) const override { class NoChangeROIGenerator (line 95) | class NoChangeROIGenerator : public ROIGenerator { method Generate (line 99) | void Generate(int width, int height, ROI* roi) const override { function GenerateRandomCropRoi (line 107) | void GenerateRandomCropRoi(RandomCropGenerator* crop_generator, int wi... class DecodeHandle (line 117) | class DecodeHandle { method DecodeHandle (line 119) | DecodeHandle() = default; class CpuDecodeHandle (line 134) | class CpuDecodeHandle final : public DecodeHandle { method CpuDecodeHandle (line 137) | CpuDecodeHandle() = default; method WarmupOnce (line 144) | void WarmupOnce(int warmup_size, unsigned char* workspace, size_t wo... method Synchronize (line 147) | void Synchronize() override { function CpuJpegDecodeRandomCropResize (line 152) | bool CpuJpegDecodeRandomCropResize(const unsigned char* data, size_t l... function OpencvDecodeRandomCropResize (line 167) | void OpencvDecodeRandomCropResize(const unsigned char* data, size_t le... function DecodeHandleFactory (line 201) | DecodeHandleFactory CreateDecodeHandleFactory(int ta... function GpuDeviceMalloc (line 208) | int GpuDeviceMalloc(void** p, size_t s) { return (int)cudaMalloc(p, s); } function GpuDeviceFree (line 210) | int GpuDeviceFree(void* p) { return (int)cudaFree(p); } function GpuPinnedMalloc (line 212) | int GpuPinnedMalloc(void** p, size_t s, unsigned int flags) { function GpuPinnedFree (line 216) | int GpuPinnedFree(void* p) { return (int)cudaFreeHost(p); } function InitNppStreamContext (line 218) | void InitNppStreamContext(NppStreamContext* ctx, int dev, cudaStream_t... class GpuDecodeHandle (line 237) | class GpuDecodeHandle final : public DecodeHandle { function DecodeHandleFactory (line 471) | DecodeHandleFactory CreateDecodeHandleFactory(int t... class Worker (line 483) | class Worker final { method Worker (line 486) | Worker(const std::function()>& handle_... method Enqueue (line 496) | void Enqueue(std::shared_ptr& work) { work_queue_.Send(work); } method PollWork (line 502) | void PollWork(const std::function()>& ... class ImageDecoderRandomCropResizeKernel (line 528) | class ImageDecoderRandomCropResizeKernel final : public Kernel { method ImageDecoderRandomCropResizeKernel (line 531) | ImageDecoderRandomCropResizeKernel() = default; FILE: oneflow/core/kernel/input_kernel.cpp type oneflow (line 22) | namespace oneflow { class InputKernel (line 26) | class InputKernel final : public Kernel { method InputKernel (line 29) | InputKernel() = default; method ForwardDataContent (line 33) | void ForwardDataContent(KernelContext* ctx) const override { method ForwardHeader (line 47) | void ForwardHeader(KernelContext* ctx) const override {} FILE: oneflow/core/kernel/kernel.cpp type oneflow (line 21) | namespace oneflow { function IsAllBlobEmpty (line 25) | bool IsAllBlobEmpty(const PbRpf& bns, KernelContext* ctx) { function ConstructKernel (line 81) | std::unique_ptr ConstructKernel(const KernelConf& conf, ... FILE: oneflow/core/kernel/kernel.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/kernel/kernel_context.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/kernel/kernel_observer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/kernel_registration.cpp type oneflow (line 19) | namespace oneflow { type kernel_registration (line 21) | namespace kernel_registration { function KernelRegistrarBuilder (line 32) | KernelRegistrarBuilder& KernelRegistrarBuilder::SetCreateFn(CreateFn... function KernelRegistrarBuilder (line 37) | KernelRegistrarBuilder& KernelRegistrarBuilder::SetIsMatchedPred(IsM... function Kernel (line 57) | Kernel* CreateKernel(const KernelConf& kernel_conf) { FILE: oneflow/core/kernel/kernel_registration.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/kernel/kernel_util.cpp type oneflow (line 25) | namespace oneflow { function AutoMemcpy (line 27) | void AutoMemcpy(ep::Stream* stream, void* dst, const void* src, size_t... function AutoMemcpy (line 51) | void AutoMemcpy(ep::Stream* stream, Blob* dst, const Blob* src) { function SyncAutoMemcpy (line 57) | void SyncAutoMemcpy(ep::Stream* stream, void* dst, const void* src, si... function AutoMemset (line 63) | void AutoMemset(ep::Stream* stream, void* dst, const char value, size_... FILE: oneflow/core/kernel/kernel_util.h function namespace (line 29) | namespace oneflow { FILE: oneflow/core/kernel/learning_rate_schedule_kernel.cpp type oneflow (line 25) | namespace oneflow { class LearningRateScheduleKernel (line 27) | class LearningRateScheduleKernel final : public Kernel { method LearningRateScheduleKernel (line 30) | LearningRateScheduleKernel() = default; method VirtualKernelInit (line 34) | void VirtualKernelInit(KernelContext* ctx) override { function ConstantLearningRate (line 52) | double ConstantLearningRate(double base_lr, double factor, int64_t tot... function LinearLearningRate (line 60) | double LinearLearningRate(double base_lr, double start_factor, double ... function ExponentialDecayedLearningRate (line 76) | double ExponentialDecayedLearningRate(const ExponentialDecayConf& conf... function InverseTimeDecayedLearningRate (line 84) | double InverseTimeDecayedLearningRate(const InverseTimeDecayConf& conf... function NaturalExpDecayedLearningRate (line 92) | double NaturalExpDecayedLearningRate(const NaturalExpDecayConf& conf, ... function PiecewiseConstantLearningRate (line 100) | double PiecewiseConstantLearningRate(const PiecewiseConstantConf& conf... function PolynomialDecayedLearningRate (line 112) | double PolynomialDecayedLearningRate(const PolynomialDecayConf& conf, ... function CosineDecayedLearningRate (line 127) | double CosineDecayedLearningRate(const CosineDecayConf& conf, double l... function CosineAnnealingDecayedLearningRate (line 138) | double CosineAnnealingDecayedLearningRate(const CosineAnnealingDecayCo... function LinearCosineDecayedLearningRate (line 152) | double LinearCosineDecayedLearningRate(const LinearCosineDecayConf& co... function PiecewiseScalingLearningRate (line 166) | double PiecewiseScalingLearningRate(const PiecewiseScalingConf& conf, ... function StepLearningRate (line 178) | double StepLearningRate(const StepConf& conf, double lr, int64_t cur_b... function MultiStepLearningRate (line 190) | double MultiStepLearningRate(const MultiStepConf& conf, double lr, int... function CosineAnnealingWarmRestartsLearningRate (line 207) | double CosineAnnealingWarmRestartsLearningRate(const CosineAnnealingWa... function SequentialScheduler (line 229) | double SequentialScheduler(const SequentialSchedulerConf& conf, const ... function GetDecayedLearningRate (line 248) | double GetDecayedLearningRate(const LearningRateDecayConf& conf, doubl... FILE: oneflow/core/kernel/nccl_send_recv_boxing_kernel.cpp type oneflow (line 27) | namespace oneflow { class CclSendRecvBoxingKernel (line 29) | class CclSendRecvBoxingKernel final : public Kernel { method CclSendRecvBoxingKernel (line 32) | CclSendRecvBoxingKernel() = default; method has_input (line 43) | const bool has_input() const { return has_input_; } method has_output (line 44) | const bool has_output() const { return has_output_; } method ccl_comm (line 45) | ccl::CclComm ccl_comm() const { return GetOrCreate().ccl_comm; } type Comm (line 48) | struct Comm { method Comm (line 49) | explicit Comm(ccl::CclComm comm) : ccl_comm(comm) {} method Init (line 53) | void Init() const { method Comm (line 61) | const Comm& GetOrCreate() const { method Comm (line 49) | explicit Comm(ccl::CclComm comm) : ccl_comm(comm) {} FILE: oneflow/core/kernel/new_kernel_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/kernel/nop_kernel.cpp type oneflow (line 18) | namespace oneflow { class NopKernel (line 20) | class NopKernel final : public Kernel { method NopKernel (line 23) | NopKernel() = default; method ForwardDataContent (line 27) | void ForwardDataContent(KernelContext* ctx) const override {} FILE: oneflow/core/kernel/output_kernel.cpp type oneflow (line 21) | namespace oneflow { class OutputKernel (line 23) | class OutputKernel final : public Kernel { method OutputKernel (line 26) | OutputKernel() = default; FILE: oneflow/core/kernel/profiler_kernel_observer.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/kernel/profiler_kernel_observer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/random_generator.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/kernel/random_generator.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/kernel/reentrant_lock_kernel.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/kernel/reentrant_lock_kernel.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/kernel/return_kernel.cpp type oneflow (line 21) | namespace oneflow { class ReturnKernel (line 23) | class ReturnKernel final : public Kernel { method ReturnKernel (line 26) | ReturnKernel() = default; FILE: oneflow/core/kernel/runtime_blob_shape_infer_helper.cpp type oneflow (line 22) | namespace oneflow { function FOR_RANGE (line 79) | FOR_RANGE(int, i, 0, input_bns.size()) { function BlobDesc (line 84) | BlobDesc* RuntimeBlobShapeInferHelper::BlobDesc4BnInOp(const std::stri... function FOR_RANGE (line 104) | FOR_RANGE(int, i, 0, op_->output_bns().size()) { function FOR_RANGE (line 118) | FOR_RANGE(int, i, 0, op_->output_bns().size()) { FILE: oneflow/core/kernel/runtime_blob_shape_infer_helper.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/kernel/shape_elem_cnt_kernel.cpp type oneflow (line 19) | namespace oneflow { class ShapeElemCntKernel (line 22) | class ShapeElemCntKernel final : public Kernel { method ShapeElemCntKernel (line 25) | ShapeElemCntKernel() = default; FILE: oneflow/core/kernel/slice_boxing_kernel.cpp type oneflow (line 23) | namespace oneflow { class SliceBoxingKernel (line 25) | class SliceBoxingKernel : public Kernel { method SliceBoxingKernel (line 28) | SliceBoxingKernel() = default; class SliceBoxingCopyKernel (line 42) | class SliceBoxingCopyKernel final : public SliceBoxingKernel { method SliceBoxingCopyKernel (line 45) | SliceBoxingCopyKernel() = default; class SliceBoxingAddKernel (line 53) | class SliceBoxingAddKernel final : public SliceBoxingKernel { method SliceBoxingAddKernel (line 56) | SliceBoxingAddKernel() = default; function SliceBoxingConf (line 83) | const SliceBoxingConf& SliceBoxingCopyKernel::GetCustomizedBoxingConf(... function FOR_RANGE (line 90) | FOR_RANGE(int64_t, i, 0, this->op_attribute().input_bns().size()) { function SliceBoxingConf (line 96) | const SliceBoxingConf& SliceBoxingAddKernel::GetCustomizedBoxingConf()... function FOR_RANGE (line 110) | FOR_RANGE(int64_t, i, 0, this->op_attribute().input_bns().size()) { FILE: oneflow/core/kernel/sync_check_kernel_observer.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/kernel/sync_check_kernel_observer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/sync_dynamic_resize_kernel.cpp type oneflow (line 30) | namespace oneflow { class CudaHostMem (line 36) | class CudaHostMem { method CudaHostMem (line 39) | CudaHostMem(const size_t size) { OF_CUDA_CHECK(cudaMallocHost(&ptr_,... class SyncDynamicResizeGPUKernel (line 50) | class SyncDynamicResizeGPUKernel final : public Kernel { method SyncDynamicResizeGPUKernel (line 53) | SyncDynamicResizeGPUKernel() = default; method IsKernelLaunchSynchronized (line 57) | bool IsKernelLaunchSynchronized() const override { return false; } method ForwardDataContent (line 59) | void ForwardDataContent(KernelContext* ctx) const override { class SyncDynamicResizeCPUKernel (line 119) | class SyncDynamicResizeCPUKernel final : public Kernel { method SyncDynamicResizeCPUKernel (line 122) | SyncDynamicResizeCPUKernel() = default; method IsKernelLaunchSynchronized (line 126) | bool IsKernelLaunchSynchronized() const override { return false; } method ForwardDataContent (line 127) | void ForwardDataContent(KernelContext* ctx) const override { FILE: oneflow/core/kernel/total_loss_instance_num_kernel.cpp type oneflow (line 18) | namespace oneflow { class TotalLossInstanceNumKernel (line 21) | class TotalLossInstanceNumKernel final : public Kernel { method TotalLossInstanceNumKernel (line 24) | TotalLossInstanceNumKernel() = default; FILE: oneflow/core/kernel/user_kernel.cpp type oneflow (line 29) | namespace oneflow { function IsAllBlobEmpty (line 33) | bool IsAllBlobEmpty(const PbRpf& bns, function FillTensorDescWithBlob (line 50) | void FillTensorDescWithBlob(const Blob* blob, user_op::NaiveTensorDesc... class UserKernelBaseContext (line 67) | class UserKernelBaseContext { method UserKernelBaseContext (line 69) | explicit UserKernelBaseContext(const KernelConf& kernel_conf) { method DeviceType (line 94) | DeviceType device_type() const { return device_type_; } method ParallelContext (line 95) | const ParallelContext& parallel_ctx() const { return parallel_ctx_; } method ArgVec (line 103) | const ArgVec& inputs() const { return inputs_; } method ArgVec (line 104) | const ArgVec& outputs() const { return outputs_; } class UserKernelInitAndCacheContext (line 116) | class UserKernelInitAndCacheContext final : public user_op::KernelInit... method UserKernelInitAndCacheContext (line 119) | explicit UserKernelInitAndCacheContext(ep::Stream* stream, const Ker... method UpdateTensorWithCorrBlob (line 147) | void UpdateTensorWithCorrBlob(const std::function... type BnTensorPair (line 539) | struct BnTensorPair { function BnTensorPair (line 544) | BnTensorPair MakeBnTensorPair(const std::string& bn) { function BnTensorPair (line 550) | BnTensorPair MakeBnTensorPair(const std::string& bn, class UserKernelComputeContext (line 560) | class UserKernelComputeContext final : public user_op::KernelComputeCo... method UserKernelComputeContext (line 562) | explicit UserKernelComputeContext(ep::Stream* stream, const KernelCo... method UpdateTensorWithCorrBlob (line 594) | bool UpdateTensorWithCorrBlob(const std::functionoutputs().size()) { FILE: oneflow/core/kernel/user_kernel.h function namespace (line 31) | namespace oneflow { FILE: oneflow/core/kernel/util/cuda_half_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/kernel/wait_and_send_ids_kernel.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/core/kernel/wait_and_send_ids_kernel.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/lazy/actor/acc_actor.cpp type oneflow (line 18) | namespace oneflow { class AccActor (line 20) | class AccActor final : public Actor { method AccActor (line 23) | AccActor() = default; FILE: oneflow/core/lazy/actor/acc_ctrl_tick_actor.cpp type oneflow (line 19) | namespace oneflow { class AccCtrlTickActor (line 21) | class AccCtrlTickActor : public Actor { method AccCtrlTickActor (line 24) | AccCtrlTickActor() method GetNaiveOrCustomizedConsumedRegstDescName (line 34) | std::pair> GetNaiveOrCustomizedC... method GetNaiveOrCustomizedProducedRegstDescName (line 38) | std::pair> GetNaiveOrCustomizedP... method IsCustomizedReadReady (line 43) | bool IsCustomizedReadReady() const override { method IsCustomizedWriteReady (line 46) | bool IsCustomizedWriteReady() const override { return produced_tick_... method NormalProcessCustomizedEordMsg (line 48) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override {} method IsCustomizedReadAlwaysUnReadyFromNow (line 49) | bool IsCustomizedReadAlwaysUnReadyFromNow() const override { FILE: oneflow/core/lazy/actor/acc_tick_actor.cpp type oneflow (line 18) | namespace oneflow { class AccTickActor (line 20) | class AccTickActor : public Actor { method AccTickActor (line 23) | AccTickActor() = default; FILE: oneflow/core/lazy/actor/actor.cpp type oneflow (line 22) | namespace oneflow { class KernelContextImpl (line 26) | class KernelContextImpl : public KernelContext, public ActorContextPro... method KernelContextImpl (line 29) | explicit KernelContextImpl(ActorContext* actor_ctx) method ActorContext (line 43) | ActorContext* GetActorContext() const override { return actor_ctx_; } method Blob (line 45) | Blob* BnInOp2Blob(const std::string& bn) const override { return bn_... method set_state (line 49) | void set_state(std::shared_ptr state) override { state_... method UpdateBnInOp2BlobFn (line 60) | void UpdateBnInOp2BlobFn(std::function fn) { function CheckInplaceRegstDescId (line 115) | void CheckInplaceRegstDescId(const TaskProto& task_proto) { function Regst (line 715) | Regst* Actor::GetSoleProducedRegst4RegstDescId(int64_t regst_desc_id) ... function Regst (line 780) | Regst* Actor::GetNaiveOrInplaceCurReadable(int64_t regst_desc_id) const { function Regst (line 786) | Regst* Actor::GetNaiveOrInplaceCurWriteable(int64_t regst_desc_id) con... function Regst (line 792) | Regst* Actor::GetNaiveCurReadable(int64_t regst_desc_id) const { function Regst (line 796) | Regst* Actor::GetNaiveCurWriteable(int64_t regst_desc_id) const { FILE: oneflow/core/lazy/actor/actor.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/lazy/actor/actor_base.cpp type oneflow (line 20) | namespace oneflow { function NewActor (line 22) | std::unique_ptr NewActor(ActorContext* actor_ctx) { FILE: oneflow/core/lazy/actor/actor_base.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/lazy/actor/actor_context.cpp type oneflow (line 19) | namespace oneflow { function NewActorContext (line 21) | std::unique_ptr NewActorContext(const TaskProto& task_pr... FILE: oneflow/core/lazy/actor/actor_context.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/lazy/actor/actor_message.cpp type oneflow (line 20) | namespace oneflow { function IsSoleBlobAndDynamicEmpty (line 24) | bool IsSoleBlobAndDynamicEmpty(Regst* regst) { function ActorMsg (line 38) | ActorMsg ActorMsg::BuildRegstMsgToConsumer(int64_t producer, int64_t c... function ActorMsg (line 53) | ActorMsg ActorMsg::BuildRegstMsgToProducer(int64_t consumer, int64_t p... function ActorMsg (line 68) | ActorMsg ActorMsg::BuildEordMsg(int64_t consumer, int64_t regst_desc_i... function ActorMsg (line 77) | ActorMsg ActorMsg::BuildCommandMsg(int64_t dst_actor_id, ActorCmd cmd) { function ActorCmd (line 88) | ActorCmd ActorMsg::actor_cmd() const { function Regst (line 93) | Regst* ActorMsg::regst() const { FILE: oneflow/core/lazy/actor/actor_message.h function ActorCmd (line 24) | enum class ActorCmd { FILE: oneflow/core/lazy/actor/actor_message_bus.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/core/lazy/actor/actor_message_bus.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/lazy/actor/boxing_zeros_actor.cpp type oneflow (line 18) | namespace oneflow { class BoxingZerosActor (line 20) | class BoxingZerosActor : public NaiveActor { method BoxingZerosActor (line 23) | BoxingZerosActor() = default; method VirtualActorInit (line 26) | void VirtualActorInit(const TaskProto& task_proto) override { method Act (line 32) | void Act() override { method VirtualAsyncSendNaiveProducedRegstMsgToConsumer (line 39) | void VirtualAsyncSendNaiveProducedRegstMsgToConsumer() override { FILE: oneflow/core/lazy/actor/callback_notify_actor.cpp type oneflow (line 18) | namespace oneflow { class CallbackNotifyActor (line 20) | class CallbackNotifyActor final : public SinkActor { method CallbackNotifyActor (line 23) | CallbackNotifyActor() = default; FILE: oneflow/core/lazy/actor/case_actor.cpp type oneflow (line 20) | namespace oneflow { class CaseActor (line 22) | class CaseActor final : public Actor { method CaseActor (line 25) | CaseActor() : case_status_(nullptr) {} method NormalProcessCustomizedEordMsg (line 39) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override {} method GetNaiveOrCustomizedConsumedRegstDescName (line 40) | std::pair> GetNaiveOrCustomizedC... method GetNaiveOrCustomizedProducedRegstDescName (line 44) | std::pair> GetNaiveOrCustomizedP... function FOR_RANGE (line 69) | FOR_RANGE(int64_t, i, 0, output_bns_size) { FILE: oneflow/core/lazy/actor/collective_boxing_actor_context.cpp type oneflow (line 19) | namespace oneflow { function StreamContext (line 64) | StreamContext* CollectiveBoxingActorContext::stream_ctx() const { retu... function TaskProto (line 66) | const TaskProto& CollectiveBoxingActorContext::task_proto() const { re... FILE: oneflow/core/lazy/actor/collective_boxing_actor_context.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/lazy/actor/copy_comm_net_actor.cpp type oneflow (line 20) | namespace oneflow { class CopyCommNetActor (line 22) | class CopyCommNetActor final : public Actor { method CopyCommNetActor (line 25) | CopyCommNetActor() = default; type RegstCtx (line 29) | struct RegstCtx { method GetNaiveOrCustomizedConsumedRegstDescName (line 38) | std::pair> GetNaiveOrCustomizedC... method NormalProcessCustomizedEordMsg (line 43) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override { is_i... FILE: oneflow/core/lazy/actor/esac_actor.cpp type oneflow (line 20) | namespace oneflow { class EsacActor (line 22) | class EsacActor final : public Actor { method EsacActor (line 25) | EsacActor() = default; method InBnId4RegstDescId (line 30) | int64_t InBnId4RegstDescId(int64_t id) const { return regst_desc_id2... method NormalProcessCustomizedEordMsg (line 39) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override {} method IsCustomizedReadAlwaysUnReadyFromNow (line 40) | bool IsCustomizedReadAlwaysUnReadyFromNow() const override { method GetNaiveOrCustomizedConsumedRegstDescName (line 44) | std::pair> GetNaiveOrCustomizedC... function FOR_RANGE (line 61) | FOR_RANGE(int64_t, i, 0, input_bns_size) { FILE: oneflow/core/lazy/actor/generic_actor_context.cpp type oneflow (line 18) | namespace oneflow { function StreamContext (line 29) | StreamContext* GenericActorContext::stream_ctx() const { return stream... function TaskProto (line 31) | const TaskProto& GenericActorContext::task_proto() const { return task... FILE: oneflow/core/lazy/actor/generic_actor_context.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/lazy/actor/input_wise_actor.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/lazy/actor/input_wise_actor.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/lazy/actor/light_actor.cpp type oneflow (line 39) | namespace oneflow { type RegstType (line 43) | enum RegstType : int8_t { type ProducedRegstState (line 50) | struct ProducedRegstState { type ConsumedRegstState (line 55) | struct ConsumedRegstState { type RegstState (line 61) | struct RegstState { type KernelInfo (line 70) | struct KernelInfo { type DebugInfo (line 76) | struct DebugInfo { method DebugInfo (line 80) | DebugInfo() : actor_id(-1), op_name(""), act_cnt(-1) {} type ArrayBaseIndex (line 84) | struct ArrayBaseIndex { method ArrayBaseIndex (line 85) | ArrayBaseIndex() { std::memset(this, 0, sizeof(*this)); } method IndexType (line 87) | inline IndexType Size() const { return size; } method Reserve (line 89) | void Reserve(IndexType new_size) { CHECK_LE(new_size, max_size); } method IndexType (line 91) | inline IndexType Lookup(int64_t v) const { method Contains (line 99) | bool Contains(int64_t v) const { method IndexType (line 106) | IndexType Add(int64_t v) { method GetValues (line 114) | void GetValues(std::vector* values) const { type MapBaseIndex (line 124) | struct MapBaseIndex { method IndexType (line 125) | inline IndexType Size() const { return index_map.size(); } method Reserve (line 127) | void Reserve(IndexType size) { index_map.reserve(size); } method IndexType (line 129) | inline IndexType Lookup(int64_t v) { method Contains (line 135) | bool Contains(int64_t v) { return index_map.count(v) > 0; } method IndexType (line 137) | IndexType Add(int64_t v) { method GetValues (line 143) | void GetValues(std::vector* values) const { type ArrayBaseStateContainer (line 152) | struct ArrayBaseStateContainer { method ArrayBaseStateContainer (line 153) | ArrayBaseStateContainer() { std::memset(this, 0, sizeof(*this)); } method Resize (line 155) | void Resize(IndexType new_size) { method IndexType (line 160) | inline IndexType Size() const { return size; } type VectorBaseStateContainer (line 172) | struct VectorBaseStateContainer { method Resize (line 173) | void Resize(IndexType new_size) { vec.resize(new_size); } method IndexType (line 175) | inline IndexType Size() const { return static_cast(vec.si... function IsInplaceRegstDesc (line 182) | bool IsInplaceRegstDesc(const RegstDescProto& regst_desc) { function GetRegstDescCount (line 186) | size_t GetRegstDescCount(const TaskProto& task) { function GetConsumerCount (line 194) | size_t GetConsumerCount(const TaskProto& task) { function NeedExecKernelWhenInplace (line 202) | bool NeedExecKernelWhenInplace(const TaskProto& task) { function IsCUDAGraphSupported (line 228) | bool IsCUDAGraphSupported(const Kernel* kernel) { class LightActor (line 237) | class LightActor : public ActorBase, public KernelContext, public Acto... method LightActor (line 240) | explicit LightActor(ActorContext* actor_ctx) method Init (line 257) | void Init(const JobDesc* job_desc, ActorContext* actor_ctx) override { method ProcessMsg (line 346) | int ProcessMsg(const ActorMsg& msg) override { method InitBnInOp2Blob (line 371) | void InitBnInOp2Blob() { method InitActMsg (line 403) | void InitActMsg() { method ResetState (line 452) | inline void ResetState() { method HandleActorMsg (line 483) | inline void HandleActorMsg(const ActorMsg& msg) { method HandleEordMsg (line 495) | void HandleEordMsg(const ActorMsg& msg) { method HandleRegstMsg (line 505) | inline void HandleRegstMsg(const ActorMsg& msg) { method ActOnce (line 557) | inline void ActOnce() { method LaunchKernel (line 621) | inline void LaunchKernel() { method SendEORDMsg (line 648) | void SendEORDMsg() { method ActorContext (line 664) | ActorContext* GetActorContext() const override { return actor_ctx_; } method Blob (line 666) | Blob* BnInOp2Blob(const std::string& bn) const override { method set_state (line 688) | void set_state(std::shared_ptr state) override { method WillForward (line 693) | void WillForward(KernelContext* kernel_ctx, const Kernel* kernel) ov... method DidForward (line 700) | void DidForward(KernelContext* kernel_ctx, const Kernel* kernel) ove... method WillForwardHeader (line 708) | void WillForwardHeader(KernelContext* kernel_ctx, const Kernel* kern... method DidForwardHeader (line 715) | void DidForwardHeader(KernelContext* kernel_ctx, const Kernel* kerne... method WillForwardDataContent (line 722) | void WillForwardDataContent(KernelContext* kernel_ctx, const Kernel*... method DidForwardDataContent (line 729) | void DidForwardDataContent(KernelContext* kernel_ctx, const Kernel* ... function ActorBase (line 763) | ActorBase* DispatchNewLightActorDebug(ActorContext* actor_ctx) { function ActorBase (line 776) | ActorBase* DispatchNewLightActorDynamicAlloc(ActorContext* actor_ctx) { function ActorBase (line 789) | ActorBase* DispatchNewLightActorMaxSize(ActorContext* actor_ctx) { function ActorBase (line 811) | ActorBase* DispatchNewLightActorIndexType(ActorContext* actor_ctx) { function ActorBase (line 824) | ActorBase* DispatchNewLightActorInplace(ActorContext* actor_ctx) { function ActorBase (line 851) | ActorBase* NewLightActorWithKernel(ActorContext* actor_ctx) { function ActorBase (line 855) | ActorBase* NewLightActorWithoutKernel(ActorContext* actor_ctx) { function ActorBase (line 859) | ActorBase* TryNewLightActorWithoutInit(ActorContext* actor_ctx) { function TryNewLightActor (line 884) | std::unique_ptr TryNewLightActor(ActorContext* actor_ctx) { FILE: oneflow/core/lazy/actor/light_actor.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/lazy/actor/naive_actor.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/lazy/actor/naive_actor.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/lazy/actor/pack_actor.cpp type oneflow (line 20) | namespace oneflow { class PackActor (line 22) | class PackActor final : public Actor { method PackActor (line 25) | PackActor() = default; FILE: oneflow/core/lazy/actor/reentrant_lock_actor.cpp type oneflow (line 19) | namespace oneflow { class ReentrantLockActor (line 21) | class ReentrantLockActor final : public Actor { method ReentrantLockActor (line 24) | ReentrantLockActor() : reentrant_lock_status_(nullptr) {} method NormalProcessCustomizedEordMsg (line 35) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override {} method GetNaiveOrCustomizedConsumedRegstDescName (line 38) | std::pair> GetNaiveOrCustomizedC... FILE: oneflow/core/lazy/actor/register_slot.cpp type oneflow (line 18) | namespace oneflow { function Regst (line 70) | Regst* RegstSlot::Front(int64_t regst_desc_id) const { function Regst (line 78) | Regst* RegstSlot::SoleFront() const { function Regst (line 86) | Regst* RegstSlot::FirstFront() const { FILE: oneflow/core/lazy/actor/register_slot.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/lazy/actor/repeat_actor.cpp type oneflow (line 19) | namespace oneflow { class RepeatActor (line 21) | class RepeatActor final : public Actor { method RepeatActor (line 24) | RepeatActor() method GetNaiveOrCustomizedConsumedRegstDescName (line 34) | std::pair> GetNaiveOrCustomizedC... method GetNaiveOrCustomizedProducedRegstDescName (line 38) | std::pair> GetNaiveOrCustomizedP... method TakeOverInplaceConsumedAndProduced (line 42) | void TakeOverInplaceConsumedAndProduced( method IsCustomizedReadReady (line 49) | bool IsCustomizedReadReady() const override { method IsCustomizedWriteReady (line 52) | bool IsCustomizedWriteReady() const override { method NormalProcessCustomizedEordMsg (line 56) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override {} method IsCustomizedReadAlwaysUnReadyFromNow (line 57) | bool IsCustomizedReadAlwaysUnReadyFromNow() const override { function FOR_RANGE (line 92) | FOR_RANGE(int64_t, i, 0, in_time_shape.NumAxes()) { FILE: oneflow/core/lazy/actor/sink_actor.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/lazy/actor/sink_actor.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/lazy/actor/source_tick_actor.cpp type oneflow (line 20) | namespace oneflow { class SourceTickActor (line 22) | class SourceTickActor final : public Actor { method SourceTickActor (line 25) | SourceTickActor() = default; method GetNaiveOrCustomizedConsumedRegstDescName (line 31) | std::pair> GetNaiveOrCustomizedC... method IsCustomizedReadAlwaysUnReadyFromNow (line 36) | bool IsCustomizedReadAlwaysUnReadyFromNow() const override { return ... FILE: oneflow/core/lazy/actor/ssp_variable_proxy_actor.cpp type oneflow (line 19) | namespace oneflow { class SspVariableProxyActor (line 21) | class SspVariableProxyActor final : public Actor { method SspVariableProxyActor (line 24) | SspVariableProxyActor() = default; method GetNaiveOrCustomizedConsumedRegstDescName (line 28) | std::pair> GetNaiveOrCustomizedC... method GetNaiveOrCustomizedProducedRegstDescName (line 32) | std::pair> GetNaiveOrCustomizedP... method IsCustomizedReadReady (line 36) | bool IsCustomizedReadReady() const override { return consumed_var_rs... method IsCustomizedWriteReady (line 37) | bool IsCustomizedWriteReady() const override { method NormalProcessCustomizedEordMsg (line 45) | void NormalProcessCustomizedEordMsg(const ActorMsg&) override {} method IsCustomizedReadAlwaysUnReadyFromNow (line 46) | bool IsCustomizedReadAlwaysUnReadyFromNow() const override { method UpdtStateAsCustomizedProducedRegst (line 54) | void UpdtStateAsCustomizedProducedRegst(Regst* regst) override { method AsyncSendCustomizedProducedRegstMsgToConsumer (line 84) | void AsyncSendCustomizedProducedRegstMsgToConsumer() override { method AsyncSendCustomizedConsumedRegstMsgToProducer (line 107) | void AsyncSendCustomizedConsumedRegstMsgToProducer() override { method ForEachCurCustomizedReadableRegst (line 114) | void ForEachCurCustomizedReadableRegst(std::function& cons... method TakeOverRefRegst (line 170) | void TakeOverRefRegst(const PbMap& prod... method TakeOverValueRegst (line 184) | void TakeOverValueRegst(const PbMap& pr... method SyncRefRegstIntoValueRegst (line 196) | void SyncRefRegstIntoValueRegst(int64_t released_piece_id) { method CopyRefToValue (line 201) | void CopyRefToValue(Regst* value_regst) { method Regst (line 215) | Regst* GetRingBufferValueRegst(int64_t value_piece_id) const { method staleness (line 219) | size_t staleness() const { return value_regst_ring_buffer_.size(); } FILE: oneflow/core/lazy/actor/tick_actor.cpp type oneflow (line 18) | namespace oneflow { class TickActor (line 20) | class TickActor final : public NaiveActor { method TickActor (line 23) | TickActor() = default; method Act (line 27) | void Act() override {} FILE: oneflow/core/lazy/actor/unpack_actor.cpp type oneflow (line 20) | namespace oneflow { class UnpackActor (line 22) | class UnpackActor final : public Actor { method UnpackActor (line 25) | UnpackActor() = default; FILE: oneflow/core/lazy/actor/wait_and_send_ids_actor.cpp type oneflow (line 21) | namespace oneflow { class WaitAndSendIdsActor (line 23) | class WaitAndSendIdsActor final : public Actor { method WaitAndSendIdsActor (line 26) | WaitAndSendIdsActor() : wait_and_send_ids_status_(nullptr) {} method GetNaiveOrCustomizedConsumedRegstDescName (line 32) | std::pair> GetNaiveOrCustomizedC... method IsCustomizedReadAlwaysUnReadyFromNow (line 38) | bool IsCustomizedReadAlwaysUnReadyFromNow() const override { return ... FILE: oneflow/core/lazy/stream_context/common/generic_stream_context.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/lazy/stream_context/cpu/cpu_stream_context.cpp type oneflow (line 27) | namespace oneflow { class CpuStreamContext (line 29) | class CpuStreamContext : public StreamContext, public KernelObserverPr... method DeviceType (line 38) | DeviceType device_type() const override { return DeviceType::kCPU; } function KernelObserver (line 65) | KernelObserver* CpuStreamContext::GetKernelObserver() { return kernel_... FILE: oneflow/core/lazy/stream_context/cuda/cuda_stream_context.cpp type oneflow (line 31) | namespace oneflow { class CudaStreamContext (line 35) | class CudaStreamContext : public StreamContext, public KernelObserverP... method DeviceType (line 42) | DeviceType device_type() const override { return DeviceType::kCUDA; } function KernelObserver (line 102) | KernelObserver* CudaStreamContext::GetKernelObserver() { return kernel... FILE: oneflow/core/lazy/stream_context/include/generic_stream_context.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/lazy/stream_context/include/stream_context.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/memory/chunk_manager.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/memory/chunk_manager.h function class (line 27) | class ChunkMgr final { FILE: oneflow/core/memory/memory_allocator.cpp type oneflow (line 25) | namespace oneflow { function GetAllocationDevice (line 29) | std::shared_ptr GetAllocationDevice(const MemoryCase& mem_... function GetAllocationOptions (line 36) | ep::AllocationOptions GetAllocationOptions(const MemoryCase& mem_case) { function InitNonPODTypeBlobIfNeed (line 84) | void InitNonPODTypeBlobIfNeed(MemoryAllocator* allocator, Blob* blob_p... function InitNonPODTypeEagerBlobObjectIfNeed (line 100) | void InitNonPODTypeEagerBlobObjectIfNeed(MemoryAllocator* allocator, FILE: oneflow/core/memory/memory_allocator.h function namespace (line 22) | namespace oneflow { function final (line 61) | struct MemoryAllocatorImpl final { FILE: oneflow/core/memory/memory_case_util.cpp type oneflow (line 20) | namespace oneflow { type memory (line 22) | namespace memory { function EqualsIgnorePinnedDevice (line 24) | bool EqualsIgnorePinnedDevice(const MemoryCase& a, const MemoryCase&... function GetPinnedHostMemoryCase (line 30) | void GetPinnedHostMemoryCase(const MemoryCase& mem_case, MemoryCase*... function MemoryCase (line 39) | MemoryCase GetPinnedHostMemoryCase(const MemoryCase& mem_case) { function GetMemCaseId (line 62) | int64_t GetMemCaseId(const MemoryCase& mem_case) { function GetUniqueMemCaseId (line 79) | int64_t GetUniqueMemCaseId(int64_t machine_id, const MemoryCase& mem... function MakeMemCaseShared (line 86) | std::shared_ptr MakeMemCaseShared(const DeviceType devic... function MemoryCase (line 101) | MemoryCase MakeHostMemCase() { function IsHostMem (line 108) | bool IsHostMem(const MemoryCase& mem_case) { return mem_case.device_... FILE: oneflow/core/memory/memory_case_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/memory/memory_zone.cpp type oneflow (line 18) | namespace oneflow { function MemZoneId (line 35) | MemZoneId GetNodeCPUMemZoneId(MemZoneId::rank_t node_index) { function EncodeMemZoneIdToInt64 (line 39) | int64_t EncodeMemZoneIdToInt64(const MemZoneId& mem_zone_id) { function MemZoneId (line 46) | MemZoneId DecodeMemZoneIdFromInt64(int64_t mem_zone_id) { FILE: oneflow/core/memory/memory_zone.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ndarray/binary_func.h function BinaryFuncTrait (line 57) | BinaryFuncTrait final { function BinaryFuncNanSum (line 72) | BinaryFuncNanSum final { function BinaryFuncAdd (line 86) | BinaryFuncAdd final { function BinaryFuncSum (line 90) | BinaryFuncSum final { function BinaryFuncSub (line 96) | BinaryFuncSub final { function BinaryFuncMul (line 102) | BinaryFuncMul final { function final (line 106) | struct BinaryFuncMul final { function BinaryFuncProd (line 110) | BinaryFuncProd final { function BinaryFuncDiv (line 116) | BinaryFuncDiv final { function BinaryFuncFloorMod (line 122) | BinaryFuncFloorMod final { function final (line 136) | struct BinaryFuncFloorMod final { function BinaryFuncFMod (line 149) | BinaryFuncFMod final { function BinaryFuncPow (line 163) | BinaryFuncPow final { function final (line 174) | struct BinaryFuncPow final { function final (line 187) | struct BinaryFuncPow final { type BinaryFuncPow (line 195) | struct BinaryFuncPow function OF_DEVICE_FUNC (line 196) | static OF_DEVICE_FUNC double Invoke(const double x, const double y) { re... function final (line 200) | struct BinaryFuncPow final { function final (line 207) | struct BinaryFuncPow final { function BinaryFuncFloorDiv (line 219) | BinaryFuncFloorDiv final { function BinaryFuncMax (line 230) | BinaryFuncMax final { function BinaryFuncMin (line 236) | BinaryFuncMin final { function BinaryFuncEQ (line 242) | BinaryFuncEQ final { function BinaryFuncNE (line 248) | BinaryFuncNE final { function BinaryFuncGT (line 254) | BinaryFuncGT final { function BinaryFuncGE (line 260) | BinaryFuncGE final { function BinaryFuncLT (line 266) | BinaryFuncLT final { function BinaryFuncLE (line 272) | BinaryFuncLE final { function BinaryFuncAND (line 278) | BinaryFuncAND final { function BinaryFuncAll (line 282) | BinaryFuncAll final { function BinaryFuncOR (line 288) | BinaryFuncOR final { function BinaryFuncAny (line 292) | BinaryFuncAny final { function BinaryFuncXOR (line 298) | BinaryFuncXOR final { function BinaryFuncBitwiseAnd (line 304) | BinaryFuncBitwiseAnd final { function BinaryFuncBitwiseOr (line 310) | BinaryFuncBitwiseOr final { function BinaryFuncBitwiseXor (line 316) | BinaryFuncBitwiseXor final { type BinaryFuncAdd (line 323) | struct BinaryFuncAdd function half (line 324) | half Invoke(const half x, const half y) { return __hadd(x, y); } function final (line 328) | struct BinaryFuncNanSum final { function final (line 336) | struct BinaryFuncSub final { function final (line 347) | struct BinaryFuncMul final { function final (line 358) | struct BinaryFuncDiv final { function final (line 369) | struct BinaryFuncMax final { function final (line 380) | struct BinaryFuncMin final { type BinaryFuncAdd (line 391) | struct BinaryFuncAdd function cuComplex (line 392) | cuComplex Invoke(const cuComplex x, const cuComplex y) { type BinaryFuncSub (line 398) | struct BinaryFuncSub function cuComplex (line 399) | cuComplex Invoke(const cuComplex x, const cuComplex y) { function final (line 405) | struct BinaryFuncMul final { type BinaryFuncAdd (line 412) | struct BinaryFuncAdd function cuDoubleComplex (line 413) | cuDoubleComplex Invoke(const cuDoubleComplex x, function final (line 420) | struct BinaryFuncSub final { FILE: oneflow/core/ndarray/cpu_concat_var_ndarray.h function namespace (line 23) | namespace oneflow { function ALWAYS_INLINE (line 60) | ALWAYS_INLINE void GetMutPtrAndMinContiguousSize(int32_t var_index, int6... function XpuShape (line 77) | XpuShape CalcConcatenatedShape(const std::vector... function CheckInputShape (line 87) | void CheckInputShape(const std::vector>& var_nda... FILE: oneflow/core/ndarray/cpu_concat_var_ndarray_test.cpp type oneflow (line 19) | namespace oneflow { type test (line 21) | namespace test { function TEST (line 23) | TEST(CpuConcatVarNdarray, two_elem_concat) { function TEST (line 35) | TEST(CpuConcatVarNdarray, two_elem_concat_assign) { FILE: oneflow/core/ndarray/cpu_ndarray.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ndarray/cpu_ndarray_builder.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ndarray/cpu_ndarray_copy.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ndarray/cpu_slice_var_ndarray.h function namespace (line 23) | namespace oneflow { function ALWAYS_INLINE (line 80) | ALWAYS_INLINE const Slice& slice(int32_t dim) const { return slices_[dim... function ALWAYS_INLINE (line 81) | ALWAYS_INLINE void GetMutPtrAndMinContiguousSize(int64_t offset, int64_t... function Shape (line 95) | static Shape BoundedSlices2Shape(const std::array& bou... FILE: oneflow/core/ndarray/cpu_slice_var_ndarray_test.cpp type oneflow (line 19) | namespace oneflow { type test (line 21) | namespace test { function TEST (line 23) | TEST(CpuSliceVarNdarray, one_elem_assign) { function TEST (line 33) | TEST(CpuSliceVarNdarray, one_elem_assign_slice_on_slice) { FILE: oneflow/core/ndarray/cpu_var_ndarray.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ndarray/cpu_var_ndarray_test.cpp type oneflow (line 19) | namespace oneflow { type test (line 21) | namespace test { function TEST (line 23) | TEST(CpuVarNdarray, one_elem_assign) { FILE: oneflow/core/ndarray/ndarray_apply_binary.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_apply_binary_core.cpp type oneflow (line 19) | namespace oneflow { type NdarrayApplyBinaryCoreWrapper (line 22) | struct NdarrayApplyBinaryCoreWrapper... method Apply (line 23) | static void Apply(ep::Stream* stream, method InplaceApply (line 28) | static void InplaceApply(ep::Stream* stream, const XpuVarNdarray& y, FILE: oneflow/core/ndarray/ndarray_apply_binary_core.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_apply_broadcast_binary.h function namespace (line 23) | namespace oneflow { function Apply (line 117) | static void Apply(ep::Stream* stream, function InplaceApply (line 125) | static void InplaceApply(ep::Stream* stream, const XpuVarNdarray& y, FILE: oneflow/core/ndarray/ndarray_apply_broadcast_binary_core.cpp type oneflow (line 18) | namespace oneflow { type NdarrayApplyBroadcastBinaryCoreWrapper (line 21) | struct NdarrayApplyBroadcastBinaryCoreWrapper (line 30) | struct NdarrayApplyBroadcastInplaceBinaryCoreWrapper& y, FILE: oneflow/core/ndarray/ndarray_apply_broadcast_binary_core.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_apply_broadcast_unary.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_apply_broadcast_unary_core.cpp type oneflow (line 18) | namespace oneflow { type NdarrayApplyBroadcastUnaryCoreWrapper (line 21) | struct NdarrayApplyBroadcastUnaryCoreWrapper& y, FILE: oneflow/core/ndarray/ndarray_apply_broadcast_unary_core.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_apply_unary.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_apply_unary_core.cpp type oneflow (line 19) | namespace oneflow { type NdarrayApplyUnaryCoreWrapper (line 22) | struct NdarrayApplyUnaryCoreWrapper f... method InplaceApply (line 23) | static void InplaceApply(ep::Stream* stream, const XpuVarNdarray&... FILE: oneflow/core/ndarray/ndarray_apply_unary_core.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_assign_core.cpp type oneflow (line 19) | namespace oneflow { type NdarrayAssignCoreWrapper (line 22) | struct NdarrayAssignCoreWrapper final { method Assign (line 23) | static void Assign(ep::Stream* stream, const XpuVarNdarray& y, method Assign (line 27) | static void Assign(ep::Stream* stream, const XpuVarNdarray& y, FILE: oneflow/core/ndarray/ndarray_assign_core.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_reduce.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ndarray/ndarray_reduce_impl.cpp type oneflow (line 21) | namespace oneflow { type NdarrayReduceCoreWrapper (line 57) | struct NdarrayReduceCoreWrapper final { function final (line 84) | struct UnaryFuncExp final { type UnaryFuncNegative (line 101) | struct UnaryFuncNegative function half (line 102) | const half Invoke(const half x) { function final (line 111) | struct UnaryFuncExp final { function final (line 118) | struct UnaryFuncNegative final { type UnaryFuncExp (line 124) | struct UnaryFuncExp function cuComplex (line 125) | const cuComplex Invoke(const cuComplex x) { function final (line 131) | struct UnaryFuncNegative final { function final (line 137) | struct UnaryFuncExp final { FILE: oneflow/core/ndarray/xpu_binary_func_ndarray.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_broadcast_ndarray.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_ndarray_assign.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_ndarray_base.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_reduced_ndarray.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_reshape_ndarray.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_shape.cpp type oneflow (line 18) | namespace oneflow { function SimplifyBroadcastShapes (line 85) | void SimplifyBroadcastShapes(const XpuShape& y, const XpuShape& b, Dim... function SimplifyBroadcastShapes (line 91) | void SimplifyBroadcastShapes(const XpuShape& y, const XpuShape& a, con... FILE: oneflow/core/ndarray/xpu_shape.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_transpose_ndarray.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_unary_func_ndarray.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_var_ndarray.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/ndarray/xpu_var_ndarray_builder.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/acc_tick_op.cpp type oneflow (line 18) | namespace oneflow { function InferBlobDescs (line 22) | Maybe InferBlobDescs(const std::function InferBlobDescs(const std::function InferLogicalOutBlobDescs( function LogicalBlobId (line 49) | LogicalBlobId BoxingIdentityOp::lbi4ibn(const std::string& input_bn) c... function LogicalBlobId (line 53) | LogicalBlobId BoxingIdentityOp::lbi4obn(const std::string& output_bn) ... FILE: oneflow/core/operator/boxing_op.cpp type oneflow (line 19) | namespace oneflow { function EraseEmptyBnInVec (line 23) | void EraseEmptyBnInVec(const std::function InferLogicalOutBlobDescs( function LogicalBlobId (line 47) | LogicalBlobId BoxingZerosOp::lbi4ibn(const std::string& input_bn) const { function LogicalBlobId (line 51) | LogicalBlobId BoxingZerosOp::lbi4obn(const std::string& output_bn) con... FILE: oneflow/core/operator/broadcast_to_compatible_with_op.cpp type oneflow (line 19) | namespace oneflow { function GetBroadcastShape (line 23) | Maybe GetBroadcastShape(const Shape& a_shape, const Shape& b_sha... function InferBlobDescs (line 38) | Maybe InferBlobDescs(const OperatorConf& op_conf, class BroadcastToCompatibleWithOp (line 55) | class BroadcastToCompatibleWithOp final : public Operator { method BroadcastToCompatibleWithOp (line 58) | BroadcastToCompatibleWithOp() = default; method InitFromOpConf (line 61) | Maybe InitFromOpConf() override { method InferLogicalOutBlobDescs (line 69) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 75) | Maybe InferOutBlobDescs( method VirtualGenKernelConf (line 82) | void VirtualGenKernelConf(std::function GetSbpSignatures( FILE: oneflow/core/operator/callback_notify_op.cpp type oneflow (line 19) | namespace oneflow { function InferBlobDescs (line 29) | Maybe InferBlobDescs(const std::function InferBlobDescs(const Operator& op, FILE: oneflow/core/operator/case_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/collective_boxing_ops.cpp type oneflow (line 19) | namespace oneflow { class CollectiveBoxingGenericOp (line 23) | class CollectiveBoxingGenericOp : public Operator { method CollectiveBoxingGenericOp (line 26) | CollectiveBoxingGenericOp() = default; method InitFromOpConf (line 30) | Maybe InitFromOpConf() override { method LogicalBlobId (line 38) | LogicalBlobId lbi4ibn(const std::string& input_bn) const override { method LogicalBlobId (line 42) | LogicalBlobId lbi4obn(const std::string& output_bn) const override { method InferLogicalOutBlobDescs (line 46) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 51) | Maybe InferOutBlobDescs( FILE: oneflow/core/operator/collective_boxing_pack_op.cpp type oneflow (line 20) | namespace oneflow { class CollectiveBoxingPackOp (line 22) | class CollectiveBoxingPackOp : public Operator { method CollectiveBoxingPackOp (line 25) | CollectiveBoxingPackOp() = default; method InferLogicalOutBlobDescs (line 30) | Maybe InferLogicalOutBlobDescs( function LogicalBlobId (line 50) | LogicalBlobId CollectiveBoxingPackOp::lbi4ibn(const std::string& input... function LogicalBlobId (line 54) | LogicalBlobId CollectiveBoxingPackOp::lbi4obn(const std::string& outpu... FILE: oneflow/core/operator/collective_boxing_unpack_op.cpp type oneflow (line 20) | namespace oneflow { class CollectiveBoxingUnpackOp (line 22) | class CollectiveBoxingUnpackOp : public Operator { method CollectiveBoxingUnpackOp (line 25) | CollectiveBoxingUnpackOp() = default; method InferLogicalOutBlobDescs (line 30) | Maybe InferLogicalOutBlobDescs( function LogicalBlobId (line 50) | LogicalBlobId CollectiveBoxingUnpackOp::lbi4ibn(const std::string& inp... function LogicalBlobId (line 54) | LogicalBlobId CollectiveBoxingUnpackOp::lbi4obn(const std::string& out... FILE: oneflow/core/operator/constant_like_op.cpp type oneflow (line 18) | namespace oneflow { function InferBlobDescs (line 22) | Maybe InferBlobDescs(const OperatorConf& op_conf, class ConstantLikeOp (line 33) | class ConstantLikeOp final : public Operator { method ConstantLikeOp (line 36) | ConstantLikeOp() = default; method InitFromOpConf (line 39) | Maybe InitFromOpConf() override { method InferLogicalOutBlobDescs (line 46) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 52) | Maybe InferOutBlobDescs( method GetSbpSignatures (line 59) | Maybe GetSbpSignatures( FILE: oneflow/core/operator/copy_comm_net_op.cpp type oneflow (line 18) | namespace oneflow { function LogicalBlobId (line 26) | LogicalBlobId CopyCommNetOp::lbi4obn(const std::string& output_bn) con... function LogicalBlobId (line 30) | LogicalBlobId CopyCommNetOp::lbi4ibn(const std::string& input_bn) const { FILE: oneflow/core/operator/copy_comm_net_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/critical_section_callback_tick_op.cpp type oneflow (line 19) | namespace oneflow { function InferBlobDescs (line 23) | Maybe InferBlobDescs(const std::function InferBlobDescs(const std::function InferBlobDescs(const std::function InferBlobDescs(const std::function InitFromOpConf() override { method InferLogicalOutBlobDescs (line 29) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 58) | Maybe InferOutBlobDescs( method GetSbpSignatures (line 105) | Maybe GetSbpSignatures( class DynamicReshapeLikeOp (line 118) | class DynamicReshapeLikeOp final : public Operator { method InitFromOpConf (line 120) | Maybe InitFromOpConf() override { method InferLogicalOutBlobDescs (line 127) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 135) | Maybe InferOutBlobDescs( method GetSbpSignatures (line 145) | Maybe GetSbpSignatures( FILE: oneflow/core/operator/esac_op.cpp type oneflow (line 19) | namespace oneflow { function InferBlobDescs (line 29) | Maybe InferBlobDescs(const OperatorConf& op_conf, FILE: oneflow/core/operator/esac_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/identity_op.cpp type oneflow (line 21) | namespace oneflow { function InferBlobDescs (line 25) | Maybe InferBlobDescs(const std::function InitFromOpConf() override { method InferLogicalOutBlobDescs (line 44) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 49) | Maybe InferOutBlobDescs( method GetSbpSignatures (line 56) | Maybe GetSbpSignatures( type IdentityOp (line 67) | struct IdentityOp {} type CopyOp (line 70) | struct CopyOp {} class LocalCastOp (line 73) | class LocalCastOp : public Operator { method LocalCastOp (line 76) | LocalCastOp() = default; method InitFromOpConf (line 79) | Maybe InitFromOpConf() override { method InferLogicalOutBlobDescs (line 84) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 89) | Maybe InferOutBlobDescs( class CastToLocalOp (line 100) | class CastToLocalOp : public LocalCastOp { method CastToLocalOp (line 103) | CastToLocalOp() = default; method InferLogicalOutBlobDescs (line 107) | Maybe InferLogicalOutBlobDescs( method InferSbpSignature (line 127) | Maybe InferSbpSignature( method InferLocalSignature (line 144) | Maybe InferLocalSignature( class CastFromLocalOp (line 163) | class CastFromLocalOp : public LocalCastOp { method CastFromLocalOp (line 166) | CastFromLocalOp() = default; method InferLogicalOutBlobDescs (line 170) | Maybe InferLogicalOutBlobDescs( method InferSbpSignature (line 186) | Maybe InferSbpSignature( method InferLocalSignature (line 201) | Maybe InferLocalSignature( FILE: oneflow/core/operator/image_decoder_random_crop_resize_op.cpp type oneflow (line 25) | namespace oneflow { function InferBlobDescs (line 29) | Maybe InferBlobDescs(const OperatorConf& op_conf, class ImageDecoderRandomCropResizeOp (line 47) | class ImageDecoderRandomCropResizeOp final : public Operator { method ImageDecoderRandomCropResizeOp (line 50) | ImageDecoderRandomCropResizeOp() = default; method InitFromOpConf (line 54) | Maybe InitFromOpConf() override { method InferLogicalOutBlobDescs (line 61) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 67) | Maybe InferOutBlobDescs( method InferInternalBlobDescs (line 73) | Maybe InferInternalBlobDescs( method GetSbpSignatures (line 84) | Maybe GetSbpSignatures( method VirtualGenKernelConf (line 95) | void VirtualGenKernelConf(std::function InferBlobParallelDesc() override { FILE: oneflow/core/operator/input_op.cpp type oneflow (line 21) | namespace oneflow { function InferInputOpNdSbpSignature (line 24) | Maybe InferInputOpNdSbpSignature(NdSbpSignature* nd_sbp_signature, FILE: oneflow/core/operator/input_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/interface_op_util.cpp type oneflow (line 19) | namespace oneflow { function CheckShape (line 23) | void CheckShape(const Shape& shape) { function GetSbpSignature (line 27) | Maybe GetSbpSignature(const InterfaceBlobConf& blob_conf, const ... function FOR_RANGE (line 118) | FOR_RANGE(int64_t, i, 0, num_axes) { nd_sbp->add_sbp_parallel()->mutab... FILE: oneflow/core/operator/interface_op_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/operator/learning_rate_schedule_op.cpp type oneflow (line 18) | namespace oneflow { class LearningRateScheduleOp (line 20) | class LearningRateScheduleOp final : public Operator { method LearningRateScheduleOp (line 23) | LearningRateScheduleOp() = default; function InferBlobDescs (line 49) | Maybe InferBlobDescs(const std::function InferLogicalOutBlobDescs( function LogicalBlobId (line 99) | LogicalBlobId NcclSendRecvBoxingOp::lbi4ibn(const std::string& input_b... function LogicalBlobId (line 103) | LogicalBlobId NcclSendRecvBoxingOp::lbi4obn(const std::string& output_... FILE: oneflow/core/operator/nccl_send_recv_boxing_op_util.cpp type oneflow (line 19) | namespace oneflow { function DfsTraverseRanks4NdSbp (line 25) | void DfsTraverseRanks4NdSbp( function NdSbpNoPartialParallel (line 51) | bool NdSbpNoPartialParallel(const NdSbp& nd_sbp) { function GetMappedParallelId (line 61) | int64_t GetMappedParallelId(const int64_t from_parallel_id, const Para... function GetRankSendRecvIntersection (line 72) | void GetRankSendRecvIntersection(int64_t parallel_id, const ParallelDe... FILE: oneflow/core/operator/nccl_send_recv_boxing_op_util.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/operator/op_conf_symbol.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/operator/op_conf_symbol.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/operator/op_conf_util.h function namespace (line 21) | namespace std { FILE: oneflow/core/operator/op_infer_cache.h function namespace (line 25) | namespace oneflow { type hash (line 53) | struct hash function const (line 54) | size_t operator()(const oneflow::OpInferCacheKey& op_infer_cache_key) co... FILE: oneflow/core/operator/operator.cpp type oneflow (line 37) | namespace oneflow { function DataType (line 41) | DataType GetDataTypeFromBnInOpVec( function CheckAndConstructOp (line 51) | Maybe CheckAndConstructOp(std::shared_ptr FillLogicalBlobDesc( function FillLogicalBlobDesc (line 177) | Maybe FillLogicalBlobDesc( function FillLogicalBlobDesc (line 191) | Maybe FillLogicalBlobDesc( function GetLogicalBlobDesc (line 205) | Maybe GetLogicalBlobDesc( function FillLogicalBlobDescSignature (line 213) | Maybe FillLogicalBlobDescSignature( function SupportNonContiguous (line 225) | Maybe SupportNonContiguous(const Operator* op) { function DebugString4LocalHint (line 1001) | std::string DebugString4LocalHint( function BlobLastUsedSignature (line 1058) | BlobLastUsedSignature* Operator::mut_blob_last_used_signature() { function BlobBackwardUsedSignature (line 1063) | BlobBackwardUsedSignature* Operator::mut_blob_backward_used_signature() { function OptLocalParallel (line 1096) | OptLocalParallel* Operator::MutOptLocalParallel(const std::string& bn_... function HasBlobDescWithField (line 1104) | bool HasBlobDescWithField(std::function GenUnRepeatedBn(const std::string& bn) { function IsCpuOnly (line 1295) | bool IsCpuOnly(const std::string& user_op_type_name) { function IsCpuOnly (line 1301) | bool IsCpuOnly(const OperatorConf& op_conf) { function ConstructOp (line 1311) | Maybe ConstructOp(const OperatorConf& op_conf, DeviceType de... function ConstructOp (line 1318) | Maybe ConstructOp(const OperatorConf& op_conf) { function LogicalBlobId (line 1418) | LogicalBlobId GenLogicalBlobId(const std::string& lbn) { function GetSbpParallelInLbnOrNothing (line 1431) | Maybe GetSbpParallelInLbnOrNothing(const std::string& lbn, SbpPa... function ParseDisableBoxingFlag (line 1451) | Maybe ParseDisableBoxingFlag(const std::string& lbn_with_hint, b... function GetInputLbnInOpCustomizedConf (line 1461) | std::string GetInputLbnInOpCustomizedConf(const OperatorConf& op_conf, function ReplaceInputLbnInOpTypeConf (line 1481) | std::string ReplaceInputLbnInOpTypeConf(PbMessage* msg, const std::str... function ReplaceInputLbnInOpCustomizedConf (line 1499) | std::string ReplaceInputLbnInOpCustomizedConf(OperatorConf* op_conf, function InferOpOutSbpParallel (line 1512) | Maybe InferOpOutSbpParallel( function InferLocalSignature (line 1533) | Maybe InferLocalSignature(Operator* op, const OpNodeSignature& u... function CheckOpInputSignature (line 1552) | Maybe CheckOpInputSignature(const Operator& op, const OpNodeSign... function ConstructAndInferOp (line 1575) | Maybe ConstructAndInferOp(const OperatorConf& op_conf, function Get1dHierarchyPhysicalShape (line 1605) | Maybe Get1dHierarchyPhysicalShape(const Shape& logical_shape, function GetNdHierarchyPhysicalShape (line 1625) | Maybe GetNdHierarchyPhysicalShape(const Shape& logical_shape, c... function GetPhysicalShape (line 1654) | Maybe GetPhysicalShape(const Shape& logical_shape, const NdSbp&... function GetPhysicalShape (line 1667) | Maybe GetPhysicalShape(const Shape& logical_shape, const NdSbp&... function GetLogicalShape (line 1673) | Maybe GetLogicalShape(const Shape& physical_shape, const NdSbp&... FILE: oneflow/core/operator/operator.h function namespace (line 35) | namespace oneflow { function final (line 358) | struct RuntimeRegstNum4OpSameOutputBlob final { function final (line 399) | struct IsInterfaceOpConf4OpTypeCase final {} function final (line 405) | struct DisableInputBoxingGroup final {} function final (line 411) | struct IsTickTockOpTypeCase final {} function OpBlobArg (line 420) | inline OpBlobArg GenOpBlobArg(const std::string& op_name, const std::str... function std (line 429) | inline std::string GenLogicalBlobName(const std::string& op_name, const ... function std (line 433) | inline std::string GenLogicalBlobName(const LogicalBlobId& lbi) { function namespace (line 467) | namespace std { FILE: oneflow/core/operator/operator_util.cpp type oneflow (line 19) | namespace oneflow { function DhwOffset (line 21) | size_t DhwOffset(const std::string& data_format) { function Get3DVecInOpConf (line 31) | std::vector Get3DVecInOpConf(const PbRf& field_vals,... function GetInDim (line 45) | int64_t GetInDim(const ShapeView& shape, const std::string& data_forma... function GetWindowedOutputSize (line 63) | void GetWindowedOutputSize(int64_t input_size, int32_t filter_size, in... function GetWindowedOutputSize (line 102) | void GetWindowedOutputSize(int64_t input_size, int32_t filter_size, in... function GetWindowedOutputSize (line 129) | void GetWindowedOutputSize(int64_t input_size, int32_t filter_size, in... function GetWindowedOutputSize (line 136) | void GetWindowedOutputSize(int64_t input_size, int32_t filter_size, in... function Get3DOutputSize (line 143) | void Get3DOutputSize(const DimVector& in, const std::vector& ... function Get3DOutputSize (line 149) | void Get3DOutputSize(const DimVector& in, const std::vector& ... function Get3DOutputSize (line 157) | void Get3DOutputSize(const DimVector& in, const std::vector& ... function Get3DOutputSize (line 186) | void Get3DOutputSize(const DimVector& in, const std::vector& ... function GetConvOutAndPad (line 206) | void GetConvOutAndPad(const ShapeView& in_blob_shape, const PbMessage&... function GetConvOutAndPad (line 226) | void GetConvOutAndPad(const ShapeView& in_blob_shape, const user_op::U... FILE: oneflow/core/operator/operator_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/operator/output_op.cpp type oneflow (line 21) | namespace oneflow { function InferOutputOpNdSbpSignature (line 24) | Maybe InferOutputOpNdSbpSignature(NdSbpSignature* nd_sbp_signature, FILE: oneflow/core/operator/output_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/reduce_sbp_util.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/operator/reduce_sbp_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/operator/reentrant_lock_op.cpp type oneflow (line 19) | namespace oneflow { function InferBlobDescs (line 30) | Maybe InferBlobDescs(const std::function InferBlobDescs(const std::function GetInclusiveAxes(const ShapeElemCntOpConf& conf, int3... function InferBlobDescs (line 60) | Maybe InferBlobDescs(const OperatorConf& op_conf, function FOR_RANGE (line 97) | FOR_RANGE(int64_t, i, 0, num_axes) { FILE: oneflow/core/operator/shape_elem_cnt_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/sink_tick_op.cpp type oneflow (line 19) | namespace oneflow { function InferBlobDescs (line 30) | Maybe InferBlobDescs(const std::function InferLogicalOutBlobDescs( class SliceBoxingCopyOp (line 47) | class SliceBoxingCopyOp final : public SliceBoxingOp { method SliceBoxingCopyOp (line 50) | SliceBoxingCopyOp() = default; method SliceBoxingConf (line 54) | const SliceBoxingConf& GetCustomizedBoxingConf() const override { class SliceBoxingAddOp (line 60) | class SliceBoxingAddOp final : public SliceBoxingOp { method SliceBoxingAddOp (line 63) | SliceBoxingAddOp() = default; method SliceBoxingConf (line 67) | const SliceBoxingConf& GetCustomizedBoxingConf() const override { function LogicalBlobId (line 84) | LogicalBlobId SliceBoxingOp::lbi4ibn(const std::string& input_bn) const { function LogicalBlobId (line 88) | LogicalBlobId SliceBoxingOp::lbi4obn(const std::string& output_bn) con... function FOR_RANGE (line 100) | FOR_RANGE(int64_t, i, 1, input_bns().size()) { function FOR_RANGE (line 104) | FOR_RANGE(int64_t, i, 0, input_bns().size()) { FILE: oneflow/core/operator/source_tick_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/operator/source_tick_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/src_subset_tick_op.cpp type oneflow (line 20) | namespace oneflow { class SrcSubsetTickOp (line 22) | class SrcSubsetTickOp final : public Operator { method SrcSubsetTickOp (line 25) | SrcSubsetTickOp() = default; function InferBlobDescs (line 49) | Maybe InferBlobDescs(const std::function InferBlobDescs(const OperatorConf& op_conf, class SyncDynamicResizeOp (line 38) | class SyncDynamicResizeOp : public Operator { method SyncDynamicResizeOp (line 41) | SyncDynamicResizeOp() = default; method InitFromOpConf (line 44) | Maybe InitFromOpConf() override { method InferLogicalOutBlobDescs (line 51) | Maybe InferLogicalOutBlobDescs( method InferOutBlobDescs (line 57) | Maybe InferOutBlobDescs( method GetSbpSignatures (line 63) | Maybe GetSbpSignatures( method VirtualGenKernelConf (line 69) | void VirtualGenKernelConf(std::function InferBlobDescs(const std::functioninputs(); } method ArgVec (line 299) | const ArgVec& outputs() const override { return op_->outputs(); } method ParallelContext (line 300) | const ParallelContext& parallel_ctx() const override { return *paral... method ParallelDesc (line 301) | const ParallelDesc& parallel_desc() const override { method JobDesc (line 304) | const JobDesc* job_desc() const override { method SbpParallel (line 309) | const SbpParallel& SbpParallel4ArgNameAndIndex(const std::string& ar... method NdSbp (line 319) | const NdSbp& NdSbp4ArgNameAndIndex(const std::string& arg_name, int3... method parallel_num (line 327) | int64_t parallel_num() const override { method has_input (line 337) | bool has_input(const std::string& arg_name, int32_t index) const ove... method has_output (line 340) | bool has_output(const std::string& arg_name, int32_t index) const ov... method input_size (line 343) | int32_t input_size(const std::string& arg_name) const override { method output_size (line 346) | int32_t output_size(const std::string& arg_name) const override { class UserOpSbpContext (line 368) | class UserOpSbpContext : public user_op::SbpContext { method UserOpSbpContext (line 372) | UserOpSbpContext(const UserOp* op, SbpSignatureList* sbp_sig_list, method ArgVec (line 394) | const ArgVec& inputs() const override { return op_->inputs(); } method ArgVec (line 395) | const ArgVec& outputs() const override { return op_->outputs(); } method NewBuilder (line 398) | user_op::UserOpSbpSignatureBuilder NewBuilder() override { method DeviceType (line 402) | DeviceType device_type() const override { return op_->device_type(); } method parallel_num (line 404) | int64_t parallel_num() const override { method hierarchy_value (line 408) | int64_t hierarchy_value() const override { return hierarchy_value_; } class UserOpInferSbpSignatureFnContext (line 417) | class UserOpInferSbpSignatureFnContext : public user_op::InferSbpSigna... method UserOpInferSbpSignatureFnContext (line 421) | UserOpInferSbpSignatureFnContext( method ArgVec (line 448) | const ArgVec& inputs() const override { return op_->inputs(); } method ArgVec (line 449) | const ArgVec& outputs() const override { return op_->outputs(); } method SbpSignature (line 450) | SbpSignature* mutable_sbp_signature() override { return signature_; } method SbpSignature (line 451) | const SbpSignature& sbp_signature_conf() const override { return sbp... method SbpParallel (line 453) | const SbpParallel& SbpParallelHint4InputArgNameAndIndex(const std::s... method DeviceType (line 463) | DeviceType device_type() const override { return op_->device_type(); } method parallel_num (line 465) | int64_t parallel_num() const override { class UserOpInferOutputBlobTimeShapeFnContext (line 478) | class UserOpInferOutputBlobTimeShapeFnContext : public user_op::InferO... method UserOpInferOutputBlobTimeShapeFnContext (line 480) | UserOpInferOutputBlobTimeShapeFnContext( method Shape (line 495) | const Shape& TimeShape4InputArgNameAndIndex(const std::string& arg_n... method Shape (line 501) | Shape* mut_output_blob_time_shape() override { return output_blob_ti... class UserOpInferNdSbpFnContext (line 509) | class UserOpInferNdSbpFnContext : public user_op::InferNdSbpFnContext { method UserOpInferNdSbpFnContext (line 512) | UserOpInferNdSbpFnContext( method NdSbpSignature (line 541) | const NdSbpSignature& nd_sbp_constraints() const override { return n... method NdSbp (line 543) | NdSbp* NdSbp4ArgNameAndIndex(const std::string& arg_name, int32_t in... method NdSbp (line 547) | const NdSbp& NdSbpHint4InputArgNameAndIndex(const std::string& arg_n... method parallel_num (line 555) | int64_t parallel_num() const override { method Shape (line 559) | const Shape& parallel_hierarchy() override { method ArgVec (line 563) | const ArgVec& inputs() const override { return op_->inputs(); } method ArgVec (line 564) | const ArgVec& outputs() const override { return op_->outputs(); } class UserOpComputeComplexityFnContext (line 576) | class UserOpComputeComplexityFnContext : public user_op::ComputeComple... method UserOpComputeComplexityFnContext (line 580) | UserOpComputeComplexityFnContext( method Shape (line 610) | const Shape& Shape4ArgNameAndIndex(const std::string& arg_name, int3... method DataType (line 618) | DataType Dtype4ArgNameAndIndex(const std::string& arg_name, int32_t ... method IsDynamic4ArgNameAndIndex (line 623) | bool IsDynamic4ArgNameAndIndex(const std::string& arg_name, int32_t ... method NdSbp (line 629) | const NdSbp NdSbp4ArgNameAndIndex(const std::string& arg_name, int32... method ArgVec (line 636) | const ArgVec& inputs() const override { return inputs_; } method ArgVec (line 637) | const ArgVec& outputs() const override { return outputs_; } method ParallelDesc (line 638) | const ParallelDesc& parallel_desc() const override { return parallel... method NdSbpSignature (line 639) | const NdSbpSignature* GetNdSbpSignature() const override { return sb... class UserOpGetNdSbpSignatureListContext (line 649) | class UserOpGetNdSbpSignatureListContext : public user_op::GetNdSbpSig... method UserOpGetNdSbpSignatureListContext (line 651) | UserOpGetNdSbpSignatureListContext( method AddNdSbpSignature (line 664) | void AddNdSbpSignature(NdSbpSignature& nd_sbp_sig) override { method Shape (line 668) | const Shape& parallel_hierarchy() override { method Shape (line 672) | const Shape& BlobShape4InputArgNameAndIndex(const std::string& arg_n... function LogicalBlobId (line 880) | LogicalBlobId UserOp::lbi4ibn(const std::string& input_bn) const { function LogicalBlobId (line 885) | LogicalBlobId UserOp::lbi4obn(const std::string& output_bn) const { function IgnoreInferNdSbpFnWhenFlatHierarchy (line 993) | bool IgnoreInferNdSbpFnWhenFlatHierarchy(const std::string& op_type_na... FILE: oneflow/core/operator/user_op.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/operator/variable_op.cpp type oneflow (line 20) | namespace oneflow { function ParseNdSbpFromConf (line 24) | Maybe ParseNdSbpFromConf(const VariableOpConf& conf, const Paral... FILE: oneflow/core/operator/variable_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/operator/wait_and_send_ids_op.cpp type oneflow (line 19) | namespace oneflow { function InferBlobDescs (line 29) | Maybe InferBlobDescs(const OperatorConf& op_conf, FILE: oneflow/core/operator/wait_and_send_ids_op.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/persistence/binary_in_stream.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/persistence/binary_in_stream_with_local_copy.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/persistence/binary_in_stream_with_local_copy.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/persistence/binary_in_stream_without_local_copy.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/persistence/binary_in_stream_without_local_copy.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/persistence/file_system.cpp type oneflow (line 25) | namespace oneflow { type fs (line 27) | namespace fs { function CreateLocalFS (line 127) | void CreateLocalFS(std::unique_ptr& fs) { function CreateHadoopFS (line 135) | void CreateHadoopFS(std::unique_ptr& fs, const std::st... function CreateFileSystemFromEnv (line 139) | void CreateFileSystemFromEnv(std::unique_ptr& fs, cons... FILE: oneflow/core/persistence/file_system.h function class (line 27) | class RandomAccessFile { FILE: oneflow/core/persistence/file_system_test.cpp type oneflow (line 21) | namespace oneflow { type fs (line 23) | namespace fs { function TestFileOperation (line 25) | void TestFileOperation(FileSystem* file_system) { function TestDirOperation (line 61) | void TestDirOperation(FileSystem* file_system) { function TestMultiThreadsDirOperation (line 89) | void TestMultiThreadsDirOperation(FileSystem* file_system) { function TestFileSystem (line 102) | void TestFileSystem(FileSystem* file_system) { function TEST (line 110) | TEST(file_system, write_and_read) { FILE: oneflow/core/persistence/hadoop/hadoop_file_system.cpp type oneflow (line 32) | namespace oneflow { type fs (line 34) | namespace fs { type internal (line 36) | namespace internal { function GetSymbolFromLibrary (line 40) | bool GetSymbolFromLibrary(void* handle, const char* symbol_name, v... function LoadLibrary (line 49) | bool LoadLibrary(const char* library_filename, void** handle) { function BindFunc (line 63) | bool BindFunc(void* handle, const char* name, std::functionhdfsHFlush(fs_, file_) == 0)... FILE: oneflow/core/persistence/hadoop/hadoop_file_system.h type hdfs_internal (line 25) | struct hdfs_internal type hdfs_internal (line 26) | typedef hdfs_internal* hdfsFS; function namespace (line 29) | namespace oneflow { FILE: oneflow/core/persistence/hadoop/hdfs.h type hdfsBuilder (line 70) | struct hdfsBuilder type tSize (line 71) | typedef int32_t tSize; type time_t (line 72) | typedef time_t tTime; type tOffset (line 73) | typedef int64_t tOffset; type tPort (line 74) | typedef uint16_t tPort; type tObjectKind (line 75) | typedef enum tObjectKind { type hdfs_internal (line 83) | struct hdfs_internal type hdfs_internal (line 84) | struct hdfs_internal type hdfsFile_internal (line 86) | struct hdfsFile_internal type hdfsFile_internal (line 87) | struct hdfsFile_internal type hadoopRzOptions (line 89) | struct hadoopRzOptions type hadoopRzBuffer (line 91) | struct hadoopRzBuffer type hdfsReadStatistics (line 111) | struct hdfsReadStatistics { type hdfsReadStatistics (line 132) | struct hdfsReadStatistics type hdfsReadStatistics (line 140) | struct hdfsReadStatistics type hdfsReadStatistics (line 162) | struct hdfsReadStatistics type hdfsBuilder (line 228) | struct hdfsBuilder type hdfsBuilder (line 245) | struct hdfsBuilder type hdfsBuilder (line 269) | struct hdfsBuilder type hdfsBuilder (line 278) | struct hdfsBuilder type hdfsBuilder (line 287) | struct hdfsBuilder type hdfsBuilder (line 298) | struct hdfsBuilder type hdfsBuilder (line 309) | struct hdfsBuilder type hdfsBuilder (line 323) | struct hdfsBuilder type hdfsFileInfo (line 627) | typedef struct { type hadoopRzOptions (line 802) | struct hadoopRzOptions type hadoopRzOptions (line 819) | struct hadoopRzOptions type hadoopRzOptions (line 828) | struct hadoopRzOptions type hadoopRzOptions (line 854) | struct hadoopRzOptions type hadoopRzBuffer (line 864) | struct hadoopRzBuffer type hadoopRzBuffer (line 877) | struct hadoopRzBuffer type hadoopRzBuffer (line 887) | struct hadoopRzBuffer FILE: oneflow/core/persistence/persistent_in_stream.cpp type oneflow (line 23) | namespace oneflow { function GetBufferSize (line 29) | size_t GetBufferSize() { FILE: oneflow/core/persistence/persistent_in_stream.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/persistence/persistent_out_stream.cpp type oneflow (line 21) | namespace oneflow { function PersistentOutStream (line 32) | PersistentOutStream& PersistentOutStream::Write(const char* s, size_t ... FILE: oneflow/core/persistence/persistent_out_stream.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/persistence/posix/posix_file_system.cpp type oneflow (line 31) | namespace oneflow { type fs (line 33) | namespace fs { class PosixRandomAccessFile (line 35) | class PosixRandomAccessFile : public RandomAccessFile { method PosixRandomAccessFile (line 41) | PosixRandomAccessFile(const std::string& fname, int fd) : fname_(f... method Read (line 44) | void Read(uint64_t offset, size_t n, char* result) const override { class PosixWritableFile (line 65) | class PosixWritableFile : public WritableFile { method PosixWritableFile (line 71) | PosixWritableFile(const std::string& fname, FILE* file) : fname_(f... method Append (line 77) | void Append(const char* data, size_t n) override { method Close (line 82) | void Close() override { method Flush (line 88) | void Flush() override { type dirent (line 130) | struct dirent type stat (line 161) | struct stat type stat (line 173) | struct stat FILE: oneflow/core/persistence/posix/posix_file_system.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/persistence/stream_scanner.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/persistence/stream_scanner.h function class (line 26) | class StreamScanner { FILE: oneflow/core/persistence/tee_persistent_log_stream.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/core/persistence/tee_persistent_log_stream.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/platform/include/ibv.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/platform/include/pthread_fork.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/platform/include/wrapper.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/platform/lib/ibv_wrapper.cpp type oneflow (line 19) | namespace oneflow { type ibv (line 21) | namespace ibv { function GetLibPaths (line 23) | std::vector GetLibPaths() { function FUNC (line 45) | FUNC LoadSymbol(const char* name, FUNC* save) { function IsAvailable (line 55) | bool IsAvailable() { return GetIBVLibraryPtr() != nullptr; } type _stubs (line 57) | namespace _stubs { function ibv_free_device_list (line 59) | void ibv_free_device_list(struct ibv_device** list) { type ibv_mr (line 63) | struct ibv_mr type ibv_pd (line 63) | struct ibv_pd function ibv_destroy_qp (line 67) | int ibv_destroy_qp(struct ibv_qp* qp) { return LoadSymbol(__func__... function ibv_query_gid (line 69) | int ibv_query_gid(struct ibv_context* context, uint8_t port_num, i... function ibv_fork_init (line 73) | int ibv_fork_init(void) { return LoadSymbol(__func__, &wrapper.ibv... function ibv_query_port_wrap (line 75) | int ibv_query_port_wrap(struct ibv_context* context, uint8_t port_... type ibv_context (line 80) | struct ibv_context type ibv_device (line 80) | struct ibv_device function ibv_destroy_cq (line 84) | int ibv_destroy_cq(struct ibv_cq* cq) { return LoadSymbol(__func__... type ibv_pd (line 86) | struct ibv_pd type ibv_context (line 86) | struct ibv_context function ibv_modify_qp (line 90) | int ibv_modify_qp(struct ibv_qp* qp, struct ibv_qp_attr* attr, int... function ibv_dealloc_pd (line 94) | int ibv_dealloc_pd(struct ibv_pd* pd) { return LoadSymbol(__func__... type ibv_device (line 96) | struct ibv_device function ibv_close_device (line 100) | int ibv_close_device(struct ibv_context* context) { type ibv_qp (line 104) | struct ibv_qp type ibv_pd (line 104) | struct ibv_pd type ibv_qp_init_attr (line 104) | struct ibv_qp_init_attr function ibv_dereg_mr (line 108) | int ibv_dereg_mr(struct ibv_mr* mr) { return LoadSymbol(__func__, ... type ibv_cq (line 110) | struct ibv_cq type ibv_context (line 110) | struct ibv_context type ibv_comp_channel (line 111) | struct ibv_comp_channel function ibv_query_device (line 116) | int ibv_query_device(struct ibv_context* context, struct ibv_devic... type ibv_device (line 120) | struct ibv_device FILE: oneflow/core/platform/lib/pthread_fork.cpp type oneflow (line 22) | namespace oneflow { type pthread_fork (line 24) | namespace pthread_fork { function IsForkedSubProcess (line 28) | bool IsForkedSubProcess() { return is_fork; } function SetIsForkedSubProcess (line 29) | static void SetIsForkedSubProcess() { is_fork = true; } function CurrentRankVmSync (line 32) | void CurrentRankVmSync() { function RegisterForkCallback (line 42) | void RegisterForkCallback() { pthread_atfork(&CurrentRankVmSync, nul... FILE: oneflow/core/platform/lib/wrapper.cpp type oneflow (line 23) | namespace oneflow { type platform (line 24) | namespace platform { type link_map (line 57) | struct link_map FILE: oneflow/core/profiler/event.cpp type oneflow (line 24) | namespace oneflow { type profiler (line 26) | namespace profiler { FILE: oneflow/core/profiler/event.h type class (line 32) | enum class function CustomEventType (line 36) | enum class CustomEventType { function namespace (line 172) | namespace nlohmann { FILE: oneflow/core/profiler/event_recorder.cpp type oneflow (line 20) | namespace oneflow { type profiler (line 21) | namespace profiler { FILE: oneflow/core/profiler/event_recorder.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/profiler/kernel.cpp type oneflow (line 23) | namespace oneflow { type profiler (line 25) | namespace profiler { function Init (line 32) | void Init() { function TraceKernelForwardDataContentStart (line 48) | void TraceKernelForwardDataContentStart(KernelContext* kernel_ctx, c... function TraceKernelForwardDataContentEnd (line 66) | void TraceKernelForwardDataContentEnd(KernelContext* kernel_ctx, con... FILE: oneflow/core/profiler/kernel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/profiler/kineto_shim.cpp type oneflow (line 22) | namespace oneflow { type profiler (line 24) | namespace profiler { function PrepareTrace (line 54) | void PrepareTrace(const bool cpuOnly, const ActivitySet& activities) { function StartTrace (line 73) | void StartTrace() { libkineto::api().activityProfiler().startTrace(); } function ActivityTraceWrapper (line 75) | ActivityTraceWrapper StopTrace() { FILE: oneflow/core/profiler/kineto_shim.h function namespace (line 25) | namespace libkineto { function namespace (line 32) | namespace oneflow { FILE: oneflow/core/profiler/profile_manager.cpp type oneflow (line 29) | namespace oneflow { type profiler (line 30) | namespace profiler { FILE: oneflow/core/profiler/profile_manager.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/profiler/profiler.cpp type oneflow (line 30) | namespace oneflow { type profiler (line 32) | namespace profiler { function NameThisHostThread (line 34) | void NameThisHostThread(const std::string& name) { function RangePush (line 46) | void RangePush(const std::string& name) { function RangePop (line 52) | void RangePop() { function LogHostMemoryUsage (line 70) | void LogHostMemoryUsage(const std::string& name) { function ProfilerStart (line 83) | void ProfilerStart() { function ProfilerStop (line 89) | void ProfilerStop() { function EnableProfiler (line 95) | void EnableProfiler(bool use_cpu, bool use_cuda, bool record_shapes,... function DisableProfilerAndReturnResult (line 105) | Maybe DisableProfilerAndReturnResult() { function StartRecord (line 116) | Maybe StartRecord(const std::string& name) { function EndRecord (line 123) | Maybe EndRecord(const std::string& event_recorder_key) { FILE: oneflow/core/profiler/profiler.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/profiler/util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/register/blob.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/core/register/blob.h function namespace (line 25) | namespace oneflow { function class (line 47) | class Blob final { function BlobDesc (line 67) | const BlobDesc* blob_desc_ptr() const { return blob_desc_; } function MutShapeView (line 104) | MutShapeView* mut_shape_view() { function MutShapeView (line 108) | MutShapeView* ForceMutShapeView() { return mut_shape_view_.get(); } function reset_dptr (line 113) | void reset_dptr(char* dptr) { dptr_ = dptr; } function set_blob_access_checker (line 124) | void set_blob_access_checker(const BlobAccessChecker* blob_access_checke... function BlobAccessChecker (line 128) | const BlobAccessChecker* blob_access_checker() { return this->blob_acces... FILE: oneflow/core/register/blob_desc.cpp type oneflow (line 18) | namespace oneflow { function CompareLbiBlobDescPair (line 20) | bool CompareLbiBlobDescPair(const LbiBlobDescPair& lhs, const LbiBlobD... function BlobDesc (line 77) | BlobDesc& BlobDesc::operator=(const BlobDesc& rhs) { FILE: oneflow/core/register/blob_desc.h function class (line 32) | class BlobDesc final { FILE: oneflow/core/register/op_blob_arg_info.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/register/register.cpp type oneflow (line 20) | namespace oneflow { function Blob (line 68) | Blob* Regst::GetBlobByOrdinal(int64_t ordinal) { return sorted_blob_ve... function Blob (line 70) | Blob* Regst::GetBlobByLbi(const LogicalBlobId& lbi) { function Blob (line 84) | Blob* Regst::GetMutSoleBlob() { function Blob (line 89) | const Blob* Regst::GetSoleBlob() const { FILE: oneflow/core/register/register.h function RegstAllocationType (line 24) | enum class RegstAllocationType { FILE: oneflow/core/register/register_desc.cpp type oneflow (line 23) | namespace oneflow { function BlobDesc (line 81) | BlobDesc* RegstDesc::AddLbi(const LogicalBlobId& lbi) { function BlobDesc (line 88) | const BlobDesc* RegstDesc::GetBlobDesc(const LogicalBlobId& lbi) const { function BlobDesc (line 96) | BlobDesc* RegstDesc::MutBlobDesc(const LogicalBlobId& lbi) { function BlobDesc (line 105) | const BlobDesc* RegstDesc::SoleBlobDesc() const { function BlobDesc (line 110) | BlobDesc* RegstDesc::MutSoleBlobDesc() { return const_cast(... function InitCtrlRegstDesc (line 205) | void InitCtrlRegstDesc(int64_t producer_task_id, RegstDescProto* ctrl_... FILE: oneflow/core/register/register_desc.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/register/register_manager.cpp type oneflow (line 29) | namespace oneflow { type PackedChunkInfo (line 33) | struct PackedChunkInfo { method PackedChunkInfo (line 37) | PackedChunkInfo(const MemoryCase& mem) { function GetDeviceByMemoryCase (line 43) | std::shared_ptr GetDeviceByMemoryCase(const MemoryCase& me... function InitDataRegst (line 48) | void InitDataRegst(Regst* regst, char* main_mem_ptr, char* separated_h... function RtRegstDesc (line 261) | const RtRegstDesc& RegstMgr::RegstDesc4RegstDescId(int64_t regst_desc_... FILE: oneflow/core/register/register_manager.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/register/runtime_register_desc.cpp type oneflow (line 20) | namespace oneflow { function BlobDesc (line 73) | const BlobDesc* RtRegstDesc::GetBlobDescFromLbi(const LogicalBlobId& l... function BlobDesc (line 82) | const BlobDesc* RtRegstDesc::GetBlobDescByOrdinal(int64_t ordinal) con... function LogicalBlobId (line 86) | const LogicalBlobId& RtRegstDesc::GetLbiByOrdinal(int64_t ordinal) con... function BlobDesc (line 90) | const BlobDesc* RtRegstDesc::GetSoleBlobDesc() const { function Shape (line 132) | const Shape& RtRegstDesc::data_regst_time_shape() const { FILE: oneflow/core/register/runtime_register_desc.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/register/tensor_slice_copier.cpp type oneflow (line 18) | namespace oneflow { function TensorSliceView (line 22) | TensorSliceView GetRawTenserSliceView(const TensorSliceView& view, Dat... FILE: oneflow/core/register/tensor_slice_copier.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/register/tensor_slice_view.cpp type oneflow (line 18) | namespace oneflow { function TensorSliceView (line 42) | TensorSliceView& TensorSliceView::operator=(const TensorSliceView& oth... function FOR_RANGE (line 66) | FOR_RANGE(int64_t, i, 0, NumAxes()) { function TensorSliceView (line 75) | TensorSliceView TensorSliceView::Intersect(const TensorSliceView& othe... function Range (line 92) | const Range& TensorSliceView::At(int64_t index) const { return range_v... function Shape (line 94) | const Shape& TensorSliceView::shape() const { return shape_; } function NdIndex (line 100) | NdIndex TensorSliceView::OffsetTo(const TensorSliceView& other) const { function TensorSliceView (line 113) | TensorSliceView TensorSliceView::Concatenate(std::vector GetCtrlPort(const EnvDesc& env_desc) { FILE: oneflow/core/rpc/lib/local.cpp type oneflow (line 24) | namespace oneflow { function TryLockResult (line 57) | TryLockResult LocalCtrlClient::TryLock(const std::string& name) { class DryRunCtrlClient (line 166) | class DryRunCtrlClient : public CtrlClient { method DryRunCtrlClient (line 169) | explicit DryRunCtrlClient(const ProcessCtx& process_ctx) method Barrier (line 176) | void Barrier(const std::string& barrier_name) override { method Barrier (line 179) | void Barrier(const std::string& barrier_name, int32_t barrier_num) o... method TryLockResult (line 184) | TryLockResult TryLock(const std::string& name) override { method NotifyDone (line 187) | void NotifyDone(const std::string& name) override { local_ctrl_clien... method WaitUntilDone (line 188) | void WaitUntilDone(const std::string& name) override { local_ctrl_cl... method PushKV (line 190) | void PushKV(const std::string& k, std::function ... method PushKV (line 193) | void PushKV(const std::string& k, const std::string& v) override { method PushKV (line 196) | void PushKV(const std::string& k, const PbMessage& msg) override { method PushMasterKV (line 199) | void PushMasterKV(const std::string& k, const PbMessage& msg) overri... method ClearKV (line 203) | void ClearKV(const std::string& k) override { local_ctrl_client_->Cl... method ClearMasterKV (line 204) | void ClearMasterKV(const std::string& k) override { local_ctrl_clien... method PullKV (line 206) | void PullKV(const std::string& k, std::functionClear(); } method IncreaseCount (line 215) | int32_t IncreaseCount(const std::string& k, int32_t v) override { method EraseCount (line 218) | void EraseCount(const std::string& k) override { local_ctrl_client_-... function SetLocalProcessCtx (line 224) | void SetLocalProcessCtx(oneflow::ProcessCtx* ctx) { FILE: oneflow/core/thread/is_main_thread_test.cpp type oneflow (line 20) | namespace oneflow { type test (line 21) | namespace test { function TEST (line 23) | TEST(IsMainThread, IsMainThread) { FILE: oneflow/core/thread/thread.cpp type oneflow (line 27) | namespace oneflow { FILE: oneflow/core/thread/thread.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/thread/thread_global_id.cpp type oneflow (line 22) | namespace oneflow { function GetThisThreadGlobalId (line 33) | int64_t GetThisThreadGlobalId() { return *MutThreadLocalUniqueGlobalId... FILE: oneflow/core/thread/thread_global_id.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/thread/thread_manager.cpp type oneflow (line 21) | namespace oneflow { function Thread (line 32) | Thread* ThreadMgr::GetThrd(int64_t thrd_id) { FILE: oneflow/core/thread/thread_manager.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/thread/thread_pool.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int32_t, i, 0, thread_num) { function FOR_RANGE (line 34) | FOR_RANGE(int32_t, i, 0, work_chans_.size()) { FILE: oneflow/core/thread/thread_pool.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/thread/thread_runtime.h function namespace (line 36) | namespace oneflow { FILE: oneflow/core/thread/thread_runtime_factory.cpp type oneflow (line 22) | namespace oneflow { type thread (line 23) | namespace thread { function CreateRuntime (line 28) | std::shared_ptr CreateRuntime() { FILE: oneflow/core/thread/thread_runtime_factory.h function namespace (line 22) | namespace thread { FILE: oneflow/core/transport/transport.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/core/transport/transport.h type TransportStatus (line 81) | struct TransportStatus { FILE: oneflow/core/transport/transport_message.h function TransportMsgType (line 26) | enum class TransportMsgType { FILE: oneflow/core/vm/access_blob_arg_cb_instruction_policy.h function namespace (line 34) | namespace vm { FILE: oneflow/core/vm/allocate_tensor_instruction_policy.cpp type oneflow (line 18) | namespace oneflow { type vm (line 19) | namespace vm { FILE: oneflow/core/vm/allocate_tensor_instruction_policy.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/vm/allocator.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/vm/barrier_instruction_policy.h function namespace (line 20) | namespace oneflow { FILE: oneflow/core/vm/bin_allocator.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/vm/bin_allocator_test.cpp type oneflow (line 23) | namespace oneflow { type vm (line 24) | namespace vm { class CudaBackendAllocator (line 26) | class CudaBackendAllocator final : public CachingAllocator { method CudaBackendAllocator (line 28) | explicit CudaBackendAllocator(int64_t device_id) : device_id_(devi... function TEST (line 60) | TEST(CudaBinAllocator, cuda_allocator) { FILE: oneflow/core/vm/caching_allocator.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/vm/control_stream_policy.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/vm/critical_section_instruction_policy.cpp type oneflow (line 26) | namespace oneflow { type vm (line 27) | namespace vm { FILE: oneflow/core/vm/critical_section_instruction_policy.h function namespace (line 30) | namespace oneflow { function DependenceVector (line 167) | const DependenceVector& input_dependences() const override { return inpu... function ForEachConstDependence (line 171) | void ForEachConstDependence(const std::function> CreateEpBackendHostAll... FILE: oneflow/core/vm/ep_d2h_stream_policy.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/ep_event.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/core/vm/ep_event.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/vm/ep_optional_event_record_status_querier.cpp type oneflow (line 18) | namespace oneflow { type vm (line 19) | namespace vm { FILE: oneflow/core/vm/ep_optional_event_record_status_querier.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/vm/ep_record_event_instruction_policy.h function namespace (line 27) | namespace oneflow { function InitInstructionStatus (line 69) | void InitInstructionStatus(Instruction* instruction) override { function Compute (line 83) | void Compute(vm::Instruction* instruction) override {} type GetRecordEventInstructionPolicy (line 94) | struct GetRecordEventInstructionPolicy FILE: oneflow/core/vm/ep_stream_policy.cpp type oneflow (line 28) | namespace oneflow { type vm (line 29) | namespace vm { function CreateEpBackendDeviceAllocator (line 33) | std::unique_ptr CreateEpBackendDeviceAllocator(Symbol... FILE: oneflow/core/vm/ep_stream_policy.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/ep_stream_policy_base.cpp type oneflow (line 27) | namespace oneflow { type vm (line 28) | namespace vm { FILE: oneflow/core/vm/ep_stream_policy_base.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/vm/event_recorded_ep_stream_policy.cpp type oneflow (line 24) | namespace oneflow { type vm (line 25) | namespace vm { FILE: oneflow/core/vm/event_recorded_ep_stream_policy.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/global_sync_instruction_policy.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/instruction.cpp type oneflow (line 27) | namespace oneflow { type vm (line 28) | namespace vm { function INTRUSIVE_FOR_EACH_PTR (line 60) | INTRUSIVE_FOR_EACH_PTR(edge, mut_in_edges()) { function INTRUSIVE_FOR_EACH_PTR (line 64) | INTRUSIVE_FOR_EACH_PTR(edge, mut_out_edges()) { function StreamPolicy (line 78) | StreamPolicy* Instruction::mut_stream_policy() { return mut_stream()... function StreamPolicy (line 80) | const StreamPolicy& Instruction::stream_policy() const { return stre... FILE: oneflow/core/vm/instruction.h function namespace (line 30) | namespace oneflow { function set_src_instruction (line 72) | void set_src_instruction(Instruction* val) { src_instruction_ = val; } function set_dst_instruction (line 73) | void set_dst_instruction(Instruction* val) { dst_instruction_ = val; } function clear_src_instruction (line 74) | void clear_src_instruction() { src_instruction_ = nullptr; } function clear_dst_instruction (line 75) | void clear_dst_instruction() { dst_instruction_ = nullptr; } function Instruction (line 76) | Instruction* mut_src_instruction() { return src_instruction_; } function Instruction (line 77) | Instruction* mut_dst_instruction() { return dst_instruction_; } function __Init__ (line 79) | void __Init__(Instruction* src_instruction, Instruction* dst_instruction) { function Stream (line 133) | Stream* mut_stream() { return stream_; } function InstructionStatusBuffer (line 134) | InstructionStatusBuffer* mut_status_buffer() { return &status_buffer_; } function InstructionPolicy (line 135) | InstructionPolicy* mut_instruction_policy() { return instruction_policy_... function InEdgeList (line 136) | InEdgeList* mut_in_edges() { return &in_edges_; } function OutEdgeList (line 137) | OutEdgeList* mut_out_edges() { return &out_edges_; } function DependenceAccessList (line 138) | DependenceAccessList* mut_access_list() { return &access_list_; } function Launched (line 143) | bool Launched() const; FILE: oneflow/core/vm/instruction_fuse_type.h function namespace (line 19) | namespace oneflow { FILE: oneflow/core/vm/instruction_policy.cpp type oneflow (line 21) | namespace oneflow { type vm (line 22) | namespace vm { FILE: oneflow/core/vm/instruction_policy.h function namespace (line 28) | namespace oneflow { FILE: oneflow/core/vm/instruction_policy_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/vm/lazy_job_instruction_policy.h function namespace (line 32) | namespace oneflow { function namespace (line 50) | namespace vm { FILE: oneflow/core/vm/lazy_job_stream_policy.cpp type oneflow (line 23) | namespace oneflow { type vm (line 24) | namespace vm { FILE: oneflow/core/vm/lazy_job_stream_policy.h function namespace (line 23) | namespace oneflow { FILE: oneflow/core/vm/naive_instruction_status_querier.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/op_call_instruction_policy.cpp type oneflow (line 35) | namespace oneflow { type vm (line 36) | namespace vm { type OpCallInstructionUtil (line 38) | struct OpCallInstructionUtil final { method Prepare (line 39) | static inline Maybe Prepare(OpCallInstructionPolicy* op_call... method Compute (line 49) | static inline Maybe Compute(OpCallInstructionPolicy* op_call... method InferTempStorageSize (line 85) | static inline void InferTempStorageSize(OpCallInstructionPolicy* o... method TryInitOpKernelStateAndCache (line 92) | static inline void TryInitOpKernelStateAndCache( method AllocateOutputBlobsMemory (line 108) | static inline Maybe AllocateOutputBlobsMemory( method TryAllocateTempStorage (line 128) | static inline Maybe TryAllocateTempStorage( method DeallocateTempStorage (line 141) | static inline void DeallocateTempStorage(OpCallInstructionPolicy* ... method OpKernelCompute (line 148) | static inline void OpKernelCompute(OpCallInstructionPolicy* op_cal... method ComputeFnForRemat (line 156) | static inline Maybe ComputeFnForRemat(OpCallInstructionPolic... method InitRematInfo (line 161) | static inline std::tuple, bool, bool>... function Recompute (line 286) | Maybe Recompute(OpCallInstructionPolicy* op_call_instruction_p... FILE: oneflow/core/vm/op_call_instruction_policy.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/vm/pinned_ep_stream_policy.cpp type oneflow (line 26) | namespace oneflow { type vm (line 27) | namespace vm { function CreatePinedEpBackendHostAllocator (line 31) | std::unique_ptr> CreatePinedEpBackendHo... FILE: oneflow/core/vm/pinned_ep_stream_policy.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/probe.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/vm/ref_cnt_instruction_status_querier.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/vm/release_tensor_instruction_policy.h function namespace (line 35) | namespace oneflow { FILE: oneflow/core/vm/remat/allocator.cpp type oneflow (line 36) | namespace oneflow { type vm (line 37) | namespace vm { function CudaMemAlignedBytes (line 41) | inline size_t CudaMemAlignedBytes(size_t bytes) { return RoundUp(byt... function IsAlignedSize (line 43) | inline bool IsAlignedSize(size_t size) { return size % kCudaMemAlloc... function bytes2Mb (line 45) | inline double bytes2Mb(size_t bytes) { return bytes * 1. / 1024 / 10... function ShouldBeHeldBySmallPiece (line 49) | inline bool ShouldBeHeldBySmallPiece(size_t size) { function GroupNumToIndexes (line 54) | std::vector GroupNumToIndexes(size_t group_num) { function get_cost (line 140) | double get_cost(const vm::RematableTensorStorage* storage) { function get_cost (line 148) | double get_cost(const vm::RematableTensorStorage* storage, size_t si... FILE: oneflow/core/vm/remat/allocator.h function namespace (line 27) | namespace oneflow { function DeviceReset (line 146) | void DeviceReset() override { allocator->DeviceReset(); } function Deallocate (line 151) | void Deallocate(char* mem_ptr, std::size_t size) override { function namespace (line 159) | namespace remat { FILE: oneflow/core/vm/remat/disjoint_set.cpp type oneflow (line 23) | namespace oneflow { type remat (line 25) | namespace remat { FILE: oneflow/core/vm/remat/disjoint_set.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/vm/remat/env.cpp type oneflow (line 23) | namespace oneflow { type remat (line 25) | namespace remat { FILE: oneflow/core/vm/remat/env.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/vm/remat/util.cpp type oneflow (line 32) | namespace oneflow { type remat (line 34) | namespace remat { function append_memory_frag_info_and_get (line 36) | double append_memory_frag_info_and_get(size_t free_mem, size_t thres... function SortKey (line 49) | std::string SortKey(const std::string& key) { function json (line 88) | json LoadTimeDataset() { function GetDatasetComputeTime (line 103) | Maybe GetDatasetComputeTime(const json& j, const vm::OpCallI... function GetComputeComplexityEstimatedBySize (line 135) | static Maybe GetComputeComplexityEstimatedBySize( function TryGetTensorTupleIndex (line 145) | int32_t TryGetTensorTupleIndex(const std::unordered_map GetComputeComplexity(const vm::OpCallInstructionPolicy... function GetComputeTime (line 239) | Maybe GetComputeTime(const vm::OpCallInstructionPolicy& oper... type vm (line 247) | namespace vm { FILE: oneflow/core/vm/remat/util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/core/vm/stream.cpp type oneflow (line 24) | namespace oneflow { type vm (line 25) | namespace vm { FILE: oneflow/core/vm/stream.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/vm/stream_create_stream_policy.h function namespace (line 29) | namespace oneflow { FILE: oneflow/core/vm/stream_get_allocator_stream_type.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/stream_policy.cpp type oneflow (line 24) | namespace oneflow { type vm (line 25) | namespace vm { FILE: oneflow/core/vm/stream_policy.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/vm/stream_record_event_instruction_policy.cpp type oneflow (line 26) | namespace oneflow { type vm (line 27) | namespace vm { FILE: oneflow/core/vm/stream_record_event_instruction_policy.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/vm/stream_wait_event_instruction_policy.cpp type oneflow (line 23) | namespace oneflow { type vm (line 24) | namespace vm { FILE: oneflow/core/vm/stream_wait_event_instruction_policy.h function namespace (line 27) | namespace oneflow { FILE: oneflow/core/vm/stream_wait_instruction_policy.cpp type oneflow (line 23) | namespace oneflow { type vm (line 24) | namespace vm { FILE: oneflow/core/vm/stream_wait_instruction_policy.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/vm/symbol_storage.cpp type oneflow (line 22) | namespace oneflow { type symbol (line 24) | namespace symbol { type detail (line 26) | namespace detail { FILE: oneflow/core/vm/symbol_storage.h function namespace (line 24) | namespace oneflow { FILE: oneflow/core/vm/sync_access_instruction_policy.cpp type oneflow (line 20) | namespace oneflow { type vm (line 21) | namespace vm { function FastCopy (line 42) | void FastCopy(char* dst, const char* src, size_t bytes) { FILE: oneflow/core/vm/sync_access_instruction_policy.h function namespace (line 32) | namespace oneflow { FILE: oneflow/core/vm/sync_vm_mode_guard.h function SyncVmMode (line 23) | enum class SyncVmMode { FILE: oneflow/core/vm/thread_ctx.cpp type oneflow (line 19) | namespace oneflow { type vm (line 20) | namespace vm { FILE: oneflow/core/vm/thread_ctx.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/vm/thread_safe_guard.h function namespace (line 25) | namespace oneflow { FILE: oneflow/core/vm/virtual_machine.cpp type oneflow (line 47) | namespace oneflow { function MicrosecondsFrom (line 52) | int MicrosecondsFrom(const T& start) { function ForEachThreadCtx (line 58) | Maybe ForEachThreadCtx(vm::VirtualMachineEngine* engine, function GetSchedulerThreadInitializer (line 66) | void GetSchedulerThreadInitializer(std::function* Initializer) { function WorkerLoop (line 70) | void WorkerLoop(vm::ThreadCtx* thread_ctx, const std::function> GetBarrierStream() { function MakeBarrierInstructions (line 111) | void MakeBarrierInstructions(vm::InstructionList* list, class SingleThreadScheduleCtx (line 155) | class SingleThreadScheduleCtx : public vm::ScheduleCtx { method SingleThreadScheduleCtx (line 157) | SingleThreadScheduleCtx() = default; method OnWorkerLoadPending (line 160) | void OnWorkerLoadPending(vm::ThreadCtx* thread_ctx) const override { function ScheduleUntilVMEmpty (line 165) | void ScheduleUntilVMEmpty(vm::VirtualMachineEngine* vm, const vm::Sche... function INTRUSIVE_FOR_EACH_PTR (line 253) | INTRUSIVE_FOR_EACH_PTR(instruction, instruction_list) { class MultiThreadScheduleCtx (line 318) | class MultiThreadScheduleCtx : public vm::ScheduleCtx { method MultiThreadScheduleCtx (line 320) | MultiThreadScheduleCtx() = default; method OnWorkerLoadPending (line 323) | void OnWorkerLoadPending(vm::ThreadCtx* thread_ctx) const override { FILE: oneflow/core/vm/virtual_machine.h function namespace (line 26) | namespace oneflow { FILE: oneflow/core/vm/virtual_machine_engine.cpp type oneflow (line 33) | namespace oneflow { type vm (line 35) | namespace vm { function INTRUSIVE_FOR_EACH (line 40) | INTRUSIVE_FOR_EACH(access, access_list) { function INTRUSIVE_FOR_EACH_PTR (line 49) | INTRUSIVE_FOR_EACH_PTR(out_edge, out_edges) { function FusableBetween (line 83) | bool FusableBetween(InstructionFuseType fuse_type, Instruction* inst... function INTRUSIVE_FOR_EACH_PTR (line 120) | INTRUSIVE_FOR_EACH_PTR(instruction, mut_local_pending_instruction_li... function INTRUSIVE_UNSAFE_FOR_EACH_PTR (line 141) | INTRUSIVE_UNSAFE_FOR_EACH_PTR(instruction, mut_lively_instruction_li... function INTRUSIVE_FOR_EACH_PTR (line 184) | INTRUSIVE_FOR_EACH_PTR(stream, mut_active_stream_list()) { function DependenceAccess (line 200) | DependenceAccess* VirtualMachineEngine::AccessDependence(OperandAcce... function INTRUSIVE_FOR_EACH_PTR (line 225) | INTRUSIVE_FOR_EACH_PTR(src_access, access_list) { function DebugDeviceReset (line 300) | std::string DebugDeviceReset(vm::Stream* stream) { function INTRUSIVE_UNSAFE_FOR_EACH_PTR (line 337) | INTRUSIVE_UNSAFE_FOR_EACH_PTR(compute_instruction, compute_instructi... FILE: oneflow/core/vm/virtual_machine_engine.h function namespace (line 30) | namespace oneflow { FILE: oneflow/core/vm/virtual_machine_scope.cpp type oneflow (line 21) | namespace oneflow { type vm (line 22) | namespace vm { FILE: oneflow/core/vm/virtual_machine_scope.h function namespace (line 18) | namespace oneflow { FILE: oneflow/core/vm/vm_object.cpp type oneflow (line 20) | namespace oneflow { type vm (line 21) | namespace vm { FILE: oneflow/core/vm/vm_object.h function namespace (line 23) | namespace oneflow { function set_access_type (line 54) | void set_access_type(OperandAccessType val) { access_type_ = val; } function set_instruction (line 55) | void set_instruction(Instruction* val) { instruction_ = val; } function set_dependence (line 56) | void set_dependence(Dependence* val) { dependence_ = val; } function clear_instruction (line 57) | void clear_instruction() { instruction_ = nullptr; } function clear_dependence (line 58) | void clear_dependence() { dependence_ = nullptr; } function Instruction (line 59) | Instruction* mut_instruction() { return instruction_; } function Dependence (line 60) | Dependence* mut_dependence() { return dependence_; } function DependenceAccessList (line 100) | DependenceAccessList* mut_access_list() { return &access_list_; } function __Init__ (line 103) | void __Init__() {} FILE: oneflow/core/vm/vm_sync.h function namespace (line 21) | namespace oneflow { FILE: oneflow/core/vm/vm_util.cpp type oneflow (line 28) | namespace oneflow { type vm (line 29) | namespace vm { function Run (line 31) | Maybe Run(vm::InstructionList* instruction_list) { function ClusterSync (line 37) | Maybe ClusterSync() { function CurrentRankSync (line 48) | Maybe CurrentRankSync() { FILE: oneflow/core/vm/vm_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/extension/python/numpy.cpp type oneflow (line 24) | namespace oneflow { type numpy (line 26) | namespace numpy { function OFDataTypeToNumpyType (line 40) | Maybe OFDataTypeToNumpyType(DataType of_data_type) { function NumpyTypeToOFDataType (line 60) | Maybe NumpyTypeToOFDataType(int np_type) { function GetOFDataTypeFromNpArray (line 80) | Maybe GetOFDataTypeFromNpArray(PyArrayObject* array) { function OFShapeToNumpyShape (line 85) | std::vector OFShapeToNumpyShape(const DimVector& fixed_vec) { function OFStrideToNumpyStride (line 93) | std::vector OFStrideToNumpyStride(const Stride& stride, cons... function PyArrayCheckLongScalar (line 101) | bool PyArrayCheckLongScalar(PyObject* obj) { function PyArrayCheckFloatScalar (line 105) | bool PyArrayCheckFloatScalar(PyObject* obj) { function PyArrayCheckBoolScalar (line 109) | bool PyArrayCheckBoolScalar(PyObject* obj) { function PyArrayCheckComplexScalar (line 113) | bool PyArrayCheckComplexScalar(PyObject* obj) { function InitNumpyCAPI (line 122) | Maybe InitNumpyCAPI() { FILE: oneflow/extension/python/numpy.h function namespace (line 22) | namespace oneflow { FILE: oneflow/extension/python/numpy_internal.h function namespace (line 35) | namespace oneflow { FILE: oneflow/extension/python/py_compute.cpp type oneflow (line 27) | namespace oneflow { type pyext (line 28) | namespace pyext { function TensorToNumpy (line 52) | void TensorToNumpy(const user_op::Tensor* tensor, PyObject** arg_ptr) { function MemToTensor (line 79) | void MemToTensor(void* array_data_ptr, const size_t size, user_op::T... function NumpyToTensor (line 95) | void NumpyToTensor(PyObject* arg, user_op::Tensor* tensor) { function MakePyInputs (line 116) | void MakePyInputs(const UserOpDef& op_def, user_op::KernelComputeCon... function GetPyOutputs (line 139) | void GetPyOutputs(const UserOpDef& op_def, user_op::KernelComputeCon... function PyRegisterKernels (line 164) | void PyRegisterKernels(PyObject* py_kernels) { function PyCompute (line 173) | void PyCompute(user_op::KernelComputeContext* ctx, const std::string... FILE: oneflow/extension/python/py_compute.h function namespace (line 22) | namespace oneflow { FILE: oneflow/extension/python/py_kernel_caller.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/extension/python/py_kernel_caller.h function namespace (line 20) | namespace oneflow { FILE: oneflow/extension/python/py_kernel_registry.cpp type oneflow (line 20) | namespace oneflow { type pyext (line 21) | namespace pyext { function RegisterPyKernelCaller (line 23) | Maybe RegisterPyKernelCaller(const std::string& op_module_name) { function RegisterPyKernels (line 41) | void RegisterPyKernels(PyObject* py_kernels) { PyRegisterKernels(py_... FILE: oneflow/extension/python/py_kernel_registry.h function namespace (line 24) | namespace oneflow { FILE: oneflow/extension/stack/foreign_stack_getter.h function namespace (line 23) | namespace oneflow { FILE: oneflow/extension/stack/python/custom_eval_frame.c function EnableCustomEvalFrame (line 34) | inline static void EnableCustomEvalFrame(PyThreadState* tstate, _PyFrame... function EnableCustomEvalFrameForCurrentThread (line 46) | void EnableCustomEvalFrameForCurrentThread(PyFrameEvalFunc eval_func) { FILE: oneflow/extension/stack/python/custom_eval_frame.h type PyObject (line 27) | typedef PyObject* (*PyFrameEvalFunc)(struct _ts*, struct _frame*, int); type PyObject (line 29) | typedef PyObject* (*PyFrameEvalFunc)(struct _frame*, int); FILE: oneflow/extension/stack/python/stack_getter.cpp function PyUnicodeToStdString (line 48) | std::string PyUnicodeToStdString(const PyObject* py_str) { function PyCodeObject (line 52) | PyCodeObject* PyFrame_GetCode(PyFrameObject* frame) { class PyFrame (line 62) | class PyFrame final : public Frame { method PyFrame (line 66) | PyFrame(PyFrameObject* frame, std::shared_ptr back) function PyStackGetter (line 85) | PyStackGetter() { function GetCurrentFrame (line 92) | std::shared_ptr GetCurrentFrame() const override { function GetFormattedStack (line 100) | std::string GetFormattedStack(std::shared_ptr frame) const overri... function PyObject (line 126) | PyObject* RecordAndEvalFrame(PyThreadState* tstate, PyFrameObject* frame, function PushFrame (line 158) | void PushFrame(PyFrameObject* frame) { function PopFrame (line 162) | void PopFrame() { function PyObject (line 169) | PyObject* RecordAndEvalFrame(PyThreadState* tstate, PyFrameObject* frame, function RegisterPyStackGetter (line 183) | void RegisterPyStackGetter() { function namespace (line 190) | namespace { FILE: oneflow/extension/stack/python/stack_getter.h function namespace (line 20) | namespace oneflow { FILE: oneflow/extension/stack/stacktrace.h type SSIZE_T (line 358) | typedef SSIZE_T ssize_t; function namespace (line 433) | namespace backward { function namespace (line 446) | namespace backward { function namespace (line 464) | namespace backward { function namespace (line 474) | namespace backward { function namespace (line 3815) | namespace Color { function class (line 3819) | class Colorize { function namespace (line 3852) | namespace Color { function class (line 3856) | class Colorize { function class (line 3866) | class Printer { function is_oneflow_file (line 3925) | static inline bool is_oneflow_file(const std::string& filename) { function print_header (line 3959) | void print_header(std::ostream& os, size_t thread_id) { function print_trace (line 3965) | void print_trace(std::ostream& os, const ResolvedTrace& trace, Colorize&... function print_snippet (line 4007) | void print_snippet(std::ostream& os, const char* indent, function class (line 4056) | class SignalHandling { function class (line 4195) | class SignalHandling { function crash_status (line 4249) | enum class crash_status { running, crashed, normal_exit, ending }; function class (line 4362) | class SignalHandling { FILE: oneflow/ir/include/OneFlow/Conversion/NVVMToCubin.h function namespace (line 23) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Conversion/OneFlowToTosa.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Extension.h function namespace (line 22) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Conversion/Conversion.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OKL/Conversion/OKLToLLVM.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/InferContext.h function namespace (line 24) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/InitContext.h function namespace (line 25) | namespace oneflow { function std (line 66) | const std::shared_ptr& Attr4Name( FILE: oneflow/ir/include/OneFlow/OKL/Kernel/JITEngine.h function namespace (line 28) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/JITOpInfer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/LauncherContext.h function namespace (line 25) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/LauncherState.h function namespace (line 29) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/RegContext.h function namespace (line 23) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/TmpBufferManager.h function class (line 27) | class TmpBufferManager { FILE: oneflow/ir/include/OneFlow/OKL/Kernel/WrapperContext.h function namespace (line 25) | namespace oneflow { FILE: oneflow/ir/include/OneFlow/OKL/OKLOps.h function namespace (line 32) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OKL/passes.h function namespace (line 23) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OKM/Conversion/Conversion.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OKM/OKMOps.h function namespace (line 26) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OKM/passes.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OneFlowDataTypeConversion.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OneFlowOpTraits.h function namespace (line 23) | namespace mlir { function LogicalResult (line 120) | static LogicalResult verifyTrait(Operation* op) { function StringRef (line 133) | static StringRef getDataTypeAttrName() { return "data_type"; } function StringRef (line 134) | static StringRef getIsDynamicAttrName() { return "is_dynamic"; } function StringRef (line 135) | static StringRef getNdSbpAttrName() { return "nd_sbp"; } function StringRef (line 136) | static StringRef getSbpAttrName() { return "parallel"; } function LogicalResult (line 138) | static LogicalResult verifyTrait(Operation* op) { FILE: oneflow/ir/include/OneFlow/OneFlowOps.h function namespace (line 34) | namespace mlir { function namespace (line 47) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OneFlowPDLLPatterns.h function namespace (line 20) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OneFlowPatternUtils.h function namespace (line 19) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OneFlowSupport.h function namespace (line 31) | namespace mlir { FILE: oneflow/ir/include/OneFlow/OneFlowUtils.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Passes.h function namespace (line 46) | namespace mlir { FILE: oneflow/ir/include/OneFlow/SBP/SBPImporter.h function namespace (line 30) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/AggregateOps.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/AutoNhwc.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/BufferHostRegister.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/CSEWithAttributesIgnored.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/ConvertInferenceOp.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/EliminateAllocOps.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/FuncOps.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/OneFlowMemPool.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/OneFlowStream.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/OutlineAndFuse.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/TraitFolder.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/include/OneFlow/Transform/TransposeHelpers.h function namespace (line 23) | namespace mlir { FILE: oneflow/ir/include/OneFlow/UserOpConversion.h function namespace (line 20) | namespace mlir { FILE: oneflow/ir/include/OneFlow/UserOpReflection.h function namespace (line 20) | namespace mlir { FILE: oneflow/ir/include/Transform/TransformDialectExtension.h function namespace (line 26) | namespace mlir { FILE: oneflow/ir/include/Transform/TransformStateExtension.h function namespace (line 22) | namespace mlir { FILE: oneflow/ir/lib/OneFlow/Conversion/NVVMToCubin.cpp function emitCudaError (line 51) | static void emitCudaError(const llvm::Twine& expr, const char* buffer, C... type mlir (line 70) | namespace mlir { type oneflow (line 71) | namespace oneflow { function LogicalResult (line 99) | LogicalResult linkLibdevice(llvm::Module& llvmModule, llvm::LLVMCont... function translateToISA (line 127) | std::optional translateToISA(llvm::Module& llvmModule, class NVVMToCubinPass (line 148) | class NVVMToCubinPass : public NVVMToCubinPassBase { method translateToLLVMIR (line 149) | std::unique_ptr translateToLLVMIR(llvm::LLVMContext&... method getDependentDialects (line 159) | void getDependentDialects(::mlir::DialectRegistry& registry) const... function createNVVMToCubinPass (line 255) | std::unique_ptr createNVVMToCubinPass() { return std::ma... function InitializeLLVMNVPTXBackend (line 257) | void InitializeLLVMNVPTXBackend() { FILE: oneflow/ir/lib/OneFlow/Conversion/OneFlowToLinalg.cpp type mlir (line 24) | namespace mlir { type oneflow (line 26) | namespace oneflow { function computeIteratorTypesAndIndexingMaps (line 30) | std::tuple, SmallVector createDimValues(OpBuilder& b, Location loc... type SoftmaxOpLowering (line 131) | struct SoftmaxOpLowering final : public OpConversionPattern createLowerOneFlowToLinalgPass() { FILE: oneflow/ir/lib/OneFlow/Conversion/OneFlowToTosa.cpp type mlir (line 53) | namespace mlir { type oneflow (line 55) | namespace oneflow { function Type (line 57) | Type convertToSignless(MLIRContext* context, Type type) { function FunctionType (line 71) | FunctionType convertToSignlessFuncType(MLIRContext* context, Functio... function isSignLessTensorOrOther (line 79) | bool isSignLessTensorOrOther(Type type) { function allSignless (line 88) | bool allSignless(mlir::TypeRange types) { function allSignless (line 95) | bool allSignless(FunctionType funcType) { function Value (line 105) | Value CreateTransposeValue(Location& loc, ConversionPatternRewriter&... function RankedTensorType (line 118) | RankedTensorType CreateTransposeType(ShapedType output, ArrayRef { method LogicalResult (line 161) | LogicalResult matchAndRewrite(Job op, OpAdaptor adaptor, type ReturnOpLowering (line 171) | struct ReturnOpLowering final : public OpConversionPattern { method LogicalResult (line 174) | LogicalResult matchAndRewrite(ReturnOp op, OpAdaptor adaptor, type InputOpLowering (line 182) | struct InputOpLowering final : public OpConversionPattern { method LogicalResult (line 185) | LogicalResult matchAndRewrite(InputOp op, OpAdaptor adaptor, type OutputOpLowering (line 196) | struct OutputOpLowering final : public OpConversionPattern { method LogicalResult (line 199) | LogicalResult matchAndRewrite(OutputOp op, OpAdaptor adaptor, type VariableOpLowering (line 208) | struct VariableOpLowering final : public OpConversionPattern { method LogicalResult (line 264) | LogicalResult matchAndRewrite(CastOp op, OpAdaptor adaptor, type ReluOpLowering (line 273) | struct ReluOpLowering final : public OpConversionPattern { method LogicalResult (line 276) | LogicalResult matchAndRewrite(ReluOp op, OpAdaptor adaptor, type BroadcastAddOpLowering (line 292) | struct BroadcastAddOpLowering final : public OpConversionPattern { method LogicalResult (line 309) | LogicalResult matchAndRewrite(Add2Op op, OpAdaptor adaptor, type AvgPool2DOpLowering (line 320) | struct AvgPool2DOpLowering final : public OpConversionPattern { method LogicalResult (line 450) | LogicalResult matchAndRewrite(MatmulOp op, OpAdaptor adaptor, type BatchMatmulOpLowering (line 469) | struct BatchMatmulOpLowering final : public OpConversionPattern { method LogicalResult (line 558) | LogicalResult matchAndRewrite(Conv2DOp op, OpAdaptor adaptor, type TransposeOpLowering (line 614) | struct TransposeOpLowering final : public OpConversionPattern { method CastInputConversion (line 635) | explicit CastInputConversion(mlir::MLIRContext* context) method matchAndRewrite (line 637) | mlir::LogicalResult matchAndRewrite(InputOp op, mlir::PatternRewri... type CastVariableConversion (line 654) | struct CastVariableConversion final : public OpRewritePattern createLowerOneFlowToTosaPass() { function createConvertToSignlessForTosaPass (line 708) | std::unique_ptr createConvertToSignlessForTosaPass() { type ConvertReturnToSignlessPattern (line 757) | struct ConvertReturnToSignlessPattern : public OpRewritePattern createCastOneFlowOpsToSignlessPass() { FILE: oneflow/ir/lib/OneFlow/OKL/Conversion/Conversion.cpp type mlir (line 25) | namespace mlir { type okl (line 26) | namespace okl { function LogicalResult (line 28) | LogicalResult LowerOKLComputeToLLVM(ModuleOp module) { FILE: oneflow/ir/lib/OneFlow/OKL/Conversion/CudaGraphSupport.cpp type mlir (line 45) | namespace mlir { type okl (line 46) | namespace okl { type TagCudaGraphSupportPattern (line 47) | struct TagCudaGraphSupportPattern final : public mlir::OpRewritePatt... method LogicalResult (line 58) | static LogicalResult CheckChild(func::FuncOp func) { method TagCudaGraphSupportPattern (line 84) | explicit TagCudaGraphSupportPattern(mlir::MLIRContext* context) method matchAndRewrite (line 86) | mlir::LogicalResult matchAndRewrite(func::FuncOp op, type TagCudaGraphSupportPass (line 103) | struct TagCudaGraphSupportPass : public TagCudaGraphSupportPassBase<... method getDependentDialects (line 106) | void getDependentDialects(DialectRegistry& registry) const override { function createTagCudaGraphSupportPass (line 112) | std::unique_ptr createTagCudaGraphSupportPass() { FILE: oneflow/ir/lib/OneFlow/OKL/Conversion/OKLToLLVM.cpp type mlir (line 41) | namespace mlir { type okl (line 42) | namespace okl { function ModuleOp (line 45) | ModuleOp GetModuleOpFromJobBodyOp(T op) { function GetPtrType (line 52) | LLVM::LLVMPointerType GetPtrType(::mlir::PatternRewriter& rewriter) { type WrapperKernelOpLowering (line 56) | struct WrapperKernelOpLowering final : public OpConversionPattern createLowerOKLToLLVMCallPass() { function createLowerLauncherToLLVMPtrPass (line 155) | std::unique_ptr createLowerLauncherToLLVMPtrPass() { FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/ComputeContext.cpp type oneflow (line 22) | namespace oneflow { type okl (line 23) | namespace okl { FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/InferContext.cpp type oneflow (line 21) | namespace oneflow { type okl (line 22) | namespace okl { function TensorDesc (line 27) | const TensorDesc* InferContext::LogicalTensorDesc4ArgNameAndIndex(co... function Shape (line 32) | const Shape& InferContext::InputShape(const std::string& arg_name, i... function Shape (line 36) | const Shape& InferContext::Shape4ArgNameAndIndex(const std::string& ... FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/JITEngine.cpp function okl_llvm_func (line 27) | void okl_llvm_func(void* launcher, int64_t index) { type oneflow (line 33) | namespace oneflow { function SharedLibs (line 35) | SharedLibs* MutSharedLibPaths() { function SharedLibs (line 40) | const SharedLibs* SharedLibPaths() { return MutSharedLibPaths(); } FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/JITOpInfer.cpp type oneflow (line 42) | namespace oneflow { type ir (line 44) | namespace ir { type jit (line 46) | namespace jit { function GetFunctionType (line 48) | static Maybe GetFunctionType(user_op::InferCon... function SetTensorDataType (line 80) | Maybe SetTensorDataType(user_op::InferContext* ctx) { function InferTensorDesc (line 120) | Maybe InferTensorDesc(user_op::InferContext* ctx) { FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/KernelLaunchOp.cpp type oneflow (line 54) | namespace oneflow { class KernelLaunchKernel (line 78) | class KernelLaunchKernel final : public user_op::OpKernel, public user... method KernelLaunchKernel (line 80) | KernelLaunchKernel() = default; method CreateOpKernelState (line 83) | std::shared_ptr CreateOpKernelState( method IsCudaGraphSupported (line 90) | bool IsCudaGraphSupported(user_op::KernelInitContext* ctx, method Compute (line 96) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 101) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/LauncherContext.cpp type oneflow (line 32) | namespace oneflow { type okl (line 33) | namespace okl { FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/LauncherState.cpp type oneflow (line 36) | namespace oneflow { type okl (line 37) | namespace okl { function GetModule (line 41) | mlir::OwningOpRef GetModule(user_op::KernelInitConte... function JITEngine (line 54) | JITEngine GetEngine(mlir::ModuleOp module) { FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/RegContext.cpp type oneflow (line 33) | namespace oneflow { type okl (line 34) | namespace okl { function GetConfWrapper (line 36) | static user_op::UserOpConfWrapper GetConfWrapper(mlir::Operation* op, function DeviceType (line 104) | DeviceType RegContext::device_type() const { return device_type_; } function ParallelContext (line 105) | const ParallelContext& RegContext::parallel_ctx() const { function ArgVec (line 116) | const ArgVec& RegContext::inputs() const { return inputs_; } function ArgVec (line 117) | const ArgVec& RegContext::outputs() const { return outputs_; } FILE: oneflow/ir/lib/OneFlow/OKL/Kernel/TmpBufferManager.cpp type oneflow (line 27) | namespace oneflow { type okl (line 28) | namespace okl { FILE: oneflow/ir/lib/OneFlow/OKL/OKLDialect.cpp type mlir (line 35) | namespace mlir { type okl (line 37) | namespace okl { FILE: oneflow/ir/lib/OneFlow/OKM/Conversion/Conversion.cpp type mlir (line 20) | namespace mlir { type okm (line 21) | namespace okm { function LogicalResult (line 23) | LogicalResult LowerWrapOpsToOKL(ModuleOp module) { FILE: oneflow/ir/lib/OneFlow/OKM/OKMDialect.cpp type mlir (line 40) | namespace mlir { type okm (line 42) | namespace okm { FILE: oneflow/ir/lib/OneFlow/OKM/passes.cpp type mlir (line 45) | namespace mlir { type okm (line 46) | namespace okm { type func_name (line 48) | namespace func_name { type ExtractOKMTensorPattern (line 59) | struct ExtractOKMTensorPattern : public mlir::OpRewritePattern createExtractOKMTensorPass() { type WrapOKMKernelPattern (line 124) | struct WrapOKMKernelPattern : public mlir::OpRewritePattern createWrapOKMKernelPass() { return std::make_u... function MemSizeFirst (line 300) | void MemSizeFirst(func::FuncOp func, mlir::PatternRewriter& rewriter) { type OptOKMMemrefPattern (line 358) | struct OptOKMMemrefPattern : public mlir::OpRewritePattern createOptOKMMemrefPass() { return std::make_un... type ConvertOKMToOKLPattern (line 392) | struct ConvertOKMToOKLPattern : public mlir::OpRewritePattern createConvertOKMToOKLPass() { FILE: oneflow/ir/lib/OneFlow/OneFlowCanonicalizers.cpp type mlir (line 20) | namespace mlir { type oneflow (line 22) | namespace oneflow { type PutSeed (line 26) | struct PutSeed : public OpRewritePattern { method PutSeed (line 27) | explicit PutSeed(MLIRContext* context) method LogicalResult (line 29) | LogicalResult matchAndRewrite(oneflow::RandomMaskLikeOp op, FILE: oneflow/ir/lib/OneFlow/OneFlowDataTypeConversion.cpp type mlir (line 19) | namespace mlir { type oneflow (line 21) | namespace oneflow { function Type (line 23) | Type getTypeFromOneFlowDataType(MLIRContext* context, ::oneflow::Dat... FILE: oneflow/ir/lib/OneFlow/OneFlowDialect.cpp type mlir (line 24) | namespace mlir { type oneflow (line 26) | namespace oneflow { FILE: oneflow/ir/lib/OneFlow/OneFlowInferReturnTypes.cpp type mlir (line 22) | namespace mlir { type oneflow (line 24) | namespace oneflow { function getBlobDescFromTensorType (line 28) | std::unique_ptr<::oneflow::BlobDesc> getBlobDescFromTensorType(Tenso... function Type (line 40) | Type getTensorTypeFromBlobDesc(MLIRContext* context, const ::oneflow... function LogicalResult (line 52) | LogicalResult ConvertUserOp(llvm::StringRef op_type_name, ::oneflow:... function getResultSize (line 65) | size_t getResultSize(DictionaryAttr attributes) { function inferReturnTypesWithOpTypeName (line 75) | ::mlir::LogicalResult inferReturnTypesWithOpTypeName( FILE: oneflow/ir/lib/OneFlow/OneFlowOpFolders.cpp type mlir (line 34) | namespace mlir { type oneflow (line 35) | namespace oneflow { function StringAttr (line 42) | StringAttr GenNewVariableOpName(MLIRContext* ctx, const std::string&... function MLIRDataTypesAreSame (line 47) | bool MLIRDataTypesAreSame(const std::vector& data_types) { function DictionaryAttrsHaveSameDataType (line 55) | bool DictionaryAttrsHaveSameDataType(const std::vector operands, function OpFoldResult (line 84) | OpFoldResult BinaryFold(MLIRContext* ctx, ArrayRef operands, function OpFoldResult (line 119) | OpFoldResult FrozenVariableOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 132) | OpFoldResult TransposeOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 141) | OpFoldResult ReshapeOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 153) | OpFoldResult ScalarAddOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 165) | OpFoldResult SqrtOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 170) | OpFoldResult BroadcastMulOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 175) | OpFoldResult BroadcastDivOp::fold(FoldAdaptor adaptor) { function OpFoldResult (line 180) | OpFoldResult BroadcastSubOp::fold(FoldAdaptor adaptor) { FILE: oneflow/ir/lib/OneFlow/OneFlowOpTraits.cpp type mlir (line 19) | namespace mlir { type OpTrait (line 21) | namespace OpTrait { function HaveIdenticalPlacement (line 26) | bool HaveIdenticalPlacement(mlir::Operation* a, mlir::Operation* b) { type impl (line 36) | namespace impl { function OpFoldResult (line 38) | OpFoldResult foldIdempotentOfIdenticalPlacement(Operation* op) { function OpFoldResult (line 47) | OpFoldResult foldInvolutionOfIdenticalPlacement(Operation* op) { function LogicalResult (line 56) | LogicalResult VerifyIsOpConfCompatible(Operation* op) { function LogicalResult (line 72) | LogicalResult VerifyIsImportCompatible(Operation* op) { function LogicalResult (line 91) | LogicalResult saveAttrToOpConf(Operation* op, ::oneflow::OperatorC... function StringAttr (line 95) | StringAttr getOpName(Operation* op) { function StringAttr (line 99) | StringAttr getDeviceTag(Operation* op) { function ArrayAttr (line 103) | ArrayAttr getDeviceName(Operation* op) { function IntegerAttr (line 108) | IntegerAttr getScopeSymbolID(Operation* op) { function ArrayAttr (line 112) | ArrayAttr getHierarchy(Operation* op) { function LogicalResult (line 117) | LogicalResult saveAttrsToNamedAttrList(Operation* op, NamedAttrLis... FILE: oneflow/ir/lib/OneFlow/OneFlowOps.cpp type mlir (line 56) | namespace mlir { type oneflow (line 58) | namespace oneflow { function OperandRange (line 60) | OperandRange UserOp::dataInputOperands() { return getDataInput(); } function OperandRange (line 61) | OperandRange UserOp::ctrlInputOperands() { return getCtrlInputs(); } function ResultRange (line 62) | ResultRange UserOp::dataOutputResults() { return getDataOutput(); } function Value (line 63) | Value UserOp::ctrlOutputResult() { return getCtrlOutput(); } function OperandRange (line 65) | OperandRange SystemOp::dataInputOperands() { return getDataInput(); } function OperandRange (line 66) | OperandRange SystemOp::ctrlInputOperands() { return getCtrlInputs(); } function ResultRange (line 67) | ResultRange SystemOp::dataOutputResults() { return getDataOutput(); } function Value (line 68) | Value SystemOp::ctrlOutputResult() { return getCtrlOutput(); } function OperandRange (line 70) | OperandRange VariableOp::dataInputOperands() { return {operand_begin... function OperandRange (line 71) | OperandRange VariableOp::ctrlInputOperands() { return getCtrlInputs(... function ResultRange (line 72) | ResultRange VariableOp::dataOutputResults() { return getOutput().dyn... function Value (line 73) | Value VariableOp::ctrlOutputResult() { return getCtrlOutput(); } function OperandRange (line 75) | OperandRange InputOp::dataInputOperands() { return getODSOperands(0); } function OperandRange (line 76) | OperandRange InputOp::ctrlInputOperands() { return getCtrlInputs(); } function ResultRange (line 77) | ResultRange InputOp::dataOutputResults() { return getOutput().dyn_ca... function Value (line 78) | Value InputOp::ctrlOutputResult() { return getCtrlOutput(); } function OperandRange (line 80) | OperandRange OutputOp::dataInputOperands() { return getODSOperands(0... function OperandRange (line 81) | OperandRange OutputOp::ctrlInputOperands() { return getCtrlInputs(); } function ResultRange (line 82) | ResultRange OutputOp::dataOutputResults() { return getOutput().dyn_c... function Value (line 83) | Value OutputOp::ctrlOutputResult() { return getCtrlOutput(); } function ParseResult (line 85) | static ParseResult parseConstantOp(OpAsmParser& parser, OperationSta... function ArrayAttr (line 95) | ArrayAttr getSI32ArrayAttr(::mlir::PatternRewriter& rewriter, ArrayR... function LogicalResult (line 103) | LogicalResult TrimRedundantCtrl(Operation* op, PatternRewriter& rewr... function IsCtrlOutTrimmed (line 127) | bool IsCtrlOutTrimmed(UserOp& op) { return !op.getCtrlOutput(); } function IsCtrlInAbsent (line 129) | bool IsCtrlInAbsent(UserOp& op) { function getValuesFromIntArrayAttribute (line 139) | static void getValuesFromIntArrayAttribute(ArrayAttr attr, SmallVect... type ConcreteUserOps (line 145) | struct ConcreteUserOps : public OpRewritePattern { method ConcreteUserOps (line 146) | explicit ConcreteUserOps(MLIRContext* context) method LogicalResult (line 148) | LogicalResult matchAndRewrite(UserOp op, PatternRewriter& rewriter... type ConcreteSystemOps (line 205) | struct ConcreteSystemOps : public OpRewritePattern { method ConcreteSystemOps (line 206) | explicit ConcreteSystemOps(MLIRContext* context) method LogicalResult (line 208) | LogicalResult matchAndRewrite(oneflow::SystemOp op, PatternRewrite... type ConvertAddOpWithArity (line 217) | struct ConvertAddOpWithArity : public OpRewritePattern { method ConvertAddOpWithArity (line 218) | explicit ConvertAddOpWithArity(MLIRContext* context) method LogicalResult (line 220) | LogicalResult matchAndRewrite(AddNOp op, PatternRewriter& rewriter... type ConcreteSystemOpPattern (line 244) | struct ConcreteSystemOpPattern : public OpRewritePattern { method ConcreteSystemOpPattern (line 245) | explicit ConcreteSystemOpPattern(MLIRContext* context) method LogicalResult (line 247) | LogicalResult matchAndRewrite(OpType op, PatternRewriter& rewriter... function ParseResult (line 286) | ParseResult Job::parse(OpAsmParser& parser, OperationState& result) { function LogicalResult (line 301) | LogicalResult Job::verify() { function LogicalResult (line 320) | LogicalResult ReturnOp::verify() { type NormalizationInferencePattern (line 338) | struct NormalizationInferencePattern : public OpRewritePattern GetCtrlIntputOperands(Operation* op) { function GetCtrlOutputResult (line 382) | llvm::Optional GetCtrlOutputResult(Operation* op) { FILE: oneflow/ir/lib/OneFlow/OneFlowRewrites.cpp type mlir (line 44) | namespace mlir { type oneflow (line 46) | namespace oneflow { function getUniqName (line 52) | std::string getUniqName(llvm::StringRef name) { function Operation (line 57) | static Operation* CopyUserOpAttrs(PatternRewriter& rewriter, Operati... function Operation (line 74) | static Operation* BuildFusedBiasAddMaskScaleOpWithRate(PatternRewrit... function Operation (line 93) | static Operation* CreateConv2dAndErasePad(PatternRewriter& rewriter,... function IntegerAttr (line 131) | IntegerAttr getSI64IntegerAttr(::mlir::PatternRewriter& rewriter, in... function Attribute (line 136) | static Attribute GetHeadSizeFromTranpose(PatternRewriter& rewriter, ... function NamedAttrList (line 142) | NamedAttrList GetUserOpCommonAttrs(MLIRContext* ctx, const std::stri... function Operation (line 153) | static Operation* CreateConv2DBatchNorm(PatternRewriter& rewriter, A... function LogicalResult (line 240) | static LogicalResult IsPaddingCouldBeAssimilatedIntoConv(PatternRewr... function LogicalResult (line 279) | static LogicalResult IsNotNestedInJit(PatternRewriter& rewriter, Ope... function LogicalResult (line 283) | static LogicalResult IsScalarTensor(PatternRewriter& rewriter, Value... function LogicalResult (line 292) | static LogicalResult IsScalarEqualSqrtDim(PatternRewriter& rewriter,... function LogicalResult (line 301) | static LogicalResult IsScalarEqualSqrtDimReciprocal(PatternRewriter&... function Attribute (line 310) | static Attribute GetReciprocal(PatternRewriter& rewriter, Attribute ... type rewrites (line 316) | namespace rewrites { function populateRewrites (line 318) | void populateRewrites(RewritePatternSet& patterns) { function GetDefaultSeed (line 330) | mlir::IntegerAttr GetDefaultSeed(::mlir::PatternRewriter& rewriter) { type constraints (line 337) | namespace constraints { function populateConstraints (line 339) | void populateConstraints(RewritePatternSet& patterns) { FILE: oneflow/ir/lib/OneFlow/OneFlowSupport.cpp type mlir (line 37) | namespace mlir { type oneflow (line 39) | namespace oneflow { type support (line 41) | namespace support { function GetInputKeys (line 43) | std::vector GetInputKeys(const std::string& op_type_n... function GetOutputKeys (line 49) | std::vector GetOutputKeys(const std::string& op_type_... function MakeDevice (line 57) | ::oneflow::Symbol<::oneflow::Device> MakeDevice(const mlir::Attrib... function __TensorToDenseElementsAttr (line 68) | mlir::DenseElementsAttr __TensorToDenseElementsAttr( function __DenseElementsAttrToTensor (line 88) | std::shared_ptr<::oneflow::one::Tensor> __DenseElementsAttrToTensor( function __DenseElementsAttrToTensor (line 116) | void __DenseElementsAttrToTensor(const mlir::DenseElementsAttr den... function TensorToDenseElementsAttr (line 162) | mlir::DenseElementsAttr TensorToDenseElementsAttr( function DenseElementsAttrToTensor (line 178) | std::shared_ptr<::oneflow::one::Tensor> DenseElementsAttrToTensor( function DenseElementsAttrToTensor (line 194) | void DenseElementsAttrToTensor(const mlir::Attribute& dense_attr, function FromMLIRTypeToOFDataType (line 212) | FailureOr<::oneflow::DataType> FromMLIRTypeToOFDataType(Type mlir_... function FromMLIRDataTypeToOFDataType (line 236) | FailureOr<::oneflow::DataType> FromMLIRDataTypeToOFDataType(::mlir... function FromMLIRAttrToOFDataType (line 261) | FailureOr<::oneflow::DataType> FromMLIRAttrToOFDataType(Attribute ... FILE: oneflow/ir/lib/OneFlow/OneFlowUtils.cpp type mlir (line 18) | namespace mlir { type oneflow (line 19) | namespace oneflow { function CheckEnableIRPrinting (line 21) | void CheckEnableIRPrinting(mlir::PassManager& pm) { function StringRef (line 28) | StringRef SanitizeIdentifier(StringRef name, SmallString<16>& buffer... FILE: oneflow/ir/lib/OneFlow/PDLL/AllocEliminationPatterns.cpp type mlir (line 31) | namespace mlir { type oneflow (line 33) | namespace oneflow { function populateAllocEliminationPatterns (line 35) | void populateAllocEliminationPatterns(RewritePatternSet& patterns) { FILE: oneflow/ir/lib/OneFlow/PDLL/ForwardOpPatterns.cpp type mlir (line 29) | namespace mlir { type oneflow (line 31) | namespace oneflow { function populateForwardOpPatterns (line 33) | void populateForwardOpPatterns(RewritePatternSet& patterns) { FILE: oneflow/ir/lib/OneFlow/PDLL/FuseConv2DBatchNormPattern.cpp type mlir (line 29) | namespace mlir { type oneflow (line 31) | namespace oneflow { function populateFuseConv2DBatchNormPattern (line 33) | void populateFuseConv2DBatchNormPattern(RewritePatternSet& patterns) { FILE: oneflow/ir/lib/OneFlow/PDLL/FuseOpsWithBackwardImplPattern.cpp type mlir (line 29) | namespace mlir { type oneflow (line 31) | namespace oneflow { function populateFuseOpsWithBackwardImplPattern (line 33) | void populateFuseOpsWithBackwardImplPattern(RewritePatternSet& patte... FILE: oneflow/ir/lib/OneFlow/PDLL/NormalizationPatterns.cpp type mlir (line 29) | namespace mlir { type oneflow (line 31) | namespace oneflow { function populateNormalizationOpPatterns (line 33) | void populateNormalizationOpPatterns(RewritePatternSet& patterns) { FILE: oneflow/ir/lib/OneFlow/Passes.cpp type mlir (line 137) | namespace mlir { type oneflow (line 138) | namespace oneflow { function GetPtr (line 140) | LLVM::LLVMPointerType GetPtr(::mlir::PatternRewriter& rewriter) { function LogicalResult (line 145) | LogicalResult DumpAssembly(::mlir::PatternRewriter& rewriter, T op, ... function DeclareKernelLaunchCInterface (line 164) | LLVM::LLVMFuncOp DeclareKernelLaunchCInterface(::mlir::PatternRewrit... function DeclareOrGetGlobalString (line 180) | LLVM::GlobalOp DeclareOrGetGlobalString(::mlir::PatternRewriter& rew... function ModuleOp (line 198) | ModuleOp GetModuleOpFromJobBodyOp(Operation* op) { function InsertKernelOFFuncOp (line 204) | func::FuncOp InsertKernelOFFuncOp(::mlir::PatternRewriter& rewriter,... function CreateGPUMemcpyOpFromMemrefCopy (line 238) | ::llvm::SmallVector<::mlir::Value, 4> CreateGPUMemcpyOpFromMemrefCopy( function HasZeroPadding (line 256) | bool HasZeroPadding(mlir::ArrayAttr padding) { function NamedAttrList (line 263) | NamedAttrList GetUserOpCommonAttrs(MLIRContext* ctx, const std::stri... type ReplaceVariablePattern (line 275) | struct ReplaceVariablePattern : public ::mlir::RewritePattern { method ReplaceVariablePattern (line 276) | explicit ReplaceVariablePattern(::mlir::MLIRContext* context) method matchAndRewrite (line 278) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, type ReplaceVariableIrPattern (line 307) | struct ReplaceVariableIrPattern : public ::mlir::RewritePattern { method ReplaceVariableIrPattern (line 308) | explicit ReplaceVariableIrPattern(::mlir::MLIRContext* context) method matchAndRewrite (line 310) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, function LogicalResult (line 375) | LogicalResult InitTransposeAttributes(Operation* op, NamedAttrList& ... function IsAddToOutputNone (line 385) | bool IsAddToOutputNone(ValueRange value) { return (int)value.size() ... function getChannelLastTransposePerm (line 387) | llvm::SmallVector getChannelLastTransposePerm() { return {0... function getChannelFirstTransposePerm (line 389) | llvm::SmallVector getChannelFirstTransposePerm() { return {... function getInputOperandTransposeOp (line 391) | llvm::SmallVector getInputOperandTransposeOp(NCHWCom... function TransposeOp (line 408) | TransposeOp getResultTransposeOp(NCHWCompatible op, Value val, Named... function IsInsertTransposeOpBefore (line 421) | bool IsInsertTransposeOpBefore(NCHWCompatible op, PatternRewriter& r... type FusedConsecutiveAddPattern (line 448) | struct FusedConsecutiveAddPattern : public OpRewritePattern { method FusedConsecutiveAddPattern (line 449) | explicit FusedConsecutiveAddPattern(mlir::MLIRContext* context) function LogicalResult (line 457) | LogicalResult TryFusedConsecutiveAdd(Op op, const SmallVector::matchAndRewrite(Ad... function LogicalResult (line 502) | LogicalResult FusedConsecutiveAddPattern::matchAndRewrite(Ad... type AutoNhwcPattern (line 507) | struct AutoNhwcPattern : public OpInterfaceRewritePattern, std::vector& wrap_ops, m... type ExtractKernelLaunchTensorPattern (line 839) | struct ExtractKernelLaunchTensorPattern : public mlir::OpRewritePatt... method ExtractArgTensors (line 840) | static func::FuncOp ExtractArgTensors(func::FuncOp op, mlir::Patte... method ExtractRetTensors (line 868) | static func::FuncOp ExtractRetTensors(func::FuncOp op, mlir::Patte... method ExtractKernelLaunchTensorPattern (line 887) | explicit ExtractKernelLaunchTensorPattern(mlir::MLIRContext* context) method matchAndRewrite (line 889) | mlir::LogicalResult matchAndRewrite(func::FuncOp op, type TrimReturnAsVoidPattern (line 903) | struct TrimReturnAsVoidPattern : public mlir::OpRewritePattern&, mlir::Operati... method IsPackagable (line 934) | virtual bool IsPackagable(mlir::Operation* op) const { method matchAndRewrite (line 940) | mlir::LogicalResult matchAndRewrite(oneflow::Job op, type KernelLaunchSimplePattern (line 964) | struct KernelLaunchSimplePattern : public KernelLaunchPattern { method KernelLaunchSimplePattern (line 965) | explicit KernelLaunchSimplePattern(mlir::MLIRContext* context) : K... method IsSameDevice (line 967) | bool IsSameDevice(std::vector& ops, mlir::Operation* o... method IsConsecutive (line 990) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... type KernelLaunchWithCudaGraphPattern (line 996) | struct KernelLaunchWithCudaGraphPattern : public KernelLaunchSimpleP... method KernelLaunchWithCudaGraphPattern (line 997) | explicit KernelLaunchWithCudaGraphPattern(mlir::MLIRContext* context) method IsOpCudaGraphSupport (line 1000) | bool IsOpCudaGraphSupport(mlir::Operation* op) const { method IsSameCudaGraphSupport (line 1006) | bool IsSameCudaGraphSupport(std::vector& ops, mlir::Op... method IsConsecutive (line 1012) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... function AddLoweringToLinalgMemRefPasses (line 1018) | void AddLoweringToLinalgMemRefPasses(PassManager& pm) { function LogicalResult (line 1043) | LogicalResult LowerModuleToLLVM(mlir::MLIRContext* context, ModuleOp... function AddLoweringLinalgOnBufferToGpuWithStdPasses (line 1064) | void AddLoweringLinalgOnBufferToGpuWithStdPasses(PassManager& pm) { function AddAdheringCubinToGpuModulePasses (line 1080) | void AddAdheringCubinToGpuModulePasses(PassManager& pm) { function AddLoweringGpuToLLVMPasses (line 1087) | void AddLoweringGpuToLLVMPasses(PassManager& pm) { function LogicalResult (line 1096) | LogicalResult LowerModuleToCUDALLVM(mlir::MLIRContext* context, Modu... function populateWrapOpsToKernelLaunchPatterns (line 1112) | void populateWrapOpsToKernelLaunchPatterns(::mlir::RewritePatternSet... function populateFuserForExistingOp (line 1127) | void populateFuserForExistingOp(::mlir::RewritePatternSet& patterns) { function populateAutoNhwcPatterns (line 1136) | void populateAutoNhwcPatterns(::mlir::RewritePatternSet& patterns) { function populateGpuHelperPatterns (line 1144) | void populateGpuHelperPatterns(::mlir::RewritePatternSet& patterns) { function populatePreConvertInferenceOp (line 1148) | void populatePreConvertInferenceOp(::mlir::RewritePatternSet& patter... function populateConvertInferenceOp (line 1152) | void populateConvertInferenceOp(::mlir::RewritePatternSet& patterns) { function populatePostConvertInferenceOp (line 1156) | void populatePostConvertInferenceOp(::mlir::RewritePatternSet& patte... type oneflow (line 445) | namespace oneflow { function GetPtr (line 140) | LLVM::LLVMPointerType GetPtr(::mlir::PatternRewriter& rewriter) { function LogicalResult (line 145) | LogicalResult DumpAssembly(::mlir::PatternRewriter& rewriter, T op, ... function DeclareKernelLaunchCInterface (line 164) | LLVM::LLVMFuncOp DeclareKernelLaunchCInterface(::mlir::PatternRewrit... function DeclareOrGetGlobalString (line 180) | LLVM::GlobalOp DeclareOrGetGlobalString(::mlir::PatternRewriter& rew... function ModuleOp (line 198) | ModuleOp GetModuleOpFromJobBodyOp(Operation* op) { function InsertKernelOFFuncOp (line 204) | func::FuncOp InsertKernelOFFuncOp(::mlir::PatternRewriter& rewriter,... function CreateGPUMemcpyOpFromMemrefCopy (line 238) | ::llvm::SmallVector<::mlir::Value, 4> CreateGPUMemcpyOpFromMemrefCopy( function HasZeroPadding (line 256) | bool HasZeroPadding(mlir::ArrayAttr padding) { function NamedAttrList (line 263) | NamedAttrList GetUserOpCommonAttrs(MLIRContext* ctx, const std::stri... type ReplaceVariablePattern (line 275) | struct ReplaceVariablePattern : public ::mlir::RewritePattern { method ReplaceVariablePattern (line 276) | explicit ReplaceVariablePattern(::mlir::MLIRContext* context) method matchAndRewrite (line 278) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, type ReplaceVariableIrPattern (line 307) | struct ReplaceVariableIrPattern : public ::mlir::RewritePattern { method ReplaceVariableIrPattern (line 308) | explicit ReplaceVariableIrPattern(::mlir::MLIRContext* context) method matchAndRewrite (line 310) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, function LogicalResult (line 375) | LogicalResult InitTransposeAttributes(Operation* op, NamedAttrList& ... function IsAddToOutputNone (line 385) | bool IsAddToOutputNone(ValueRange value) { return (int)value.size() ... function getChannelLastTransposePerm (line 387) | llvm::SmallVector getChannelLastTransposePerm() { return {0... function getChannelFirstTransposePerm (line 389) | llvm::SmallVector getChannelFirstTransposePerm() { return {... function getInputOperandTransposeOp (line 391) | llvm::SmallVector getInputOperandTransposeOp(NCHWCom... function TransposeOp (line 408) | TransposeOp getResultTransposeOp(NCHWCompatible op, Value val, Named... function IsInsertTransposeOpBefore (line 421) | bool IsInsertTransposeOpBefore(NCHWCompatible op, PatternRewriter& r... type FusedConsecutiveAddPattern (line 448) | struct FusedConsecutiveAddPattern : public OpRewritePattern { method FusedConsecutiveAddPattern (line 449) | explicit FusedConsecutiveAddPattern(mlir::MLIRContext* context) function LogicalResult (line 457) | LogicalResult TryFusedConsecutiveAdd(Op op, const SmallVector::matchAndRewrite(Ad... function LogicalResult (line 502) | LogicalResult FusedConsecutiveAddPattern::matchAndRewrite(Ad... type AutoNhwcPattern (line 507) | struct AutoNhwcPattern : public OpInterfaceRewritePattern, std::vector& wrap_ops, m... type ExtractKernelLaunchTensorPattern (line 839) | struct ExtractKernelLaunchTensorPattern : public mlir::OpRewritePatt... method ExtractArgTensors (line 840) | static func::FuncOp ExtractArgTensors(func::FuncOp op, mlir::Patte... method ExtractRetTensors (line 868) | static func::FuncOp ExtractRetTensors(func::FuncOp op, mlir::Patte... method ExtractKernelLaunchTensorPattern (line 887) | explicit ExtractKernelLaunchTensorPattern(mlir::MLIRContext* context) method matchAndRewrite (line 889) | mlir::LogicalResult matchAndRewrite(func::FuncOp op, type TrimReturnAsVoidPattern (line 903) | struct TrimReturnAsVoidPattern : public mlir::OpRewritePattern&, mlir::Operati... method IsPackagable (line 934) | virtual bool IsPackagable(mlir::Operation* op) const { method matchAndRewrite (line 940) | mlir::LogicalResult matchAndRewrite(oneflow::Job op, type KernelLaunchSimplePattern (line 964) | struct KernelLaunchSimplePattern : public KernelLaunchPattern { method KernelLaunchSimplePattern (line 965) | explicit KernelLaunchSimplePattern(mlir::MLIRContext* context) : K... method IsSameDevice (line 967) | bool IsSameDevice(std::vector& ops, mlir::Operation* o... method IsConsecutive (line 990) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... type KernelLaunchWithCudaGraphPattern (line 996) | struct KernelLaunchWithCudaGraphPattern : public KernelLaunchSimpleP... method KernelLaunchWithCudaGraphPattern (line 997) | explicit KernelLaunchWithCudaGraphPattern(mlir::MLIRContext* context) method IsOpCudaGraphSupport (line 1000) | bool IsOpCudaGraphSupport(mlir::Operation* op) const { method IsSameCudaGraphSupport (line 1006) | bool IsSameCudaGraphSupport(std::vector& ops, mlir::Op... method IsConsecutive (line 1012) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... function AddLoweringToLinalgMemRefPasses (line 1018) | void AddLoweringToLinalgMemRefPasses(PassManager& pm) { function LogicalResult (line 1043) | LogicalResult LowerModuleToLLVM(mlir::MLIRContext* context, ModuleOp... function AddLoweringLinalgOnBufferToGpuWithStdPasses (line 1064) | void AddLoweringLinalgOnBufferToGpuWithStdPasses(PassManager& pm) { function AddAdheringCubinToGpuModulePasses (line 1080) | void AddAdheringCubinToGpuModulePasses(PassManager& pm) { function AddLoweringGpuToLLVMPasses (line 1087) | void AddLoweringGpuToLLVMPasses(PassManager& pm) { function LogicalResult (line 1096) | LogicalResult LowerModuleToCUDALLVM(mlir::MLIRContext* context, Modu... function populateWrapOpsToKernelLaunchPatterns (line 1112) | void populateWrapOpsToKernelLaunchPatterns(::mlir::RewritePatternSet... function populateFuserForExistingOp (line 1127) | void populateFuserForExistingOp(::mlir::RewritePatternSet& patterns) { function populateAutoNhwcPatterns (line 1136) | void populateAutoNhwcPatterns(::mlir::RewritePatternSet& patterns) { function populateGpuHelperPatterns (line 1144) | void populateGpuHelperPatterns(::mlir::RewritePatternSet& patterns) { function populatePreConvertInferenceOp (line 1148) | void populatePreConvertInferenceOp(::mlir::RewritePatternSet& patter... function populateConvertInferenceOp (line 1152) | void populateConvertInferenceOp(::mlir::RewritePatternSet& patterns) { function populatePostConvertInferenceOp (line 1156) | void populatePostConvertInferenceOp(::mlir::RewritePatternSet& patte... type mlir (line 443) | namespace mlir { type oneflow (line 138) | namespace oneflow { function GetPtr (line 140) | LLVM::LLVMPointerType GetPtr(::mlir::PatternRewriter& rewriter) { function LogicalResult (line 145) | LogicalResult DumpAssembly(::mlir::PatternRewriter& rewriter, T op, ... function DeclareKernelLaunchCInterface (line 164) | LLVM::LLVMFuncOp DeclareKernelLaunchCInterface(::mlir::PatternRewrit... function DeclareOrGetGlobalString (line 180) | LLVM::GlobalOp DeclareOrGetGlobalString(::mlir::PatternRewriter& rew... function ModuleOp (line 198) | ModuleOp GetModuleOpFromJobBodyOp(Operation* op) { function InsertKernelOFFuncOp (line 204) | func::FuncOp InsertKernelOFFuncOp(::mlir::PatternRewriter& rewriter,... function CreateGPUMemcpyOpFromMemrefCopy (line 238) | ::llvm::SmallVector<::mlir::Value, 4> CreateGPUMemcpyOpFromMemrefCopy( function HasZeroPadding (line 256) | bool HasZeroPadding(mlir::ArrayAttr padding) { function NamedAttrList (line 263) | NamedAttrList GetUserOpCommonAttrs(MLIRContext* ctx, const std::stri... type ReplaceVariablePattern (line 275) | struct ReplaceVariablePattern : public ::mlir::RewritePattern { method ReplaceVariablePattern (line 276) | explicit ReplaceVariablePattern(::mlir::MLIRContext* context) method matchAndRewrite (line 278) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, type ReplaceVariableIrPattern (line 307) | struct ReplaceVariableIrPattern : public ::mlir::RewritePattern { method ReplaceVariableIrPattern (line 308) | explicit ReplaceVariableIrPattern(::mlir::MLIRContext* context) method matchAndRewrite (line 310) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, function LogicalResult (line 375) | LogicalResult InitTransposeAttributes(Operation* op, NamedAttrList& ... function IsAddToOutputNone (line 385) | bool IsAddToOutputNone(ValueRange value) { return (int)value.size() ... function getChannelLastTransposePerm (line 387) | llvm::SmallVector getChannelLastTransposePerm() { return {0... function getChannelFirstTransposePerm (line 389) | llvm::SmallVector getChannelFirstTransposePerm() { return {... function getInputOperandTransposeOp (line 391) | llvm::SmallVector getInputOperandTransposeOp(NCHWCom... function TransposeOp (line 408) | TransposeOp getResultTransposeOp(NCHWCompatible op, Value val, Named... function IsInsertTransposeOpBefore (line 421) | bool IsInsertTransposeOpBefore(NCHWCompatible op, PatternRewriter& r... type FusedConsecutiveAddPattern (line 448) | struct FusedConsecutiveAddPattern : public OpRewritePattern { method FusedConsecutiveAddPattern (line 449) | explicit FusedConsecutiveAddPattern(mlir::MLIRContext* context) function LogicalResult (line 457) | LogicalResult TryFusedConsecutiveAdd(Op op, const SmallVector::matchAndRewrite(Ad... function LogicalResult (line 502) | LogicalResult FusedConsecutiveAddPattern::matchAndRewrite(Ad... type AutoNhwcPattern (line 507) | struct AutoNhwcPattern : public OpInterfaceRewritePattern, std::vector& wrap_ops, m... type ExtractKernelLaunchTensorPattern (line 839) | struct ExtractKernelLaunchTensorPattern : public mlir::OpRewritePatt... method ExtractArgTensors (line 840) | static func::FuncOp ExtractArgTensors(func::FuncOp op, mlir::Patte... method ExtractRetTensors (line 868) | static func::FuncOp ExtractRetTensors(func::FuncOp op, mlir::Patte... method ExtractKernelLaunchTensorPattern (line 887) | explicit ExtractKernelLaunchTensorPattern(mlir::MLIRContext* context) method matchAndRewrite (line 889) | mlir::LogicalResult matchAndRewrite(func::FuncOp op, type TrimReturnAsVoidPattern (line 903) | struct TrimReturnAsVoidPattern : public mlir::OpRewritePattern&, mlir::Operati... method IsPackagable (line 934) | virtual bool IsPackagable(mlir::Operation* op) const { method matchAndRewrite (line 940) | mlir::LogicalResult matchAndRewrite(oneflow::Job op, type KernelLaunchSimplePattern (line 964) | struct KernelLaunchSimplePattern : public KernelLaunchPattern { method KernelLaunchSimplePattern (line 965) | explicit KernelLaunchSimplePattern(mlir::MLIRContext* context) : K... method IsSameDevice (line 967) | bool IsSameDevice(std::vector& ops, mlir::Operation* o... method IsConsecutive (line 990) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... type KernelLaunchWithCudaGraphPattern (line 996) | struct KernelLaunchWithCudaGraphPattern : public KernelLaunchSimpleP... method KernelLaunchWithCudaGraphPattern (line 997) | explicit KernelLaunchWithCudaGraphPattern(mlir::MLIRContext* context) method IsOpCudaGraphSupport (line 1000) | bool IsOpCudaGraphSupport(mlir::Operation* op) const { method IsSameCudaGraphSupport (line 1006) | bool IsSameCudaGraphSupport(std::vector& ops, mlir::Op... method IsConsecutive (line 1012) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... function AddLoweringToLinalgMemRefPasses (line 1018) | void AddLoweringToLinalgMemRefPasses(PassManager& pm) { function LogicalResult (line 1043) | LogicalResult LowerModuleToLLVM(mlir::MLIRContext* context, ModuleOp... function AddLoweringLinalgOnBufferToGpuWithStdPasses (line 1064) | void AddLoweringLinalgOnBufferToGpuWithStdPasses(PassManager& pm) { function AddAdheringCubinToGpuModulePasses (line 1080) | void AddAdheringCubinToGpuModulePasses(PassManager& pm) { function AddLoweringGpuToLLVMPasses (line 1087) | void AddLoweringGpuToLLVMPasses(PassManager& pm) { function LogicalResult (line 1096) | LogicalResult LowerModuleToCUDALLVM(mlir::MLIRContext* context, Modu... function populateWrapOpsToKernelLaunchPatterns (line 1112) | void populateWrapOpsToKernelLaunchPatterns(::mlir::RewritePatternSet... function populateFuserForExistingOp (line 1127) | void populateFuserForExistingOp(::mlir::RewritePatternSet& patterns) { function populateAutoNhwcPatterns (line 1136) | void populateAutoNhwcPatterns(::mlir::RewritePatternSet& patterns) { function populateGpuHelperPatterns (line 1144) | void populateGpuHelperPatterns(::mlir::RewritePatternSet& patterns) { function populatePreConvertInferenceOp (line 1148) | void populatePreConvertInferenceOp(::mlir::RewritePatternSet& patter... function populateConvertInferenceOp (line 1152) | void populateConvertInferenceOp(::mlir::RewritePatternSet& patterns) { function populatePostConvertInferenceOp (line 1156) | void populatePostConvertInferenceOp(::mlir::RewritePatternSet& patte... type oneflow (line 445) | namespace oneflow { function GetPtr (line 140) | LLVM::LLVMPointerType GetPtr(::mlir::PatternRewriter& rewriter) { function LogicalResult (line 145) | LogicalResult DumpAssembly(::mlir::PatternRewriter& rewriter, T op, ... function DeclareKernelLaunchCInterface (line 164) | LLVM::LLVMFuncOp DeclareKernelLaunchCInterface(::mlir::PatternRewrit... function DeclareOrGetGlobalString (line 180) | LLVM::GlobalOp DeclareOrGetGlobalString(::mlir::PatternRewriter& rew... function ModuleOp (line 198) | ModuleOp GetModuleOpFromJobBodyOp(Operation* op) { function InsertKernelOFFuncOp (line 204) | func::FuncOp InsertKernelOFFuncOp(::mlir::PatternRewriter& rewriter,... function CreateGPUMemcpyOpFromMemrefCopy (line 238) | ::llvm::SmallVector<::mlir::Value, 4> CreateGPUMemcpyOpFromMemrefCopy( function HasZeroPadding (line 256) | bool HasZeroPadding(mlir::ArrayAttr padding) { function NamedAttrList (line 263) | NamedAttrList GetUserOpCommonAttrs(MLIRContext* ctx, const std::stri... type ReplaceVariablePattern (line 275) | struct ReplaceVariablePattern : public ::mlir::RewritePattern { method ReplaceVariablePattern (line 276) | explicit ReplaceVariablePattern(::mlir::MLIRContext* context) method matchAndRewrite (line 278) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, type ReplaceVariableIrPattern (line 307) | struct ReplaceVariableIrPattern : public ::mlir::RewritePattern { method ReplaceVariableIrPattern (line 308) | explicit ReplaceVariableIrPattern(::mlir::MLIRContext* context) method matchAndRewrite (line 310) | ::mlir::LogicalResult matchAndRewrite(::mlir::Operation* op0, function LogicalResult (line 375) | LogicalResult InitTransposeAttributes(Operation* op, NamedAttrList& ... function IsAddToOutputNone (line 385) | bool IsAddToOutputNone(ValueRange value) { return (int)value.size() ... function getChannelLastTransposePerm (line 387) | llvm::SmallVector getChannelLastTransposePerm() { return {0... function getChannelFirstTransposePerm (line 389) | llvm::SmallVector getChannelFirstTransposePerm() { return {... function getInputOperandTransposeOp (line 391) | llvm::SmallVector getInputOperandTransposeOp(NCHWCom... function TransposeOp (line 408) | TransposeOp getResultTransposeOp(NCHWCompatible op, Value val, Named... function IsInsertTransposeOpBefore (line 421) | bool IsInsertTransposeOpBefore(NCHWCompatible op, PatternRewriter& r... type FusedConsecutiveAddPattern (line 448) | struct FusedConsecutiveAddPattern : public OpRewritePattern { method FusedConsecutiveAddPattern (line 449) | explicit FusedConsecutiveAddPattern(mlir::MLIRContext* context) function LogicalResult (line 457) | LogicalResult TryFusedConsecutiveAdd(Op op, const SmallVector::matchAndRewrite(Ad... function LogicalResult (line 502) | LogicalResult FusedConsecutiveAddPattern::matchAndRewrite(Ad... type AutoNhwcPattern (line 507) | struct AutoNhwcPattern : public OpInterfaceRewritePattern, std::vector& wrap_ops, m... type ExtractKernelLaunchTensorPattern (line 839) | struct ExtractKernelLaunchTensorPattern : public mlir::OpRewritePatt... method ExtractArgTensors (line 840) | static func::FuncOp ExtractArgTensors(func::FuncOp op, mlir::Patte... method ExtractRetTensors (line 868) | static func::FuncOp ExtractRetTensors(func::FuncOp op, mlir::Patte... method ExtractKernelLaunchTensorPattern (line 887) | explicit ExtractKernelLaunchTensorPattern(mlir::MLIRContext* context) method matchAndRewrite (line 889) | mlir::LogicalResult matchAndRewrite(func::FuncOp op, type TrimReturnAsVoidPattern (line 903) | struct TrimReturnAsVoidPattern : public mlir::OpRewritePattern&, mlir::Operati... method IsPackagable (line 934) | virtual bool IsPackagable(mlir::Operation* op) const { method matchAndRewrite (line 940) | mlir::LogicalResult matchAndRewrite(oneflow::Job op, type KernelLaunchSimplePattern (line 964) | struct KernelLaunchSimplePattern : public KernelLaunchPattern { method KernelLaunchSimplePattern (line 965) | explicit KernelLaunchSimplePattern(mlir::MLIRContext* context) : K... method IsSameDevice (line 967) | bool IsSameDevice(std::vector& ops, mlir::Operation* o... method IsConsecutive (line 990) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... type KernelLaunchWithCudaGraphPattern (line 996) | struct KernelLaunchWithCudaGraphPattern : public KernelLaunchSimpleP... method KernelLaunchWithCudaGraphPattern (line 997) | explicit KernelLaunchWithCudaGraphPattern(mlir::MLIRContext* context) method IsOpCudaGraphSupport (line 1000) | bool IsOpCudaGraphSupport(mlir::Operation* op) const { method IsSameCudaGraphSupport (line 1006) | bool IsSameCudaGraphSupport(std::vector& ops, mlir::Op... method IsConsecutive (line 1012) | bool IsConsecutive(std::vector& ops, mlir::Operation* ... function AddLoweringToLinalgMemRefPasses (line 1018) | void AddLoweringToLinalgMemRefPasses(PassManager& pm) { function LogicalResult (line 1043) | LogicalResult LowerModuleToLLVM(mlir::MLIRContext* context, ModuleOp... function AddLoweringLinalgOnBufferToGpuWithStdPasses (line 1064) | void AddLoweringLinalgOnBufferToGpuWithStdPasses(PassManager& pm) { function AddAdheringCubinToGpuModulePasses (line 1080) | void AddAdheringCubinToGpuModulePasses(PassManager& pm) { function AddLoweringGpuToLLVMPasses (line 1087) | void AddLoweringGpuToLLVMPasses(PassManager& pm) { function LogicalResult (line 1096) | LogicalResult LowerModuleToCUDALLVM(mlir::MLIRContext* context, Modu... function populateWrapOpsToKernelLaunchPatterns (line 1112) | void populateWrapOpsToKernelLaunchPatterns(::mlir::RewritePatternSet... function populateFuserForExistingOp (line 1127) | void populateFuserForExistingOp(::mlir::RewritePatternSet& patterns) { function populateAutoNhwcPatterns (line 1136) | void populateAutoNhwcPatterns(::mlir::RewritePatternSet& patterns) { function populateGpuHelperPatterns (line 1144) | void populateGpuHelperPatterns(::mlir::RewritePatternSet& patterns) { function populatePreConvertInferenceOp (line 1148) | void populatePreConvertInferenceOp(::mlir::RewritePatternSet& patter... function populateConvertInferenceOp (line 1152) | void populateConvertInferenceOp(::mlir::RewritePatternSet& patterns) { function populatePostConvertInferenceOp (line 1156) | void populatePostConvertInferenceOp(::mlir::RewritePatternSet& patte... FILE: oneflow/ir/lib/OneFlow/SBP/SBPAttributes.cpp function LogicalResult (line 28) | LogicalResult parseSBP(AsmParser& parser, ArrayAttr& args) { function printSBP (line 71) | void printSBP(AsmPrinter& printer, ArrayAttr args) { printer << args; } type mlir (line 75) | namespace mlir { type sbp (line 77) | namespace sbp { FILE: oneflow/ir/lib/OneFlow/SBP/SBPDialect.cpp type mlir (line 22) | namespace mlir { type sbp (line 24) | namespace sbp { FILE: oneflow/ir/lib/OneFlow/SBP/SBPImporter.cpp type mlir (line 21) | namespace mlir { type oneflow (line 22) | namespace oneflow { FILE: oneflow/ir/lib/OneFlow/Transform/AggregateOps.cpp type mlir (line 33) | namespace mlir { type oneflow (line 34) | namespace oneflow { type AggregateComputeOpsPattern (line 36) | struct AggregateComputeOpsPattern : public mlir::OpRewritePattern createAggregateComputeOpsPass() { FILE: oneflow/ir/lib/OneFlow/Transform/AutoNHWCOps.cpp type mlir (line 19) | namespace mlir { type oneflow (line 21) | namespace oneflow { FILE: oneflow/ir/lib/OneFlow/Transform/AutoNhwc.cpp type mlir (line 22) | namespace mlir { type oneflow (line 24) | namespace oneflow { class AutoNhwcPass (line 28) | class AutoNhwcPass : public AutoNhwcPassBase { method runOnOperation (line 29) | void runOnOperation() override { function createAutoNhwcPass (line 39) | std::unique_ptr createAutoNhwcPass() { return std::make_unique... FILE: oneflow/ir/lib/OneFlow/Transform/BufferHostRegister.cpp type mlir (line 23) | namespace mlir { type oneflow (line 25) | namespace oneflow { class BufferHostRegisterPass (line 29) | class BufferHostRegisterPass : public BufferHostRegisterPassBase { method runOnOperation (line 44) | void runOnOperation() override { function createBufferHostRegisterPass (line 53) | std::unique_ptr createBufferHostRegisterPass() { function createGpuCopyArgPass (line 57) | std::unique_ptr createGpuCopyArgPass() { return std::make_uniq... FILE: oneflow/ir/lib/OneFlow/Transform/CSEWithAttributesIgnored.cpp type mlir (line 23) | namespace mlir { type oneflow (line 25) | namespace oneflow { type EraseAttributes (line 32) | struct EraseAttributes : public mlir::OpInterfaceRewritePattern state)... method runOnOperation (line 89) | void runOnOperation() override { class CSEPutAttributes (line 100) | class CSEPutAttributes : public CSEPutAttributesBase state) { state... method runOnOperation (line 105) | void runOnOperation() override { function createCSEWithAttributesIgnored (line 118) | std::unique_ptr createCSEWithAttributesIgnored() { function createCSEPutAttributes (line 122) | std::unique_ptr createCSEPutAttributes() { return std::make_un... function createCSEPasses (line 124) | std::pair, std::unique_ptr> createCSEPas... function registerCSEPasses (line 130) | void registerCSEPasses(std::shared_ptr state) { FILE: oneflow/ir/lib/OneFlow/Transform/ConvertInferenceOp.cpp type mlir (line 23) | namespace mlir { type oneflow (line 25) | namespace oneflow { class PreConvertInferenceOpPass (line 29) | class PreConvertInferenceOpPass : public PreConvertInferenceOpPassBa... method runOnOperation (line 30) | void runOnOperation() override { class ConvertInferenceOpPass (line 38) | class ConvertInferenceOpPass : public ConvertInferenceOpPassBase createPreConvertInferenceOpPass() { function createConvertInferenceOpPass (line 64) | std::unique_ptr createConvertInferenceOpPass() { function createPostConvertInferenceOpPass (line 68) | std::unique_ptr createPostConvertInferenceOpPass() { FILE: oneflow/ir/lib/OneFlow/Transform/EliminateAllocOps.cpp type mlir (line 20) | namespace mlir { type oneflow (line 21) | namespace oneflow { class EliminateAllocOpsPass (line 24) | class EliminateAllocOpsPass : public EliminateAllocOpsPassBase createEliminateAllocOpsPass() { FILE: oneflow/ir/lib/OneFlow/Transform/FuncOps.cpp type mlir (line 24) | namespace mlir { type func (line 26) | namespace func { type FuncConversionToOneFlow (line 27) | struct FuncConversionToOneFlow final : public OpConversionPattern { method LogicalResult (line 55) | LogicalResult matchAndRewrite(Job op, OpAdaptor adaptor, type ReturnConversionToFunc (line 64) | struct ReturnConversionToFunc final : public OpConversionPattern createOneFlowJobToFuncPass() { function createFuncToOneFlowJobPass (line 116) | std::unique_ptr createFuncToOneFlowJobPass() { FILE: oneflow/ir/lib/OneFlow/Transform/GroupMatMulOps.cpp type mlir (line 18) | namespace mlir { type oneflow (line 20) | namespace oneflow { function isLinearMatmulOp (line 23) | bool isLinearMatmulOp(OpTy op) { function Value (line 33) | Value MatmulOp::matMulGetX() { return getA(); } function Value (line 35) | Value MatmulOp::matMulGetW() { return getB(); } function Value (line 37) | Value MatmulOp::matMulGetY() { return getOut(); } function Value (line 41) | Value BroadcastMatmulOp::matMulGetX() { return getA(); } function Value (line 43) | Value BroadcastMatmulOp::matMulGetW() { return getB(); } function Value (line 45) | Value BroadcastMatmulOp::matMulGetY() { return getOut(); } function Value (line 51) | Value BiasAddOp::biasAddGetBias() { return getB(); } function Value (line 53) | Value BiasAddOp::biasAddGetOut() { return getOut(); } function Value (line 55) | Value BroadcastAddOp::biasAddGetBias() { return getY(); } function Value (line 57) | Value BroadcastAddOp::biasAddGetOut() { return getZ(); } function Value (line 61) | Value FusedMatmulBiasOp::matMulGetX() { return getX(); } function Value (line 63) | Value FusedMatmulBiasOp::matMulGetW() { return getWeight(); } function Value (line 65) | Value FusedMatmulBiasOp::matMulGetY() { return getOut(); } function shouldGroupFusedMatmulBiasOp (line 69) | bool shouldGroupFusedMatmulBiasOp(FusedMatmulBiasOp& op) { function Value (line 80) | Value FusedMatmulBiasOp::biasAddGetBias() { return getBias(); } function Value (line 82) | Value FusedMatmulBiasOp::biasAddGetOut() { return getOut(); } FILE: oneflow/ir/lib/OneFlow/Transform/JITPasses.cpp type mlir (line 27) | namespace mlir { type oneflow (line 29) | namespace oneflow { function NamedAttrList (line 46) | NamedAttrList GetJitOpAttributes(Builder& rewriter, StringRef op_nam... function isOneFlowOp (line 64) | bool isOneFlowOp(Operation* op) { return llvm::dyn_cast createOutlineJitFunctionPass() { FILE: oneflow/ir/lib/OneFlow/Transform/OneFlowMemPool.cpp type mlir (line 40) | namespace mlir { type oneflow (line 41) | namespace oneflow { function Type (line 44) | Type getMemPoolElemType(MLIRContext* ctx) { return IntegerType::get(... type AllocOpInfo (line 48) | struct AllocOpInfo { function getAllocInfoList (line 56) | std::vector getAllocInfoList(T op) { function replaceAllocwithSubview (line 77) | void replaceAllocwithSubview(func::FuncOp func, OpBuilder& builder, function isMemPool (line 95) | bool isMemPool(Operation* op) { type InsertOneFlowMemPoolPattern (line 103) | struct InsertOneFlowMemPoolPattern final : public OpRewritePattern getAllocOp(func::FuncOp func) con... method MemRefType (line 117) | MemRefType getNullMemType(mlir::PatternRewriter& rewriter) const { method InsertOneFlowMemPoolPattern (line 122) | explicit InsertOneFlowMemPoolPattern(mlir::MLIRContext* context) method matchAndRewrite (line 124) | mlir::LogicalResult matchAndRewrite(func::FuncOp op, class InsertOneFlowMemPoolPass (line 160) | class InsertOneFlowMemPoolPass : public InsertOneFlowMemPoolPassBase... method runOnOperation (line 161) | void runOnOperation() override { class FoldAllocToSubviewPass (line 170) | class FoldAllocToSubviewPass : public FoldAllocToSubviewPassBase createInsertOneFlowMemPoolPass() { function createFoldAllocToSubviewPass (line 216) | std::unique_ptr createFoldAllocToSubviewPass() { FILE: oneflow/ir/lib/OneFlow/Transform/OneFlowStream.cpp type mlir (line 29) | namespace mlir { type oneflow (line 30) | namespace oneflow { type MgpuToOneFlowStreamPattern (line 34) | struct MgpuToOneFlowStreamPattern final : public OpRewritePattern createAppendOneFlowStreamPass() { function createMgpuToOneFlowStreamPass (line 141) | std::unique_ptr createMgpuToOneFlowStreamPass() { FILE: oneflow/ir/lib/OneFlow/Transform/OutlineAndFuse.cpp type mlir (line 34) | namespace mlir { type oneflow (line 35) | namespace oneflow { class WrapOpsToKernelLaunchPass (line 39) | class WrapOpsToKernelLaunchPass : public WrapOpsToKernelLaunchPassBa... method WrapOpsToKernelLaunchPass (line 41) | WrapOpsToKernelLaunchPass() = default; method WrapOpsToKernelLaunchPass (line 42) | WrapOpsToKernelLaunchPass(const WrapOpsToKernelLaunchPass& other) method getDependentDialects (line 45) | void getDependentDialects(DialectRegistry& registry) const override { method runOnOperation (line 49) | void runOnOperation() override { class FuseIntoExistingOpPass (line 62) | class FuseIntoExistingOpPass : public FuseIntoExistingOpPassBase { method runOnOperation (line 155) | void runOnOperation() override { type GroupNormActivationPattern (line 163) | struct GroupNormActivationPattern : public OpRewritePattern createWrapOpsToKernelLaunchPass() { function createFuseIntoExistingOpPass (line 226) | std::unique_ptr createFuseIntoExistingOpPass() { function createGroupMatMul (line 230) | std::unique_ptr createGroupMatMul() { return std::make_unique<... function createFuseForwardOps (line 232) | std::unique_ptr createFuseForwardOps() { return std::make_uniq... function createFuseOpsWithBackwardImpl (line 233) | std::unique_ptr createFuseOpsWithBackwardImpl() { function createFuseNormalizationOps (line 237) | std::unique_ptr createFuseNormalizationOps() { FILE: oneflow/ir/lib/OneFlow/Transform/TraitFolder.cpp type mlir (line 28) | namespace mlir { type oneflow (line 29) | namespace oneflow { class TestOneFlowTraitFolderPass (line 31) | class TestOneFlowTraitFolderPass method runOnOperation (line 33) | void runOnOperation() override { function createTestOneFlowTraitFolderPass (line 42) | std::unique_ptr createTestOneFlowTraitFolderPass() { FILE: oneflow/ir/lib/OneFlow/TransposeHelpers.cpp type mlir (line 18) | namespace mlir { type oneflow (line 20) | namespace oneflow { function RankedTensorType (line 22) | RankedTensorType getNHWCType(RankedTensorType t) { function RankedTensorType (line 27) | RankedTensorType getNHWCType(Type t) { return getNHWCType(t.cast getNHWCResultTypes(NCHWCompatible op) { FILE: oneflow/ir/lib/OneFlow/UserOpConversion.cpp type mlir (line 22) | namespace mlir { type oneflow (line 24) | namespace oneflow { type user_op (line 26) | namespace user_op { function LogicalResult (line 28) | LogicalResult saveAttrDictionaryToOpConf(DictionaryAttr attributes, function LogicalResult (line 47) | LogicalResult doConvertUserOpAttributes(llvm::StringRef op_type_na... function LogicalResult (line 169) | LogicalResult ConvertUserOpAttributes(llvm::StringRef op_type_name... function LogicalResult (line 184) | LogicalResult ConvertUserOpAttributes(Operation* op, ::oneflow::Op... function LogicalResult (line 209) | LogicalResult ConvertUserOpAttributes(Operation* op, ::oneflow::Op... function LogicalResult (line 394) | LogicalResult ConvertUserOpInputs(llvm::StringRef op_type_name, Va... function getAttrAsShape (line 422) | ::oneflow::ShapeProto getAttrAsShape(mlir::Attribute& attr) { function getAttrAsStride (line 430) | ::oneflow::Int64ListProto getAttrAsStride(mlir::Attribute& attr) { function getParallelConfFromAttrDictionary (line 438) | ::oneflow::ParallelConf getParallelConfFromAttrDictionary(Dictiona... function getParallelConfFromAttrs (line 461) | ::oneflow::ParallelConf getParallelConfFromAttrs(Attribute device_... function getDeviceTypeFromAttrDictionary (line 477) | ::oneflow::DeviceType getDeviceTypeFromAttrDictionary(DictionaryAt... function queryAttrType (line 499) | ::oneflow::AttrType queryAttrType(const std::string& op_type_name,... FILE: oneflow/ir/lib/OneFlow/UserOpReflection.cpp type mlir (line 22) | namespace mlir { type oneflow (line 24) | namespace oneflow { type user_op (line 26) | namespace user_op { function StringRef (line 94) | StringRef GetSegmentSizeAttr() { function StringRef (line 99) | StringRef GetSegmentSizeAttr() { function ArrayAttr (line 120) | ArrayAttr GetUserOpArgSizes(Use... function ArrayAttr (line 125) | ArrayAttr GetUserOpArgSizes(User... function LogicalResult (line 130) | LogicalResult GetUserOpFilteredSegmentKeyAndSizes(UserOp op, std::... function Source (line 145) | Source GetOpSourceByName(Operation* op, const std::string& to_find) { function LogicalResult (line 182) | LogicalResult GetFilteredSegmentKeyAndSizes(Operation* op, std::ve... function LogicalResult (line 229) | LogicalResult GetFilteredSegmentKeyAndSizes(llvm::StringRef op_typ... function GetOutputLbn (line 315) | llvm::Optional GetOutputLbn(OpResult result) { FILE: oneflow/ir/lib/Transform/TransformDialectExtension.cpp type MemrefCopyOpFoldPatterns (line 40) | struct MemrefCopyOpFoldPatterns final : public OpRewritePattern> {} class TransformDialectInterpreterPass (line 31) | class TransformDialectInterpreterPass method TransformDialectInterpreterPass (line 37) | TransformDialectInterpreterPass() = default; method TransformDialectInterpreterPass (line 38) | TransformDialectInterpreterPass(const TransformDialectInterpreterPass&... method StringRef (line 41) | StringRef getArgument() const override { return "oneflow-transform-dia... method StringRef (line 43) | StringRef getDescription() const override { method findOperationsByName (line 47) | void findOperationsByName(Operation* root, StringRef name, method createParameterMapping (line 54) | void createParameterMapping(MLIRContext& context, ArrayRef values, method createOpResultMapping (line 64) | void createOpResultMapping(Operation* root, StringRef name, method numberOfSetOptions (line 73) | unsigned numberOfSetOptions(const Option& ops, const List... method runOnOperation (line 82) | void runOnOperation() override { type TransformDialectEraseSchedulePass (line 185) | struct TransformDialectEraseSchedulePass method StringRef (line 189) | StringRef getArgument() const final { return "oneflow-transform-dialec... method StringRef (line 191) | StringRef getDescription() const final { return "erase transform diale... method runOnOperation (line 193) | void runOnOperation() override { type mlir (line 205) | namespace mlir { type oneflow (line 206) | namespace oneflow { type transform_dialect (line 207) | namespace transform_dialect { function registerTransformDialectEraseSchedulePass (line 209) | void registerTransformDialectEraseSchedulePass() { function registerTransformDialectInterpreterPass (line 213) | void registerTransformDialectInterpreterPass() { FILE: oneflow/ir/lib/Transform/TransformStateExtension.cpp function LogicalResult (line 20) | LogicalResult mlir::oneflow::transform_dialect::TransformStateExtension:... FILE: oneflow/ir/oneflow-extension/include/OneFlow/JITOpInfer.h function namespace (line 21) | namespace oneflow { FILE: oneflow/ir/oneflow-extension/include/OneFlow/OneFlowLRJITRegistry.h function namespace (line 31) | namespace mlir { type std (line 35) | typedef std::pair, std::function<... function class (line 38) | class LRJITRegistry final { FILE: oneflow/ir/oneflow-extension/include/OneFlow/OneFlowRoundTrip.h function namespace (line 21) | namespace oneflow { FILE: oneflow/ir/oneflow-extension/include/PyAst/Ast.h function namespace (line 24) | namespace pyast { function class (line 53) | class stmt { function class (line 74) | class expr { function class (line 97) | class FunctionDef : public stmt { function class (line 118) | class Return : public stmt { function class (line 131) | class Assign : public stmt { function class (line 149) | class If : public stmt { function class (line 170) | class Raise : public stmt { function class (line 188) | class Assert : public stmt { function class (line 205) | class Expr : public stmt { function class (line 218) | class BoolOp : public expr { function boolop_t (line 231) | boolop_t get_op() { return op; } function classof (line 234) | static bool classof(const expr* c) { return c->get_kind() == kBoolOp; } function class (line 241) | class BinOp : public expr { function class (line 275) | class Lambda : public expr { function class (line 293) | class Compare : public expr { FILE: oneflow/ir/oneflow-extension/include/PyAst/AstMlirGen.h function class (line 72) | class BuilderWithSymbolTable { function class (line 88) | class MLIRGenImpl : public BuilderWithSymbolTable { FILE: oneflow/ir/oneflow-extension/ir_pass.cpp type oneflow (line 27) | namespace oneflow { class RoundTripOneFlowJobWrapper (line 58) | class RoundTripOneFlowJobWrapper : public mlir::oneflow::RoundTripOneF... method RoundTripOneFlowJobWrapper (line 60) | explicit RoundTripOneFlowJobWrapper(::oneflow::Job* job) method Job (line 63) | const Job* job() const override { return job_; } method IsLastIRPass (line 65) | bool IsLastIRPass() const override { return IsLastIRPassForIRPassTyp... method UpdateJob (line 67) | void UpdateJob(::oneflow::Job* new_job) override { method DumpLog (line 72) | void DumpLog(const std::string& filename, const std::string& content... method InputBns4OpName (line 84) | std::pair, std::vector> InputB... method OutputLbns4OpName (line 100) | std::vector OutputLbns4OpName(const std::string& op_nam... method ReplaceInputLbnInOpCustomizedConf (line 109) | std::string ReplaceInputLbnInOpCustomizedConf(::oneflow::OperatorCon... method QueryLogicalBlob (line 115) | void QueryLogicalBlob( method TopoForEachOpConf (line 125) | void TopoForEachOpConf( method LogDir (line 131) | std::string LogDir() { function SortJob (line 149) | void SortJob(Job& job) { class IRRoundTrip (line 184) | class IRRoundTrip class IRRoundTrip (line 185) | class IRRoundTrip function ConvertJobToTosaIR (line 187) | Maybe ConvertJobToTosaIR(Job* job) { function SaveJobToIR (line 192) | Maybe SaveJobToIR(Job* job, const std::string& path) { function ConvertJobToIR (line 200) | Maybe ConvertJobToIR(Job* job) { function LoadJobFromIR (line 206) | Maybe LoadJobFromIR(Job* job, const std::string& path) { FILE: oneflow/ir/oneflow-extension/lr_jit.cpp type LLVMInitializer (line 79) | struct LLVMInitializer { method LLVMInitializer (line 80) | LLVMInitializer() { function lowerToLLVMDialect (line 86) | static mlir::LogicalResult lowerToLLVMDialect(mlir::ModuleOp module) { function GenModuleForTest (line 104) | static mlir::OwningOpRef GenModuleForTest(mlir::MLIRCont... function GenModule (line 116) | static mlir::OwningOpRef GenModule(mlir::MLIRContext& co... function LRJITRegistry_Store_ (line 127) | static LRJITRegistry_Store_ GenFunc(pyast::FunctionDef& ast, bool is_dum... FILE: oneflow/ir/oneflow-extension/mlir_jit_op.cpp type oneflow (line 38) | namespace oneflow { function GetSbpFn (line 42) | Maybe GetSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/ir/oneflow-extension/mlir_jit_op_kernel.cpp type oneflow (line 37) | namespace oneflow { function OpaqueMemRefDescriptor (line 44) | OpaqueMemRefDescriptor CreateMemRefDescriptor(user_op::Tensor* tensor) { function OpaqueMemRefDescriptor (line 59) | OpaqueMemRefDescriptor CreateMutMemRefDescriptor(user_op::Tensor* tens... function GetMLIRCInterface (line 82) | std::string GetMLIRCInterface(const std::string& func_name) { function GetMLIRCInterfaceArgs (line 86) | llvm::SmallVector GetMLIRCInterfaceArgs( function getDialectRegistry (line 106) | mlir::DialectRegistry getDialectRegistry() { function WithMlirContext (line 116) | void WithMlirContext( function inferOneFlowMemPoolSize (line 143) | size_t inferOneFlowMemPoolSize(user_op::InferContext* ctx) { class MlirJitCpuKernel (line 161) | class MlirJitCpuKernel final : public user_op::OpKernel { method MlirJitCpuKernel (line 163) | MlirJitCpuKernel() = default; method Compute (line 167) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 179) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class MlirJitGpuKernel (line 203) | class MlirJitGpuKernel final : public user_op::OpKernel { method MlirJitGpuKernel (line 205) | MlirJitGpuKernel() = default; method Compute (line 209) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 225) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/ir/oneflow-lite/OneFlowLiteCompileMain.cpp type mlir (line 34) | namespace mlir { type oneflow (line 35) | namespace oneflow { type lite (line 36) | namespace lite { function LogicalResult (line 38) | LogicalResult Compile(int argc, char** argv) { function main (line 97) | int main(int argc, char** argv) { FILE: oneflow/ir/oneflow-lite/include/OneFlow/ConvertToLiteExecutable.h function namespace (line 26) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/FlatbufferUtils.h function namespace (line 48) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/OneFlowLiteUtils.h function namespace (line 26) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/FoldVariable.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/InferPlacement.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/InsertTransferOp.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/Lowering/LoweringAscend.h function namespace (line 28) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/Lowering/LoweringAscendUtils.h function ge (line 40) | inline ge::Shape convertAscendShape(ArrayRef shape) { function Optional (line 44) | inline Optional convertAscendElementType(Type type) { function Optional (line 83) | inline Optional convertAscendElementType(::mlir::oneflow::... function ge (line 100) | inline ge::TensorDesc convertAscendType(Type type) { function ge (line 112) | inline ge::TensorDesc convertAscendType(::mlir::oneflow::DataType type, ... FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/LoweringLaunchJob.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/MemoryPlanning.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/oneflow-lite/include/OneFlow/Transform/PartitionLaunchJob.h function namespace (line 21) | namespace mlir { FILE: oneflow/ir/oneflow-lite/lib/OneFlow/ConvertToLiteExecutable.cpp type mlir (line 53) | namespace mlir { type oneflow (line 54) | namespace oneflow { type lite (line 56) | namespace lite { function flatbuffers_vec_ref_t (line 58) | static flatbuffers_vec_ref_t createLiteOpAttrs(FlatbufferBuilder& ... function flatbuffers_vec_ref_t (line 135) | static flatbuffers_vec_ref_t createLiteVariableOpAttrs(FlatbufferB... function oneflow_lite_OpDef_ref_t (line 184) | static oneflow_lite_OpDef_ref_t createLiteVariableOpDef( function oneflow_lite_OpDef_ref_t (line 203) | static oneflow_lite_OpDef_ref_t createLiteOpDef( function oneflow_lite_TensorDef_ref_t (line 237) | static oneflow_lite_TensorDef_ref_t createLiteTensorDef(Flatbuffer... function oneflow_lite_BufferSegmentDef_ref_t (line 256) | static oneflow_lite_BufferSegmentDef_ref_t createLiteBufferSegment... function LogicalResult (line 268) | LogicalResult ConvertToLiteExecutable(MLIRContext* context, Module... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/FlatbufferUtils.cpp type mlir (line 32) | namespace mlir { type oneflow (line 33) | namespace oneflow { type lite (line 34) | namespace lite { function cloneBufferIntoContiguousBytes (line 44) | static SmallVector cloneBufferIntoContiguousBytes(Fla... function flatbuffers_uint8_vec_ref_t (line 57) | flatbuffers_uint8_vec_ref_t FlatbufferBuilder::streamUint8Vec( function DenseIntElementsAttr (line 66) | DenseIntElementsAttr FlatbufferBuilder::getBufferAttr(MLIRContext*... function LogicalResult (line 78) | LogicalResult FlatbufferBuilder::copyToStream(llvm::raw_ostream& o... function LogicalResult (line 98) | LogicalResult FlatbufferBuilder::printJsonToStream(bool pretty, bo... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/OneFlowLiteUtils.cpp type mlir (line 46) | namespace mlir { type oneflow (line 47) | namespace oneflow { type lite (line 49) | namespace lite { function Operation (line 51) | Operation* getEntryJobOp(ModuleOp module) { return getEntryJobOp(m... function Operation (line 53) | Operation* getEntryJobOp(Operation* op) { function StringAttr (line 62) | StringAttr getValueDevice(Value value) { function getLiteStringElementType (line 74) | Optional getLiteStringElementType(Type type) { function getLiteStringElementType (line 95) | Optional getLiteStringElementType(::mlir::oneflow::Data... function getUserOpAttrType (line 112) | Optional<::oneflow::AttrType> getUserOpAttrType(StringRef opName, ... function serializeI32Attr (line 124) | void serializeI32Attr(FlatbufferBuilder& builder, Attribute attrib... function serializeI64Attr (line 130) | void serializeI64Attr(FlatbufferBuilder& builder, Attribute attrib... function serializeBoolAttr (line 136) | void serializeBoolAttr(FlatbufferBuilder& builder, Attribute attri... function serializeF32Attr (line 142) | void serializeF32Attr(FlatbufferBuilder& builder, Attribute attrib... function serializeF64Attr (line 149) | void serializeF64Attr(FlatbufferBuilder& builder, Attribute attrib... function serializeStringAttr (line 156) | void serializeStringAttr(FlatbufferBuilder& builder, Attribute att... function serializeShapeAttr (line 163) | void serializeShapeAttr(FlatbufferBuilder& builder, Attribute attr... function serializeStrideAttr (line 173) | void serializeStrideAttr(FlatbufferBuilder& builder, Attribute att... function serializeDataTypeAttr (line 177) | void serializeDataTypeAttr(FlatbufferBuilder& builder, Attribute a... function serializeI32sAttr (line 189) | void serializeI32sAttr(FlatbufferBuilder& builder, Attribute attri... function serializeI64sAttr (line 199) | void serializeI64sAttr(FlatbufferBuilder& builder, Attribute attri... function serializeF32sAttr (line 209) | void serializeF32sAttr(FlatbufferBuilder& builder, Attribute attri... function serializeDataTypesAttr (line 219) | void serializeDataTypesAttr(FlatbufferBuilder& builder, Attribute ... function serializeShapesAttr (line 234) | void serializeShapesAttr(FlatbufferBuilder& builder, Attribute att... function serializeStridesAttr (line 250) | void serializeStridesAttr(FlatbufferBuilder& builder, Attribute at... function serializeStringsAttr (line 254) | void serializeStringsAttr(FlatbufferBuilder& builder, Attribute at... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/FoldVariable.cpp type mlir (line 18) | namespace mlir { type oneflow (line 19) | namespace oneflow { type lite (line 20) | namespace lite { type FoldVariablePass (line 22) | struct FoldVariablePass : public PassWrapper createLiteFoldVariablePass() { FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/InferPlacement.cpp type mlir (line 26) | namespace mlir { type oneflow (line 27) | namespace oneflow { type lite (line 28) | namespace lite { function CanScheduleOnTarget (line 30) | static bool CanScheduleOnTarget(Operation* op, StringRef target) { type InferPlacementPass (line 39) | struct InferPlacementPass : public PassWrapper createLiteInferPlacementPass(StringRef... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/InsertTransferOp.cpp type mlir (line 26) | namespace mlir { type oneflow (line 27) | namespace oneflow { type lite (line 28) | namespace lite { type InsertTransferOpPass (line 30) | struct InsertTransferOpPass : public PassWrapper createLiteInsertTransferOpPass() { FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/Lowering/LoweringAscend.cpp type mlir (line 43) | namespace mlir { type oneflow (line 44) | namespace oneflow { type lite (line 45) | namespace lite { class AscendValue (line 47) | class AscendValue { method AscendValue (line 49) | AscendValue() = default; method AscendValue (line 50) | AscendValue(const std::shared_ptr& op, const ge::T... method AscendValue (line 53) | AscendValue(const std::shared_ptr& op, const ge::T... method AscendValue (line 57) | AscendValue(const AscendValue&) = default; method StringRef (line 63) | StringRef getComponentName() const { return componentName_; } method getComponentIndex (line 64) | int getComponentIndex() const { return componentIndex_; } method StringRef (line 66) | StringRef getComponentNameAndIndex() const { method setOperation (line 72) | void setOperation(const std::shared_ptr& op) { op_... method setType (line 73) | void setType(const ge::TensorDesc& type) { type_ = type; } method setComponentName (line 74) | void setComponentName(StringRef componentName) { componentName_ ... method setComponentIndex (line 75) | void setComponentIndex(int componentIndex) { componentIndex_ = c... class AscendCompiler (line 84) | class AscendCompiler { method AscendCompiler (line 86) | AscendCompiler() = default; method AscendValue (line 106) | AscendValue getValue(Value value) const { method createOp (line 113) | std::shared_ptr createOp(llvm::Twine opName) { function LogicalResult (line 360) | LogicalResult loweringAscend(OpBuilder& builder, Operation* callee... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/LoweringLaunchJob.cpp type mlir (line 31) | namespace mlir { type oneflow (line 32) | namespace oneflow { type lite (line 33) | namespace lite { type LoweringLaunchJobPass (line 35) | struct LoweringLaunchJobPass : public PassWrapper createLiteLoweringLaunchJobPass(String... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/MemoryPlanning.cpp type mlir (line 30) | namespace mlir { type oneflow (line 31) | namespace oneflow { type lite (line 32) | namespace lite { function LogicalResult (line 46) | LogicalResult LiteBufferStrategy::insertValue(Value value, int seg... class ValueLiveness (line 55) | class ValueLiveness { method ValueLiveness (line 57) | ValueLiveness() = default; method addValue (line 59) | void addValue(Value value, size_t liveStart, size_t liveEnd) { method isLivenessOverlap (line 63) | bool isLivenessOverlap(Value lhs, Value rhs) { type LiveRange (line 71) | struct LiveRange { type MemoryPlanningPass (line 78) | struct MemoryPlanningPass : public PassWrapper createLiteMemoryPlanningPass(LiteBuffe... FILE: oneflow/ir/oneflow-lite/lib/OneFlow/Transform/PartitionLaunchJob.cpp type mlir (line 27) | namespace mlir { type oneflow (line 28) | namespace oneflow { type lite (line 29) | namespace lite { type PartitionLaunchJobPass (line 31) | struct PartitionLaunchJobPass method needPartition (line 35) | bool needPartition(StringRef device) const { return device == "t... function createLitePartitionLaunchJobPass (line 152) | std::unique_ptr createLitePartitionLaunchJobPass() { FILE: oneflow/ir/oneflow-opt/oneflow-opt.cpp function main (line 34) | int32_t main(int32_t argc, char** argv) { FILE: oneflow/ir/oneflow-runner/oneflow-runner.cpp function main (line 40) | int main(int argc, char** argv) { FILE: oneflow/ir/oneflow-translate/include/OneFlow/MLIROneFlowTranslation.h function namespace (line 37) | namespace mlir { FILE: oneflow/ir/oneflow-translate/lib/OneFlow/Importer.cpp type mlir (line 69) | namespace mlir { type oneflow (line 71) | namespace oneflow { function SizeVec (line 79) | SizeVec GetSizesFromArgs(UserOpArgs args, UserOpArgDefs arg_defs) { function GetOutputLbns (line 91) | std::vector GetOutputLbns(const ::oneflow::OperatorConf... function LogicalResult (line 113) | LogicalResult IsAttrBelong2Op(const std::string& op_type_name, const... function LogicalResult (line 118) | LogicalResult Importer::AddUserOpInputOutputSegments(const ::oneflow... function GetDataTypeAttr (line 138) | llvm::Optional GetDataTypeAttr(MLIRCont... function ArrayAttr (line 165) | ArrayAttr Importer::GetAttrFromShape(const ::oneflow::ShapeProto& sh... function ArrayAttr (line 170) | ArrayAttr Importer::GetAttrFromStride(const ::oneflow::Int64ListProt... function LogicalResult (line 175) | LogicalResult Importer::namedAttributesFromUserOp(const ::oneflow::O... function LogicalResult (line 287) | LogicalResult Importer::AddOperandSegmentSizes(int32_t input_lbns_size... function LogicalResult (line 295) | LogicalResult Importer::AddResultSegmentSizes(int32_t output_lbns_size, function LogicalResult (line 304) | LogicalResult Importer::AppendCtrlOutType(llvm::SmallVector& ... function LogicalResult (line 309) | LogicalResult Importer::AddOpConf(const ::oneflow::OperatorConf& op, function LogicalResult (line 324) | LogicalResult ParseNdSbpFromAttr(::llvm::ArrayRef nd_sbp_attr, function Attribute (line 352) | Attribute ConvertNdSbpToAttr(Builder& builder, const ::oneflow::NdSbp&... function LogicalResult (line 370) | LogicalResult ValidateUserOpConf(const ::oneflow::OperatorConf& op_con... function LogicalResult (line 388) | LogicalResult Importer::ProcessUserOp(const ::oneflow::OperatorConf& o... function LogicalResult (line 461) | LogicalResult ConvertCtrlInputs(Operation* op, ::oneflow::OperatorConf... function LogicalResult (line 472) | LogicalResult ConvertUserOpInputs(Operation* op, StringRef op_name, function LogicalResult (line 502) | LogicalResult ConvertUserOpOutputs(Operation* op, StringRef op_name, function LogicalResult (line 523) | LogicalResult ConvertDT(::mlir::oneflow::DataType data_type_mlir, ::on... function LogicalResult (line 547) | LogicalResult ConvertDTFromAttr(Attribute attr, ::oneflow::DataType& d... function LogicalResult (line 558) | LogicalResult ConvertVariableOpConf(VariableOp op, ::oneflow::Operator... function LogicalResult (line 630) | LogicalResult ConvertInputOpConf(InputOp op, ::oneflow::OperatorConf* ... function LogicalResult (line 676) | LogicalResult ConvertOutputOpConf(OutputOp op, ::oneflow::OperatorConf... FILE: oneflow/ir/oneflow-translate/lib/OneFlow/MLIROneFlowTranslation.cpp type mlir (line 74) | namespace mlir { type oneflow (line 76) | namespace oneflow { class JobImporter (line 80) | class JobImporter : Importer { method JobImporter (line 82) | JobImporter(RoundTripOneFlowJobWrapperInterface& job_wrapper, MLIR... function LogicalResult (line 118) | LogicalResult JobImporter::AppendCtrlInOperand(const ::oneflow::Oper... function LogicalResult (line 133) | LogicalResult JobImporter::AppendDataInOperand(const std::string& lbn, function LogicalResult (line 145) | LogicalResult JobImporter::InsertOpResults(const ::oneflow::Operator... function LogicalResult (line 174) | LogicalResult JobImporter::AddDeviceName(const ::oneflow::OperatorCo... function Type (line 189) | Type JobImporter::GetTensorTypeOfLbn(const std::string& lbn) { function LogicalResult (line 203) | LogicalResult JobImporter::ProcessSystemOp(const ::oneflow::Operator... function LogicalResult (line 256) | LogicalResult JobImporter::ProcessVariableOp(const ::oneflow::Operat... function LogicalResult (line 371) | LogicalResult JobImporter::ProcessInputOp(const ::oneflow::OperatorC... function LogicalResult (line 459) | LogicalResult JobImporter::ProcessOutputOp(const ::oneflow::Operator... function LogicalResult (line 546) | LogicalResult JobImporter::ProcessJob() { function UpdatePlacement (line 603) | void UpdatePlacement(OpType* op, AdaptorType& adaptor, ::oneflow::Jo... function LogicalResult (line 619) | LogicalResult JobImporter::TryToUpdateJob() { function LogicalResult (line 709) | LogicalResult JobImporter::ConvertUserOp(Operation* op, ::oneflow::J... function LogicalResult (line 735) | LogicalResult JobImporter::ConvertSystemOp(Operation* op, ::oneflow:... function LogicalResult (line 751) | LogicalResult JobImporter::ConvertVariableOp(VariableOp op, ::oneflo... function LogicalResult (line 758) | LogicalResult JobImporter::ConvertInputOp(InputOp op, ::oneflow::Job... function LogicalResult (line 765) | LogicalResult JobImporter::ConvertOutputOp(OutputOp op, ::oneflow::J... function Type (line 772) | Type JobImporter::GetInterfaceBlobConfType(const ::oneflow::Interfac... function DumpMLIR (line 783) | void DumpMLIR(RoundTripOneFlowJobWrapperInterface& job_wrapper, Modu... function LogicalResult (line 791) | LogicalResult ApplyRoundTripPatterns(RoundTripOneFlowJobWrapperInter... function TranslateOneFlowJobToModule (line 885) | OwningOpRef TranslateOneFlowJobToModule(llvm::StringRef st... function RoundTripOneFlowJob (line 896) | void RoundTripOneFlowJob( function ConvertJobToTosaIR (line 930) | std::string ConvertJobToTosaIR(RoundTripOneFlowJobWrapperInterface& ... function ConvertJobToIR (line 961) | std::string ConvertJobToIR(RoundTripOneFlowJobWrapperInterface& job_... function SaveJobToIR (line 989) | void SaveJobToIR(RoundTripOneFlowJobWrapperInterface& job_wrapper, c... function LoadJobFromIR (line 1024) | void LoadJobFromIR(RoundTripOneFlowJobWrapperInterface& job_wrapper,... function registerFromOneFlowJobTranslation (line 1040) | void registerFromOneFlowJobTranslation() { FILE: oneflow/ir/oneflow-translate/oneflow-translate.cpp function main (line 22) | int32_t main(int32_t argc, char** argv) { FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_batchnorm_relu.py function do_nhwc_bacth_norm (line 29) | def do_nhwc_bacth_norm(test_case, with_cuda): class TestNhwcConv (line 59) | class TestNhwcConv(oneflow.unittest.MLIRTestCase): method setUp (line 60) | def setUp(self): method test_nhwc_conv_graph (line 64) | def test_nhwc_conv_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_bias_add.py function do_nhwc_bias_add (line 26) | def do_nhwc_bias_add(test_case, with_cuda): class TestNhwcBiasAdd (line 52) | class TestNhwcBiasAdd(oneflow.unittest.MLIRTestCase): method setUp (line 53) | def setUp(self): method test_nhwc_bias_add_graph (line 57) | def test_nhwc_bias_add_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_conv.py function do_nhwc_conv (line 29) | def do_nhwc_conv(test_case, with_cuda, with_bias): class TestNhwcConv (line 54) | class TestNhwcConv(oneflow.unittest.MLIRTestCase): method setUp (line 55) | def setUp(self): method test_nhwc_conv_graph (line 60) | def test_nhwc_conv_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_conv2d_maxpool2d.py function do_nhwc_conv_maxpool (line 29) | def do_nhwc_conv_maxpool(test_case, with_cuda, with_bias): class TestNhwcConvMaxPool (line 58) | class TestNhwcConvMaxPool(oneflow.unittest.MLIRTestCase): method setUp (line 59) | def setUp(self): method test_nhwc_conv_graph (line 63) | def test_nhwc_conv_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_conv_relu_add.py function do_nhwc_conv (line 29) | def do_nhwc_conv(test_case, with_cuda, with_bias): class TestNhwcConv (line 56) | class TestNhwcConv(oneflow.unittest.MLIRTestCase): method setUp (line 57) | def setUp(self): method test_nhwc_conv_graph (line 61) | def test_nhwc_conv_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_lenet.py class LeNet (line 30) | class LeNet(nn.Module): method __init__ (line 31) | def __init__(self): method forward (line 39) | def forward(self, x): function do_lenet (line 51) | def do_lenet(test_case, with_cuda): class TestLeNet (line 76) | class TestLeNet(oneflow.unittest.MLIRTestCase): method setUp (line 77) | def setUp(self): method test_nhwc_lenet_graph (line 81) | def test_nhwc_lenet_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_maxpool_2d.py function do_nhwc_maxpool_2d (line 29) | def do_nhwc_maxpool_2d(test_case, with_cuda, with_return_induces): class TestNhwcMaxPool2d (line 71) | class TestNhwcMaxPool2d(oneflow.unittest.MLIRTestCase): method setUp (line 72) | def setUp(self): method test_nhwc_maxpool_2d_graph (line 76) | def test_nhwc_maxpool_2d_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_resnet.py function conv3x3 (line 37) | def conv3x3( function conv1x1 (line 53) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: class BasicBlock (line 58) | class BasicBlock(nn.Module): method __init__ (line 61) | def __init__( method forward (line 88) | def forward(self, x: Tensor) -> Tensor: class Bottleneck (line 107) | class Bottleneck(nn.Module): method __init__ (line 116) | def __init__( method forward (line 142) | def forward(self, x: Tensor) -> Tensor: class ResNet (line 165) | class ResNet(nn.Module): method __init__ (line 166) | def __init__( method _make_layer (line 231) | def _make_layer( method _forward_impl (line 279) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 297) | def forward(self, x: Tensor) -> Tensor: function _resnet (line 301) | def _resnet( function resnet50 (line 313) | def resnet50(pretrained: bool = False, progress: bool = True, **kwargs: ... function do_resnet (line 323) | def do_resnet(test_case): class TestResNet (line 348) | class TestResNet(oneflow.unittest.MLIRTestCase): method setUp (line 349) | def setUp(self): method test_nhwc_resnet_graph (line 354) | def test_nhwc_resnet_graph(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_nhwc_transpose_eliminate.py function do_eliminate_transpose (line 29) | def do_eliminate_transpose(test_case, with_cuda): class TestNhwcEliminateTranspose (line 51) | class TestNhwcEliminateTranspose(oneflow.unittest.MLIRTestCase): method setUp (line 52) | def setUp(self): method test_eliminate_transpose (line 56) | def test_eliminate_transpose(test_case): FILE: oneflow/ir/test/OneFlow/auto_nhwc/test_resnet101_benchmark.py function conv3x3 (line 42) | def conv3x3( function conv1x1 (line 58) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: class BasicBlock (line 63) | class BasicBlock(nn.Module): method __init__ (line 66) | def __init__( method forward (line 93) | def forward(self, x: Tensor) -> Tensor: class Bottleneck (line 112) | class Bottleneck(nn.Module): method __init__ (line 121) | def __init__( method forward (line 147) | def forward(self, x: Tensor) -> Tensor: class ResNet (line 170) | class ResNet(nn.Module): method __init__ (line 171) | def __init__( method _make_layer (line 236) | def _make_layer( method _forward_impl (line 284) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 302) | def forward(self, x: Tensor) -> Tensor: function _resnet (line 306) | def _resnet( function resnet101 (line 318) | def resnet101(pretrained: bool = False, progress: bool = True, **kwargs:... function bench (line 336) | def bench(forward: Callable, x, n=1000): class ResNetEvalGraph (line 357) | class ResNetEvalGraph(nn.Graph): method __init__ (line 358) | def __init__(self, model): method build (line 363) | def build(self, x): function main (line 369) | def main(): FILE: oneflow/ir/test/OneFlow/cuda_code_gen/test_fuser_cast_scale.py class CastModule (line 29) | class CastModule(flow.nn.Module): method __init__ (line 30) | def __init__(self): method forward (line 33) | def forward(self, x, scale): function do_relu_graph (line 38) | def do_relu_graph(test_case, data, with_cuda): class TestFuseCastScale (line 61) | class TestFuseCastScale(oneflow.unittest.MLIRTestCase): method setUp (line 62) | def setUp(self): method test_relu_graph (line 67) | def test_relu_graph(test_case): FILE: oneflow/ir/test/OneFlow/cuda_code_gen/test_matmul.py class MatMulModule (line 29) | class MatMulModule(flow.nn.Module): method __init__ (line 30) | def __init__(self): method forward (line 35) | def forward(self, x): function do_matmul_graph (line 39) | def do_matmul_graph(test_case, with_cuda=False): class TestFuseCastScale (line 61) | class TestFuseCastScale(oneflow.unittest.MLIRTestCase): method setUp (line 62) | def setUp(self): method test_relu_graph (line 67) | def test_relu_graph(test_case): FILE: oneflow/ir/test/OneFlow/folding/test_conv_bn.py function _test_fuse_conv_bn (line 28) | def _test_fuse_conv_bn(test_case): class TestFuseConvBn (line 52) | class TestFuseConvBn(oneflow.unittest.MLIRTestCase): method setUp (line 53) | def setUp(self): method test_fuse_conv_bn (line 57) | def test_fuse_conv_bn(test_case): FILE: oneflow/ir/test/OneFlow/folding/test_simple_multiply.py class MultiplyModel (line 27) | class MultiplyModel(nn.Module): method __init__ (line 28) | def __init__(self, dtype=flow.float32): method forward (line 34) | def forward(self): class MultiplyModelComplex (line 38) | class MultiplyModelComplex(MultiplyModel): method __init__ (line 39) | def __init__(self, dtype=flow.float32): method forward (line 43) | def forward(self): class MultiplyModelWithInput (line 47) | class MultiplyModelWithInput(MultiplyModel): method __init__ (line 48) | def __init__(self, dtype=flow.float32): method forward (line 51) | def forward(self, a: flow.Tensor, b: flow.Tensor): function _test_fold_multiply (line 56) | def _test_fold_multiply(test_case, module, with_cuda, *args, dtype=onefl... class TestFoldMultiply (line 82) | class TestFoldMultiply(oneflow.unittest.MLIRTestCase): method setUp (line 83) | def setUp(self): method test_fold_multiply (line 87) | def test_fold_multiply(test_case): method test_fold_multiply_cuda (line 94) | def test_fold_multiply_cuda(test_case): method test_fold_multiply_complex (line 100) | def test_fold_multiply_complex(test_case): method test_fold_multiply_complex_cuda (line 107) | def test_fold_multiply_complex_cuda(test_case): method test_fold_multiply_with_input (line 113) | def test_fold_multiply_with_input(test_case): method test_fold_multiply_with_input_cuda (line 124) | def test_fold_multiply_with_input_cuda(test_case): FILE: oneflow/ir/test/OneFlow/fuse/test_cast_optimal_pass.py function _cast_optimal_pass (line 28) | def _cast_optimal_pass(test_case, dtype): class TestCastOpOptimalPass (line 45) | class TestCastOpOptimalPass(flow.unittest.MLIRTestCase): method setUp (line 46) | def setUp(self): method test_case_optimal_pass (line 54) | def test_case_optimal_pass(test_case): FILE: oneflow/ir/test/OneFlow/fuse/test_fuse_pad_conv.py function do_pad_conv_graph (line 28) | def do_pad_conv_graph(test_case, with_cuda, with_bias, with_nchw=True): class TestFusePadConv (line 62) | class TestFusePadConv(oneflow.unittest.MLIRTestCase): method setUp (line 63) | def setUp(self): method test_pad_conv_graph_cuda (line 72) | def test_pad_conv_graph_cuda(test_case): method test_pad_conv_graph_cpu (line 77) | def test_pad_conv_graph_cpu(test_case): FILE: oneflow/ir/test/OneFlow/kernel_launch/test_resnet.py function _test_okl_resnet (line 32) | def _test_okl_resnet(test_case): class TestOKLResNet (line 56) | class TestOKLResNet(flow.unittest.MLIRTestCase): method setUp (line 57) | def setUp(self): method test_okl_resnet (line 66) | def test_okl_resnet(test_case): FILE: oneflow/ir/test/OneFlow/networks/resnet50.py function conv3x3 (line 22) | def conv3x3( function conv1x1 (line 38) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: class BasicBlock (line 43) | class BasicBlock(nn.Module): method __init__ (line 46) | def __init__( method forward (line 73) | def forward(self, x: Tensor) -> Tensor: class Bottleneck (line 92) | class Bottleneck(nn.Module): method __init__ (line 95) | def __init__( method forward (line 121) | def forward(self, x: Tensor) -> Tensor: class ResNet (line 144) | class ResNet(nn.Module): method __init__ (line 145) | def __init__( method _make_layer (line 210) | def _make_layer( method _forward_impl (line 258) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 275) | def forward(self, x: Tensor) -> Tensor: function _resnet (line 279) | def _resnet( function resnet50 (line 289) | def resnet50(**kwargs: Any) -> ResNet: FILE: oneflow/ir/test/OneFlow/psig/test_2nd_basic_parse.py function _test_nd_basic_parse (line 27) | def _test_nd_basic_parse(test_case): class TestBasicParse (line 60) | class TestBasicParse(flow.unittest.MLIRTestCase): method setUp (line 61) | def setUp(self): method test_nd_basic_parse (line 64) | def test_nd_basic_parse(test_case): FILE: oneflow/ir/test/OneFlow/psig/test_basic_parse.py function _test_1nd_basic_parse (line 27) | def _test_1nd_basic_parse(test_case): class TestBasicParse (line 59) | class TestBasicParse(flow.unittest.MLIRTestCase): method setUp (line 60) | def setUp(self): method test_1nd_basic_parse (line 63) | def test_1nd_basic_parse(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_conv_bn_auto_nhwc.py function _test_fuse_conv_bn (line 28) | def _test_fuse_conv_bn(test_case, with_cuda): class TestFuseConvBn (line 56) | class TestFuseConvBn(oneflow.unittest.MLIRTestCase): method setUp (line 57) | def setUp(self): method test_fuse_conv_bn_cuda (line 66) | def test_fuse_conv_bn_cuda(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fuse_bias_add_dropout.py function do_bias_add_dropout_graph (line 30) | def do_bias_add_dropout_graph(test_case, with_cuda, prob): class TestBiasAddDropout (line 59) | class TestBiasAddDropout(oneflow.unittest.MLIRTestCase): method setUp (line 60) | def setUp(self): method test_bias_add_dropout_graph (line 65) | def test_bias_add_dropout_graph(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fuse_bias_add_gelu.py function do_bias_add_gelu_graph (line 29) | def do_bias_add_gelu_graph(test_case, with_cuda): class TestBiasAddGelu (line 55) | class TestBiasAddGelu(oneflow.unittest.MLIRTestCase): method setUp (line 56) | def setUp(self): method test_bias_add_gelu_graph (line 61) | def test_bias_add_gelu_graph(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fuse_bn_add_relu.py function do_normalization_add_relu_graph (line 30) | def do_normalization_add_relu_graph(test_case, with_cuda): class TestNormalizationAddRelu (line 73) | class TestNormalizationAddRelu(oneflow.unittest.MLIRTestCase): method setUp (line 74) | def setUp(self): method test_normalization_add_relu_graph (line 79) | def test_normalization_add_relu_graph(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fuse_gelu.py class GEGLU (line 35) | class GEGLU(nn.Module): method __init__ (line 44) | def __init__( method forward (line 50) | def forward(self, hidden_states): class GraphToRun (line 55) | class GraphToRun(flow.nn.Graph): method __init__ (line 56) | def __init__(self, gelu_mod): method build (line 60) | def build(self, hidden_states): function do_fused_gelu_graph (line 64) | def do_fused_gelu_graph(test_case, dev, fuse_linear=False): class TestFusedGelu (line 79) | class TestFusedGelu(oneflow.unittest.MLIRTestCase): method setUp (line 80) | def setUp(self): method test_fused_gelu_graph (line 85) | def test_fused_gelu_graph(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fuse_scale_tril.py function _test_fused_scale_tril (line 29) | def _test_fused_scale_tril( class FusedScaleTrilTestCase (line 59) | class FusedScaleTrilTestCase(flow.unittest.MLIRTestCase): method setUp (line 60) | def setUp(self): method test_fused_scale_tril (line 65) | def test_fused_scale_tril(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fused_matmul_bias.py function _matmul_bias0 (line 28) | def _matmul_bias0(x, weight, bias): function _matmul_bias1 (line 34) | def _matmul_bias1(x, w, bias): function do_fused_matmul_bias_graph (line 38) | def do_fused_matmul_bias_graph(test_case, dev): class TestGroupMatMulBias (line 67) | class TestGroupMatMulBias(oneflow.unittest.MLIRTestCase): method setUp (line 68) | def setUp(self): method test_fused_matmul_bias_graph (line 74) | def test_fused_matmul_bias_graph(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_fused_multi_head_attention_inference.py function _ref (line 33) | def _ref(query, key, value, num_heads, causal=False): function _ref2 (line 57) | def _ref2(query, key, value, num_heads, causal=False): function _fused_mha (line 102) | def _fused_mha(query, key, value, num_heads, causal=False): class GraphToRun (line 108) | class GraphToRun(flow.nn.Graph): method __init__ (line 109) | def __init__(self, ref=None, num_heads=None, causal=False): method build (line 115) | def build(self, query, key, value): function _test_fused_multi_head_attention_inference (line 119) | def _test_fused_multi_head_attention_inference( class TestFusedMultiHeadAttentionInference (line 163) | class TestFusedMultiHeadAttentionInference(flow.unittest.MLIRTestCase): method setUp (line 164) | def setUp(self): method test_multi_head_attention_inference (line 170) | def test_multi_head_attention_inference(test_case): FILE: oneflow/ir/test/OneFlow/with_cuda/test_graph_save_and_load.py class InferGraph (line 33) | class InferGraph(flow.nn.Graph): method __init__ (line 34) | def __init__(self, placement_arg=None): method build (line 44) | def build(self, image): class GraphSaveTestCase (line 52) | class GraphSaveTestCase(flow.unittest.MLIRTestCase): method setUp (line 53) | def setUp(self): method test_save_and_load (line 56) | def test_save_and_load(self): FILE: oneflow/ir/test/lit.cfg.py function add_runtime (line 103) | def add_runtime(name): FILE: oneflow/maybe/error.h function namespace (line 33) | namespace oneflow { function Error (line 120) | decltype(auto) Error(U&& se) { function StackSize (line 124) | static std::size_t StackSize(const ValueType& se) { return PointedTraits... function ConstRefExceptVoid (line 126) | static ConstRefExceptVoid StackElem(const T& se, std::si... function Abort (line 142) | [[noreturn]] static void Abort(U&& se) { function string (line 154) | struct MessageFormatTrait { function string_view (line 171) | struct MessageFormatTrait { function error_ (line 205) | StackedError(ErrorType error) // NOLINT(google-explicit-constructor) FILE: oneflow/maybe/error_test.cpp type oneflow (line 26) | namespace oneflow { type maybe (line 27) | namespace maybe { type StackedErrorTraits> (line 30) | struct StackedErrorTraits> type StackedErrorTraits> (line 31) | struct StackedErrorTraits> function TEST (line 36) | TEST(StackedError, SimpleStackedError) { function TEST (line 62) | TEST(StackedError, SimpleNoStackError) { FILE: oneflow/maybe/just.h function namespace (line 26) | namespace oneflow { FILE: oneflow/maybe/just_test.cpp function TEST (line 26) | TEST(Just, MaybeBasic) { function TEST (line 57) | TEST(Just, MaybeVoid) { function TEST (line 101) | TEST(Just, MaybeRef) { function TEST (line 134) | TEST(Just, MaybeErrorPtr) { type oneflow (line 166) | namespace oneflow { type maybe (line 167) | namespace maybe { type JustTraits (line 170) | struct JustTraits { method ValueNotFoundError (line 172) | static simple::StackedError ValueNotFoundError(U&&) { method Value (line 177) | static decltype(auto) Value(U&& v) { function TEST (line 185) | TEST(Just, Optional) { function TEST (line 205) | TEST(Just, Ptr) { function TEST (line 225) | TEST(Just, WithMsg) { function TEST (line 280) | TEST(Just, JustOpt) { function TEST (line 303) | TEST(Just, NoStack) { FILE: oneflow/maybe/maybe.h function namespace (line 29) | namespace oneflow { function Error (line 98) | decltype(auto) Error() & { return this->Base::template Value(); } function Error (line 100) | decltype(auto) Error() && { return std::move(*this).Base::template Value... function Base (line 111) | MaybeStorage(const E& err) : Base(err) {} function Value (line 118) | void Value() const {} function Value (line 179) | decltype(auto) Value() const& { return Base::Value(); } FILE: oneflow/maybe/maybe_test.cpp function TEST (line 23) | TEST(Maybe, Basic) { function TEST (line 46) | TEST(Maybe, NonPOD) { function TEST (line 79) | TEST(Maybe, Reference) { function TEST (line 117) | TEST(Maybe, Void) { function TEST (line 135) | TEST(Maybe, PtrError) { function TEST (line 146) | TEST(Maybe, NoStack) { function TEST (line 159) | TEST(Maybe, Monadic) { FILE: oneflow/maybe/optional.h function namespace (line 27) | namespace oneflow { function Init (line 136) | void Init() { function Reset (line 149) | void Reset() { has_ = false; } function Destory (line 151) | void Destory() {} function CopyConstruct (line 165) | void CopyConstruct(const OptionalStorage& s) { function Copy (line 170) | void Copy(const OptionalStorage& s) { CopyConstruct(s); } function Init (line 189) | void Init() { value_ = nullptr; } function T (line 191) | T Value() { return *value_; } function Reset (line 197) | void Reset() { value_ = nullptr; } function Destory (line 199) | void Destory() {} function Construct (line 201) | void Construct(T v) { value_ = &v; } function T (line 203) | T Emplace(T v) { function CopyConstruct (line 208) | void CopyConstruct(const OptionalStorage& s) { value_ = s.value_; } function Copy (line 210) | void Copy(const OptionalStorage& s) { CopyConstruct(s); } type OptionalPrivateScope (line 213) | struct OptionalPrivateScope { function OF_MAYBE_NODISCARD_TYPE (line 261) | OF_MAYBE_NODISCARD_TYPE Optional { function ValueOr (line 437) | decltype(auto) ValueOr(const T& v) const& { function Reset (line 472) | void Reset() { storage_.Reset(); } function namespace (line 509) | namespace std { FILE: oneflow/maybe/optional_test.cpp function TEST (line 25) | TEST(Optional, Scalar) { function TEST (line 76) | TEST(Optional, NonScalar) { function TEST (line 161) | TEST(Optional, Reference) { function TEST (line 214) | TEST(Optional, Hash) { function TEST (line 236) | TEST(Optional, Compare) { function TEST (line 282) | TEST(Optional, Monadic) { FILE: oneflow/maybe/type_traits.h function namespace (line 26) | namespace oneflow { FILE: oneflow/maybe/type_traits_test.cpp function TEST (line 23) | TEST(TypeTraits, Basics) { FILE: oneflow/maybe/utility.h function namespace (line 23) | namespace oneflow { function namespace (line 78) | namespace std { FILE: oneflow/maybe/utility_test.cpp function TEST (line 22) | TEST(Utility, NullOpt) { FILE: oneflow/maybe/variant.h function namespace (line 31) | namespace oneflow { function Destory (line 420) | void Destory() { function namespace (line 436) | namespace std { FILE: oneflow/maybe/variant_test.cpp function TEST (line 24) | TEST(Variant, Basics) { function TEST (line 77) | TEST(Variant, NonPOD) { function TEST (line 116) | TEST(Variant, Optional) { function TEST (line 146) | TEST(Variant, BinarySearchVisit) { function TEST (line 183) | TEST(Variant, Compare) { function TEST (line 226) | TEST(Variant, UniquePtr) { FILE: oneflow/user/data/batch_dataset.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/data/batch_random_shuffle_dataset.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/data/coco_data_reader.cpp type oneflow (line 26) | namespace oneflow { type data (line 27) | namespace data { function FOR_RANGE (line 135) | FOR_RANGE(size_t, i, 0, keypoints.size() / 3) { FILE: oneflow/user/data/coco_data_reader.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/data/coco_dataset.cpp type oneflow (line 21) | namespace oneflow { type data (line 22) | namespace data { FILE: oneflow/user/data/coco_dataset.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/data/coco_parser.cpp type oneflow (line 20) | namespace oneflow { type data (line 21) | namespace data { FILE: oneflow/user/data/coco_parser.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/data/data_reader.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/data/dataset.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/data/distributed_training_dataset.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/data/distributed_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/data/gpt_dataset.cpp type oneflow (line 26) | namespace oneflow { type data (line 28) | namespace data { function GetSplitDocIndices (line 32) | void GetSplitDocIndices(std::vector* doc_indices, const std:... function GetNumEpochs (line 52) | size_t GetNumEpochs(size_t num_samples, size_t seq_length, size_t to... function GetNumCompleteEpochs (line 61) | size_t GetNumCompleteEpochs(size_t num_samples, size_t seq_length, s... type stat (line 128) | struct stat function FOR_RANGE (line 199) | FOR_RANGE(size_t, i, 0, num_epochs) { function FOR_RANGE (line 209) | FOR_RANGE(size_t, i, 0, total_num_samples) { FILE: oneflow/user/data/gpt_dataset.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/data/group_batch_dataset.h function namespace (line 22) | namespace data { FILE: oneflow/user/data/ofrecord_data_reader.h function namespace (line 25) | namespace oneflow { FILE: oneflow/user/data/ofrecord_dataset.h function namespace (line 28) | namespace oneflow { FILE: oneflow/user/data/ofrecord_image_classification_data_reader.h function namespace (line 27) | namespace oneflow { FILE: oneflow/user/data/ofrecord_image_classification_dataset.cpp type oneflow (line 24) | namespace oneflow { type data (line 26) | namespace data { function DecodeImageFromOFRecord (line 32) | void DecodeImageFromOFRecord(const OFRecord& record, const std::stri... function DecodeLabelFromFromOFRecord (line 60) | void DecodeLabelFromFromOFRecord(const OFRecord& record, const std::... function LoadWorker (line 77) | void LoadWorker(Dataset* record_dataset, function DecodeWorker (line 96) | void DecodeWorker(const std::string& image_feature_name, const std::... function GetNumLocalDecodeThreads (line 116) | int32_t GetNumLocalDecodeThreads(int32_t num_decode_threads_per_mach... FILE: oneflow/user/data/ofrecord_image_classification_dataset.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/data/ofrecord_image_classification_parser.h function namespace (line 25) | namespace oneflow { FILE: oneflow/user/data/ofrecord_parser.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/data/parser.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/data/random_shuffle_dataset.h function namespace (line 24) | namespace data { FILE: oneflow/user/image/crop_window.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/image/image_util.cpp type oneflow (line 19) | namespace oneflow { function GenCvMat4ImageBuffer (line 41) | cv::Mat GenCvMat4ImageBuffer(const TensorBuffer& image_buffer) { function GenCvMat4ImageTensor (line 61) | cv::Mat GenCvMat4ImageTensor(const user_op::Tensor* image_tensor, int ... function CvMatConvertToDataType (line 92) | void CvMatConvertToDataType(const cv::Mat& src, cv::Mat* dst, DataType... function GetCvInterpolationFlag (line 102) | int GetCvInterpolationFlag(const std::string& interp_type, int org_w, ... function CheckInterpolationValid (line 123) | bool CheckInterpolationValid(const std::string& interp_type, std::ostr... FILE: oneflow/user/image/image_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/image/jpeg_decoder.cpp type oneflow (line 22) | namespace oneflow { class LibjpegCtx (line 24) | class LibjpegCtx { method LibjpegCtx (line 26) | explicit LibjpegCtx(struct jpeg_decompress_struct* compress_info) type jpeg_decompress_struct (line 30) | struct jpeg_decompress_struct type jpeg_decompress_struct (line 35) | struct jpeg_decompress_struct function JpegPartialDecodeRandomCropImage (line 38) | bool JpegPartialDecodeRandomCropImage(const unsigned char* data, size_... function OpenCvPartialDecodeRandomCropImage (line 104) | void OpenCvPartialDecodeRandomCropImage(const unsigned char* data, siz... FILE: oneflow/user/image/jpeg_decoder.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/image/jpeg_decoder_test.cpp type oneflow (line 27) | namespace oneflow { function GenerateImage (line 30) | void GenerateImage(std::vector& jpg, int w, int h) { function TEST (line 64) | TEST(JPEG, decoder) { FILE: oneflow/user/image/random_crop_generator.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/user/image/random_crop_generator.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/acc_kernel.cpp type oneflow (line 20) | namespace oneflow { class AccKernel (line 24) | class AccKernel final : public user_op::OpKernel { method AccKernel (line 26) | AccKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 41) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/activation_kernels.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/kernels/adaptive_avg_pool_cpu_kernel.cpp type oneflow (line 19) | namespace oneflow { function AvgForwardCompute (line 24) | void AvgForwardCompute(user_op::KernelComputeContext* ctx, const int32... function AvgBackwardCompute (line 83) | void AvgBackwardCompute(user_op::KernelComputeContext* ctx, const int3... class AdaptivePool1DCpuKernel (line 142) | class AdaptivePool1DCpuKernel final : public user_op::OpKernel { method AdaptivePool1DCpuKernel (line 144) | AdaptivePool1DCpuKernel() = default; method Compute (line 148) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 156) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AdaptivePool2DCpuKernel (line 160) | class AdaptivePool2DCpuKernel final : public user_op::OpKernel { method AdaptivePool2DCpuKernel (line 162) | AdaptivePool2DCpuKernel() = default; method Compute (line 166) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 174) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AdaptivePool3DCpuKernel (line 178) | class AdaptivePool3DCpuKernel final : public user_op::OpKernel { method AdaptivePool3DCpuKernel (line 180) | AdaptivePool3DCpuKernel() = default; method Compute (line 184) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 192) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AdaptivePool1DCpuGradKernel (line 195) | class AdaptivePool1DCpuGradKernel final : public user_op::OpKernel { method AdaptivePool1DCpuGradKernel (line 197) | AdaptivePool1DCpuGradKernel() = default; method Compute (line 201) | void Compute(user_op::KernelComputeContext* ctx) const override { Av... method AlwaysComputeWhenAllOutputsEmpty (line 202) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AdaptivePool2DCpuGradKernel (line 206) | class AdaptivePool2DCpuGradKernel final : public user_op::OpKernel { method AdaptivePool2DCpuGradKernel (line 208) | AdaptivePool2DCpuGradKernel() = default; method Compute (line 212) | void Compute(user_op::KernelComputeContext* ctx) const override { Av... method AlwaysComputeWhenAllOutputsEmpty (line 213) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AdaptivePool3DCpuGradKernel (line 216) | class AdaptivePool3DCpuGradKernel final : public user_op::OpKernel { method AdaptivePool3DCpuGradKernel (line 218) | AdaptivePool3DCpuGradKernel() = default; method Compute (line 222) | void Compute(user_op::KernelComputeContext* ctx) const override { Av... method AlwaysComputeWhenAllOutputsEmpty (line 223) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/adaptive_max_pool_cpu_kernel.cpp type oneflow (line 18) | namespace oneflow { function AdapativeMaxPoolForward (line 22) | void AdapativeMaxPoolForward(user_op::KernelComputeContext* ctx) { function AdaptiveMaxPoolBackward (line 88) | void AdaptiveMaxPoolBackward(user_op::KernelComputeContext* ctx) { class AdaptiveMaxPoolNDCpuKernel (line 132) | class AdaptiveMaxPoolNDCpuKernel final : public user_op::OpKernel { method AdaptiveMaxPoolNDCpuKernel (line 134) | AdaptiveMaxPoolNDCpuKernel() = default; method Compute (line 138) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 142) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AdaptiveMaxPoolNDGradCpuKernel (line 146) | class AdaptiveMaxPoolNDGradCpuKernel final : public user_op::OpKernel { method AdaptiveMaxPoolNDGradCpuKernel (line 148) | AdaptiveMaxPoolNDGradCpuKernel() = default; method Compute (line 152) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 156) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/adaptive_pool_kernel_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/add_n_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { function NewAddPrimitive (line 28) | std::unique_ptr NewAddPrimitive(Context* ctx) { class AddNKernel (line 33) | class AddNKernel : public OpKernel, public CudaGraphSupport { method AddNKernel (line 36) | AddNKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 39) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... method Compute (line 42) | void Compute(KernelComputeContext* ctx) const override { function AddPrimitiveExists (line 61) | auto AddPrimitiveExists() { FILE: oneflow/user/kernels/affine_grid_kernel.cpp type oneflow (line 23) | namespace oneflow { function GetBlasTransposeType (line 27) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 31) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewAffineGridMatmulPrimitive (line 41) | std::unique_ptr NewAffineGridMatmulPrimitive(Co... function AffineGridMatmulPrimitiveExists (line 47) | auto AffineGridMatmulPrimitiveExists() { function NewAffineGridGradMatmulPrimitive (line 55) | std::unique_ptr NewAffineGridGradMatmulPrimitiv... function AffineGridGradMatmulPrimitiveExists (line 61) | auto AffineGridGradMatmulPrimitiveExists() { class AffineGridKernel (line 71) | class AffineGridKernel final : public user_op::OpKernel { method AffineGridKernel (line 73) | AffineGridKernel() = default; method Compute (line 77) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 123) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class AffineGridGradKernel (line 146) | class AffineGridGradKernel final : public user_op::OpKernel { method AffineGridGradKernel (line 148) | AffineGridGradKernel() = default; method Compute (line 152) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 196) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/affine_grid_kernel.h function namespace (line 22) | namespace oneflow { function kCUDA (line 94) | struct GenerateBaseGridImp { FILE: oneflow/user/kernels/arange_kernel.cpp type oneflow (line 22) | namespace oneflow { type user_op (line 23) | namespace user_op { class ArangeOpKernelCache (line 24) | class ArangeOpKernelCache final : public user_op::OpKernelCache { method ArangeOpKernelCache (line 26) | ArangeOpKernelCache(int32_t lower, int32_t upper) : lower_(lower),... method lower (line 29) | int32_t lower() const { return lower_; } method upper (line 30) | int32_t upper() const { return upper_; } class ArangeKernel (line 37) | class ArangeKernel final : public OpKernel, public CudaGraphSupport { method ArangeKernel (line 39) | ArangeKernel() = default; method InitOpKernelCache (line 41) | std::shared_ptr InitOpKernelCache( method Compute (line 74) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 110) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/arange_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 20) | namespace user_op { type ArangeFunctor (line 22) | struct ArangeFunctor final { FILE: oneflow/user/kernels/arange_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/arg_sort_kernel.cpp type oneflow (line 19) | namespace oneflow { class CpuArgSortKernel (line 22) | class CpuArgSortKernel final : public user_op::OpKernel { method CpuArgSortKernel (line 24) | CpuArgSortKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 62) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/arg_where_kernel.cpp type oneflow (line 21) | namespace oneflow { class ArgWhereKernel (line 26) | class ArgWhereKernel final : public user_op::OpKernel { method ArgWhereKernel (line 28) | ArgWhereKernel() = default; method Compute (line 32) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 44) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method NdimCompute (line 51) | static void NdimCompute(user_op::KernelComputeContext* ctx) { function GetWorkspaceBytesSize (line 65) | size_t GetWorkspaceBytesSize(int64_t elem_cnt) { type SwitchUtil (line 71) | struct SwitchUtil type SwitchUtil (line 74) | struct SwitchUtil { type SwitchUtil (line 89) | struct SwitchUtil { function InferTempStorageBytesSize (line 104) | size_t InferTempStorageBytesSize(user_op::InferContext* ctx) { FILE: oneflow/user/kernels/arg_where_kernel_util.cpp type oneflow (line 21) | namespace oneflow { type ArgWhereKernelUtil (line 24) | struct ArgWhereKernelUtil { method ArgWhere (line 25) | static void ArgWhere(ep::Stream* stream, const ShapeView& input_shap... method GetWorkspaceBytesSize (line 50) | static size_t GetWorkspaceBytesSize(ep::Stream* stream, int64_t elem... FILE: oneflow/user/kernels/arg_where_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/argmax_kernel.cpp type oneflow (line 20) | namespace oneflow { class CpuArgMaxKernel (line 23) | class CpuArgMaxKernel final : public user_op::OpKernel { method CpuArgMaxKernel (line 25) | CpuArgMaxKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 59) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/as_strided_kernel.cpp type oneflow (line 24) | namespace oneflow { type AsStridedFunctor (line 31) | struct AsStridedFunctor final { method FOR_RANGE (line 36) | FOR_RANGE(int64_t, i, 0, output_num) { type AsStridedGradFunctor (line 47) | struct AsStridedGradFunctor final { method FOR_RANGE (line 52) | FOR_RANGE(int64_t, i, 0, dy_num) { class CpuAsStridedKernel (line 65) | class CpuAsStridedKernel final : public user_op::OpKernel { method CpuAsStridedKernel (line 67) | CpuAsStridedKernel() = default; method Compute (line 72) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 87) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuAsStridedGradKernel (line 91) | class CpuAsStridedGradKernel final : public user_op::OpKernel { method CpuAsStridedGradKernel (line 93) | CpuAsStridedGradKernel() = default; method Compute (line 98) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 116) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/assign_if_kernel.cpp type oneflow (line 19) | namespace oneflow { class AssignIfCPUKernel (line 24) | class AssignIfCPUKernel final : public user_op::OpKernel { method AssignIfCPUKernel (line 26) | AssignIfCPUKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 43) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } FILE: oneflow/user/kernels/assign_kernel.cpp type oneflow (line 19) | namespace oneflow { class AssignKernel (line 23) | class AssignKernel final : public user_op::OpKernel { method AssignKernel (line 25) | AssignKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 41) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } FILE: oneflow/user/kernels/avg_pool_kernel.cpp type oneflow (line 18) | namespace oneflow { type AvgPoolOpKernelCache (line 20) | struct AvgPoolOpKernelCache final : public user_op::OpKernelCache { method AvgPoolOpKernelCache (line 22) | explicit AvgPoolOpKernelCache(const AvgPoolParams3D& params_3d) : pa... method AvgPoolParams3D (line 23) | const AvgPoolParams3D& GetParams3D() const { return params_3d; } function CreateAvgOpKernelCache (line 26) | std::shared_ptr CreateAvgOpKernelCache(user_op::... type AvgPoolKernelUtil (line 45) | struct AvgPoolKernelUtil { method Avgpool1dForward (line 46) | static void Avgpool1dForward(ep::Stream* stream, const NdIndexOffset... method Avgpool1dBackward (line 55) | static void Avgpool1dBackward(ep::Stream* stream, const NdIndexOffse... method Avgpool2dForward (line 64) | static void Avgpool2dForward(ep::Stream* stream, const NdIndexOffset... method Avgpool2dBackward (line 75) | static void Avgpool2dBackward(ep::Stream* stream, const NdIndexOffse... method Avgpool3dForward (line 86) | static void Avgpool3dForward(ep::Stream* stream, const NdIndexOffset... method Avgpool3dBackward (line 98) | static void Avgpool3dBackward(ep::Stream* stream, const NdIndexOffse... class AvgPool1dKernel (line 112) | class AvgPool1dKernel final : public user_op::OpKernel { method AvgPool1dKernel (line 114) | AvgPool1dKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 117) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 118) | std::shared_ptr InitOpKernelCache( method Compute (line 124) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool1dGradKernel (line 152) | class AvgPool1dGradKernel final : public user_op::OpKernel { method AvgPool1dGradKernel (line 154) | AvgPool1dGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 157) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 158) | std::shared_ptr InitOpKernelCache( method Compute (line 164) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool2dKernel (line 194) | class AvgPool2dKernel final : public user_op::OpKernel { method AvgPool2dKernel (line 196) | AvgPool2dKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 199) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 200) | std::shared_ptr InitOpKernelCache( method Compute (line 206) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool2dGradKernel (line 235) | class AvgPool2dGradKernel final : public user_op::OpKernel { method AvgPool2dGradKernel (line 237) | AvgPool2dGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 240) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 241) | std::shared_ptr InitOpKernelCache( method Compute (line 247) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool3dKernel (line 279) | class AvgPool3dKernel final : public user_op::OpKernel { method AvgPool3dKernel (line 281) | AvgPool3dKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 284) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 285) | std::shared_ptr InitOpKernelCache( method Compute (line 291) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool3dGradKernel (line 321) | class AvgPool3dGradKernel final : public user_op::OpKernel { method AvgPool3dGradKernel (line 323) | AvgPool3dGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 326) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 327) | std::shared_ptr InitOpKernelCache( method Compute (line 333) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/avg_pool_kernel_util.cpp type oneflow (line 18) | namespace oneflow { function GetAvg3DVec (line 20) | std::vector GetAvg3DVec(const std::vector& original_... function GetAvg3DPadVec (line 33) | std::vector GetAvg3DPadVec(const std::vector& origin... function GetNoDilationWindowedOutputShape (line 46) | const int64_t GetNoDilationWindowedOutputShape(int64_t input_size, int... function GetNoDilation3DOutputShape (line 60) | void GetNoDilation3DOutputShape(const DimVector& in, const std::vector... function Shape (line 105) | Shape AvgPoolParams3D::GetYShape() const { function Shape (line 127) | Shape AvgPoolParams3D::GetXShape5D() const { function Shape (line 131) | Shape AvgPoolParams3D::GetYShape5D() const { FILE: oneflow/user/kernels/avg_pool_kernel_util.h function namespace (line 29) | namespace oneflow { function Avgpool1dBackwardCompute (line 174) | void Avgpool1dBackwardCompute(const NdIndexOffsetHelper index_he... function Avgpool2dForwardCompute (line 212) | void Avgpool2dForwardCompute( function Avgpool2dBackwardCompute (line 259) | void Avgpool2dBackwardCompute( function Avgpool3dForwardCompute (line 303) | void Avgpool3dForwardCompute( function Avgpool3dBackwardCompute (line 355) | void Avgpool3dBackwardCompute( function HalfAvgpool1dForwardCompute (line 434) | void HalfAvgpool1dForwardCompute(const NdIndexOffsetHelper index... function HalfAvgpool1dBackwardCompute (line 472) | void HalfAvgpool1dBackwardCompute( function HalfAvgpool2dForwardCompute (line 506) | void HalfAvgpool2dForwardCompute( function HalfAvgpool2dBackwardCompute (line 552) | void HalfAvgpool2dBackwardCompute( function HalfAvgpool3dForwardCompute (line 596) | void HalfAvgpool3dForwardCompute( function HalfAvgpool3dBackwardCompute (line 648) | void HalfAvgpool3dBackwardCompute( FILE: oneflow/user/kernels/batch_gather_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 22) | namespace user_op { class BatchGatherKernel (line 25) | class BatchGatherKernel final : public user_op::OpKernel, public use... method BatchGatherKernel (line 27) | BatchGatherKernel() = default; method Compute (line 31) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 43) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/batch_gather_kernel_util.cpp type oneflow (line 18) | namespace oneflow { function Shape (line 22) | Shape GetFlatShape(const ShapeView& shape, const int64_t axis) { function BatchGatherForward (line 30) | void BatchGatherForward(ep::Stream* stream, const Blob* in, const Blob... function BatchGatherBackward (line 39) | void BatchGatherBackward(ep::Stream* stream, const Blob* out_diff, con... type BatchGatherSwitchUtil (line 50) | struct BatchGatherSwitchUtil final { type BatchGatherKernelUtilImpl (line 78) | struct BatchGatherKernelUtilImpl final { function FOR_RANGE (line 94) | FOR_RANGE(int64_t, batch_idx, 0, batch_num) { function FOR_RANGE (line 112) | FOR_RANGE(int64_t, batch_idx, 0, batch_num) { FILE: oneflow/user/kernels/batch_gather_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/batch_norm_kernel_utils.h function lastPow2 (line 33) | static int lastPow2(unsigned int n) { function T (line 46) | T ceil_div(T a, T b) { function float (line 77) | struct AccumulateType { function double (line 81) | struct AccumulateType { function getNumThreads (line 89) | static int32_t getNumThreads(int64_t nElem) { function device_sqrt (line 101) | float device_sqrt(float val) { function device_sqrt (line 106) | double device_sqrt(double val) { function __forceinline__ (line 111) | __forceinline__ T inv_std(T var, double eps) { function getMSB (line 120) | int32_t getMSB(int32_t val) { return 31 - __clz(val); } FILE: oneflow/user/kernels/bernoulli_kernel.cpp type oneflow (line 22) | namespace oneflow { class BernoulliKerenl (line 25) | class BernoulliKerenl final : public user_op::OpKernel { method BernoulliKerenl (line 27) | BernoulliKerenl() = default; method CreateOpKernelState (line 30) | std::shared_ptr CreateOpKernelState( method Compute (line 39) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 71) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/bias_add_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewPrimitive (line 25) | std::unique_ptr NewPrimitiv... class BiasAddUserKernel (line 31) | class BiasAddUserKernel final : public user_op::OpKernel, public user_... method BiasAddUserKernel (line 33) | BiasAddUserKernel() = default; method Compute (line 38) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 56) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function PrimitiveExists (line 59) | auto PrimitiveExists() { FILE: oneflow/user/kernels/binary_cross_entropy_kernel.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 20) | namespace user_op { function ComputeBinaryCrossEntropyOut (line 26) | void ComputeBinaryCrossEntropyOut(int64_t elem_cnt, const T* input, ... function ComputeBinaryCrossEntropyGradOut (line 41) | void ComputeBinaryCrossEntropyGradOut(int64_t elem_cnt, const T* inp... class BinaryCrossEntropyKernel (line 54) | class BinaryCrossEntropyKernel final : public user_op::OpKernel { method BinaryCrossEntropyKernel (line 56) | BinaryCrossEntropyKernel() = default; method Compute (line 61) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 76) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class BinaryCrossEntropyGradKernel (line 80) | class BinaryCrossEntropyGradKernel final : public user_op::OpKernel { method BinaryCrossEntropyGradKernel (line 82) | BinaryCrossEntropyGradKernel() = default; method Compute (line 87) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 103) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/binary_cross_entropy_with_logits_kernel.cpp type oneflow (line 22) | namespace oneflow { type user_op (line 23) | namespace user_op { function T (line 29) | inline T ComputeMaxVal(const T x) { function T (line 35) | inline T CalSigmoid(const T x) { function ComputeBinaryCrossEntropyWithLogitsOut (line 41) | void ComputeBinaryCrossEntropyWithLogitsOut(int64_t elem_cnt, const ... function ComputeBinaryCrossEntropyWithLogitsGradOut (line 65) | void ComputeBinaryCrossEntropyWithLogitsGradOut(int64_t elem_cnt, co... class BinaryCrossEntropyWithLogitsKernel (line 89) | class BinaryCrossEntropyWithLogitsKernel final : public user_op::OpK... method BinaryCrossEntropyWithLogitsKernel (line 91) | BinaryCrossEntropyWithLogitsKernel() = default; method Compute (line 96) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 133) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class BinaryCrossEntropyWithLogitsGradKernel (line 137) | class BinaryCrossEntropyWithLogitsGradKernel final : public user_op:... method BinaryCrossEntropyWithLogitsGradKernel (line 139) | BinaryCrossEntropyWithLogitsGradKernel() = default; method Compute (line 144) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 182) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function GenFwInferTmpSizeFn (line 186) | user_op::InferTmpSizeFn GenFwInferTmpSizeFn() { function GenBwInferTmpSizeFn (line 196) | user_op::InferTmpSizeFn GenBwInferTmpSizeFn() { FILE: oneflow/user/kernels/binary_cross_entropy_with_logits_mean_kernel_util.h function namespace (line 17) | namespace oneflow { FILE: oneflow/user/kernels/binary_cross_entropy_with_logits_reduce_mean.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 21) | namespace user_op { function T (line 27) | inline T ComputeMaxVal(const T x) { function T (line 33) | inline T CalSigmoid(const T x) { type ComputeBinaryCrossEntropyWithLogitsReduceMeanOutFunctor (line 39) | struct ComputeBinaryCrossEntropyWithLogitsReduceMeanOutFunctor { method ComputeType (line 40) | inline ComputeType Compute(int64_t elem_cnt, const INPUT_T* input,... function ComputeBinaryCrossEntropyWithLogitsReduceMeanOut (line 55) | void ComputeBinaryCrossEntropyWithLogitsReduceMeanOut(int64_t elem_c... function ComputeBinaryCrossEntropyWithLogitsReduceMeanGradOut (line 68) | void ComputeBinaryCrossEntropyWithLogitsReduceMeanGradOut(int64_t el... class BinaryCrossEntropyWithLogitsReduceMeanKernel (line 82) | class BinaryCrossEntropyWithLogitsReduceMeanKernel final : public us... method BinaryCrossEntropyWithLogitsReduceMeanKernel (line 84) | BinaryCrossEntropyWithLogitsReduceMeanKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 86) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... method InitOpKernelCache (line 88) | std::shared_ptr InitOpKernelCache( method Compute (line 95) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... class BinaryCrossEntropyWithLogitsReduceMeanGradKernel (line 120) | class BinaryCrossEntropyWithLogitsReduceMeanGradKernel final : publi... method BinaryCrossEntropyWithLogitsReduceMeanGradKernel (line 122) | BinaryCrossEntropyWithLogitsReduceMeanGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 124) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... method InitOpKernelCache (line 126) | std::shared_ptr InitOpKernelCache( method Compute (line 133) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... FILE: oneflow/user/kernels/bincount_kernel.cpp type oneflow (line 20) | namespace oneflow { function BinCountComputeWeight (line 23) | void BinCountComputeWeight(const IDX* in_ptr, const T* weight, T* out_... function BinCountCompute (line 31) | void BinCountCompute(const IDX* in_ptr, T* out_ptr, int64_t size) { class CpuBinCountKernel (line 39) | class CpuBinCountKernel final : public user_op::OpKernel { method CpuBinCountKernel (line 41) | CpuBinCountKernel() = default; method Compute (line 45) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 63) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/broadcast_div_grad_kernel.cpp type oneflow (line 21) | namespace oneflow { class BroadcastDivGradKernel (line 26) | class BroadcastDivGradKernel final : public user_op::OpKernel { method BroadcastDivGradKernel (line 28) | BroadcastDivGradKernel() = default; method Compute (line 32) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 82) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/broadcast_like_kernel.cpp type oneflow (line 21) | namespace oneflow { class BroadcastLikeKernel (line 26) | class BroadcastLikeKernel final : public user_op::OpKernel, public use... method BroadcastLikeKernel (line 28) | BroadcastLikeKernel() = default; method Compute (line 32) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 43) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/cast_kernel.cpp type oneflow (line 24) | namespace oneflow { type user_op (line 26) | namespace user_op { function NewBroadcastPrimitive (line 31) | std::unique_ptr NewBroadca... class CastKernel (line 40) | class CastKernel final : public OpKernel, public user_op::CudaGraphS... method CastKernel (line 42) | CastKernel() = default; method Compute (line 46) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 75) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function BroadcastPrimitiveExists (line 78) | auto BroadcastPrimitiveExists() { FILE: oneflow/user/kernels/cast_to_static_shape_kernel.cpp type oneflow (line 19) | namespace oneflow { class CastToStaticShapeKernel (line 23) | class CastToStaticShapeKernel final : public user_op::OpKernel { method CastToStaticShapeKernel (line 25) | CastToStaticShapeKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 45) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/categorical_ordinal_encode_kernel.cpp type oneflow (line 19) | namespace oneflow { class CategoricalOrdinalEncodeKernel (line 22) | class CategoricalOrdinalEncodeKernel final : public user_op::OpKernel { method CategoricalOrdinalEncodeKernel (line 24) | CategoricalOrdinalEncodeKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 42) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } FILE: oneflow/user/kernels/categorical_ordinal_encode_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type CategoricalOrdinalEncodeKernelUtil (line 21) | struct CategoricalOrdinalEncodeKernelUtil { method Encode (line 22) | static void Encode(ep::Stream* stream, int64_t capacity, T* table, T... FILE: oneflow/user/kernels/categorical_ordinal_encode_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/clip_by_value_kernel.cpp type oneflow (line 23) | namespace oneflow { function GetDtypeMatchedValue (line 31) | float GetDtypeMatchedValue(double floating, int64_t integral) { function GetDtypeMatchedValue (line 36) | double GetDtypeMatchedValue(double floating, int64_t integral) { function GetDtypeMatchedValue (line 41) | int8_t GetDtypeMatchedValue(double floating, int64_t integral) { function GetDtypeMatchedValue (line 46) | int32_t GetDtypeMatchedValue(double floating, int64_t integral) { function GetDtypeMatchedValue (line 51) | int64_t GetDtypeMatchedValue(double floating, int64_t integral) { function half (line 57) | half GetDtypeMatchedValue(double floating, int64_t integral) { function float16 (line 67) | float16 GetDtypeMatchedValue(double floating, int64_t integral) { type ClipKernelUtil (line 74) | struct ClipKernelUtil { method Forward (line 76) | static void Forward(ep::Stream* stream, F clip_func, const int64_t n... method Backward (line 81) | static void Backward(ep::Stream* stream, F clip_func, const int64_t ... class ClipByScalarKernel (line 88) | class ClipByScalarKernel final : public user_op::OpKernel, public user... method ClipByScalarKernel (line 90) | ClipByScalarKernel() = default; method Compute (line 94) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 106) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ClipByScalarMinKernel (line 110) | class ClipByScalarMinKernel final : public user_op::OpKernel, public u... method ClipByScalarMinKernel (line 112) | ClipByScalarMinKernel() = default; method Compute (line 116) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 125) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ClipByScalarMaxKernel (line 129) | class ClipByScalarMaxKernel final : public user_op::OpKernel, public u... method ClipByScalarMaxKernel (line 131) | ClipByScalarMaxKernel() = default; method Compute (line 135) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 144) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ClipByScalarGradKernel (line 148) | class ClipByScalarGradKernel final : public user_op::OpKernel, public ... method ClipByScalarGradKernel (line 150) | ClipByScalarGradKernel() = default; method Compute (line 154) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 167) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ClipByScalarMinGradKernel (line 171) | class ClipByScalarMinGradKernel final : public user_op::OpKernel, publ... method ClipByScalarMinGradKernel (line 173) | ClipByScalarMinGradKernel() = default; method Compute (line 177) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 187) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ClipByScalarMaxGradKernel (line 191) | class ClipByScalarMaxGradKernel final : public user_op::OpKernel, publ... method ClipByScalarMaxGradKernel (line 193) | ClipByScalarMaxGradKernel() = default; method Compute (line 197) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 207) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/clip_by_value_kernel.h function namespace (line 22) | namespace oneflow { function min_value (line 66) | ClipByMinGradFunctor(T min) : min_value(min) {} function OF_DEVICE_FUNC (line 67) | OF_DEVICE_FUNC T operator()(T value, T grad) { function max_value (line 75) | ClipByMaxGradFunctor(T max) : max_value(max) {} function OF_DEVICE_FUNC (line 76) | OF_DEVICE_FUNC T operator()(T value, T grad) { FILE: oneflow/user/kernels/coco_reader_kernel.cpp type oneflow (line 19) | namespace oneflow { class COCOReaderWrapper (line 23) | class COCOReaderWrapper final : public user_op::OpKernelState { method COCOReaderWrapper (line 25) | explicit COCOReaderWrapper(user_op::KernelInitContext* ctx) : reader... method Read (line 28) | void Read(user_op::KernelComputeContext* ctx) { reader_.Read(ctx); } class COCOReaderKernel (line 34) | class COCOReaderKernel final : public user_op::OpKernel { method COCOReaderKernel (line 36) | COCOReaderKernel() = default; method CreateOpKernelState (line 39) | std::shared_ptr CreateOpKernelState( method Compute (line 46) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 51) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_all_gather.cpp type oneflow (line 23) | namespace oneflow { type ccl (line 25) | namespace ccl { function AllGatherImpl (line 29) | Maybe AllGatherImpl(const void* in, void* out, size_t elem_cnt... class CpuAllGather (line 82) | class CpuAllGather final : public AllGather { method CpuAllGather (line 85) | CpuAllGather() : datatype_(kInvalidDataType) {} method Init (line 88) | void Init(DataType datatype) override { this->datatype_ = datatype; } method Launch (line 90) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... method Launch (line 98) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_all_reduce.cpp type oneflow (line 23) | namespace oneflow { type ccl (line 25) | namespace ccl { type AllReduceImpl (line 30) | struct AllReduceImpl final { method Call (line 31) | static Maybe Call(const void* void_in, void* void_out, size_... class CpuAllReduce (line 131) | class CpuAllReduce final : public AllReduce { method CpuAllReduce (line 134) | CpuAllReduce() : datatype_(kInvalidDataType), reduce_type_(kInvali... method Init (line 137) | void Init(DataType datatype, ReduceType reduce_type) override { method Launch (line 142) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... method Launch (line 151) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_broadcast.cpp type oneflow (line 23) | namespace oneflow { type ccl (line 25) | namespace ccl { class CpuBroadcastImpl (line 28) | class CpuBroadcastImpl final : public Broadcast { method CpuBroadcastImpl (line 31) | CpuBroadcastImpl() : size_of_dtype_(0) {} method Init (line 34) | void Init(DataType datatype) override { method Launch (line 39) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_collective_communication_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/cpu/cpu_communication_context.cpp type oneflow (line 19) | namespace oneflow { type ccl (line 21) | namespace ccl { FILE: oneflow/user/kernels/collective_communication/cpu/cpu_communication_context.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/cpu/cpu_recv.cpp type oneflow (line 22) | namespace oneflow { type ccl (line 24) | namespace ccl { class CpuRecvImpl (line 27) | class CpuRecvImpl final : public Recv { method CpuRecvImpl (line 30) | CpuRecvImpl() : size_of_dtype_(0) {} method Init (line 33) | void Init(DataType datatype) override { method Launch (line 38) | void Launch(ep::Stream* stream, void* out, size_t elem_cnt, int64_... method Launch (line 43) | void Launch(ep::Stream* stream, void* out, size_t elem_cnt, int64_... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_reduce.cpp type oneflow (line 24) | namespace oneflow { type ccl (line 26) | namespace ccl { type ReduceImpl (line 31) | struct ReduceImpl final { method Call (line 32) | static Maybe Call(const void* void_in, void* void_out, size_... class CpuReduce (line 158) | class CpuReduce final : public Reduce { method CpuReduce (line 161) | CpuReduce() : datatype_(kInvalidDataType), reduce_type_(kInvalidRe... method Init (line 164) | void Init(DataType datatype, ReduceType reduce_type) override { method Launch (line 169) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_reduce_scatter.cpp type oneflow (line 23) | namespace oneflow { type ccl (line 25) | namespace ccl { type ReduceScatterImpl (line 30) | struct ReduceScatterImpl final { method Call (line 31) | static Maybe Call(const void* void_in, void* void_out, size_... class CpuReduceScatter (line 103) | class CpuReduceScatter final : public ReduceScatter { method CpuReduceScatter (line 106) | CpuReduceScatter() : datatype_(kInvalidDataType), reduce_type_(kIn... method Init (line 109) | void Init(DataType datatype, ReduceType reduce_type) override { method Launch (line 114) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... method Launch (line 123) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cpu/cpu_send.cpp type oneflow (line 22) | namespace oneflow { type ccl (line 24) | namespace ccl { class CpuSendImpl (line 27) | class CpuSendImpl final : public Send { method CpuSendImpl (line 30) | CpuSendImpl() : size_of_dtype_(0) {} method Init (line 33) | void Init(DataType datatype) override { method Launch (line 38) | void Launch(ep::Stream* stream, const void* in, size_t elem_cnt, i... method Launch (line 43) | void Launch(ep::Stream* stream, const void* in, size_t elem_cnt, i... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_all_gather.cpp type oneflow (line 21) | namespace oneflow { type ccl (line 23) | namespace ccl { class CudaAllGather (line 25) | class CudaAllGather final : public AllGather { method CudaAllGather (line 28) | CudaAllGather() : nccl_datatype_() {} method Init (line 31) | void Init(DataType datatype) override { this->nccl_datatype_ = Get... method Launch (line 33) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... method Launch (line 43) | virtual void Launch(ep::Stream* stream, const void* in, void* out,... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_all_reduce.cpp type oneflow (line 21) | namespace oneflow { type ccl (line 23) | namespace ccl { function ncclRedOp_t (line 27) | inline ncclRedOp_t GetNcclReduceType(ReduceType reduce_type) { class CudaAllReduce (line 39) | class CudaAllReduce final : public AllReduce { method CudaAllReduce (line 42) | CudaAllReduce() : nccl_datatype_(), nccl_reduce_op_() {} method Init (line 45) | void Init(DataType datatype, ReduceType reduce_type) override { method Launch (line 50) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... method Launch (line 60) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_all_to_all.cpp type oneflow (line 22) | namespace oneflow { type ccl (line 24) | namespace ccl { class CudaAllToAll (line 26) | class CudaAllToAll final : public AllToAll { method CudaAllToAll (line 29) | CudaAllToAll() method Init (line 33) | void Init(DataType send_dtype, DataType recv_dtype, size_t paralle... method Launch (line 41) | void Launch(ep::Stream* stream, void* send, int64_t send_count, vo... method Launch (line 64) | void Launch(ep::Stream* stream, void* send, const void* send_count... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_broadcast.cpp type oneflow (line 21) | namespace oneflow { type ccl (line 23) | namespace ccl { class CudaBroadcast (line 25) | class CudaBroadcast final : public Broadcast { method CudaBroadcast (line 28) | CudaBroadcast() : nccl_datatype_() {} method Init (line 31) | void Init(DataType datatype) override { this->nccl_datatype_ = Get... method Launch (line 33) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_communication_context.cpp type oneflow (line 21) | namespace oneflow { type ccl (line 23) | namespace ccl { function FOR_RANGE (line 27) | FOR_RANGE(int64_t, parallel_id, 0, parallel_desc->parallel_num()) { FILE: oneflow/user/kernels/collective_communication/cuda/cuda_communication_context.h function namespace (line 27) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/cuda/cuda_recv.cpp type oneflow (line 22) | namespace oneflow { type ccl (line 24) | namespace ccl { class CudaRecv (line 26) | class CudaRecv final : public Recv { method CudaRecv (line 29) | CudaRecv() : nccl_datatype_() {} method Init (line 32) | void Init(DataType datatype) override { this->nccl_datatype_ = Get... method Launch (line 34) | void Launch(ep::Stream* stream, void* out, size_t elem_cnt, int64_... method Launch (line 44) | void Launch(ep::Stream* stream, void* out, size_t elem_cnt, int64_... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_reduce.cpp type oneflow (line 21) | namespace oneflow { type ccl (line 23) | namespace ccl { function ncclRedOp_t (line 27) | inline ncclRedOp_t GetNcclReduceType(ReduceType reduce_type) { class CudaReduce (line 39) | class CudaReduce final : public Reduce { method CudaReduce (line 42) | CudaReduce() : nccl_datatype_(), nccl_reduce_op_() {} method Init (line 45) | void Init(DataType datatype, ReduceType reduce_type) override { method Launch (line 50) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_reduce_scatter.cpp type oneflow (line 21) | namespace oneflow { type ccl (line 23) | namespace ccl { function ncclRedOp_t (line 27) | inline ncclRedOp_t GetNcclReduceType(ReduceType reduce_type) { class CudaReduceScatter (line 39) | class CudaReduceScatter final : public ReduceScatter { method CudaReduceScatter (line 42) | CudaReduceScatter() : nccl_datatype_(), nccl_reduce_op_() {} method Init (line 45) | void Init(DataType datatype, ReduceType reduce_type) override { method Launch (line 50) | void Launch(ep::Stream* stream, const void* in, void* out, size_t ... method Launch (line 60) | virtual void Launch(ep::Stream* stream, const void* in, void* out,... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_send.cpp type oneflow (line 22) | namespace oneflow { type ccl (line 24) | namespace ccl { class CudaSend (line 26) | class CudaSend final : public Send { method CudaSend (line 29) | CudaSend() : nccl_datatype_() {} method Init (line 32) | void Init(DataType datatype) override { this->nccl_datatype_ = Get... method Launch (line 34) | void Launch(ep::Stream* stream, const void* in, size_t elem_cnt, i... method Launch (line 44) | void Launch(ep::Stream* stream, const void* in, size_t elem_cnt, i... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_send_recv_util.cpp type oneflow (line 22) | namespace oneflow { type ccl (line 24) | namespace ccl { function RawGetNcclCommAndPeerNcclRank (line 26) | std::pair RawGetNcclCommAndPeerNcclRank(int64_t... FILE: oneflow/user/kernels/collective_communication/cuda/cuda_send_recv_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/all_gather.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/all_reduce.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/all_to_all.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/broadcast.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/collective_communication.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/communication_context.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/recv.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/reduce.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/reduce_scatter.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/collective_communication/include/send.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/combined_margin_loss_kernel.cpp type oneflow (line 21) | namespace oneflow { class CombinedMarginLossOpKernelCache (line 25) | class CombinedMarginLossOpKernelCache final : public user_op::OpKernel... method CombinedMarginLossOpKernelCache (line 27) | CombinedMarginLossOpKernelCache(int64_t lower, int64_t upper) : lowe... method lower (line 30) | int64_t lower() const { return lower_; } method upper (line 31) | int64_t upper() const { return upper_; } function CreateCombinedMarginLossOpKernelCache (line 38) | std::shared_ptr CreateCombinedMarginLossOpKern... class CombinedMarginLossCpuKernel (line 62) | class CombinedMarginLossCpuKernel final : public user_op::OpKernel { method CombinedMarginLossCpuKernel (line 64) | CombinedMarginLossCpuKernel() = default; method InitOpKernelCache (line 67) | std::shared_ptr InitOpKernelCache( method Compute (line 73) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 108) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CombinedMarginLossGradCpuKernel (line 123) | class CombinedMarginLossGradCpuKernel final : public user_op::OpKernel { method CombinedMarginLossGradCpuKernel (line 125) | CombinedMarginLossGradCpuKernel() = default; method InitOpKernelCache (line 128) | std::shared_ptr InitOpKernelCache( method Compute (line 134) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 167) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/communicate_util.cpp type oneflow (line 23) | namespace oneflow { function IsSendAndRecvRegistered (line 34) | bool IsSendAndRecvRegistered(DeviceType device_type) { function Send (line 38) | Maybe Send(const void* in, size_t elem_cnt, DataType dtype, int6... function Recv (line 52) | Maybe Recv(void* out, size_t elem_cnt, DataType dtype, int64_t s... FILE: oneflow/user/kernels/communicate_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/complex_kernels.cpp type oneflow (line 27) | namespace oneflow { type user_op (line 28) | namespace user_op { FILE: oneflow/user/kernels/concat_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewCopyNdPrimitive (line 25) | std::unique_ptr NewCopyNdPrimitive(Context* ctx) { class ConcatKernel (line 29) | class ConcatKernel final : public user_op::OpKernel, public user_op::C... method ConcatKernel (line 31) | ConcatKernel() = default; method InferShape (line 35) | void InferShape(user_op::KernelInferContext* ctx) const override { method Compute (line 57) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 89) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function CopyNdPrimitiveExists (line 92) | auto CopyNdPrimitiveExists() { FILE: oneflow/user/kernels/constant_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 21) | namespace user_op { function NewFillPrimitive (line 26) | std::unique_ptr NewFillPrimitive(Context* ctx) { class ConstantKernel (line 31) | class ConstantKernel final : public OpKernel { method ConstantKernel (line 33) | ConstantKernel() = default; method Compute (line 37) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 53) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function FillPrimitiveExists (line 56) | auto FillPrimitiveExists() { FILE: oneflow/user/kernels/conv_cudnn_kernels.cpp type oneflow (line 27) | namespace oneflow { type CudnnConvArgsAndAlgo (line 32) | struct CudnnConvArgsAndAlgo final { method CudnnConvArgsAndAlgo (line 38) | CudnnConvArgsAndAlgo(const user_op::Tensor* x, const user_op::Tensor... method CudnnConvArgsAndAlgo (line 76) | CudnnConvArgsAndAlgo() = delete; function InferTmpSizeWithCudnn (line 81) | size_t InferTmpSizeWithCudnn(const user_op::TensorDesc* x, const user_... function CudnnTensorDesc (line 117) | CudnnTensorDesc* GetBiasCudnnTensorDesc(const std::string& data_format... function CudnnTensorDesc (line 129) | CudnnTensorDesc* GetBiasCudnnTensorDesc<3>(const std::string& data_for... type ConvCudnnOpKernelCache (line 140) | struct ConvCudnnOpKernelCache final : public user_op::OpKernelCache { class ConvGpuKernel (line 145) | class ConvGpuKernel final : public user_op::OpKernel, public user_op::... method ConvGpuKernel (line 147) | ConvGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 150) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method CreateConvCudnnOpKernelCache (line 152) | std::shared_ptr CreateConvCudnnOpKernelCache( method InitOpKernelCache (line 168) | std::shared_ptr InitOpKernelCache( method Compute (line 174) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method IsCudaGraphSupported (line 218) | bool IsCudaGraphSupported(user_op::KernelInitContext* ctx, class ConvDataGradGpuKernel (line 255) | class ConvDataGradGpuKernel final : public user_op::OpKernel, public u... method ConvDataGradGpuKernel (line 258) | ConvDataGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 261) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 264) | void Compute(user_op::KernelComputeContext* ctx) const override { method IsCudaGraphSupported (line 298) | bool IsCudaGraphSupported(user_op::KernelInitContext* ctx, class ConvFilterGradGpuKernel (line 328) | class ConvFilterGradGpuKernel final : public user_op::OpKernel, public... method ConvFilterGradGpuKernel (line 331) | ConvFilterGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 334) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 337) | void Compute(user_op::KernelComputeContext* ctx) const override { method IsCudaGraphSupported (line 364) | bool IsCudaGraphSupported(user_op::KernelInitContext* ctx, type ConvBiasGradState (line 387) | struct ConvBiasGradState final : public user_op::OpKernelState { class ConvBiasGradGpuKernel (line 391) | class ConvBiasGradGpuKernel final : public user_op::OpKernel, public u... method ConvBiasGradGpuKernel (line 393) | ConvBiasGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 396) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method CreateConvBiasGradState (line 398) | std::shared_ptr CreateConvBiasGradState( method Compute (line 421) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/conv_kernels.cpp type oneflow (line 22) | namespace oneflow { function GetBlasTransposeType (line 26) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 30) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewChannelsFirstMatmulPrimitive (line 40) | std::unique_ptr NewChannelsFirstMatmulPrimitive... function ChannelsFirstMatmulPrimitiveExists (line 46) | auto ChannelsFirstMatmulPrimitiveExists() { function NewChannelsLastMatmulPrimitive (line 54) | std::unique_ptr NewChannelsLastMatmulPrimitive(... function ChannelsLastMatmulPrimitiveExists (line 60) | auto ChannelsLastMatmulPrimitiveExists() { function NewConvDataGradTransATransBMatmulPrimitive (line 68) | std::unique_ptr NewConvDataGradTransATransBMatm... function ConvDataGradTransATransBMatmulPrimitiveExists (line 74) | auto ConvDataGradTransATransBMatmulPrimitiveExists() { function NewConvDataGradTransANoTransBMatmulPrimitive (line 82) | std::unique_ptr NewConvDataGradTransANoTransBMa... function ConvDataGradTransANoTransBMatmulPrimitiveExists (line 88) | auto ConvDataGradTransANoTransBMatmulPrimitiveExists() { function NewConvWeightGradTransATransBMatmulPrimitive (line 96) | std::unique_ptr NewConvWeightGradTransATransBMa... function ConvWeightGradTransATransBMatmulPrimitiveExists (line 102) | auto ConvWeightGradTransATransBMatmulPrimitiveExists() { function NewConvWeightGradNoTransATransBMatmulPrimitive (line 110) | std::unique_ptr NewConvWeightGradNoTransATransB... function ConvWeightGradNoTransATransBMatmulPrimitiveExists (line 117) | auto ConvWeightGradNoTransATransBMatmulPrimitiveExists() { function NewConvBiasGradNoTransANoTransBMatmulPrimitive (line 126) | std::unique_ptr NewConvBiasGradNoTransANoTransB... function ConvBiasGradNoTransANoTransBMatmulPrimitiveExists (line 133) | auto ConvBiasGradNoTransANoTransBMatmulPrimitiveExists() { function NewConvBiasGradTransANoTransBMatmulPrimitive (line 142) | std::unique_ptr NewConvBiasGradTransANoTransBMa... function ConvBiasGradTransANoTransBMatmulPrimitiveExists (line 148) | auto ConvBiasGradTransANoTransBMatmulPrimitiveExists() { function T (line 168) | T* GetImgMutDptr(user_op::Tensor* tensor, int64_t idx) { function T (line 173) | const T* GetImgDptr(const user_op::Tensor* tensor, int64_t idx) { function CalcElemNumOfColBuf (line 177) | size_t CalcElemNumOfColBuf(const ShapeView& out_shape, const ShapeView... class ColBufWriter (line 188) | class ColBufWriter { method ColBufWriter (line 190) | ColBufWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... class Im2ColWriter (line 222) | class Im2ColWriter final : public ColBufWriter { method Im2ColWriter (line 224) | Im2ColWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... method DHWCWrite (line 229) | void DHWCWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method CDHWWrite (line 233) | void CDHWWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method InvalidDFunc (line 236) | void InvalidDFunc() override { method InvalidHFunc (line 239) | void InvalidHFunc() override { method InvalidWFunc (line 242) | void InvalidWFunc() override { method NextImCSize (line 245) | void NextImCSize() override { this->src_ptr_ += this->c_size_; } class Col2ImWriter (line 249) | class Col2ImWriter final : public ColBufWriter { method Col2ImWriter (line 251) | Col2ImWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... method DHWCWrite (line 256) | void DHWCWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method CDHWWrite (line 260) | void CDHWWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method InvalidDFunc (line 263) | void InvalidDFunc() override { this->src_ptr_ += this->od_size_; } method InvalidHFunc (line 264) | void InvalidHFunc() override { this->src_ptr_ += this->oh_size_; } method InvalidWFunc (line 265) | void InvalidWFunc() override { this->src_ptr_ += this->ow_size_; } method NextImCSize (line 266) | void NextImCSize() override { this->dst_ptr_ += this->c_size_; } class ColBufUtil (line 273) | class ColBufUtil final { method ColBufUtil (line 275) | ColBufUtil(const ShapeView& in_shape, const ShapeView& out_shape, in... method FOR_RANGE (line 292) | FOR_RANGE(int64_t, od, 0, od_num_) { type ConvKernelUtil (line 332) | struct ConvKernelUtil final { method NCDHWIm2Col (line 334) | static void NCDHWIm2Col(const T* in_dptr, const ShapeView& in_shape, method NDHWCIm2Col (line 344) | static void NDHWCIm2Col(const T* in_dptr, const ShapeView& in_shape, method NCDHWCol2Im (line 355) | static void NCDHWCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method NDHWCCol2Im (line 365) | static void NDHWCCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method DoNCDWHFunc (line 377) | static void DoNCDWHFunc(const ShapeView& weight_shape, ColBufUtil... method DoNDWHCFunc (line 390) | static void DoNDWHCFunc(const ShapeView& weight_shape, ColBufUtil... type ConvOpKernelCache (line 405) | struct ConvOpKernelCache final : public user_op::OpKernelCache { function CreateConvOpKernelCache (line 424) | std::shared_ptr> CreateConvOpKernelCache(user_op:... function InitBiasMulBuf (line 479) | void InitBiasMulBuf(T* dptr, int64_t num) { class ConvCpuKernel (line 484) | class ConvCpuKernel final : public user_op::OpKernel { method ConvCpuKernel (line 486) | ConvCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 489) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 491) | std::shared_ptr InitOpKernelCache( method Compute (line 497) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class ConvDataGradCpuKernel (line 615) | class ConvDataGradCpuKernel final : public user_op::OpKernel { method ConvDataGradCpuKernel (line 618) | ConvDataGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 621) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 623) | std::shared_ptr InitOpKernelCache( method Compute (line 629) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class ConvBiasGradCpuKernel (line 782) | class ConvBiasGradCpuKernel final : public user_op::OpKernel { method ConvBiasGradCpuKernel (line 785) | ConvBiasGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 788) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 791) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/convert_memory_format_kernel.cpp type oneflow (line 19) | namespace oneflow { class ConvertMemoryFormatKernel (line 21) | class ConvertMemoryFormatKernel final : public user_op::OpKernel { method ConvertMemoryFormatKernel (line 23) | ConvertMemoryFormatKernel() = default; method Compute (line 27) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 36) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/convert_memory_format_util.cpp type oneflow (line 24) | namespace oneflow { function NewPermutePrimitive (line 26) | std::unique_ptr NewPermutePrimitive(DeviceType... function NewMemcpyPrimitive (line 31) | std::unique_ptr NewMemcpyPrimitive(DeviceType d... function ComputeIdentity (line 36) | void ComputeIdentity(ep::Stream* stream, int ndim, const int64_t* shap... function ComputeContiguousToChannelsLast (line 46) | void ComputeContiguousToChannelsLast(ep::Stream* stream, int ndim, con... function ComputeChannelsLastToContiguous (line 59) | void ComputeChannelsLastToContiguous(ep::Stream* stream, int ndim, con... function ConvertMemoryFormat (line 80) | void ConvertMemoryFormat(ep::Stream* stream, const user_op::Tensor* in... function ConvertMemoryFormat (line 87) | void ConvertMemoryFormat(ep::Stream* stream, int ndim, const int64_t* ... function ConvertMemoryFormat (line 94) | void ConvertMemoryFormat(ep::Stream* stream, const ShapeView& shape, D... FILE: oneflow/user/kernels/convert_memory_format_util.h function namespace (line 18) | namespace oneflow { FILE: oneflow/user/kernels/copy_data_content_kernel.cpp type oneflow (line 21) | namespace oneflow { class CopyDataContentKernel (line 25) | class CopyDataContentKernel final : public user_op::OpKernel, public u... method CopyDataContentKernel (line 27) | CopyDataContentKernel() = default; method Compute (line 31) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 61) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/copy_hd_kernel.cpp type oneflow (line 19) | namespace oneflow { class CopyHdKernel (line 23) | class CopyHdKernel final : public user_op::OpKernel { method CopyHdKernel (line 25) | CopyHdKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 58) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/copy_kernel.cpp type oneflow (line 19) | namespace oneflow { class CopyKernel (line 23) | class CopyKernel final : public user_op::OpKernel { method CopyKernel (line 25) | CopyKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 45) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/count_not_finite_kernel.cpp type oneflow (line 18) | namespace oneflow { class MultiCountNotFiniteCpuKernel (line 21) | class MultiCountNotFiniteCpuKernel final : public user_op::OpKernel { method MultiCountNotFiniteCpuKernel (line 23) | MultiCountNotFiniteCpuKernel() = default; method Compute (line 27) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 40) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/ctc_greedy_decoder.cpp type oneflow (line 22) | namespace oneflow { type CTCGreedyDecoderFunctor (line 26) | struct CTCGreedyDecoderFunctor final { method FOR_RANGE (line 31) | FOR_RANGE(int64_t, b, 0, batch_size) { CHECK_GE(max_input_length, in... method FOR_RANGE (line 34) | FOR_RANGE(int64_t, b, 0, batch_size) { FILE: oneflow/user/kernels/ctc_greedy_decoder.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/ctc_loss_kernel.cpp type oneflow (line 19) | namespace oneflow { class CtcLossKernel (line 22) | class CtcLossKernel final : public user_op::OpKernel { method CtcLossKernel (line 24) | CtcLossKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 57) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CtcLossGradKernel (line 74) | class CtcLossGradKernel final : public user_op::OpKernel { method CtcLossGradKernel (line 76) | CtcLossGradKernel() = default; method Compute (line 80) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 117) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/ctc_loss_kernel_util.cpp type oneflow (line 18) | namespace oneflow { function get_target_prime (line 21) | int64_t get_target_prime(const TARGET* targets_ptr, const IDX* target_... type CtcLossKernelUtil (line 39) | struct CtcLossKernelUtil final { function FOR_RANGE (line 69) | FOR_RANGE(int64_t, b, 0, batch_size) { function FOR_RANGE (line 73) | FOR_RANGE(int32_t, b, 0, batch_size) { function FOR_RANGE (line 143) | FOR_RANGE(int64_t, i, 0, elem_cnt) { grad_ptr[i] = neginf; } function FOR_RANGE (line 145) | FOR_RANGE(int64_t, b, 0, batch_size) { FILE: oneflow/user/kernels/ctc_loss_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/cufft_plan_cache.h function namespace (line 36) | namespace oneflow { type CuFFTParams (line 147) | struct CuFFTParams { function class (line 188) | class CuFFTConfig { FILE: oneflow/user/kernels/cum_backward_kernel.cpp type oneflow (line 19) | namespace oneflow { function CumProdBackward (line 23) | void CumProdBackward(const T* dy_ptr, T* dx_ptr, const T* output_ptr, ... class CpuCumProdGradKernel (line 90) | class CpuCumProdGradKernel final : public user_op::OpKernel { method CpuCumProdGradKernel (line 92) | CpuCumProdGradKernel() = default; method Compute (line 96) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 120) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/cum_forward_kernel.cpp type oneflow (line 19) | namespace oneflow { function CumForward (line 23) | void CumForward(const T* in_ptr, T* out_ptr, int64_t up_space, int64_t... class CpuCumKernel (line 42) | class CpuCumKernel : public user_op::OpKernel { method CpuCumKernel (line 44) | CpuCumKernel() = default; method Compute (line 48) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 67) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/cutlass_conv_tuner.cpp type oneflow (line 28) | namespace oneflow { function IsWeakerAlginOperation (line 32) | bool IsWeakerAlginOperation(const cutlass::library::Operation* lhs, type Conv2dOperationCacheKey (line 51) | struct Conv2dOperationCacheKey { method Conv2dOperationCacheKey (line 55) | Conv2dOperationCacheKey(cutlass::library::ConvFunctionalKey function... type Conv2dProblemSizeHasher (line 83) | struct Conv2dProblemSizeHasher { type Conv2dConfigurationHasher (line 108) | struct Conv2dConfigurationHasher { type Conv2dOperationCacheKeyHasher (line 125) | struct Conv2dOperationCacheKeyHasher { function GetTensorSize (line 146) | size_t GetTensorSize(cutlass::library::NumericTypeID element, cutlass:... type CutlassConvTuner::Impl (line 164) | struct CutlassConvTuner::Impl { function CutlassConvTuner (line 360) | const CutlassConvTuner& CutlassConvTuner::Get() { FILE: oneflow/user/kernels/cutlass_conv_tuner.h function namespace (line 30) | namespace oneflow { FILE: oneflow/user/kernels/deconv_cpu_kernel.cpp type oneflow (line 22) | namespace oneflow { function GetBlasTransposeType (line 26) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 30) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewDeconvTransATransBMatmulPrimitive (line 40) | std::unique_ptr NewDeconvTransATransBMatmulPrim... function NewDeconvTransANoTransBMatmulPrimitive (line 46) | std::unique_ptr NewDeconvTransANoTransBMatmulPr... function DeconvTransATransBMatmulPrimitiveExists (line 51) | auto DeconvTransATransBMatmulPrimitiveExists() { function DeconvTransANoTransBMatmulPrimitiveExists (line 58) | auto DeconvTransANoTransBMatmulPrimitiveExists() { function T (line 72) | T* GetImgMutDptr(user_op::Tensor* tensor, int64_t idx) { function T (line 77) | const T* GetImgDptr(const user_op::Tensor* tensor, int64_t idx) { function CalcElemNumOfColBuf (line 81) | size_t CalcElemNumOfColBuf(const ShapeView& out_shape, const ShapeView... class ColBufWriter (line 91) | class ColBufWriter { method ColBufWriter (line 93) | ColBufWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... class Col2ImWriter (line 125) | class Col2ImWriter final : public ColBufWriter { method Col2ImWriter (line 127) | Col2ImWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... method DHWCWrite (line 132) | void DHWCWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method CDHWWrite (line 136) | void CDHWWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method InvalidDFunc (line 139) | void InvalidDFunc() override { this->src_ptr_ += this->od_size_; } method InvalidHFunc (line 140) | void InvalidHFunc() override { this->src_ptr_ += this->oh_size_; } method InvalidWFunc (line 141) | void InvalidWFunc() override { this->src_ptr_ += this->ow_size_; } method NextImCSize (line 142) | void NextImCSize() override { this->dst_ptr_ += this->c_size_; } class ColBufUtil (line 149) | class ColBufUtil final { method ColBufUtil (line 151) | ColBufUtil(const ShapeView& in_shape, const ShapeView& out_shape, in... method FOR_RANGE (line 168) | FOR_RANGE(int64_t, od, 0, od_num_) { type DeconvKernelUtil (line 208) | struct DeconvKernelUtil final { method NCDHWCol2Im (line 210) | static void NCDHWCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method NDHWCCol2Im (line 220) | static void NDHWCCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method DoNCDWHFunc (line 232) | static void DoNCDWHFunc(const ShapeView& weight_shape, ColBufUtil... method DoNDWHCFunc (line 245) | static void DoNDWHCFunc(const ShapeView& weight_shape, ColBufUtil... type DeconvOpKernelCache (line 260) | struct DeconvOpKernelCache final : public user_op::OpKernelCache { method Update (line 276) | void Update(const ShapeView& x_shape, const ShapeView& out_shape) { function CreateDeconvOpKernelCache (line 293) | std::shared_ptr> CreateDeconvOpKernelCache(user... class DeconvCpuKernel (line 345) | class DeconvCpuKernel final : public user_op::OpKernel { method DeconvCpuKernel (line 348) | DeconvCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 351) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCacheWithFlags (line 353) | void InitOpKernelCacheWithFlags( method Compute (line 366) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/deconv_cudnn_kernel.cpp type CudnnDeConvArgsAndAlgo (line 29) | struct CudnnDeConvArgsAndAlgo final { method CudnnDeConvArgsAndAlgo (line 36) | CudnnDeConvArgsAndAlgo(const user_op::Tensor* x, const user_op::Tensor... method CudnnDeConvArgsAndAlgo (line 71) | CudnnDeConvArgsAndAlgo() = delete; function InferTmpSizeWithCudnn (line 76) | size_t InferTmpSizeWithCudnn(const user_op::TensorDesc* x, const user_op... class DeConvGpuKernel (line 110) | class DeConvGpuKernel final : public user_op::OpKernel { method DeConvGpuKernel (line 112) | DeConvGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 115) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return false; } method Compute (line 118) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/deform_conv_kernel.cpp type oneflow (line 24) | namespace oneflow { function T (line 29) | T get_coordinate_weight(const T* im_data, int height, int width, T y, ... function T (line 56) | T DeformableIm2ColBilinear(const T* bottom_data, const int data_width,... function T (line 83) | T GetGradientWeight(T argmax_h, T argmax_w, const int h, const int w, ... function T (line 104) | T GetCoordinateWeight(T argmax_h, T argmax_w, const int height, const ... function T (line 144) | T bilinear_interpolate(const T* in, int height, int width, T h, T w) { function DeformableIm2Col (line 172) | void DeformableIm2Col(int n, const T* input, const T* offset, const T*... function DeformableCol2Im (line 219) | void DeformableCol2Im(int n, const T* col, const T* offset_data, const... function DeformableCol2ImCoord (line 271) | void DeformableCol2ImCoord(int n, const T* col_data, const T* im_data,... function GetBlasTransposeType (line 350) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 354) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewPermutePrimitive (line 363) | std::unique_ptr NewPermutePrimitive(Context* c... class DeformableConv2dCpuKernel (line 368) | class DeformableConv2dCpuKernel final : public user_op::OpKernel { method DeformableConv2dCpuKernel (line 370) | DeformableConv2dCpuKernel() = default; method Compute (line 375) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 439) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class DeformableConv2dInputGradCpuKernel (line 443) | class DeformableConv2dInputGradCpuKernel final : public user_op::OpKer... method DeformableConv2dInputGradCpuKernel (line 445) | DeformableConv2dInputGradCpuKernel() = default; method Compute (line 451) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 526) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class DeformableConv2dParamGradCpuKernel (line 530) | class DeformableConv2dParamGradCpuKernel final : public user_op::OpKer... method DeformableConv2dParamGradCpuKernel (line 532) | DeformableConv2dParamGradCpuKernel() = default; method Compute (line 538) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 609) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/det_kernel.cpp type oneflow (line 19) | namespace oneflow { function BatchCount (line 23) | static inline size_t BatchCount(const user_op::Tensor* batched_matrice... function MatrixStride (line 31) | static inline size_t MatrixStride(const user_op::Tensor* batched_matri... class DetKernel (line 40) | class DetKernel final : public user_op::OpKernel { method DetKernel (line 42) | DetKernel() = default; method Compute (line 46) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 66) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/diag_kernel.cpp type oneflow (line 22) | namespace oneflow { type DiagFunctor (line 26) | struct DiagFunctor final { method FOR_RANGE (line 30) | FOR_RANGE(int32_t, i, 0, size) { out_buf[i * stride] = in_buf[i]; } method FOR_RANGE (line 32) | FOR_RANGE(int32_t, i, 0, size) { out_buf[i] = in_buf[i * stride]; } type DiagGradFunctor (line 38) | struct DiagGradFunctor final { method FOR_RANGE (line 42) | FOR_RANGE(int32_t, i, 0, dx_cnt) { dx_buf[i] = dy_buf[i * stride]; } method FOR_RANGE (line 44) | FOR_RANGE(int32_t, i, 0, dy_cnt) { dx_buf[i * stride] = dy_buf[i]; } FILE: oneflow/user/kernels/diag_kernel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/diagonal_kernel.cpp type oneflow (line 23) | namespace oneflow { type DiagonalFunctor (line 27) | struct DiagonalFunctor final { method FOR_RANGE (line 31) | FOR_RANGE(int32_t, index, 0, size * dim2) { type DiagonalGradFunctor (line 40) | struct DiagonalGradFunctor final { method FOR_RANGE (line 44) | FOR_RANGE(int32_t, index, 0, size * dim2) { class CpuDiagonalKernel (line 55) | class CpuDiagonalKernel final : public user_op::OpKernel { method CpuDiagonalKernel (line 57) | CpuDiagonalKernel() = default; method Compute (line 62) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 84) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuDiagonalBackwardKernel (line 88) | class CpuDiagonalBackwardKernel final : public user_op::OpKernel { method CpuDiagonalBackwardKernel (line 90) | CpuDiagonalBackwardKernel() = default; method Compute (line 95) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 119) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/dim_gather_kernel_util.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 21) | namespace user_op { type DimGatherFunctor (line 24) | struct DimGatherFunctor final { FILE: oneflow/user/kernels/dim_gather_kernel_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/dim_gather_kernels.cpp type oneflow (line 22) | namespace oneflow { type user_op (line 23) | namespace user_op { function ConvertShape2Array (line 28) | void ConvertShape2Array(const ShapeView& shape_view, IDX_T* array, i... class DimGatherKernel (line 35) | class DimGatherKernel final : public user_op::OpKernel { method DimGatherKernel (line 37) | DimGatherKernel() = default; method Compute (line 41) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 62) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/dim_scatter_kernel_util.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 21) | namespace user_op { type DimScatterFunctor (line 24) | struct DimScatterFunctor final { FILE: oneflow/user/kernels/dim_scatter_kernel_util.h function namespace (line 30) | namespace oneflow { FILE: oneflow/user/kernels/dim_scatter_kernels.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 21) | namespace user_op { class DimScatterKernel (line 24) | class DimScatterKernel final : public user_op::OpKernel { method DimScatterKernel (line 26) | DimScatterKernel() = default; method Compute (line 30) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 74) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/dim_scatter_scalar_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 20) | namespace user_op { type DimScatterScalarFunctor (line 23) | struct DimScatterScalarFunctor f... FILE: oneflow/user/kernels/dim_scatter_scalar_kernel_util.h function namespace (line 28) | namespace oneflow { FILE: oneflow/user/kernels/dim_scatter_scalar_kernels.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 20) | namespace user_op { class DimScatterScalarKernel (line 23) | class DimScatterScalarKernel final : public user_op::OpKernel { method DimScatterScalarKernel (line 25) | DimScatterScalarKernel() = default; method Compute (line 29) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 75) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/distributions/common.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/distributions/exponential_distribution.cpp type oneflow (line 24) | namespace oneflow { function make64BitsFrom32Bits (line 26) | static uint64_t make64BitsFrom32Bits(uint32_t hi, uint32_t lo) { function T (line 31) | static T uniform_real(V val, T from, T to) { FILE: oneflow/user/kernels/distributions/exponential_distribution.h function namespace (line 26) | namespace oneflow { FILE: oneflow/user/kernels/distributions/exponential_kernel.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/user/kernels/distributions/exponential_kernel.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/distributions/multinomial_with_replacement_kernel.cpp type oneflow (line 22) | namespace oneflow { function InferTmpSizeForCpuKernel (line 26) | static size_t InferTmpSizeForCpuKernel(user_op::InferContext* ctx) { function T (line 33) | static T uniform_real(V val, T from, T to) { function make64BitsFrom32Bits (line 42) | static uint64_t make64BitsFrom32Bits(uint32_t hi, uint32_t lo) { class MultinomialWithReplacementCpuKernel (line 49) | class MultinomialWithReplacementCpuKernel final : public user_op::OpKe... method MultinomialWithReplacementCpuKernel (line 51) | MultinomialWithReplacementCpuKernel() = default; method CreateOpKernelState (line 54) | std::shared_ptr CreateOpKernelState( method Compute (line 64) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 149) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/distributions/normal_distribution.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/kernels/distributions/normal_distribution.h function namespace (line 27) | namespace oneflow { FILE: oneflow/user/kernels/distributions/normal_kernel.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/kernels/distributions/normal_kernel.h function namespace (line 25) | namespace oneflow { FILE: oneflow/user/kernels/distributions/uniform_distribution.cpp type oneflow (line 19) | namespace oneflow { class CPUUniformDistributionImpl (line 22) | class CPUUniformDistributionImpl class CPUUniformDistributionImpl::value>::type> (line 25) | class CPUUniformDistributionImpl(random_di... FILE: oneflow/user/kernels/distributions/uniform_int_distribution.h function namespace (line 26) | namespace oneflow { FILE: oneflow/user/kernels/distributions/uniform_int_kernel.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/user/kernels/distributions/uniform_int_kernel.h function namespace (line 25) | namespace oneflow { FILE: oneflow/user/kernels/distributions/uniform_kernel.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/user/kernels/distributions/uniform_kernel.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/dot_kernel.cpp type oneflow (line 21) | namespace oneflow { function NewMatmulPrimitive (line 28) | std::unique_ptr NewMatmulPrimitive(Context* ctx) { function MatmulPrimitiveExists (line 34) | auto MatmulPrimitiveExists() { class DotKernel (line 40) | class DotKernel final : public user_op::OpKernel { method DotKernel (line 42) | DotKernel() = default; method Compute (line 46) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 55) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/dropout_kernel.cpp type oneflow (line 23) | namespace oneflow { function MaskAndScale (line 28) | void MaskAndScale(ep::Stream* stream, const int64_t n, float scale, co... function FusedDropoutKernel (line 34) | void FusedDropoutKernel(ep::Stream* stream, const int64_t elem_cnt, class DropoutKernelCPU (line 50) | class DropoutKernelCPU final : public user_op::OpKernel { method DropoutKernelCPU (line 52) | DropoutKernelCPU() = default; method CreateOpKernelState (line 55) | std::shared_ptr CreateOpKernelState( method Compute (line 64) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 95) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class DropoutGradKernelCPU (line 114) | class DropoutGradKernelCPU final : public user_op::OpKernel { method DropoutGradKernelCPU (line 116) | DropoutGradKernelCPU() = default; method Compute (line 120) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 128) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/dropout_kernel.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/dynamic_loss_scale_schedule_kernel.cpp type oneflow (line 18) | namespace oneflow { class DynamicLossScaleScheduleCpuKernel (line 20) | class DynamicLossScaleScheduleCpuKernel final : public user_op::OpKern... method DynamicLossScaleScheduleCpuKernel (line 22) | DynamicLossScaleScheduleCpuKernel() = default; method Compute (line 26) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 56) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } FILE: oneflow/user/kernels/eager_b_to_s_kernel.cpp type oneflow (line 28) | namespace oneflow { function GetAllSplitNdSbp (line 32) | Maybe> GetAllSplitNdSbp(int64_t axis, int64_t ndim) { function GetAllBroadcastNdSbp (line 42) | Maybe> GetAllBroadcastNdSbp(int64_t ndim) { class EagerBToSOpKernelCache (line 52) | class EagerBToSOpKernelCache final : public user_op::OpKernelCache { method EagerBToSOpKernelCache (line 54) | explicit EagerBToSOpKernelCache(user_op::KernelCacheContext* ctx) { ... method Init (line 72) | void Init(user_op::KernelCacheContext* ctx) { function InferEagerBToSKernelTmpBufferSize (line 138) | size_t InferEagerBToSKernelTmpBufferSize(user_op::InferContext* ctx) { class EagerBToSKernel (line 156) | class EagerBToSKernel final : public user_op::OpKernel { method EagerBToSKernel (line 158) | EagerBToSKernel() = default; method InitOpKernelCacheWithFlags (line 161) | void InitOpKernelCacheWithFlags( method Compute (line 168) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 221) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_ccl_kernel.cpp type oneflow (line 27) | namespace oneflow { function AllReduceCollectiveCommunicationExists (line 31) | auto AllReduceCollectiveCommunicationExists() { function ReduceScatterCollectiveCommunicationExists (line 40) | auto ReduceScatterCollectiveCommunicationExists() { function AllGatherCollectiveCommunicationExists (line 49) | auto AllGatherCollectiveCommunicationExists() { function ReduceCollectiveCommunicationExists (line 58) | auto ReduceCollectiveCommunicationExists() { function BroadcastCollectiveCommunicationExists (line 67) | auto BroadcastCollectiveCommunicationExists() { class EagerCclOpKernelCache (line 76) | class EagerCclOpKernelCache final : public user_op::OpKernelCache { method EagerCclOpKernelCache (line 78) | explicit EagerCclOpKernelCache(user_op::KernelCacheContext* ctx) { I... method Init (line 86) | void Init(user_op::KernelCacheContext* ctx) { function InitEagerCclOpKernelCache (line 97) | void InitEagerCclOpKernelCache(user_op::KernelCacheContext* ctx, class EagerCclAllReduceKernel (line 106) | class EagerCclAllReduceKernel final : public user_op::OpKernel { method EagerCclAllReduceKernel (line 108) | EagerCclAllReduceKernel() = default; method InitOpKernelCacheWithFlags (line 111) | void InitOpKernelCacheWithFlags( method Compute (line 118) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 135) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class EagerCclReduceScatterKernel (line 142) | class EagerCclReduceScatterKernel final : public user_op::OpKernel { method EagerCclReduceScatterKernel (line 144) | EagerCclReduceScatterKernel() = default; method InitOpKernelCacheWithFlags (line 147) | void InitOpKernelCacheWithFlags( method Compute (line 155) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 172) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class EagerCclAllGatherKernel (line 179) | class EagerCclAllGatherKernel final : public user_op::OpKernel { method EagerCclAllGatherKernel (line 181) | EagerCclAllGatherKernel() = default; method InitOpKernelCacheWithFlags (line 184) | void InitOpKernelCacheWithFlags( method Compute (line 192) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 204) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class EagerCclReduceKernel (line 211) | class EagerCclReduceKernel final : public user_op::OpKernel { method EagerCclReduceKernel (line 213) | EagerCclReduceKernel() = default; method InitOpKernelCacheWithFlags (line 216) | void InitOpKernelCacheWithFlags( method Compute (line 224) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 249) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class EagerCclBroadcastKernel (line 256) | class EagerCclBroadcastKernel final : public user_op::OpKernel { method EagerCclBroadcastKernel (line 258) | EagerCclBroadcastKernel() = default; method InitOpKernelCacheWithFlags (line 261) | void InitOpKernelCacheWithFlags( method Compute (line 268) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method ComputeForOneInput (line 274) | void ComputeForOneInput(user_op::KernelComputeContext* ctx, const us... method AlwaysComputeWhenAllOutputsEmpty (line 296) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class EagerCclTouchKernel (line 303) | class EagerCclTouchKernel final : public user_op::OpKernel { method EagerCclTouchKernel (line 305) | EagerCclTouchKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 314) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class EagerCclS2SCpuOpKernelCache (line 324) | class EagerCclS2SCpuOpKernelCache final : public user_op::OpKernelCache { method EagerCclS2SCpuOpKernelCache (line 326) | explicit EagerCclS2SCpuOpKernelCache(user_op::KernelCacheContext* ct... method parallel_desc (line 329) | Symbol parallel_desc() const { return parallel_desc_; } method Init (line 332) | void Init(user_op::KernelCacheContext* ctx) { function InferEagerCclS2SCpuKernelTmpBufferSize (line 342) | size_t InferEagerCclS2SCpuKernelTmpBufferSize(user_op::InferContext* c... function RawGroupP2PPair (line 350) | Maybe>> RawGroupP2PPair( class EagerCclS2SCPUKernel (line 367) | class EagerCclS2SCPUKernel final : public user_op::OpKernel { method EagerCclS2SCPUKernel (line 369) | EagerCclS2SCPUKernel() = default; method InitOpKernelCacheWithFlags (line 372) | void InitOpKernelCacheWithFlags( method Compute (line 382) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 490) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_p_to_b_kernel.cpp type oneflow (line 27) | namespace oneflow { class EagerPToBOpKernelCache (line 31) | class EagerPToBOpKernelCache final : public user_op::OpKernelCache { method EagerPToBOpKernelCache (line 33) | explicit EagerPToBOpKernelCache(user_op::KernelCacheContext* ctx) { ... method Init (line 39) | void Init(user_op::KernelCacheContext* ctx) { function InferEagerPToBKernelTmpBufferSize (line 60) | size_t InferEagerPToBKernelTmpBufferSize(user_op::InferContext* ctx) { class EagerPToBKernel (line 69) | class EagerPToBKernel final : public user_op::OpKernel { method EagerPToBKernel (line 71) | EagerPToBKernel() = default; method InitOpKernelCacheWithFlags (line 74) | void InitOpKernelCacheWithFlags( method Compute (line 81) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 120) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_p_to_s_kernel.cpp type oneflow (line 30) | namespace oneflow { function GetAllSplitNdSbp (line 34) | Maybe> GetAllSplitNdSbp(int64_t axis, int64_t ndim) { function GetAllPartialSumNdSbp (line 44) | Maybe> GetAllPartialSumNdSbp(int64_t ndim) { class EagerPToSOpKernelCache (line 54) | class EagerPToSOpKernelCache final : public user_op::OpKernelCache { method EagerPToSOpKernelCache (line 56) | explicit EagerPToSOpKernelCache(user_op::KernelCacheContext* ctx) : ... method elem_cnt_of_this_chunk (line 61) | int64_t elem_cnt_of_this_chunk() const { return elem_cnt_of_this_chu... method Init (line 73) | void Init(user_op::KernelCacheContext* ctx) { function InferEagerPToSKernelTmpBufferSize (line 120) | size_t InferEagerPToSKernelTmpBufferSize(user_op::InferContext* ctx) { class EagerPToSKernel (line 138) | class EagerPToSKernel final : public user_op::OpKernel { method EagerPToSKernel (line 140) | EagerPToSKernel() = default; method InitOpKernelCacheWithFlags (line 143) | void InitOpKernelCacheWithFlags( method Compute (line 150) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 196) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_s_to_b_kernel.cpp type oneflow (line 28) | namespace oneflow { function GetAllSplitNdSbp (line 32) | Maybe> GetAllSplitNdSbp(int64_t axis, int64_t ndim) { function GetAllBroadcastNdSbp (line 42) | Maybe> GetAllBroadcastNdSbp(int64_t ndim) { class EagerSToBOpKernelCache (line 52) | class EagerSToBOpKernelCache final : public user_op::OpKernelCache { method EagerSToBOpKernelCache (line 54) | explicit EagerSToBOpKernelCache(user_op::KernelCacheContext* ctx) { ... method Init (line 72) | void Init(user_op::KernelCacheContext* ctx) { function InferEagerSToBKernelTmpBufferSize (line 120) | size_t InferEagerSToBKernelTmpBufferSize(user_op::InferContext* ctx) { class EagerSToBKernel (line 138) | class EagerSToBKernel final : public user_op::OpKernel { method EagerSToBKernel (line 140) | EagerSToBKernel() = default; method InitOpKernelCacheWithFlags (line 143) | void InitOpKernelCacheWithFlags( method Compute (line 150) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 196) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_s_to_p_kernel.cpp type oneflow (line 28) | namespace oneflow { function GetAllSplitNdSbp (line 32) | Maybe> GetAllSplitNdSbp(int64_t axis, int64_t ndim) { function GetAllPartialSumNdSbp (line 42) | Maybe> GetAllPartialSumNdSbp(int64_t ndim) { class EagerSToPOpKernelCache (line 52) | class EagerSToPOpKernelCache final : public user_op::OpKernelCache { method EagerSToPOpKernelCache (line 54) | explicit EagerSToPOpKernelCache(user_op::KernelCacheContext* ctx) { ... method Init (line 72) | void Init(user_op::KernelCacheContext* ctx) { function InferEagerSToPKernelTmpBufferSize (line 138) | size_t InferEagerSToPKernelTmpBufferSize(user_op::InferContext* ctx) { class EagerSToPKernel (line 155) | class EagerSToPKernel final : public user_op::OpKernel { method EagerSToPKernel (line 157) | EagerSToPKernel() = default; method InitOpKernelCacheWithFlags (line 160) | void InitOpKernelCacheWithFlags( method Compute (line 167) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 228) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_s_to_s_kernel.cpp type oneflow (line 27) | namespace oneflow { function ContainsEmptySlice (line 31) | bool ContainsEmptySlice(const std::vector& slices) { function GetAllSplitNdSbp (line 36) | Maybe> GetAllSplitNdSbp(int64_t axis, int64_t ndim) { class EagerNaiveSToSOpKernelCache (line 46) | class EagerNaiveSToSOpKernelCache final : public user_op::OpKernelCache { method EagerNaiveSToSOpKernelCache (line 48) | explicit EagerNaiveSToSOpKernelCache(user_op::KernelCacheContext* ct... method Init (line 66) | void Init(user_op::KernelCacheContext* ctx) { function InferNaiveSToSKernelTmpBufferSize (line 119) | size_t InferNaiveSToSKernelTmpBufferSize(user_op::InferContext* ctx) { class EagerNaiveSToSKernel (line 138) | class EagerNaiveSToSKernel final : public user_op::OpKernel { method EagerNaiveSToSKernel (line 140) | EagerNaiveSToSKernel() = default; method InitOpKernelCacheWithFlags (line 143) | void InitOpKernelCacheWithFlags( method Compute (line 150) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 196) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eager_symmetric_s_to_p_kernel.cpp type oneflow (line 24) | namespace oneflow { function NewMemsetPrimitive (line 29) | std::unique_ptr NewMemsetPrimitive(Context* ctx) { function MemsetPrimitiveExists (line 33) | auto MemsetPrimitiveExists() { function GetAllSplitNdSbp (line 39) | Maybe> GetAllSplitNdSbp(int64_t axis, int64_t ndim) { function GetAllPartialSumNdSbp (line 49) | Maybe> GetAllPartialSumNdSbp(int64_t ndim) { class EagerSymmetricSToPOpKernelCache (line 59) | class EagerSymmetricSToPOpKernelCache final : public user_op::OpKernel... method EagerSymmetricSToPOpKernelCache (line 61) | explicit EagerSymmetricSToPOpKernelCache(user_op::KernelCacheContext... method Init (line 69) | void Init(user_op::KernelCacheContext* ctx) { class EagerSymmetricSToPKernel (line 101) | class EagerSymmetricSToPKernel final : public user_op::OpKernel { method EagerSymmetricSToPKernel (line 103) | EagerSymmetricSToPKernel() = default; method InitOpKernelCacheWithFlags (line 106) | void InitOpKernelCacheWithFlags( method Compute (line 115) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 133) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/elementwise_maximum_minimum_kernel.cpp type oneflow (line 18) | namespace oneflow { type ElemwiseXimumGradFunctor (line 21) | struct ElemwiseXimumGradFunctor final { method XPU_1D_KERNEL_LOOP (line 24) | XPU_1D_KERNEL_LOOP(idx, elem_cnt) { type ElemwiseXimumFunctor (line 31) | struct ElemwiseXimumFunctor final { method FOR_RANGE (line 33) | FOR_RANGE(int64_t, idx, 0, elem_cnt) { z[idx] = Opt()(x[idx], y[i... FILE: oneflow/user/kernels/elementwise_maximum_minimum_kernel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/elementwise_primitive_kernel.h function namespace (line 27) | namespace oneflow { function Compute (line 88) | void Compute(user_op::KernelComputeContext* ctx) const override { function namespace (line 117) | namespace { FILE: oneflow/user/kernels/embedding_kernel.cpp type oneflow (line 21) | namespace oneflow { class CpuEmbeddingRenormKernel (line 24) | class CpuEmbeddingRenormKernel final : public user_op::OpKernel { method CpuEmbeddingRenormKernel (line 26) | CpuEmbeddingRenormKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 48) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuEmbeddingKernel (line 52) | class CpuEmbeddingKernel final : public user_op::OpKernel { method CpuEmbeddingKernel (line 54) | CpuEmbeddingKernel() = default; method Compute (line 58) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 77) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuEmbeddingGradKernel (line 81) | class CpuEmbeddingGradKernel final : public user_op::OpKernel { method CpuEmbeddingGradKernel (line 83) | CpuEmbeddingGradKernel() = default; method Compute (line 87) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 112) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/embedding_kernel_util.cpp type oneflow (line 19) | namespace oneflow { type EmbeddingReNormFunctor (line 22) | struct EmbeddingReNormFunctor final { type EmbeddingFunctor (line 48) | struct EmbeddingFunctor final { type EmbeddingGradFunctor (line 63) | struct EmbeddingGradFunctor final { FILE: oneflow/user/kernels/embedding_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/empty_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 22) | namespace user_op { class EmptyKernel (line 24) | class EmptyKernel final : public OpKernel { method EmptyKernel (line 26) | EmptyKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 39) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/erfinv_kernel.cpp type oneflow (line 18) | namespace oneflow { class CpuErfinvKernel (line 21) | class CpuErfinvKernel final : public user_op::OpKernel { method CpuErfinvKernel (line 23) | CpuErfinvKernel() = default; method Compute (line 27) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 67) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/expand_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewPrimitive (line 25) | std::unique_ptr NewPrimitive... function PrimitiveExists (line 34) | auto PrimitiveExists() { class ExpandKernel (line 43) | class ExpandKernel final : public user_op::OpKernel, public user_op::C... method ExpandKernel (line 45) | ExpandKernel() = default; method Compute (line 49) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 78) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/eye_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 21) | namespace user_op { class EyeKernel (line 23) | class EyeKernel final : public OpKernel { method EyeKernel (line 25) | EyeKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 40) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/eye_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 20) | namespace user_op { type EyeFunctor (line 23) | struct EyeFunctor final { FILE: oneflow/user/kernels/eye_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/fake_quantization_kernel.cpp type oneflow (line 20) | namespace oneflow { function FakeQuantizationPerLayerSymmetric (line 23) | void FakeQuantizationPerLayerSymmetric(const T* in_ptr, const T scale, function FakeQuantizationPerLayerAffine (line 37) | void FakeQuantizationPerLayerAffine(const T* in_ptr, const T scale, co... function FakeQuantizationPerLayerCambricon (line 52) | void FakeQuantizationPerLayerCambricon(const T* in_ptr, const T shift, class CpuFakeQuantizationKernel (line 67) | class CpuFakeQuantizationKernel final : public user_op::OpKernel { method CpuFakeQuantizationKernel (line 69) | CpuFakeQuantizationKernel() = default; method Compute (line 73) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 125) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/fft_kernel_util.cpp type oneflow (line 23) | namespace oneflow { function _conj_symmetry_cpu (line 26) | static void _conj_symmetry_cpu(T* data_out, const Shape& shape, const ... type FillConjSymmetryUtil (line 51) | struct FillConjSymmetryUtil { method FillConjSymmetryForward (line 52) | static void FillConjSymmetryForward(ep::Stream* stream, T* data_out,... type ComplexConvertUtil (line 62) | struct ComplexConvertUtil { method ConvertToDoubleSized (line 63) | static void ConvertToDoubleSized(ep::Stream* stream, const complex_t... method ConvertComplexToReal (line 84) | static void ConvertComplexToReal(ep::Stream* stream, const complex_t... type FftC2CKernelUtil (line 94) | struct FftC2CKernelUtil { method FftC2CForward (line 95) | static void FftC2CForward(ep::Stream* stream, const T* data_in, T* d... type FftR2CKernelUtil (line 108) | struct FftR2CKernelUtil { method FftR2CForward (line 109) | static void FftR2CForward(ep::Stream* stream, const IN* data_in, OUT... type FftC2RKernelUtil (line 121) | struct FftC2RKernelUtil { method FftC2RForward (line 122) | static void FftC2RForward(ep::Stream* stream, const IN* data_in, OUT... type FftStftKernelUtil (line 135) | struct FftStftKernelUtil { method FftStftForward (line 136) | static void FftStftForward(ep::Stream* stream, const IN* data_in, OU... type FillConjSymmetryUtil> (line 155) | struct FillConjSymmetryUtil> type FillConjSymmetryUtil> (line 156) | struct FillConjSymmetryUtil> type ComplexConvertUtil> (line 158) | struct ComplexConvertUtil> type ComplexConvertUtil> (line 159) | struct ComplexConvertUtil> type FftC2CKernelUtil, float> (line 161) | struct FftC2CKernelUtil, float> type FftC2CKernelUtil, double> (line 162) | struct FftC2CKernelUtil, double> type FftR2CKernelUtil> (line 164) | struct FftR2CKernelUtil> type FftR2CKernelUtil> (line 165) | struct FftR2CKernelUtil> type FftC2RKernelUtil, float> (line 167) | struct FftC2RKernelUtil, float> type FftC2RKernelUtil, double> (line 168) | struct FftC2RKernelUtil, double> type FftStftKernelUtil> (line 170) | struct FftStftKernelUtil> type FftStftKernelUtil> (line 171) | struct FftStftKernelUtil> FILE: oneflow/user/kernels/fft_kernel_util.h function namespace (line 28) | namespace oneflow { FILE: oneflow/user/kernels/fft_kernels.cpp type oneflow (line 23) | namespace oneflow { class FftC2CKernel (line 26) | class FftC2CKernel final : public user_op::OpKernel { method FftC2CKernel (line 28) | FftC2CKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 32) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 33) | void Compute(user_op::KernelComputeContext* ctx) const override { class FftR2CKernel (line 62) | class FftR2CKernel final : public user_op::OpKernel { method FftR2CKernel (line 64) | FftR2CKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 68) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 69) | void Compute(user_op::KernelComputeContext* ctx) const override { class FftC2RKernel (line 97) | class FftC2RKernel final : public user_op::OpKernel { method FftC2RKernel (line 99) | FftC2RKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 103) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 104) | void Compute(user_op::KernelComputeContext* ctx) const override { class StftCpuKernel (line 131) | class StftCpuKernel final : public user_op::OpKernel { method StftCpuKernel (line 133) | StftCpuKernel() = default; method Compute (line 138) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 183) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/fill_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewFillPrimitive (line 24) | std::unique_ptr NewFillPrimitive(Context* ctx) { class FillKernel (line 31) | class FillKernel final : public user_op::OpKernel { method FillKernel (line 33) | FillKernel() = default; method Compute (line 37) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 50) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function FillPrimitiveExists (line 53) | auto FillPrimitiveExists() { class FillTensorCpuKernel (line 60) | class FillTensorCpuKernel final : public user_op::OpKernel { method FillTensorCpuKernel (line 62) | FillTensorCpuKernel() = default; method Compute (line 66) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 75) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/flip_kernel.cpp type oneflow (line 20) | namespace oneflow { type SIZE_V (line 26) | struct SIZE_V { type VIS (line 30) | struct VIS { function FlipCpuForward (line 35) | void FlipCpuForward(const int32_t element, const int64_t total_dims, c... class FlipCpuKernel (line 56) | class FlipCpuKernel final : public user_op::OpKernel { method FlipCpuKernel (line 58) | FlipCpuKernel() = default; method Compute (line 62) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 86) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/fold_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { class FoldOpKernelState (line 30) | class FoldOpKernelState : public OpKernelState { method FoldOpKernelState (line 33) | FoldOpKernelState(const ShapeView& input_shape, const std::vector<... method ParamType (line 39) | const ParamType& params() const { return params_; } function CreateFoldOpKernelState (line 46) | std::shared_ptr> CreateFoldOp... class FoldKernel (line 57) | class FoldKernel final : public OpKernel { method FoldKernel (line 59) | FoldKernel() = default; method Compute (line 63) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 83) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/fold_kernel_util.cpp type oneflow (line 17) | namespace oneflow { type user_op (line 19) | namespace user_op { type FoldKernelUtil (line 24) | struct FoldKernelUtil { method Forward (line 26) | static void Forward(ep::Stream* stream, const void* raw_params, co... FILE: oneflow/user/kernels/fold_kernel_util.h function namespace (line 28) | namespace oneflow { FILE: oneflow/user/kernels/frac_kernel.cpp type oneflow (line 18) | namespace oneflow { class CpuFracKernel (line 21) | class CpuFracKernel final : public user_op::OpKernel { method CpuFracKernel (line 23) | CpuFracKernel() = default; method Compute (line 27) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 35) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/fused_cast_scale_kernel.cpp type oneflow (line 18) | namespace oneflow { class FusedCastScaleCpuKernel (line 21) | class FusedCastScaleCpuKernel final : public user_op::OpKernel { method FusedCastScaleCpuKernel (line 23) | FusedCastScaleCpuKernel() = default; method Compute (line 27) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 38) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/fused_clip_grad.h function Compute (line 34) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/fused_clip_grad_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/fused_weighted_sum_kernel.cpp type oneflow (line 20) | namespace oneflow { class FusedWeightedSumKernel (line 25) | class FusedWeightedSumKernel final : public user_op::OpKernel { method FusedWeightedSumKernel (line 27) | FusedWeightedSumKernel() = default; method Compute (line 32) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 59) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/gather_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { function Shape (line 27) | Shape GetFlatShape(ShapeView shape, int64_t axis) { class GatherOpKernelCache (line 31) | class GatherOpKernelCache final : public user_op::OpKernelCache { method GatherOpKernelCache (line 33) | GatherOpKernelCache(int64_t lower, int64_t upper) : lower_(lower),... method lower (line 36) | int64_t lower() const { return lower_; } method upper (line 37) | int64_t upper() const { return upper_; } function CheckNdSbp (line 44) | void CheckNdSbp(const Shape& hierarchy, int64_t gather_axis, const N... class GatherKernel (line 62) | class GatherKernel final : public user_op::OpKernel, public user_op:... method GatherKernel (line 64) | GatherKernel() = default; method InitOpKernelCache (line 67) | std::shared_ptr InitOpKernelCache( method Compute (line 86) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 109) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/gather_kernel_util.cpp type oneflow (line 18) | namespace oneflow { function Shape (line 22) | Shape GetFlatShape(const ShapeView& shape, int64_t axis) { function GatherForward (line 30) | void GatherForward(ep::Stream* stream, const Blob* indices, const Blob... type GatherSwitchUtil (line 39) | struct GatherSwitchUtil final { type GatherKernelUtilImpl (line 66) | struct GatherKernelUtilImpl final { function FOR_RANGE (line 79) | FOR_RANGE(int64_t, outer_idx, 0, outer_dim_size) { FILE: oneflow/user/kernels/gather_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/generate_random_batch_permutation_indices_kernel.cpp type oneflow (line 21) | namespace oneflow { class GenerateRandomBatchPermutationIndicesCPUKernel (line 23) | class GenerateRandomBatchPermutationIndicesCPUKernel final : public us... method GenerateRandomBatchPermutationIndicesCPUKernel (line 25) | GenerateRandomBatchPermutationIndicesCPUKernel() = default; method CreateOpKernelState (line 28) | std::shared_ptr CreateOpKernelState( method Compute (line 35) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 43) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/gpt_data_loader_kernel.cpp type oneflow (line 21) | namespace oneflow { function GetNumShards (line 28) | size_t GetNumShards(const Shape& hierarchy, const NdSbp& nd_sbp) { function GetShardIndex (line 39) | size_t GetShardIndex(const Shape& hierarchy, const NdSbp& nd_sbp, size... class GPTDataLoader (line 59) | class GPTDataLoader final : public OpKernelState { method GPTDataLoader (line 61) | GPTDataLoader(KernelInitContext* ctx) : batch_cnt_(0) { method GetBatch (line 77) | void GetBatch(size_t iter, user_op::Tensor* tokens) const { method NextBatch (line 90) | void NextBatch(user_op::Tensor* tokens) { class GPTDataLoaderKernel (line 106) | class GPTDataLoaderKernel final : public OpKernel { method GPTDataLoaderKernel (line 108) | GPTDataLoaderKernel() = default; method CreateOpKernelState (line 111) | std::shared_ptr CreateOpKernelState(KernelInitContext... method Compute (line 117) | void Compute(KernelComputeContext* ctx, OpKernelState* state, method AlwaysComputeWhenAllOutputsEmpty (line 132) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/greater_inplace_kernel.cpp type oneflow (line 19) | namespace oneflow { class GreaterInplaceKernel (line 22) | class GreaterInplaceKernel final : public user_op::OpKernel { method GreaterInplaceKernel (line 24) | GreaterInplaceKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 51) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ScalarGreaterInplaceKernel (line 80) | class ScalarGreaterInplaceKernel final : public user_op::OpKernel { method ScalarGreaterInplaceKernel (line 82) | ScalarGreaterInplaceKernel() = default; method Compute (line 86) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 106) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/greater_inplace_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type GreaterInplaceKernelUtil (line 21) | struct GreaterInplaceKernelUtil { method Forward (line 22) | static void Forward(ep::Stream* stream, const int64_t n, const T* x,... type ScalarGreaterInplaceKernelUtil (line 28) | struct ScalarGreaterInplaceKernelUtil { method Forward (line 29) | static void Forward(ep::Stream* stream, const int64_t n, const T* x,... FILE: oneflow/user/kernels/greater_inplace_kernel_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/grid_sample_kernel.cpp type oneflow (line 23) | namespace oneflow { class GridSampleKernel (line 26) | class GridSampleKernel final : public user_op::OpKernel { method GridSampleKernel (line 28) | GridSampleKernel() = default; method Compute (line 32) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 69) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class GridSampleGradKernel (line 86) | class GridSampleGradKernel final : public user_op::OpKernel { method GridSampleGradKernel (line 88) | GridSampleGradKernel() = default; method Compute (line 92) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 134) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/grid_sample_kernel_util.cpp type oneflow (line 19) | namespace oneflow { type GridSampleKernelUtil (line 22) | struct GridSampleKernelUtil f... method Forward4D (line 23) | static void Forward4D(user_op::KernelComputeContext* ctx, const user... method Forward5D (line 34) | static void Forward5D(user_op::KernelComputeContext* ctx, const user... method Backward4D (line 46) | static void Backward4D(user_op::KernelComputeContext* ctx, const use... method Backward5D (line 60) | static void Backward5D(user_op::KernelComputeContext* ctx, const use... FILE: oneflow/user/kernels/grid_sample_kernel_util.h function GridSamplerInterpolation (line 30) | enum class GridSamplerInterpolation { kBilinear = 0, kNearest, kBicubic }; FILE: oneflow/user/kernels/group_conv_kernel.cpp type oneflow (line 22) | namespace oneflow { function GetBlasTransposeType (line 26) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 30) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewChannelsFirstMatmulPrimitive (line 40) | std::unique_ptr NewChannelsFirstMatmulPrimitive... function ChannelsFirstMatmulPrimitiveExists (line 46) | auto ChannelsFirstMatmulPrimitiveExists() { function NewChannelsLastMatmulPrimitive (line 54) | std::unique_ptr NewChannelsLastMatmulPrimitive(... function ChannelsLastMatmulPrimitiveExists (line 60) | auto ChannelsLastMatmulPrimitiveExists() { function NewConvDataGradTransATransBMatmulPrimitive (line 68) | std::unique_ptr NewConvDataGradTransATransBMatm... function ConvDataGradTransATransBMatmulPrimitiveExists (line 74) | auto ConvDataGradTransATransBMatmulPrimitiveExists() { function NewConvDataGradTransANoTransBMatmulPrimitive (line 82) | std::unique_ptr NewConvDataGradTransANoTransBMa... function ConvDataGradTransANoTransBMatmulPrimitiveExists (line 88) | auto ConvDataGradTransANoTransBMatmulPrimitiveExists() { function NewConvWeightGradTransATransBMatmulPrimitive (line 96) | std::unique_ptr NewConvWeightGradTransATransBMa... function ConvWeightGradTransATransBMatmulPrimitiveExists (line 102) | auto ConvWeightGradTransATransBMatmulPrimitiveExists() { function NewConvWeightGradNoTransATransBMatmulPrimitive (line 110) | std::unique_ptr NewConvWeightGradNoTransATransB... function ConvWeightGradNoTransATransBMatmulPrimitiveExists (line 117) | auto ConvWeightGradNoTransATransBMatmulPrimitiveExists() { function T (line 138) | T* GetImgMutDptr(user_op::Tensor* tensor, int64_t idx) { function T (line 143) | const T* GetImgDptr(const user_op::Tensor* tensor, int64_t idx) { function CalcElemNumOfColBuf (line 147) | size_t CalcElemNumOfColBuf(const ShapeView& out_shape, const ShapeView... class ColBufWriter (line 158) | class ColBufWriter { method ColBufWriter (line 160) | ColBufWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... class Im2ColWriter (line 192) | class Im2ColWriter final : public ColBufWriter { method Im2ColWriter (line 194) | Im2ColWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... method DHWCWrite (line 199) | void DHWCWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method CDHWWrite (line 203) | void CDHWWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method InvalidDFunc (line 206) | void InvalidDFunc() override { method InvalidHFunc (line 209) | void InvalidHFunc() override { method InvalidWFunc (line 212) | void InvalidWFunc() override { method NextImCSize (line 215) | void NextImCSize() override { this->src_ptr_ += this->c_size_; } class Col2ImWriter (line 219) | class Col2ImWriter final : public ColBufWriter { method Col2ImWriter (line 221) | Col2ImWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... method DHWCWrite (line 226) | void DHWCWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method CDHWWrite (line 230) | void CDHWWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method InvalidDFunc (line 233) | void InvalidDFunc() override { this->src_ptr_ += this->od_size_; } method InvalidHFunc (line 234) | void InvalidHFunc() override { this->src_ptr_ += this->oh_size_; } method InvalidWFunc (line 235) | void InvalidWFunc() override { this->src_ptr_ += this->ow_size_; } method NextImCSize (line 236) | void NextImCSize() override { this->dst_ptr_ += this->c_size_; } class ColBufUtil (line 243) | class ColBufUtil final { method ColBufUtil (line 245) | ColBufUtil(const ShapeView& in_shape, const ShapeView& out_shape, in... method FOR_RANGE (line 266) | FOR_RANGE(int64_t, od, 0, od_num_) { type ConvKernelUtil (line 306) | struct ConvKernelUtil final { method NCDHWIm2Col (line 308) | static void NCDHWIm2Col(const T* in_dptr, const ShapeView& in_shape, method NDHWCIm2Col (line 320) | static void NDHWCIm2Col(const T* in_dptr, const ShapeView& in_shape, method NCDHWCol2Im (line 333) | static void NCDHWCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method NDHWCCol2Im (line 345) | static void NDHWCCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method DoNCDWHFunc (line 359) | static void DoNCDWHFunc(const ShapeView& weight_shape, ColBufUtil... method DoNDWHCFunc (line 372) | static void DoNDWHCFunc(const ShapeView& weight_shape, ColBufUtil... type ConvOpKernelCache (line 387) | struct ConvOpKernelCache final : public user_op::OpKernelCache { function CreateConvOpKernelCache (line 406) | std::shared_ptr> CreateConvOpKernelCache(user_op:... function InitBiasMulBuf (line 461) | void InitBiasMulBuf(T* dptr, int64_t num) { class ConvCpuKernel (line 465) | class ConvCpuKernel final : public user_op::OpKernel { method ConvCpuKernel (line 467) | ConvCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 470) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 472) | std::shared_ptr InitOpKernelCache( method Compute (line 478) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class ConvDataGradCpuKernel (line 592) | class ConvDataGradCpuKernel final : public user_op::OpKernel { method ConvDataGradCpuKernel (line 595) | ConvDataGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 598) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 600) | std::shared_ptr InitOpKernelCache( method Compute (line 606) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/group_deconv_kernel.cpp type oneflow (line 22) | namespace oneflow { function GetBlasTransposeType (line 26) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 30) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewDeconvTransATransBMatmulPrimitive (line 40) | std::unique_ptr NewDeconvTransATransBMatmulPrim... function NewDeconvTransANoTransBMatmulPrimitive (line 46) | std::unique_ptr NewDeconvTransANoTransBMatmulPr... function DeconvTransATransBMatmulPrimitiveExists (line 51) | auto DeconvTransATransBMatmulPrimitiveExists() { function DeconvTransANoTransBMatmulPrimitiveExists (line 58) | auto DeconvTransANoTransBMatmulPrimitiveExists() { function T (line 72) | T* GetImgMutDptr(user_op::Tensor* tensor, int64_t idx) { function T (line 77) | const T* GetImgDptr(const user_op::Tensor* tensor, int64_t idx) { function CalcElemNumOfColBuf (line 81) | size_t CalcElemNumOfColBuf(const ShapeView& out_shape, const ShapeView... class ColBufWriter (line 91) | class ColBufWriter { method ColBufWriter (line 93) | ColBufWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... class Col2ImWriter (line 125) | class Col2ImWriter final : public ColBufWriter { method Col2ImWriter (line 127) | Col2ImWriter(const T* src_ptr, T* dst_ptr, int64_t c_size, int64_t i... method DHWCWrite (line 132) | void DHWCWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method CDHWWrite (line 136) | void CDHWWrite(int64_t c, int64_t id, int64_t ih, int64_t iw) overri... method InvalidDFunc (line 139) | void InvalidDFunc() override { this->src_ptr_ += this->od_size_; } method InvalidHFunc (line 140) | void InvalidHFunc() override { this->src_ptr_ += this->oh_size_; } method InvalidWFunc (line 141) | void InvalidWFunc() override { this->src_ptr_ += this->ow_size_; } method NextImCSize (line 142) | void NextImCSize() override { this->dst_ptr_ += this->c_size_; } class ColBufUtil (line 149) | class ColBufUtil final { method ColBufUtil (line 151) | ColBufUtil(const ShapeView& in_shape, const ShapeView& out_shape, in... method FOR_RANGE (line 172) | FOR_RANGE(int64_t, od, 0, od_num_) { type DeconvKernelUtil (line 212) | struct DeconvKernelUtil final { method NCDHWCol2Im (line 214) | static void NCDHWCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method NDHWCCol2Im (line 226) | static void NDHWCCol2Im(const T* col_buf_ptr, const ShapeView& in_sh... method DoNCDWHFunc (line 240) | static void DoNCDWHFunc(const ShapeView& weight_shape, ColBufUtil... method DoNDWHCFunc (line 253) | static void DoNDWHCFunc(const ShapeView& weight_shape, ColBufUtil... type DeconvOpKernelCache (line 268) | struct DeconvOpKernelCache final : public user_op::OpKernelCache { method Update (line 286) | void Update(const ShapeView& x_shape, const ShapeView& out_shape) { function CreateDeconvOpKernelCache (line 303) | std::shared_ptr> CreateDeconvOpKernelCache(user... class DeconvCpuKernel (line 357) | class DeconvCpuKernel final : public user_op::OpKernel { method DeconvCpuKernel (line 360) | DeconvCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 363) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCacheWithFlags (line 365) | void InitOpKernelCacheWithFlags( method Compute (line 378) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/image_batch_align_kernel.cpp type oneflow (line 21) | namespace oneflow { function CopyFromTensorBuffer (line 26) | void CopyFromTensorBuffer(T* image_ptr, const TensorBuffer& image_buff... type ImageCopier (line 43) | struct ImageCopier final { class ImageBatchAlignKernel (line 53) | class ImageBatchAlignKernel final : public user_op::OpKernel { method ImageBatchAlignKernel (line 55) | ImageBatchAlignKernel() = default; method Compute (line 59) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 97) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/image_decode_kernel.cpp type oneflow (line 21) | namespace oneflow { function DecodeImage (line 25) | void DecodeImage(const TensorBuffer& raw_bytes, TensorBuffer* image_bu... class ImageDecodeKernel (line 63) | class ImageDecodeKernel final : public user_op::OpKernel { method ImageDecodeKernel (line 65) | ImageDecodeKernel() = default; method Compute (line 69) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 84) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/image_object_preprocess_kernels.cpp type oneflow (line 22) | namespace oneflow { type FlipCode (line 26) | enum class FlipCode : int8_t { function CvFlipCode (line 38) | int CvFlipCode(FlipCode flip_code) { function FlipImage (line 50) | void FlipImage(TensorBuffer* image_buffer, FlipCode flip_code) { function FlipBoxes (line 56) | void FlipBoxes(TensorBuffer* boxes_buffer, int32_t image_width, int32_... function ScaleBoxes (line 83) | void ScaleBoxes(TensorBuffer* boxes_buffer, T scale_w, T scale_h) { function FlipPolygons (line 101) | void FlipPolygons(TensorBuffer* polygons_buffer, int32_t image_width, ... function ScalePolygons (line 118) | void ScalePolygons(TensorBuffer* poly_buffer, T scale_w, T scale_h) { function ImageNormalizeByChannel (line 134) | void ImageNormalizeByChannel(TensorBuffer* image_buffer, const std::ve... function PolygonsToMask (line 155) | void PolygonsToMask(const TensorBuffer& polys, const TensorBuffer& pol... class ImageFlipKernel (line 218) | class ImageFlipKernel final : public user_op::OpKernel { method ImageFlipKernel (line 220) | ImageFlipKernel() = default; method Compute (line 224) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 240) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ObjectBboxFlipKernel (line 243) | class ObjectBboxFlipKernel final : public user_op::OpKernel { method ObjectBboxFlipKernel (line 245) | ObjectBboxFlipKernel() = default; method Compute (line 249) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 274) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ObjectBboxScaleKernel (line 277) | class ObjectBboxScaleKernel final : public user_op::OpKernel { method ObjectBboxScaleKernel (line 279) | ObjectBboxScaleKernel() = default; method Compute (line 283) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 304) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ObjectSegmentationPolygonFlipKernel (line 307) | class ObjectSegmentationPolygonFlipKernel final : public user_op::OpKe... method ObjectSegmentationPolygonFlipKernel (line 309) | ObjectSegmentationPolygonFlipKernel() = default; method Compute (line 313) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 338) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ObjectSegmentationPolygonScaleKernel (line 341) | class ObjectSegmentationPolygonScaleKernel final : public user_op::OpK... method ObjectSegmentationPolygonScaleKernel (line 343) | ObjectSegmentationPolygonScaleKernel() = default; method Compute (line 347) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 369) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ImageNormalize (line 372) | class ImageNormalize final : public user_op::OpKernel { method ImageNormalize (line 374) | ImageNormalize() = default; method Compute (line 378) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 395) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ObjectSegmentationPolygonToMask (line 398) | class ObjectSegmentationPolygonToMask final : public user_op::OpKernel { method ObjectSegmentationPolygonToMask (line 400) | ObjectSegmentationPolygonToMask() = default; method Compute (line 404) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 426) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function MakeInplaceProposalFn (line 431) | std::function(const user_op::InferContext&, user_op::AddIn... FILE: oneflow/user/kernels/image_preprocess_kernels.cpp type oneflow (line 27) | namespace oneflow { function CMN1Sample (line 60) | void CMN1Sample(int64_t C, int64_t in_H, int64_t in_W, int64_t out_H, ... function GetMirrorVec (line 80) | std::vector GetMirrorVec(user_op::KernelComputeContext* ctx) { class CMNAttr (line 95) | class CMNAttr final : public user_op::OpKernelState { method CMNAttr (line 97) | CMNAttr(user_op::KernelInitContext* ctx) { class CropMirrorNormalizeFromStaticShapeToFloatKernel (line 120) | class CropMirrorNormalizeFromStaticShapeToFloatKernel final : public u... method CropMirrorNormalizeFromStaticShapeToFloatKernel (line 122) | CropMirrorNormalizeFromStaticShapeToFloatKernel() = default; method CreateOpKernelState (line 125) | std::shared_ptr CreateOpKernelState( method Compute (line 131) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 193) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CropMirrorNormalizeFromTensorBufferToFloatKernel (line 202) | class CropMirrorNormalizeFromTensorBufferToFloatKernel final : public ... method CropMirrorNormalizeFromTensorBufferToFloatKernel (line 204) | CropMirrorNormalizeFromTensorBufferToFloatKernel() = default; method CreateOpKernelState (line 207) | std::shared_ptr CreateOpKernelState( method Compute (line 213) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 284) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class RandBoolGen (line 295) | class RandBoolGen final : public user_op::OpKernelState { method RandBoolGen (line 297) | explicit RandBoolGen(float prob, int64_t seed) : dis_(prob), rng_(se... method GetNextBool (line 300) | bool GetNextBool() { return dis_(rng_); } class CoinFlipKernel (line 309) | class CoinFlipKernel final : public user_op::OpKernel { method CoinFlipKernel (line 311) | CoinFlipKernel() = default; method CreateOpKernelState (line 314) | std::shared_ptr CreateOpKernelState( method Compute (line 323) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 332) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function ImageRandomCropImpl (line 342) | void ImageRandomCropImpl(const TensorBuffer* in_buffer, TensorBuffer* ... class ImageRandomCropKernel (line 372) | class ImageRandomCropKernel final : public user_op::OpKernel { method ImageRandomCropKernel (line 374) | ImageRandomCropKernel() = default; method CreateOpKernelState (line 377) | std::shared_ptr CreateOpKernelState( method Compute (line 383) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 398) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/image_resize_kernels.cpp type oneflow (line 22) | namespace oneflow { function GetTargetResizedSize4ImageBuffer (line 27) | std::pair GetTargetResizedSize4ImageBuffer(const TensorBuffer& i... function CheckMatSizeMatch (line 74) | bool CheckMatSizeMatch(const cv::Mat& mat, const bool resize_longer, c... function ImageTargetResize (line 97) | void ImageTargetResize(const TensorBuffer& image_buffer, TensorBuffer*... class ImageResizeToFixedSizeKernel (line 116) | class ImageResizeToFixedSizeKernel final : public user_op::OpKernel { method ImageResizeToFixedSizeKernel (line 118) | ImageResizeToFixedSizeKernel() = default; method Compute (line 122) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 176) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ImageResizeKeepAspectRatioKernel (line 179) | class ImageResizeKeepAspectRatioKernel final : public user_op::OpKernel { method ImageResizeKeepAspectRatioKernel (line 181) | ImageResizeKeepAspectRatioKernel() = default; method Compute (line 185) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 222) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/image_target_resize_kernel.cpp type oneflow (line 22) | namespace oneflow { function GetTargetResizedSize4ImageBuffer (line 27) | std::pair GetTargetResizedSize4ImageBuffer(const TensorBuffer& i... function ImageTargetResize (line 58) | void ImageTargetResize(const TensorBuffer& image_buffer, TensorBuffer*... class ImageTargetResizeKernel (line 82) | class ImageTargetResizeKernel final : public user_op::OpKernel { method ImageTargetResizeKernel (line 84) | ImageTargetResizeKernel() = default; method Compute (line 88) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 119) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/in_top_k_kernel.cpp type oneflow (line 19) | namespace oneflow { class InTopkKernel (line 22) | class InTopkKernel final : public user_op::OpKernel { method InTopkKernel (line 24) | InTopkKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 42) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/in_top_k_kernel_util.cpp type oneflow (line 19) | namespace oneflow { type InTopkKernelUtil (line 22) | struct InTopkKernelUtil { method InTopk (line 23) | static void InTopk(ep::Stream* stream, const int instance_num, const... FILE: oneflow/user/kernels/in_top_k_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/index_add_kernel.cpp type oneflow (line 19) | namespace oneflow { function index_add_cpu_kernel (line 23) | void index_add_cpu_kernel(const int64_t n, const T* input, const Index... class IndexAddCpuKernel (line 38) | class IndexAddCpuKernel final : public user_op::OpKernel { method IndexAddCpuKernel (line 40) | IndexAddCpuKernel() = default; method Compute (line 45) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 71) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/indexed_slices_reduce_sum_kernel.cpp type oneflow (line 19) | namespace oneflow { class IndexedSlicesReduceSumKernel (line 24) | class IndexedSlicesReduceSumKernel final : public user_op::OpKernel { method IndexedSlicesReduceSumKernel (line 26) | IndexedSlicesReduceSumKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 46) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function GenInferTmpSizeFn (line 50) | user_op::InferTmpSizeFn GenInferTmpSizeFn() { FILE: oneflow/user/kernels/indexed_slices_reduce_sum_kernel_util.cpp type oneflow (line 20) | namespace oneflow { function GetUniqueIdxSize (line 23) | int64_t GetUniqueIdxSize(int64_t n) { FILE: oneflow/user/kernels/indexed_slices_reduce_sum_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/inv_kernels.cpp type oneflow (line 19) | namespace oneflow { function BatchCount (line 23) | static inline size_t BatchCount(const user_op::Tensor* batched_matrice... function MatrixStride (line 31) | static inline size_t MatrixStride(const user_op::Tensor* batched_matri... class CpuInvKernel (line 40) | class CpuInvKernel final : public user_op::OpKernel { method CpuInvKernel (line 42) | CpuInvKernel() = default; method Compute (line 46) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 65) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/kl_div_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 21) | namespace user_op { function ComputeKLDivOut (line 27) | void ComputeKLDivOut(int64_t elem_cnt, const T* input, const T* targ... function ComputeKLDivGradOut (line 40) | void ComputeKLDivGradOut(int64_t elem_cnt, const T* input, const T* ... class KLDivKernel (line 50) | class KLDivKernel : public SimpleLossKernel (line 21) | struct L1L2RegularizeGradientKernelUtil { method RegularizeGradient (line 22) | static void RegularizeGradient(ep::Stream* stream, int64_t n, const ... FILE: oneflow/user/kernels/l1_l2_regularize_gradient_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/l2_normalize_kernel.cpp type oneflow (line 19) | namespace oneflow { function L2NormalizeForward (line 24) | static void L2NormalizeForward(const int32_t n, const int32_t c, const... function L2NormalizeBackward (line 41) | static void L2NormalizeBackward(const int32_t n, const int32_t c, cons... class CpuL2NormalizeKernel (line 69) | class CpuL2NormalizeKernel final : public user_op::OpKernel { method CpuL2NormalizeKernel (line 71) | CpuL2NormalizeKernel() = default; method Compute (line 75) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 90) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuL2NormalizeGradKernel (line 103) | class CpuL2NormalizeGradKernel final : public user_op::OpKernel { method CpuL2NormalizeGradKernel (line 105) | CpuL2NormalizeGradKernel() = default; method Compute (line 109) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 122) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/layer_norm_cpu_kernel.cpp type oneflow (line 18) | namespace oneflow { class LayerNormCpuKernel (line 21) | class LayerNormCpuKernel final : public user_op::OpKernel { method LayerNormCpuKernel (line 23) | LayerNormCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 27) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { TO... class LayerNormGradCpuKernel (line 41) | class LayerNormGradCpuKernel final : public user_op::OpKernel { method LayerNormGradCpuKernel (line 43) | LayerNormGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 47) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 48) | void Compute(user_op::KernelComputeContext* ctx) const override { TO... class FuseLayerNormGradCpuKernel (line 61) | class FuseLayerNormGradCpuKernel final : public user_op::OpKernel { method FuseLayerNormGradCpuKernel (line 63) | FuseLayerNormGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 67) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 68) | void Compute(user_op::KernelComputeContext* ctx) const override { TO... class LayerNormParamGradCpuKernel (line 81) | class LayerNormParamGradCpuKernel final : public user_op::OpKernel { method LayerNormParamGradCpuKernel (line 83) | LayerNormParamGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 87) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 88) | void Compute(user_op::KernelComputeContext* ctx) const override { TO... FILE: oneflow/user/kernels/lerp_kernel.cpp type oneflow (line 19) | namespace oneflow { class LerpKernel (line 22) | class LerpKernel final : public user_op::OpKernel { method LerpKernel (line 24) | LerpKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 48) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class LerpGradKernel (line 73) | class LerpGradKernel final : public user_op::OpKernel { method LerpGradKernel (line 75) | LerpGradKernel() = default; method Compute (line 79) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 106) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ScalarLerpKernel (line 126) | class ScalarLerpKernel final : public user_op::OpKernel { method ScalarLerpKernel (line 128) | ScalarLerpKernel() = default; method Compute (line 132) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 157) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ScalarLerpGradKernel (line 181) | class ScalarLerpGradKernel final : public user_op::OpKernel { method ScalarLerpGradKernel (line 183) | ScalarLerpGradKernel() = default; method Compute (line 187) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 217) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/lerp_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type LerpKernelUtil (line 21) | struct LerpKernelUtil { method Forward (line 22) | static void Forward(ep::Stream* stream, const int64_t n, const T* st... method Backward (line 27) | static void Backward(ep::Stream* stream, const int64_t n, const T* s... type ScalarLerpKernelUtil (line 40) | struct ScalarLerpKernelUtil { method Forward (line 41) | static void Forward(ep::Stream* stream, const int64_t n, const T* st... method Backward (line 47) | static void Backward(ep::Stream* stream, const int64_t n, const T* s... FILE: oneflow/user/kernels/lerp_kernel_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/linalg_cross_kernel.cpp type oneflow (line 20) | namespace oneflow { class CpuLinalgCrossKernel (line 23) | class CpuLinalgCrossKernel final : public user_op::OpKernel { method CpuLinalgCrossKernel (line 25) | CpuLinalgCrossKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 82) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/log_softmax_kernel.cpp type oneflow (line 21) | namespace oneflow { function NewLogSoftmaxPrimitive (line 26) | std::unique_ptr NewLogSoftmaxPrimitive(Cont... function LogSoftmaxPrimitiveExists (line 32) | auto LogSoftmaxPrimitiveExists() { function NewLogSoftmaxBackwardPrimitive (line 39) | std::unique_ptr NewLogSoftmaxBackwa... function LogSoftmaxBackwardPrimitiveExists (line 45) | auto LogSoftmaxBackwardPrimitiveExists() { class LogSoftmaxKernel (line 52) | class LogSoftmaxKernel final : public user_op::OpKernel, public user_o... method LogSoftmaxKernel (line 54) | LogSoftmaxKernel() = default; method Compute (line 60) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 70) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class LogSoftmaxGradKernel (line 73) | class LogSoftmaxGradKernel final : public user_op::OpKernel, public us... method LogSoftmaxGradKernel (line 75) | LogSoftmaxGradKernel() = default; method Compute (line 81) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 95) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/logical_not_kernel.cpp type oneflow (line 21) | namespace oneflow { function NewLogicalNotPrimitive (line 26) | std::unique_ptr NewLogicalNotPrimitiv... class LogicalNotKernel (line 33) | class LogicalNotKernel final : public user_op::OpKernel, public user_o... method LogicalNotKernel (line 35) | LogicalNotKernel() = default; method Compute (line 39) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 51) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function LogicalNotPrimitiveExists (line 54) | auto LogicalNotPrimitiveExists() { FILE: oneflow/user/kernels/loss_kernel_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/masked_fill_kernel.cpp type oneflow (line 20) | namespace oneflow { class MaskedFillKernel (line 23) | class MaskedFillKernel final : public user_op::OpKernel { method MaskedFillKernel (line 25) | MaskedFillKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 47) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/math_binary_broadcast_kernels.cpp type oneflow (line 24) | namespace oneflow { function NewBroadcastElementwiseBinaryPrimitive (line 27) | std::enable_if_t NewMemcpyPrimitive(Context* ctx) { function NewMatmulPrimitive (line 75) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewMatmulPrimitive (line 85) | std::unique_ptr NewMatmulPrimitive(Context* ctx) { function NewBatchMatmulPrimitive (line 92) | std::unique_ptr NewBatchMatmulPrimitive(Co... function NewBroadcastMatmulPrimitive (line 101) | std::unique_ptr NewBroadcastMatmulPrim... function MemcpyPrimitiveExists (line 112) | auto MemcpyPrimitiveExists() { function MatmulPrimitiveExists (line 118) | auto MatmulPrimitiveExists() { function BatchMatmulPrimitiveExists (line 124) | auto BatchMatmulPrimitiveExists() { function BroadcastMatmulPrimitiveExists (line 130) | auto BroadcastMatmulPrimitiveExists() { class MatmulKernel (line 137) | class MatmulKernel final : public user_op::OpKernel, public user_op::C... method MatmulKernel (line 139) | MatmulKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 142) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 145) | void Compute(user_op::KernelComputeContext* ctx) const override { class BatchMatmulKernel (line 196) | class BatchMatmulKernel final : public user_op::OpKernel, public user_... method BatchMatmulKernel (line 198) | BatchMatmulKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 201) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 204) | void Compute(user_op::KernelComputeContext* ctx) const override { class BroadcastMatmulKernel (line 260) | class BroadcastMatmulKernel final : public user_op::OpKernel, public u... method BroadcastMatmulKernel (line 262) | BroadcastMatmulKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 265) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 268) | void Compute(user_op::KernelComputeContext* ctx) const override { function NewMatmulPrimitiveForBroadcastMatmulGradB (line 310) | std::unique_ptr NewMatmulPrimitiveForBroadcastM... class BroadcastMatmulGradBKernel (line 315) | class BroadcastMatmulGradBKernel final : public user_op::OpKernel, method BroadcastMatmulGradBKernel (line 318) | BroadcastMatmulGradBKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 321) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 324) | void Compute(user_op::KernelComputeContext* ctx) const override { function PrimitiveExistsForBroadcastMatmulGradB (line 352) | auto PrimitiveExistsForBroadcastMatmulGradB() { FILE: oneflow/user/kernels/matrix_vector_product_kernel.cpp type oneflow (line 23) | namespace oneflow { function GetBlasTransposeType (line 27) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMemcpyPrimitive (line 32) | std::unique_ptr NewMemcpyPrimitive(Context* ctx) { function NewMatmulPrimitive (line 37) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewMatrixVectorProductPrimitive (line 47) | std::unique_ptr NewMatrixVectorProductPrimitive... function NewMatrixVectorProductGradAPrimitive (line 53) | std::unique_ptr NewMatrixVectorProductGradAPrim... function NewMatrixVectorProductGradBPrimitive (line 59) | std::unique_ptr NewMatrixVectorProductGradBPrim... function MatrixVectorProductPrimitiveExists (line 64) | auto MatrixVectorProductPrimitiveExists() { function MatrixVectorProductGradAPrimitiveExists (line 71) | auto MatrixVectorProductGradAPrimitiveExists() { function MatrixVectorProductGradBPrimitiveExists (line 78) | auto MatrixVectorProductGradBPrimitiveExists() { class MatrixVectorProductKernel (line 85) | class MatrixVectorProductKernel final : public user_op::OpKernel, publ... method MatrixVectorProductKernel (line 87) | MatrixVectorProductKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 90) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 93) | void Compute(user_op::KernelComputeContext* ctx) const override { class MatrixVectorProductGradAKernel (line 121) | class MatrixVectorProductGradAKernel final : public user_op::OpKernel, method MatrixVectorProductGradAKernel (line 124) | MatrixVectorProductGradAKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 127) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 130) | void Compute(user_op::KernelComputeContext* ctx) const override { class MatrixVectorProductGradBKernel (line 153) | class MatrixVectorProductGradBKernel final : public user_op::OpKernel, method MatrixVectorProductGradBKernel (line 156) | MatrixVectorProductGradBKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 159) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 162) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/max_pool_kernel.cpp type oneflow (line 18) | namespace oneflow { type PoolOpKernelCache (line 20) | struct PoolOpKernelCache final : public user_op::OpKernelCache { method PoolOpKernelCache (line 22) | explicit PoolOpKernelCache(const MaxPoolParams3D& params_3d) : param... method MaxPoolParams3D (line 23) | const MaxPoolParams3D& GetParams3D() const { return params_3d; } function CreatePoolOpKernelCache (line 26) | std::shared_ptr CreatePoolOpKernelCache(user_op::Ke... function Maxpool2dForwardComputeCLast (line 46) | void Maxpool2dForwardComputeCLast(const NdIndexOffsetHelper& i... type PoolKernelUtil (line 99) | struct PoolKernelUtil { method Maxpool1dForward (line 100) | static void Maxpool1dForward(ep::Stream* stream, const NdIndexOffset... method Maxpool1dBackward (line 109) | static void Maxpool1dBackward(ep::Stream* stream, const NdIndexOffse... method Maxpool2dForwardCFirst (line 117) | static void Maxpool2dForwardCFirst(ep::Stream* stream, method Maxpool2dBackwardCFirst (line 129) | static void Maxpool2dBackwardCFirst(ep::Stream* stream, method Maxpool2dForwardCLast (line 139) | static void Maxpool2dForwardCLast(ep::Stream* stream, method Maxpool2dBackwardCLast (line 152) | static void Maxpool2dBackwardCLast(ep::Stream* stream, method Maxpool3dForward (line 162) | static void Maxpool3dForward(ep::Stream* stream, const NdIndexOffset... method Maxpool3dBackward (line 175) | static void Maxpool3dBackward(ep::Stream* stream, const NdIndexOffse... class MaxPool1dKernel (line 187) | class MaxPool1dKernel final : public user_op::OpKernel { method MaxPool1dKernel (line 189) | MaxPool1dKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 192) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 193) | std::shared_ptr InitOpKernelCache( method Compute (line 199) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool1dGradKernel (line 229) | class MaxPool1dGradKernel final : public user_op::OpKernel { method MaxPool1dGradKernel (line 231) | MaxPool1dGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 234) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 235) | std::shared_ptr InitOpKernelCache( method Compute (line 241) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool2dKernel (line 273) | class MaxPool2dKernel final : public user_op::OpKernel { method MaxPool2dKernel (line 275) | MaxPool2dKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 278) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 279) | std::shared_ptr InitOpKernelCache( method Compute (line 285) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool2dGradKernel (line 334) | class MaxPool2dGradKernel final : public user_op::OpKernel { method MaxPool2dGradKernel (line 336) | MaxPool2dGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 339) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 340) | std::shared_ptr InitOpKernelCache( method Compute (line 346) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool3dKernel (line 398) | class MaxPool3dKernel final : public user_op::OpKernel { method MaxPool3dKernel (line 400) | MaxPool3dKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 403) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 404) | std::shared_ptr InitOpKernelCache( method Compute (line 410) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool3dGradKernel (line 443) | class MaxPool3dGradKernel final : public user_op::OpKernel { method MaxPool3dGradKernel (line 445) | MaxPool3dGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 448) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 449) | std::shared_ptr InitOpKernelCache( method Compute (line 455) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/max_pool_kernel_util.cpp type oneflow (line 19) | namespace oneflow { function GetWindowedOutputShape (line 21) | void GetWindowedOutputShape(int64_t input_size, int32_t filter_size, i... function Get3DOutputShape (line 36) | void Get3DOutputShape(const DimVector& in, const std::vector&... function Shape (line 82) | Shape MaxPoolParams3D::GetYShape() const { function Shape (line 104) | Shape MaxPoolParams3D::GetXShape5D() const { function Shape (line 108) | Shape MaxPoolParams3D::GetYShape5D() const { FILE: oneflow/user/kernels/max_pool_kernel_util.h function namespace (line 29) | namespace oneflow { function Maxpool1dBackwardCompute (line 171) | void Maxpool1dBackwardCompute(const NdIndexOffsetHelper index_he... function Maxpool2dForwardComputeCFirst (line 192) | void Maxpool2dForwardComputeCFirst( function Maxpool2dBackwardComputeCFirst (line 242) | void Maxpool2dBackwardComputeCFirst( function Maxpool2dBackwardComputeCLast (line 264) | void Maxpool2dBackwardComputeCLast(const NdIndexOffsetHelper ind... function Maxpool3dForwardCompute (line 285) | void Maxpool3dForwardCompute( function Maxpool3dBackwardCompute (line 335) | void Maxpool3dBackwardCompute(const NdIndexOffsetHelper index_he... FILE: oneflow/user/kernels/max_unpool_kernel.cpp type oneflow (line 21) | namespace oneflow { function MaxUnpoolNdForwardOrBackward (line 25) | void MaxUnpoolNdForwardOrBackward(const NdIndexOffsetHelper& i... type UnpoolKernelUtil (line 41) | struct UnpoolKernelUtil { method MaxUnpoolNdForward (line 42) | static void MaxUnpoolNdForward(ep::Stream* stream, method MaxUnpoolNdBackward (line 51) | static void MaxUnpoolNdBackward(ep::Stream* stream, class MaxUnpoolNdKernel (line 62) | class MaxUnpoolNdKernel final : public user_op::OpKernel { method MaxUnpoolNdKernel (line 64) | MaxUnpoolNdKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 67) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 70) | void Compute(user_op::KernelComputeContext* ctx) const override { class MaxUnpoolNdGradKernel (line 111) | class MaxUnpoolNdGradKernel final : public user_op::OpKernel { method MaxUnpoolNdGradKernel (line 113) | MaxUnpoolNdGradKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 116) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 119) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/max_unpool_kernel_util.cpp type oneflow (line 20) | namespace oneflow { function GetWindowedOutputShape (line 23) | void GetWindowedOutputShape(int64_t input_size, int32_t filter_size, i... function Get3DOutputShape (line 29) | void Get3DOutputShape(const DimVector& in, const std::vector&... function Shape (line 67) | Shape MaxUnpoolParams3D::GetYShape() const { function Shape (line 83) | Shape MaxUnpoolParams3D::GetXShape5D() const { function Shape (line 87) | Shape MaxUnpoolParams3D::GetYShape5D() const { FILE: oneflow/user/kernels/max_unpool_kernel_util.h type small_vector (line 41) | typedef small_vector FixedDimVector; type UnpoolKernelUtil (line 75) | struct UnpoolKernelUtil { FILE: oneflow/user/kernels/median_kernel.cpp type oneflow (line 19) | namespace oneflow { class CpuMedianKernel (line 22) | class CpuMedianKernel final : public user_op::OpKernel { method CpuMedianKernel (line 24) | CpuMedianKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 42) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/median_with_indices_kernel.cpp type oneflow (line 20) | namespace oneflow { class CpuMedianWithIndicesKernel (line 23) | class CpuMedianWithIndicesKernel final : public user_op::OpKernel { method CpuMedianWithIndicesKernel (line 25) | CpuMedianWithIndicesKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 69) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/min_max_observer_kernel.cpp type oneflow (line 20) | namespace oneflow { function GenQuantScaleSymmetric (line 23) | void GenQuantScaleSymmetric(const T* in_ptr, const int32_t quantizatio... function GenQuantScaleAffine (line 37) | void GenQuantScaleAffine(const T* in_ptr, const int32_t quantization_bit, function GenQuantScaleCambricon (line 49) | void GenQuantScaleCambricon(const T* in_ptr, const int32_t quantizatio... class CpuMinMaxObserverKernel (line 61) | class CpuMinMaxObserverKernel final : public user_op::OpKernel { method CpuMinMaxObserverKernel (line 63) | CpuMinMaxObserverKernel() = default; method Compute (line 67) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 116) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/mode_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewMemcpyPrimitive (line 23) | std::unique_ptr NewMemcpyPrimitive(Context* ctx) { class CpuModeKernel (line 28) | class CpuModeKernel final : public user_op::OpKernel { method CpuModeKernel (line 30) | CpuModeKernel() = default; method Compute (line 34) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 88) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/model_update_kernel_util.cpp type oneflow (line 19) | namespace oneflow { function T (line 25) | T Fastpow(T a, int64_t b) { function SumSquares2 (line 36) | void SumSquares2(int64_t n, const T* src0, T* dst0, const T* src1, T* ... type SGDUpdateKernelUtil (line 44) | struct SGDUpdateKernelUtil { type SGDUpdateKernelUtil (line 71) | struct SGDUpdateKernelUtil type SGDUpdateKernelUtil (line 72) | struct SGDUpdateKernelUtil type IndexedSlicesSGDUpdateKernelUtil (line 75) | struct IndexedSlicesSGDUpdateKernelUtil { function FOR_RANGE (line 90) | FOR_RANGE(int64_t, i, 0, n) { type MomentumUpdateKernelUtil (line 112) | struct MomentumUpdateKernelUtil { type MomentumUpdateKernelUtil (line 136) | struct MomentumUpdateKernelUtil type MomentumUpdateKernelUtil (line 137) | struct MomentumUpdateKernelUtil type IndexedSlicesMomentumMdUpdateKernelUtil (line 140) | struct IndexedSlicesMomentumMdUpdateKernelUtil (line 179) | struct AdamUpdateKernelUtil { function FOR_RANGE (line 204) | FOR_RANGE(int64_t, i, 0, n) { type AdamUpdateKernelUtil (line 218) | struct AdamUpdateKernelUtil type AdamUpdateKernelUtil (line 219) | struct AdamUpdateKernelUtil type IndexedSlicesAdamMdUpdateKernelUtil (line 222) | struct IndexedSlicesAdamMdUpdateKernelUtil { method Update (line 223) | static void Update(ep::Stream* stream, float beta1, float beta2, flo... type AdagradUpdateKernelUtil (line 264) | struct AdagradUpdateKernelUtil { function FOR_RANGE (line 286) | FOR_RANGE(int64_t, i, 0, n) { type AdagradUpdateKernelUtil (line 292) | struct AdagradUpdateKernelUtil type AdagradUpdateKernelUtil (line 293) | struct AdagradUpdateKernelUtil type LambUpdateKernelUtil (line 296) | struct LambUpdateKernelUtil { function FOR_RANGE (line 320) | FOR_RANGE(int64_t, i, 0, n) { function FOR_RANGE (line 331) | FOR_RANGE(int64_t, i, 0, n) { type LambUpdateKernelUtil (line 336) | struct LambUpdateKernelUtil type LambUpdateKernelUtil (line 337) | struct LambUpdateKernelUtil type BiasCorrectionFactorKernelUtil (line 340) | struct BiasCorrectionFactorKernelUtil { type RmsPropUpdateKernelUtil (line 352) | struct RmsPropUpdateKernelUtil { function FOR_RANGE (line 371) | FOR_RANGE(int64_t, i, 0, n) { function FOR_RANGE (line 377) | FOR_RANGE(int64_t, i, 0, n) { type RmsPropUpdateKernelUtil (line 385) | struct RmsPropUpdateKernelUtil type RmsPropUpdateKernelUtil (line 386) | struct RmsPropUpdateKernelUtil type LarsUpdateKernelUtil (line 389) | struct LarsUpdateKernelUtil { function FOR_RANGE (line 407) | FOR_RANGE(int64_t, i, 0, n) { function FOR_RANGE (line 422) | FOR_RANGE(int64_t, i, 0, n) { type LarsUpdateKernelUtil (line 428) | struct LarsUpdateKernelUtil type LarsUpdateKernelUtil (line 429) | struct LarsUpdateKernelUtil type FtrlUpdateKernelUtil (line 432) | struct FtrlUpdateKernelUtil { type FtrlUpdateKernelUtil (line 456) | struct FtrlUpdateKernelUtil type FtrlUpdateKernelUtil (line 457) | struct FtrlUpdateKernelUtil type AdadeltaUpdateKernelUtil (line 460) | struct AdadeltaUpdateKernelUtil { type AdadeltaUpdateKernelUtil (line 484) | struct AdadeltaUpdateKernelUtil type AdadeltaUpdateKernelUtil (line 485) | struct AdadeltaUpdateKernelUtil FILE: oneflow/user/kernels/model_update_kernel_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/model_update_kernels.cpp type oneflow (line 22) | namespace oneflow { class TmpBufferManager (line 27) | class TmpBufferManager final { method TmpBufferManager (line 30) | TmpBufferManager(void* ptr, const int64_t num_indices, const int64_t... method UniqueWorkspaceBytes (line 50) | int64_t UniqueWorkspaceBytes() const { return unique_workspace_bytes... method GetTotalBufferSize (line 51) | size_t GetTotalBufferSize() const { return total_buffer_size_; } method K (line 52) | K* UniqueDiffIndicesPtr() const { method T (line 56) | T* UniqueDiffValuesPtr() const { class IndexedSlicesUpdateOpKernelCache (line 81) | class IndexedSlicesUpdateOpKernelCache final : public user_op::OpKerne... method IndexedSlicesUpdateOpKernelCache (line 83) | IndexedSlicesUpdateOpKernelCache(int64_t lower, int64_t upper) : low... method lower (line 86) | int64_t lower() const { return lower_; } method upper (line 87) | int64_t upper() const { return upper_; } function CreateIndexedSlicesUpdateOpKernelCache (line 94) | std::shared_ptr CreateIndexedSlicesUpdateOpKer... class SGDUpdateKernel (line 114) | class SGDUpdateKernel final : public user_op::OpKernel, public user_op... method SGDUpdateKernel (line 116) | SGDUpdateKernel() = default; method Compute (line 120) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 157) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } function GenInferTmpSizeFn (line 176) | user_op::InferTmpSizeFn GenInferTmpSizeFn() { class IndexedSlicesSGDUpdateKernel (line 188) | class IndexedSlicesSGDUpdateKernel final : public user_op::OpKernel { method IndexedSlicesSGDUpdateKernel (line 190) | IndexedSlicesSGDUpdateKernel() = default; method InitOpKernelCache (line 193) | std::shared_ptr InitOpKernelCache( method Compute (line 201) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 237) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MomentumUpdateKernel (line 258) | class MomentumUpdateKernel final : public user_op::OpKernel, public us... method MomentumUpdateKernel (line 260) | MomentumUpdateKernel() = default; method Compute (line 264) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 303) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class IndexedSlicesMomentumUpdateKernel (line 322) | class IndexedSlicesMomentumUpdateKernel final : public user_op::OpKern... method IndexedSlicesMomentumUpdateKernel (line 324) | IndexedSlicesMomentumUpdateKernel() = default; method InitOpKernelCache (line 327) | std::shared_ptr InitOpKernelCache( method Compute (line 335) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 379) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class AdamUpdateKernel (line 400) | class AdamUpdateKernel final : public user_op::OpKernel, public user_o... method AdamUpdateKernel (line 402) | AdamUpdateKernel() = default; method Compute (line 406) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 483) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class AdagradUpdateKernel (line 502) | class AdagradUpdateKernel final : public user_op::OpKernel, public use... method AdagradUpdateKernel (line 504) | AdagradUpdateKernel() = default; method Compute (line 508) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 552) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class IndexedSlicesAdamUpdateKernel (line 570) | class IndexedSlicesAdamUpdateKernel final : public user_op::OpKernel { method IndexedSlicesAdamUpdateKernel (line 572) | IndexedSlicesAdamUpdateKernel() = default; method InitOpKernelCache (line 574) | std::shared_ptr InitOpKernelCache( method Compute (line 582) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 659) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class LambTmpBufferManager (line 680) | class LambTmpBufferManager final { method LambTmpBufferManager (line 683) | LambTmpBufferManager(void* ptr, const int64_t n) : ptr_(ptr) { method GetNormBufferSize (line 692) | size_t GetNormBufferSize() const { return norm_buffer_bytes_; } method GetTotalBufferSize (line 693) | size_t GetTotalBufferSize() const { return total_buffer_size_; } method T (line 695) | T* AdamDiffPtr() const { method T (line 699) | T* NormBufferPtr() const { class LambUpdateKernel (line 714) | class LambUpdateKernel final : public user_op::OpKernel, public user_o... method LambUpdateKernel (line 716) | LambUpdateKernel() = default; method Compute (line 720) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 784) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } function LambGenInferTmpSizeFn (line 788) | user_op::InferTmpSizeFn LambGenInferTmpSizeFn() { class BiasCorrectionFactorKernel (line 813) | class BiasCorrectionFactorKernel final : public user_op::OpKernel, method BiasCorrectionFactorKernel (line 816) | BiasCorrectionFactorKernel() = default; method Compute (line 820) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 827) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class RmsPropUpdateKernel (line 840) | class RmsPropUpdateKernel final : public user_op::OpKernel, public use... method RmsPropUpdateKernel (line 842) | RmsPropUpdateKernel() = default; method Compute (line 846) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 888) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class LarsTmpBufferManager (line 907) | class LarsTmpBufferManager final { method LarsTmpBufferManager (line 910) | LarsTmpBufferManager(void* ptr, const int64_t n) : ptr_(ptr) { method GetTotalBufferSize (line 919) | size_t GetTotalBufferSize() const { return total_buffer_size_; } method GetDataTmpBufferSize (line 920) | size_t GetDataTmpBufferSize() const { return data_tmp_size_; } method T (line 922) | T* ModelDiffPtr() const { method T (line 927) | T* DataTmpPtr() const { class LarsUpdateKernel (line 942) | class LarsUpdateKernel final : public user_op::OpKernel, public user_o... method LarsUpdateKernel (line 944) | LarsUpdateKernel() = default; method Compute (line 948) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 983) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } function LarsGenInferTmpSizeFn (line 987) | user_op::InferTmpSizeFn LarsGenInferTmpSizeFn() { class FtrlUpdateKernel (line 1012) | class FtrlUpdateKernel final : public user_op::OpKernel, public user_o... method FtrlUpdateKernel (line 1014) | FtrlUpdateKernel() = default; method Compute (line 1018) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 1063) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class AdadeltaUpdateKernel (line 1082) | class AdadeltaUpdateKernel final : public user_op::OpKernel, public us... method AdadeltaUpdateKernel (line 1084) | AdadeltaUpdateKernel() = default; method Compute (line 1088) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 1127) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } FILE: oneflow/user/kernels/moving_average_min_max_observer_kernel.cpp type oneflow (line 20) | namespace oneflow { function GenQuantScalePerLayerSymmetric (line 23) | void GenQuantScalePerLayerSymmetric(const T* in, const int64_t current... function GenQuantScalePerLayerAffine (line 52) | void GenQuantScalePerLayerAffine(const T* in, const int64_t current_tr... function GenQuantScalePerLayerCambricon (line 82) | void GenQuantScalePerLayerCambricon(const T* in, const int64_t current... class CpuMovingAverageMinMaxObserverKernel (line 110) | class CpuMovingAverageMinMaxObserverKernel final : public user_op::OpK... method CpuMovingAverageMinMaxObserverKernel (line 112) | CpuMovingAverageMinMaxObserverKernel() = default; method Compute (line 116) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 160) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/multi_reduce_kernel_util.h function namespace (line 24) | namespace oneflow { function OF_DEVICE_FUNC (line 53) | OF_DEVICE_FUNC T operator()(const T& x, const T& y) const { return x + y; } function OF_DEVICE_FUNC (line 58) | OF_DEVICE_FUNC T operator()(const T& x, const T& y) const { return x > y... function OF_DEVICE_FUNC (line 63) | OF_DEVICE_FUNC T operator()(const T& x, const T& y) const { return x < y... function OF_DEVICE_FUNC (line 68) | OF_DEVICE_FUNC T operator()(const T& x) const { return x < GetZeroVal... function OF_DEVICE_FUNC (line 73) | OF_DEVICE_FUNC T operator()(const T& x) const { function OF_DEVICE_FUNC (line 80) | OF_DEVICE_FUNC T operator()(const T& x) const { return x * x; } function explicit (line 85) | explicit AbsPow(const T& base) : base_(base) {} function OF_DEVICE_FUNC (line 87) | OF_DEVICE_FUNC T operator()(const T& x) { FILE: oneflow/user/kernels/multi_reduce_kernels.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/user/kernels/multi_reduce_kernels.h function Compute (line 35) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelState*, function Ximum (line 71) | enum class Ximum { FILE: oneflow/user/kernels/multi_tensor_model_update_kernel.cpp type oneflow (line 20) | namespace oneflow { class MultiTensorSGDUpdateKernel (line 25) | class MultiTensorSGDUpdateKernel final : public user_op::OpKernel, method MultiTensorSGDUpdateKernel (line 28) | MultiTensorSGDUpdateKernel() = default; method Compute (line 33) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 86) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MultiTensorMomentumUpdateKernel (line 103) | class MultiTensorMomentumUpdateKernel final : public user_op::OpKernel, method MultiTensorMomentumUpdateKernel (line 106) | MultiTensorMomentumUpdateKernel() = default; method Compute (line 111) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 170) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MultiTensorAdamUpdateKernel (line 188) | class MultiTensorAdamUpdateKernel final : public user_op::OpKernel, method MultiTensorAdamUpdateKernel (line 191) | MultiTensorAdamUpdateKernel() = default; method Compute (line 196) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 281) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MultiTensorSGDUpdateWithCastKernel (line 298) | class MultiTensorSGDUpdateWithCastKernel final : public user_op::OpKer... method MultiTensorSGDUpdateWithCastKernel (line 301) | MultiTensorSGDUpdateWithCastKernel() = default; method Compute (line 306) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 361) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MultiTensorMomentumUpdateWithCastKernel (line 378) | class MultiTensorMomentumUpdateWithCastKernel final : public user_op::... method MultiTensorMomentumUpdateWithCastKernel (line 381) | MultiTensorMomentumUpdateWithCastKernel() = default; method Compute (line 386) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 447) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MultiTensorAdamUpdateWithCastKernel (line 465) | class MultiTensorAdamUpdateWithCastKernel final : public user_op::OpKe... method MultiTensorAdamUpdateWithCastKernel (line 468) | MultiTensorAdamUpdateWithCastKernel() = default; method Compute (line 473) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 560) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class MultiTensorYoloV5WeightUpdateKernel (line 577) | class MultiTensorYoloV5WeightUpdateKernel final : public user_op::OpKe... method MultiTensorYoloV5WeightUpdateKernel (line 580) | MultiTensorYoloV5WeightUpdateKernel() = default; method Compute (line 585) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 611) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } FILE: oneflow/user/kernels/multi_tensor_model_update_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/mutable_cast_once_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { function NewCastPrimitive (line 28) | std::unique_ptr NewCastPrimitive(Context* ctx) { class MutableCastOnceOpKernelState (line 35) | class MutableCastOnceOpKernelState final : public OpKernelState { method MutableCastOnceOpKernelState (line 37) | MutableCastOnceOpKernelState() : cast_once_flag_(false) {} method SetDone (line 39) | void SetDone() { method IsDone (line 43) | bool IsDone() { return cast_once_flag_; } class MutableCastOnce (line 49) | class MutableCastOnce final : public OpKernel { method MutableCastOnce (line 51) | MutableCastOnce() = default; method CreateOpKernelState (line 54) | std::shared_ptr CreateOpKernelState(KernelInitConte... method Compute (line 59) | void Compute(KernelComputeContext* ctx, user_op::OpKernelState* st... method AlwaysComputeWhenAllOutputsEmpty (line 73) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function CastPrimitiveExists (line 76) | auto CastPrimitiveExists() { FILE: oneflow/user/kernels/narrow_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 22) | namespace user_op { function NewCopyNdPrimitive (line 27) | std::unique_ptr NewCopyNdPrimitive(Context* c... function NewMemsetPrimitive (line 32) | std::unique_ptr NewMemsetPrimitive(Context* c... function CopyNdPrimitiveExists (line 36) | auto CopyNdPrimitiveExists() { function MemsetPrimitiveExists (line 42) | auto MemsetPrimitiveExists() { class NarrowKernel (line 50) | class NarrowKernel final : public user_op::OpKernel { method NarrowKernel (line 52) | NarrowKernel() = default; method Compute (line 56) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 81) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class NarrowGradKernel (line 84) | class NarrowGradKernel final : public user_op::OpKernel { method NarrowGradKernel (line 86) | NarrowGradKernel() = default; method Compute (line 90) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 123) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/nccl_logical_2d_sbp_kernels.cpp type oneflow (line 30) | namespace oneflow { function AllReduceCollectiveCommunicationExists (line 34) | auto AllReduceCollectiveCommunicationExists() { function AllGatherCollectiveCommunicationExists (line 43) | auto AllGatherCollectiveCommunicationExists() { function AllToAllCollectiveCommunicationExists (line 52) | auto AllToAllCollectiveCommunicationExists() { class CclLogical2DSameDim0KernelCommState (line 61) | class CclLogical2DSameDim0KernelCommState : public user_op::OpKernelSt... method CclLogical2DSameDim0KernelCommState (line 63) | explicit CclLogical2DSameDim0KernelCommState(user_op::KernelInitCont... method num_ranks (line 77) | int64_t num_ranks() { method Init (line 85) | void Init() { class CclLogical2DSameDim0AllGatherNoncontinuousKernelState (line 105) | class CclLogical2DSameDim0AllGatherNoncontinuousKernelState method CclLogical2DSameDim0AllGatherNoncontinuousKernelState (line 108) | explicit CclLogical2DSameDim0AllGatherNoncontinuousKernelState(user_... method src_split_axis (line 112) | int64_t src_split_axis() const { return src_split_axis_; } method set_src_split_axis (line 113) | void set_src_split_axis(int64_t split_axis) { src_split_axis_ = spli... class CclLogical2DSameDim0All2AllKernelState (line 119) | class CclLogical2DSameDim0All2AllKernelState : public CclLogical2DSame... method CclLogical2DSameDim0All2AllKernelState (line 121) | explicit CclLogical2DSameDim0All2AllKernelState(user_op::KernelInitC... method src_split_axis (line 125) | int64_t src_split_axis() const { return src_split_axis_; } method set_src_split_axis (line 126) | void set_src_split_axis(int64_t split_axis) { src_split_axis_ = spli... method dst_split_axis (line 127) | int64_t dst_split_axis() const { return dst_split_axis_; } method set_dst_split_axis (line 128) | void set_dst_split_axis(int64_t split_axis) { dst_split_axis_ = spli... class CclLogical2DSameDim0AllReduce (line 135) | class CclLogical2DSameDim0AllReduce final : public user_op::OpKernel { method CclLogical2DSameDim0AllReduce (line 137) | CclLogical2DSameDim0AllReduce() = default; method CreateOpKernelState (line 140) | std::shared_ptr CreateOpKernelState( method Compute (line 146) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 165) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 166) | bool IsKernelLaunchSynchronized() const override { class CclLogical2DSameDim0AllGather (line 172) | class CclLogical2DSameDim0AllGather final : public user_op::OpKernel { method CclLogical2DSameDim0AllGather (line 174) | CclLogical2DSameDim0AllGather() = default; method CreateOpKernelState (line 177) | std::shared_ptr CreateOpKernelState( method Compute (line 183) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 202) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 203) | bool IsKernelLaunchSynchronized() const override { class CclLogical2DSameDim0AllGatherNoncontinuous (line 210) | class CclLogical2DSameDim0AllGatherNoncontinuous final : public user_o... method CclLogical2DSameDim0AllGatherNoncontinuous (line 212) | CclLogical2DSameDim0AllGatherNoncontinuous() = default; method CreateOpKernelState (line 215) | std::shared_ptr CreateOpKernelState( method Compute (line 227) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 277) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 278) | bool IsKernelLaunchSynchronized() const override { function Infer2DSameDim0AllGatherNoncontinuousKernelTmpBufferSize (line 284) | size_t Infer2DSameDim0AllGatherNoncontinuousKernelTmpBufferSize(user_o... class CclLogical2DSameDim0All2All (line 291) | class CclLogical2DSameDim0All2All final : public user_op::OpKernel { method CclLogical2DSameDim0All2All (line 293) | CclLogical2DSameDim0All2All() = default; method CreateOpKernelState (line 296) | std::shared_ptr CreateOpKernelState( method Compute (line 313) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 401) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 402) | bool IsKernelLaunchSynchronized() const override { function Infer2DSameDim0All2AllKernelTmpBufferSize (line 408) | size_t Infer2DSameDim0All2AllKernelTmpBufferSize(user_op::InferContext... class CclLogical2DSameDim1KernelCommState (line 426) | class CclLogical2DSameDim1KernelCommState final : public user_op::OpKe... method CclLogical2DSameDim1KernelCommState (line 428) | explicit CclLogical2DSameDim1KernelCommState(user_op::KernelInitCont... class CclLogical2DSameDim1AllReduce (line 459) | class CclLogical2DSameDim1AllReduce final : public user_op::OpKernel { method CclLogical2DSameDim1AllReduce (line 461) | CclLogical2DSameDim1AllReduce() = default; method CreateOpKernelState (line 464) | std::shared_ptr CreateOpKernelState( method Compute (line 470) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 490) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 491) | bool IsKernelLaunchSynchronized() const override { FILE: oneflow/user/kernels/nccl_logical_fusion_kernel.cpp type oneflow (line 33) | namespace oneflow { function GetTmpBufferSizeByNcclType (line 37) | size_t GetTmpBufferSizeByNcclType(const std::string& nccl_type, size_t... function GetTensorByteSize (line 68) | size_t GetTensorByteSize(const user_op::TensorDesc& tensor_desc) { function GetTensorByteSize (line 73) | size_t GetTensorByteSize(const user_op::Tensor& tensor) { class CclLogicalFusionKernelState (line 77) | class CclLogicalFusionKernelState : public user_op::OpKernelState { method CclLogicalFusionKernelState (line 79) | explicit CclLogicalFusionKernelState(user_op::KernelInitContext* ctx) method ccl_comm (line 92) | ccl::CclComm ccl_comm() { method num_ranks (line 97) | int64_t num_ranks() { method src_split_axis (line 103) | int64_t src_split_axis(int32_t i) const { method dst_split_axis (line 108) | int64_t dst_split_axis(int32_t i) const { method nccl_num (line 114) | int32_t nccl_num() const { return nccl_num_; } method tmp_buffer_offset (line 115) | size_t tmp_buffer_offset(int32_t i) { method tmp_buffer_size (line 121) | size_t tmp_buffer_size(int32_t i) { method InitComm (line 128) | void InitComm() { method UpdateOrCheckEqCommKey (line 155) | void UpdateOrCheckEqCommKey(const std::string& val) { method UpdateSplitAxisByNcclType (line 163) | void UpdateSplitAxisByNcclType(const std::string& nccl_type, const i... method InitSplitAxisAndTmpBufferOffset (line 196) | void InitSplitAxisAndTmpBufferOffset(user_op::KernelInitContext* ctx) { class CclLogicalFusionKernel (line 261) | class CclLogicalFusionKernel final : public user_op::OpKernel { method CclLogicalFusionKernel (line 264) | CclLogicalFusionKernel() = default; method CreateOpKernelState (line 267) | std::shared_ptr CreateOpKernelState( method AlwaysComputeWhenAllOutputsEmpty (line 275) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 276) | bool IsKernelLaunchSynchronized() const override { function DoPackBeforeNcclGroup (line 329) | void DoPackBeforeNcclGroup(void* pack_to_ptr, const std::string& nccl_... function DoNcclComputeByNcclTypeInGroup (line 403) | void DoNcclComputeByNcclTypeInGroup(const void* pack_to_ptr, void* unp... function DoUnpackAfterNcclGroup (line 528) | void DoUnpackAfterNcclGroup(void* unpack_from_ptr, const std::string& ... function InferNcclLogicalFusionKernelTmpBufferSize (line 679) | size_t InferNcclLogicalFusionKernelTmpBufferSize(user_op::InferContext... FILE: oneflow/user/kernels/nccl_logical_kernels.cpp type oneflow (line 33) | namespace oneflow { function AllReduceCollectiveCommunicationExists (line 37) | auto AllReduceCollectiveCommunicationExists() { function ReduceScatterCollectiveCommunicationExists (line 46) | auto ReduceScatterCollectiveCommunicationExists() { function AllGatherCollectiveCommunicationExists (line 55) | auto AllGatherCollectiveCommunicationExists() { function ReduceCollectiveCommunicationExists (line 64) | auto ReduceCollectiveCommunicationExists() { function BroadcastCollectiveCommunicationExists (line 73) | auto BroadcastCollectiveCommunicationExists() { function AllToAllCollectiveCommunicationExists (line 82) | auto AllToAllCollectiveCommunicationExists() { class NcclLogicalKernelCommState (line 91) | class NcclLogicalKernelCommState : public user_op::OpKernelState { method NcclLogicalKernelCommState (line 93) | explicit NcclLogicalKernelCommState(user_op::KernelInitContext* ctx) class NcclLogicalAllGatherNoncontinuousKernelState (line 119) | class NcclLogicalAllGatherNoncontinuousKernelState : public NcclLogica... method NcclLogicalAllGatherNoncontinuousKernelState (line 121) | explicit NcclLogicalAllGatherNoncontinuousKernelState(user_op::Kerne... method src_split_axis (line 125) | int64_t src_split_axis() const { return src_split_axis_; } method set_src_split_axis (line 126) | void set_src_split_axis(int64_t split_axis) { src_split_axis_ = spli... class NcclLogicalReduceScatterNoncontinuousKernelState (line 132) | class NcclLogicalReduceScatterNoncontinuousKernelState : public NcclLo... method NcclLogicalReduceScatterNoncontinuousKernelState (line 134) | explicit NcclLogicalReduceScatterNoncontinuousKernelState(user_op::K... method dst_split_axis (line 138) | int64_t dst_split_axis() const { return dst_split_axis_; } method set_dst_split_axis (line 139) | void set_dst_split_axis(int64_t split_axis) { dst_split_axis_ = spli... class NcclLogicalS2SKernelState (line 145) | class NcclLogicalS2SKernelState : public NcclLogicalKernelCommState { method NcclLogicalS2SKernelState (line 147) | explicit NcclLogicalS2SKernelState(user_op::KernelInitContext* ctx) method src_split_axis (line 151) | int64_t src_split_axis() const { return src_split_axis_; } method set_src_split_axis (line 152) | void set_src_split_axis(int64_t split_axis) { src_split_axis_ = spli... method dst_split_axis (line 153) | int64_t dst_split_axis() const { return dst_split_axis_; } method set_dst_split_axis (line 154) | void set_dst_split_axis(int64_t split_axis) { dst_split_axis_ = spli... class CclLogicalAllReduceKernel (line 161) | class CclLogicalAllReduceKernel final : public user_op::OpKernel { method CclLogicalAllReduceKernel (line 163) | CclLogicalAllReduceKernel() = default; method CreateOpKernelState (line 166) | std::shared_ptr CreateOpKernelState( method Compute (line 172) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 192) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 193) | bool IsKernelLaunchSynchronized() const override { class CclLogicalReduceScatterKernel (line 199) | class CclLogicalReduceScatterKernel final : public user_op::OpKernel { method CclLogicalReduceScatterKernel (line 201) | CclLogicalReduceScatterKernel() = default; method CreateOpKernelState (line 204) | std::shared_ptr CreateOpKernelState( method Compute (line 210) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 231) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 232) | bool IsKernelLaunchSynchronized() const override { class CclLogicalAllGatherKernel (line 238) | class CclLogicalAllGatherKernel final : public user_op::OpKernel { method CclLogicalAllGatherKernel (line 240) | CclLogicalAllGatherKernel() = default; method CreateOpKernelState (line 243) | std::shared_ptr CreateOpKernelState( method Compute (line 249) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 268) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 269) | bool IsKernelLaunchSynchronized() const override { class CclLogicalAllGatherNoncontinuous (line 276) | class CclLogicalAllGatherNoncontinuous final : public user_op::OpKernel { method CclLogicalAllGatherNoncontinuous (line 278) | CclLogicalAllGatherNoncontinuous() = default; method CreateOpKernelState (line 281) | std::shared_ptr CreateOpKernelState( method Compute (line 293) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 340) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 341) | bool IsKernelLaunchSynchronized() const override { function InferAllGatherNoncontinuousKernelTmpBufferSize (line 347) | size_t InferAllGatherNoncontinuousKernelTmpBufferSize(user_op::InferCo... class CclLogicalReduceScatterNoncontinuous (line 354) | class CclLogicalReduceScatterNoncontinuous final : public user_op::OpK... method CclLogicalReduceScatterNoncontinuous (line 356) | CclLogicalReduceScatterNoncontinuous() = default; method CreateOpKernelState (line 359) | std::shared_ptr CreateOpKernelState( method Compute (line 371) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 415) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 416) | bool IsKernelLaunchSynchronized() const override { function InferReduceScatterNoncontinuousKernelTmpBufferSize (line 422) | size_t InferReduceScatterNoncontinuousKernelTmpBufferSize(user_op::Inf... class CclLogicalS2SKernel (line 429) | class CclLogicalS2SKernel final : public user_op::OpKernel { method CclLogicalS2SKernel (line 431) | CclLogicalS2SKernel() = default; method CreateOpKernelState (line 434) | std::shared_ptr CreateOpKernelState( method Compute (line 451) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 539) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 540) | bool IsKernelLaunchSynchronized() const override { function InferS2SKernelTmpBufferSize (line 546) | size_t InferS2SKernelTmpBufferSize(user_op::InferContext* ctx) { FILE: oneflow/user/kernels/nccl_logical_send_recv_kernel.cpp type oneflow (line 35) | namespace oneflow { class CclLogicalSendRecvState (line 37) | class CclLogicalSendRecvState final : public user_op::OpKernelState { method src_nd_sbp_has_no_partial_parallel (line 46) | bool src_nd_sbp_has_no_partial_parallel() const { return src_nd_sbp_... method ccl_comm (line 49) | ccl::CclComm ccl_comm() const { return GetOrCreateComm().ccl_comm; } type Comm (line 52) | struct Comm { method Comm (line 53) | Comm(ccl::CclComm comm) : ccl_comm(comm) {} method Comm (line 58) | const Comm& GetOrCreateComm() const { method Comm (line 53) | Comm(ccl::CclComm comm) : ccl_comm(comm) {} class CclLogicalSendRecv (line 135) | class CclLogicalSendRecv final : public user_op::OpKernel { method CclLogicalSendRecv (line 138) | CclLogicalSendRecv() = default; method CreateOpKernelState (line 141) | std::shared_ptr CreateOpKernelState( method AlwaysComputeWhenAllOutputsEmpty (line 149) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method IsKernelLaunchSynchronized (line 150) | bool IsKernelLaunchSynchronized() const override { function InferTmpBufferSize (line 252) | size_t InferTmpBufferSize(user_op::InferContext* ctx) { FILE: oneflow/user/kernels/nd_index_slice_kernels.cpp type oneflow (line 18) | namespace oneflow { type GatherNdFunctor (line 21) | struct GatherNdFunctor final { type ScatterNdAddFunctor (line 30) | struct ScatterNdAddFunctor final { type ScatterNdUpdateFunctor (line 39) | struct ScatterNdUpdateFunctor final { type ScatterNdUpdateWithStrideFunctor (line 48) | struct ScatterNdUpdateWithStrideFunctor final { type FillByNdIndexFunctor (line 57) | struct FillByNdIndexFunctor final { FILE: oneflow/user/kernels/nd_index_slice_kernels.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/nd_index_slice_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/nll_kernel.cpp type oneflow (line 21) | namespace oneflow { class NLLKernelCache (line 25) | class NLLKernelCache final : public user_op::OpKernelCache { method NLLKernelCache (line 27) | NLLKernelCache(int64_t class_start, int64_t num_classes) method class_start (line 31) | int64_t class_start() const { return class_start_; } method num_classes (line 32) | int64_t num_classes() const { return num_classes_; } function CreateNLLKernelCache (line 39) | std::shared_ptr CreateNLLKernelCache(user_op::... class NLLKernel (line 70) | class NLLKernel final : public user_op::OpKernel { method NLLKernel (line 72) | NLLKernel() = default; method InitOpKernelCache (line 75) | std::shared_ptr InitOpKernelCache( method AlwaysComputeWhenAllOutputsEmpty (line 81) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 83) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class NLLGradKernel (line 119) | class NLLGradKernel final : public user_op::OpKernel { method NLLGradKernel (line 121) | NLLGradKernel() = default; method InitOpKernelCache (line 124) | std::shared_ptr InitOpKernelCache( method AlwaysComputeWhenAllOutputsEmpty (line 130) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 132) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/nll_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type NLLKernelUtil (line 21) | struct NLLKernelUtil { method Forward (line 22) | static void Forward(ep::Stream* stream, const int32_t num_samples, c... method Backward (line 41) | static void Backward(ep::Stream* stream, const int32_t num_samples, ... type NLLKernelUtil (line 58) | struct NLLKernelUtil type NLLKernelUtil (line 59) | struct NLLKernelUtil type NLLKernelUtil (line 60) | struct NLLKernelUtil type NLLKernelUtil (line 61) | struct NLLKernelUtil FILE: oneflow/user/kernels/nll_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/nms_kernel.cpp type oneflow (line 20) | namespace oneflow { function T (line 25) | __inline__ T IoU(T const* const a, T const* const b) { class NmsCpuKernel (line 36) | class NmsCpuKernel final : public user_op::OpKernel { method NmsCpuKernel (line 38) | NmsCpuKernel() = default; method Compute (line 43) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 62) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/nop_kernel.cpp type oneflow (line 18) | namespace oneflow { class NopKernel (line 22) | class NopKernel final : public user_op::OpKernel { method NopKernel (line 24) | NopKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 31) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/normalization_kernel.cpp type oneflow (line 18) | namespace oneflow { function ComputeMeanAndVar (line 21) | static void ComputeMeanAndVar(const T* input_ptr, T* mean_ptr, T* inv_... function Normalize (line 71) | static void Normalize(const T* input_ptr, const T* mean_ptr, const T* ... function AddToOutput (line 96) | static void AddToOutput(const T* add_to_output_ptr, T* output_ptr, con... function AddRelu (line 101) | static void AddRelu(const T* addend_ptr, int32_t* mask_ptr, T* output_... function Relu (line 133) | static void Relu(int32_t* mask_ptr, T* output_ptr, const int64_t elem_... function AddReluGrad (line 164) | static void AddReluGrad(const T* dy_ptr, const int32_t* mask_ptr, T* a... function ReluGrad (line 191) | static void ReluGrad(const T* dy_ptr, const int32_t* mask_ptr, T* relu... function InferGradTmpSizeForCpuKernel (line 217) | static size_t InferGradTmpSizeForCpuKernel(user_op::InferContext* ctx) { function ForEachFast (line 228) | static inline void ForEachFast(const T* data, const int64_t batch_size... function ForEachFast (line 240) | static inline void ForEachFast(const T* in_data1, const T* in_data2, c... function ForEachFast (line 256) | static inline void ForEachFast(const T* in_data, T* out_data, const in... class NormalizationInferenceCpuKernel (line 272) | class NormalizationInferenceCpuKernel final : public user_op::OpKernel { method NormalizationInferenceCpuKernel (line 274) | NormalizationInferenceCpuKernel() = default; method Compute (line 279) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 327) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class NormalizationTrainCpuKernel (line 351) | class NormalizationTrainCpuKernel final : public user_op::OpKernel { method NormalizationTrainCpuKernel (line 353) | NormalizationTrainCpuKernel() = default; method Compute (line 358) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 439) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class NormalizationGradCpuKernel (line 474) | class NormalizationGradCpuKernel final : public user_op::OpKernel { method NormalizationGradCpuKernel (line 476) | NormalizationGradCpuKernel() = default; method Compute (line 481) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 579) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/ofrecord_decoder_kernels.cpp type oneflow (line 33) | namespace oneflow { function DecodeOneRawOFRecord (line 38) | void DecodeOneRawOFRecord(const Feature& feature, T* dptr, int64_t sam... class OFRecordRawDecoderKernel (line 81) | class OFRecordRawDecoderKernel final : public user_op::OpKernel { method OFRecordRawDecoderKernel (line 83) | OFRecordRawDecoderKernel() = default; method Compute (line 87) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 110) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class OFRecordBytesDecoderKernel (line 128) | class OFRecordBytesDecoderKernel final : public user_op::OpKernel { method OFRecordBytesDecoderKernel (line 130) | OFRecordBytesDecoderKernel() = default; method Compute (line 134) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 157) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function DecodeRandomCropImageFromOneRecord (line 168) | void DecodeRandomCropImageFromOneRecord(const OFRecord& record, Tensor... class OFRecordImageDecoderRandomCropKernel (line 210) | class OFRecordImageDecoderRandomCropKernel final : public user_op::OpK... method OFRecordImageDecoderRandomCropKernel (line 212) | OFRecordImageDecoderRandomCropKernel() = default; method CreateOpKernelState (line 215) | std::shared_ptr CreateOpKernelState( method Compute (line 221) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 242) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class OFRecordImageDecoderKernel (line 251) | class OFRecordImageDecoderKernel final : public user_op::OpKernel { method OFRecordImageDecoderKernel (line 253) | OFRecordImageDecoderKernel() = default; method Compute (line 257) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 274) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/ofrecord_image_classification_reader_kernel.cpp type oneflow (line 19) | namespace oneflow { class OFRecordImageClassificationReaderKernelState (line 23) | class OFRecordImageClassificationReaderKernelState final : public user... method OFRecordImageClassificationReaderKernelState (line 25) | explicit OFRecordImageClassificationReaderKernelState(user_op::Kerne... method Read (line 29) | void Read(user_op::KernelComputeContext* ctx) { reader_.Read(ctx); } class OFRecordImageClassificationReaderKernel (line 37) | class OFRecordImageClassificationReaderKernel final : public user_op::... method OFRecordImageClassificationReaderKernel (line 39) | OFRecordImageClassificationReaderKernel() = default; method CreateOpKernelState (line 42) | std::shared_ptr CreateOpKernelState( method Compute (line 48) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 54) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/ofrecord_reader_kernel.cpp type oneflow (line 19) | namespace oneflow { class OFRecordReaderWrapper (line 23) | class OFRecordReaderWrapper final : public user_op::OpKernelState { method OFRecordReaderWrapper (line 25) | explicit OFRecordReaderWrapper(user_op::KernelInitContext* ctx) : re... method Read (line 28) | void Read(user_op::KernelComputeContext* ctx) { reader_.Read(ctx); } class OFRecordReaderKernel (line 36) | class OFRecordReaderKernel final : public user_op::OpKernel { method OFRecordReaderKernel (line 38) | OFRecordReaderKernel() = default; method CreateOpKernelState (line 41) | std::shared_ptr CreateOpKernelState( method Compute (line 48) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 53) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/one_hot_kernel.cpp type oneflow (line 20) | namespace oneflow { class CpuOneHotKernel (line 23) | class CpuOneHotKernel final : public user_op::OpKernel { method CpuOneHotKernel (line 25) | CpuOneHotKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 55) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/ones_like_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 22) | namespace user_op { function NewFillPrimitive (line 27) | std::unique_ptr NewFillPrimitive(Context* ctx) { class OnesLikeKernel (line 32) | class OnesLikeKernel final : public user_op::OpKernel { method OnesLikeKernel (line 34) | OnesLikeKernel() = default; method Compute (line 38) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 47) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function FillPrimitiveExists (line 50) | auto FillPrimitiveExists() { FILE: oneflow/user/kernels/op_kernel_wrapper.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/p2p_comm_kernel.cpp type oneflow (line 24) | namespace oneflow { function SendCollectiveCommunicationExists (line 30) | auto SendCollectiveCommunicationExists() { function RecvCollectiveCommunicationExists (line 38) | auto RecvCollectiveCommunicationExists() { class SendKernel (line 48) | class SendKernel final : public user_op::OpKernel { method SendKernel (line 50) | SendKernel() = default; method Compute (line 54) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 61) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class RecvKernel (line 64) | class RecvKernel final : public user_op::OpKernel { method RecvKernel (line 66) | RecvKernel() = default; method Compute (line 70) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 77) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/pack_kernel.cpp type oneflow (line 20) | namespace oneflow { class PackKernel (line 25) | class PackKernel final : public user_op::OpKernel { method PackKernel (line 27) | PackKernel() = default; method CreateOpKernelState (line 30) | std::shared_ptr CreateOpKernelState( method Compute (line 37) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 64) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/pad_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { function NewConstantPadPrimitive (line 28) | std::unique_ptr NewConstantPadPrimitive(... function ConstantPadPrimitiveExists (line 34) | auto ConstantPadPrimitiveExists() { class PadKernel (line 42) | class PadKernel final : public OpKernel, public CudaGraphSupport { method PadKernel (line 44) | PadKernel() = default; method Compute (line 48) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 74) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/pocketfft_hdronly.h function namespace (line 106) | namespace pocketfft { function cmplx (line 389) | static cmplx calc(size_t x, size_t n, Thigh ang) { type util (line 445) | struct util // hack to avoid duplicate symbols function POCKETFFT_NOINLINE (line 462) | static POCKETFFT_NOINLINE double cost_guess(size_t n) { function POCKETFFT_NOINLINE (line 480) | static POCKETFFT_NOINLINE size_t good_size_cmplx(size_t n) { function POCKETFFT_NOINLINE (line 504) | static POCKETFFT_NOINLINE size_t good_size_real(size_t n) { function prod (line 525) | static size_t prod(const shape_t& shape) { function shutdown (line 776) | void shutdown() { function restart (line 781) | void restart() { function thread_pool (line 787) | inline thread_pool& get_pool() { type fctdata (line 845) | struct fctdata { function add_factor (line 854) | void add_factor(size_t factor) { fact.push_back({factor, nullptr, nullpt... function twsize (line 1496) | size_t twsize() const { function comp_twiddle (line 1507) | void comp_twiddle() { type fctdata (line 1544) | struct fctdata { function add_factor (line 1553) | void add_factor(size_t factor) { fact.push_back({factor, nullptr, nullpt... function twsize (line 2260) | size_t twsize() const { function comp_twiddle (line 2271) | void comp_twiddle() { function exec (line 2471) | void exec(T c[], T0 fct, bool fwd) const { function exec (line 2491) | void exec(T c[], T0 fct, bool ortho, int /*type*/, bool /*cosine*/) const { function exec (line 2522) | void exec(T c[], T0 fct, bool /*ortho*/, int /*type*/, bool /*cosine*/) ... function exec (line 2550) | void exec(T c[], T0 fct, bool ortho, int type, bool cosine) const { function exec (line 2612) | void exec(T c[], T0 fct, bool /*ortho*/, int /*type*/, bool cosine) const { function find_in_cache (line 2687) | auto find_in_cache = [&]() -> std::shared_ptr { function class (line 2724) | class arr_info { function advance_i (line 2768) | void advance_i() { function advance (line 2815) | void advance(size_t n) { function iofs (line 2824) | ptrdiff_t iofs(size_t i) const { return p_i[0] + ptrdiff_t(i) * str_i; } function iofs (line 2825) | ptrdiff_t iofs(size_t j, size_t i) const { return p_i[j] + ptrdiff_t(i) ... function oofs (line 2826) | ptrdiff_t oofs(size_t i) const { return p_o[0] + ptrdiff_t(i) * str_o; } function oofs (line 2827) | ptrdiff_t oofs(size_t j, size_t i) const { return p_o[j] + ptrdiff_t(i) ... function class (line 2835) | class simple_iter { function class (line 2858) | class rev_iter { type ExecC2C (line 3043) | struct ExecC2C { type ExecHartley (line 3080) | struct ExecHartley { type ExecDcst (line 3090) | struct ExecDcst { type ExecR2R (line 3209) | struct ExecR2R { function ExecDcst (line 3245) | const ExecDcst exec{ortho, type, true}; function ExecDcst (line 3263) | const ExecDcst exec{ortho, type, false}; function newaxes (line 3295) | auto newaxes = shape_t{axes.begin(), --axes.end()}; function newaxes (line 3329) | auto newaxes = shape_t{axes.begin(), --axes.end()}; FILE: oneflow/user/kernels/pocketfftplan.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/prelu_kernel.cpp type oneflow (line 19) | namespace oneflow { class CpuPReluKernel (line 22) | class CpuPReluKernel final : public user_op::OpKernel { method CpuPReluKernel (line 24) | CpuPReluKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 44) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuPReluGradKernel (line 56) | class CpuPReluGradKernel final : public user_op::OpKernel { method CpuPReluGradKernel (line 58) | CpuPReluGradKernel() = default; method Compute (line 62) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 91) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/quantization_kernel.cpp type oneflow (line 20) | namespace oneflow { function QuantizationPerLayerSymmetric (line 23) | void QuantizationPerLayerSymmetric(const T* in_ptr, const T scale, con... function QuantizationPerLayerAffine (line 36) | void QuantizationPerLayerAffine(const T* in_ptr, const T scale, const ... function QuantizationPerLayerCambricon (line 51) | void QuantizationPerLayerCambricon(const T* in_ptr, const T shift, con... class CpuQuantizationKernel (line 65) | class CpuQuantizationKernel final : public user_op::OpKernel { method CpuQuantizationKernel (line 67) | CpuQuantizationKernel() = default; method Compute (line 71) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 122) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/random_crop_kernel_state.cpp type oneflow (line 19) | namespace oneflow { function CreateRandomCropKernelState (line 21) | std::shared_ptr CreateRandomCropKernelState( FILE: oneflow/user/kernels/random_crop_kernel_state.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/random_mask_generator.cpp type oneflow (line 18) | namespace oneflow { class RandomMaskGenerator (line 28) | class RandomMaskGenerator FILE: oneflow/user/kernels/random_mask_generator.h function namespace (line 27) | namespace oneflow { FILE: oneflow/user/kernels/random_mask_like_kernel.cpp type oneflow (line 18) | namespace oneflow { FILE: oneflow/user/kernels/random_mask_like_kernel.h function namespace (line 25) | namespace oneflow { FILE: oneflow/user/kernels/random_seed_util.cpp type oneflow (line 21) | namespace oneflow { function GetOpKernelRandomSeed (line 23) | Maybe GetOpKernelRandomSeed(const user_op::KernelInitContext... function GetRandomSeedForRank (line 35) | Maybe GetRandomSeedForRank(const ParallelDesc& placement, co... function GetOpKernelRandomSeedInCurrentRank (line 63) | Maybe GetOpKernelRandomSeedInCurrentRank(const user_op::Kern... function GetGeneratorForLazyOrGlobal (line 73) | Maybe GetGeneratorForLazyOrGlobal(const std::shared_pt... function GetGeneratorForLazyOrGlobal (line 116) | Maybe GetGeneratorForLazyOrGlobal(const std::shared_pt... FILE: oneflow/user/kernels/random_seed_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/randperm_kernel.cpp type oneflow (line 28) | namespace oneflow { class CpuRandPermKernelCache (line 29) | class CpuRandPermKernelCache final : public user_op::OpKernelCache { method CpuRandPermKernelCache (line 31) | CpuRandPermKernelCache(int32_t lower, int32_t upper) : lower_(lower)... method lower (line 34) | int32_t lower() const { return lower_; } method upper (line 35) | int32_t upper() const { return upper_; } class CpuRandPermKernel (line 42) | class CpuRandPermKernel final : public user_op::OpKernel { method CpuRandPermKernel (line 44) | CpuRandPermKernel() = default; method InitOpKernelCache (line 46) | std::shared_ptr InitOpKernelCache( method CreateOpKernelState (line 64) | std::shared_ptr CreateOpKernelState( method Compute (line 73) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 97) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/raw_reader_kernel.cpp type oneflow (line 23) | namespace oneflow { type Block (line 27) | struct Block { type BatchReaderRequest (line 32) | struct BatchReaderRequest { class BatchReader (line 37) | class BatchReader { method BatchReader (line 40) | BatchReader(std::vector>&& files, method SubmitRequest (line 93) | void SubmitRequest(BatchReaderRequest&& request) { method WaitCompleted (line 97) | void WaitCompleted(BatchReaderRequest* request) { type Worker (line 103) | struct Worker { method Close (line 107) | void Close() { function GetNumShards (line 122) | size_t GetNumShards(const Shape& hierarchy, const NdSbp& nd_sbp) { function GetShardIndex (line 133) | size_t GetShardIndex(const Shape& hierarchy, const NdSbp& nd_sbp, size... class BatchGenerator (line 153) | class BatchGenerator { method BatchGenerator (line 156) | BatchGenerator() = default; class SequentialBatchGenerator (line 162) | class SequentialBatchGenerator : public BatchGenerator { method SequentialBatchGenerator (line 165) | SequentialBatchGenerator(size_t shard_index, size_t num_shards, size... method Next (line 175) | void Next(size_t* blocks) override { class RandomShuffleBatchGenerator (line 193) | class RandomShuffleBatchGenerator : public BatchGenerator { method RandomShuffleBatchGenerator (line 196) | RandomShuffleBatchGenerator(size_t shard_index, size_t num_shards, s... method Next (line 210) | void Next(size_t* blocks) override { class RawReaderKernelState (line 236) | class RawReaderKernelState final : public user_op::OpKernelState { method RawReaderKernelState (line 239) | explicit RawReaderKernelState(user_op::KernelInitContext* ctx) { method Next (line 358) | void Next(user_op::KernelComputeContext* ctx) { class RawReaderKernel (line 406) | class RawReaderKernel final : public user_op::OpKernel { method RawReaderKernel (line 408) | RawReaderKernel() = default; method CreateOpKernelState (line 411) | std::shared_ptr CreateOpKernelState( method Compute (line 418) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 423) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/reduce_kernel.cpp type oneflow (line 30) | namespace oneflow { function GetBlasTransposeType (line 34) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 38) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewReduceMatmulTransAPrimitive (line 48) | std::unique_ptr NewReduceMatmulTransAPrimitive(... function NewReduceMatmulNoTransAPrimitive (line 55) | std::unique_ptr NewReduceMatmulNoTransAPrimitiv... function NewFillPrimitive (line 62) | std::unique_ptr NewFillPrimitive(Context* ctx) { function ReduceMatmulTransAPrimitiveExists (line 67) | auto ReduceMatmulTransAPrimitiveExists() { function ReduceMatmulNoTransAPrimitiveExists (line 74) | auto ReduceMatmulNoTransAPrimitiveExists() { function FillPrimitiveExists (line 81) | auto FillPrimitiveExists() { class ReduceKernel (line 88) | class ReduceKernel final : public user_op::OpKernel, public user_op::C... method ReduceKernel (line 90) | ReduceKernel() = default; method Compute (line 94) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 123) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function RegularAxis (line 224) | std::vector RegularAxis(const std::vector& axis) { function GetReduceSumLayout (line 230) | void GetReduceSumLayout(const std::vector& axis, const ShapeV... class ReduceSumHalfKernel (line 243) | class ReduceSumHalfKernel final : public user_op::OpKernel, public use... method ReduceSumHalfKernel (line 245) | ReduceSumHalfKernel() = default; method Compute (line 249) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 314) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ReduceSumFloatCudaKernel (line 347) | class ReduceSumFloatCudaKernel final : public user_op::OpKernel, publi... method ReduceSumFloatCudaKernel (line 349) | ReduceSumFloatCudaKernel() = default; method Compute (line 353) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 403) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/reduce_like_kernels.cpp type oneflow (line 25) | namespace oneflow { function GetBlasTransposeType (line 29) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMatmulPrimitive (line 33) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewReduceMatmulTransAPrimitive (line 43) | std::unique_ptr NewReduceMatmulTransAPrimitive(... function NewReduceMatmulNoTransAPrimitive (line 50) | std::unique_ptr NewReduceMatmulNoTransAPrimitiv... function ReduceMatmulTransAPrimitiveExists (line 56) | auto ReduceMatmulTransAPrimitiveExists() { function ReduceMatmulNoTransAPrimitiveExists (line 63) | auto ReduceMatmulNoTransAPrimitiveExists() { function ReduceSumLikeInferTmpSize (line 70) | size_t ReduceSumLikeInferTmpSize(user_op::InferContext* ctx) { class ReduceSumLikeOpKernel (line 79) | class ReduceSumLikeOpKernel final : public user_op::OpKernel, public u... method ReduceSumLikeOpKernel (line 81) | ReduceSumLikeOpKernel() = default; method Compute (line 85) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 114) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function RegularAxis (line 142) | std::vector RegularAxis(const std::vector& axis) { function GetReduceSumLayout (line 148) | void GetReduceSumLayout(const std::vector& axis, const ShapeV... class ReduceSumLikeHalfKernel (line 159) | class ReduceSumLikeHalfKernel final : public user_op::OpKernel, public... method ReduceSumLikeHalfKernel (line 161) | ReduceSumLikeHalfKernel() = default; method Compute (line 165) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 234) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/reflection_pad_kernels.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 22) | namespace user_op { class ReflectionPad1dKernel (line 25) | class ReflectionPad1dKernel final : public OpKernel { method ReflectionPad1dKernel (line 27) | ReflectionPad1dKernel() = default; method Compute (line 31) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 57) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReflectionPad1dGradKernel (line 61) | class ReflectionPad1dGradKernel final : public OpKernel { method ReflectionPad1dGradKernel (line 63) | ReflectionPad1dGradKernel() = default; method Compute (line 67) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 96) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReflectionPad2dKernel (line 100) | class ReflectionPad2dKernel final : public OpKernel { method ReflectionPad2dKernel (line 102) | ReflectionPad2dKernel() = default; method Compute (line 106) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 135) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReflectionPad2dGradKernel (line 139) | class ReflectionPad2dGradKernel final : public OpKernel { method ReflectionPad2dGradKernel (line 141) | ReflectionPad2dGradKernel() = default; method Compute (line 145) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 177) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/reflection_pad_kernels_util.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 20) | namespace user_op { type ReflectionPad1dFunctor (line 23) | struct ReflectionPad1dFunctor final { type ReflectionPad1dGradFunctor (line 37) | struct ReflectionPad1dGradFunctor final { type ReflectionPad2dFunctor (line 51) | struct ReflectionPad2dFunctor final { type ReflectionPad2dGradFunctor (line 66) | struct ReflectionPad2dGradFunctor final { FILE: oneflow/user/kernels/reflection_pad_kernels_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/repeat_interleave_kernel.cpp type oneflow (line 21) | namespace oneflow { class CpuRepeatInterLeaveKernel (line 24) | class CpuRepeatInterLeaveKernel final : public user_op::OpKernel { method CpuRepeatInterLeaveKernel (line 26) | CpuRepeatInterLeaveKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 45) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/replication_pad_kernels.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 22) | namespace user_op { class ReplicationPad1dKernel (line 25) | class ReplicationPad1dKernel final : public OpKernel { method ReplicationPad1dKernel (line 27) | ReplicationPad1dKernel() = default; method Compute (line 31) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 55) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReplicationPad1dGradKernel (line 59) | class ReplicationPad1dGradKernel final : public OpKernel { method ReplicationPad1dGradKernel (line 61) | ReplicationPad1dGradKernel() = default; method Compute (line 65) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 92) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReplicationPad2dKernel (line 96) | class ReplicationPad2dKernel final : public OpKernel { method ReplicationPad2dKernel (line 98) | ReplicationPad2dKernel() = default; method Compute (line 102) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 131) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReplicationPad2dGradKernel (line 135) | class ReplicationPad2dGradKernel final : public OpKernel { method ReplicationPad2dGradKernel (line 137) | ReplicationPad2dGradKernel() = default; method Compute (line 141) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 173) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/replication_pad_kernels_util.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 20) | namespace user_op { type ReplicationPad1dFunctor (line 23) | struct ReplicationPad1dFunctor final { type ReplicationPad1dGradFunctor (line 37) | struct ReplicationPad1dGradFunctor final { type ReplicationPad2dFunctor (line 51) | struct ReplicationPad2dFunctor final { type ReplicationPad2dGradFunctor (line 66) | struct ReplicationPad2dGradFunctor final { FILE: oneflow/user/kernels/replication_pad_kernels_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/roc_auc_score_kernel.cpp type oneflow (line 18) | namespace oneflow { function RocAucScore (line 23) | double RocAucScore(size_t n, const L* label, const P* pred, float* buf... class RocAucScoreKernel (line 75) | class RocAucScoreKernel final : public user_op::OpKernel { method RocAucScoreKernel (line 78) | RocAucScoreKernel() = default; method Compute (line 82) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 93) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/roll_kernel.cpp type oneflow (line 21) | namespace oneflow { class CpuRollKernel (line 24) | class CpuRollKernel final : public user_op::OpKernel { method CpuRollKernel (line 26) | CpuRollKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 56) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/roll_kernel_utils.h function namespace (line 21) | namespace oneflow { function getShiftedIndex (line 43) | int32_t getShiftedIndex(const int32_t global_index, const int32_t* shifts, function OF_DEVICE_FUNC (line 64) | OF_DEVICE_FUNC int32_t switchGetShiftedIndex(const int32_t global_index,... function initStride (line 87) | static void initStride(STRIDE& stride, const SHAPE& dim_vec, const int32... function transformShifts (line 91) | static void transformShifts(int32_t* shifts, int32_t* shape, int n) { function computeParams (line 95) | static void computeParams(const ShapeView& in_shape, const std::vector CreateOpKernelState( method Compute (line 56) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 93) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/same_padding_kernel.cpp type oneflow (line 21) | namespace oneflow { function NewFillPrimitive (line 26) | std::unique_ptr NewFillPrimitive(Context* ctx) { function NewCopyNdPrimitive (line 32) | std::unique_ptr NewCopyNdPrimitive(Context* ctx) { function FillPrimitiveExists (line 39) | auto FillPrimitiveExists() { function CopyNdPrimitiveExists (line 45) | auto CopyNdPrimitiveExists() { class SamePaddingKernel (line 53) | class SamePaddingKernel final : public user_op::OpKernel { method SamePaddingKernel (line 55) | SamePaddingKernel() = default; method Compute (line 59) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 99) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class SamePaddingGradKernel (line 106) | class SamePaddingGradKernel final : public user_op::OpKernel { method SamePaddingGradKernel (line 108) | SamePaddingGradKernel() = default; method Compute (line 112) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 148) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/scalar_bitwise_kernels.cpp type oneflow (line 20) | namespace oneflow { function NewBinaryPrimitive (line 25) | std::unique_ptr NewBinaryPr... function PrimitiveExists (line 35) | auto PrimitiveExists() { class ScalarBitwiseKernel (line 43) | class ScalarBitwiseKernel final : public user_op::OpKernel, public use... method ScalarBitwiseKernel (line 45) | ScalarBitwiseKernel() = default; method Compute (line 49) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 66) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/scalar_by_tensor_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewBroadcastElementwiseBinaryPrimitive (line 25) | std::unique_ptr NewBroadcas... function BroadcastElementwiseBinaryPrimitiveExists (line 35) | auto BroadcastElementwiseBinaryPrimitiveExists() { class ScalarByTensorKernel (line 43) | class ScalarByTensorKernel final : public user_op::OpKernel, public us... method ScalarByTensorKernel (line 45) | ScalarByTensorKernel() = default; method Compute (line 49) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 66) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/scalar_logical_kernels.cpp type oneflow (line 20) | namespace oneflow { function NewBinaryPrimitive (line 25) | std::unique_ptr NewBinaryPr... function PrimitiveExists (line 35) | auto PrimitiveExists() { class ScalarLogicalKernel (line 43) | class ScalarLogicalKernel final : public user_op::OpKernel, public use... method ScalarLogicalKernel (line 45) | ScalarLogicalKernel() = default; method Compute (line 49) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 73) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/scalar_math_kernels.cpp type oneflow (line 21) | namespace oneflow { function NewBroadcastElementwiseBinaryPrimitive (line 26) | std::unique_ptr NewBroadcas... function BroadcastElementwiseBinaryPrimitiveExists (line 36) | auto BroadcastElementwiseBinaryPrimitiveExists() { function NewBroadcastElementwiseAttrBinaryPrimitive (line 44) | std::unique_ptr function BroadcastElementwiseAttrBinaryPrimitiveExists (line 62) | auto BroadcastElementwiseAttrBinaryPrimitiveExists() { class ScalarMathKernel (line 72) | class ScalarMathKernel final : public user_op::OpKernel, public user_o... method ScalarMathKernel (line 74) | ScalarMathKernel() = default; method Compute (line 78) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 108) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ScalarReverseMathKernel (line 112) | class ScalarReverseMathKernel final : public user_op::OpKernel { method ScalarReverseMathKernel (line 114) | ScalarReverseMathKernel() = default; method Compute (line 118) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 141) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class ScalarPowGradKernel (line 174) | class ScalarPowGradKernel final : public user_op::OpKernel { method ScalarPowGradKernel (line 176) | ScalarPowGradKernel() = default; method Compute (line 180) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 198) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/scaled_dot_product_attention_kernel.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/kernels/scaled_dot_product_attention_util.h function namespace (line 25) | namespace oneflow { FILE: oneflow/user/kernels/search_sorted_kernel.cpp type oneflow (line 20) | namespace oneflow { class CpuSearchSortedKernel (line 23) | class CpuSearchSortedKernel final : public user_op::OpKernel { method CpuSearchSortedKernel (line 25) | CpuSearchSortedKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 54) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class CpuSearchSortedScalarKernel (line 71) | class CpuSearchSortedScalarKernel final : public user_op::OpKernel { method CpuSearchSortedScalarKernel (line 73) | CpuSearchSortedScalarKernel() = default; method Compute (line 77) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 93) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/sigmoid_cross_entropy_kernel.cpp type oneflow (line 18) | namespace oneflow { type ElemwiseSigmoidCrossEntropyGradFunctor (line 22) | struct ElemwiseSigmoidCrossEntropyGradFunctor (line 32) | struct ElemwiseSigmoidCrossEntropyFunctor()(predict... FILE: oneflow/user/kernels/sigmoid_cross_entropy_kernel.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/slice_kernel.cpp type oneflow (line 26) | namespace oneflow { function GetSizeInSlice (line 33) | int64_t GetSizeInSlice(const int64_t start, const int64_t end, const i... class SliceContext (line 38) | class SliceContext final { type SplitInfo (line 40) | struct SplitInfo { method SliceContext (line 57) | SliceContext() : axis_bitset_(0) {} method PushSplitInfo (line 59) | Maybe PushSplitInfo(int64_t split_axis, int64_t lower, int64_t... method IsAxisPushed (line 73) | bool IsAxisPushed(int64_t split_axis) const { function ConstructSliceParamsLarge (line 84) | void ConstructSliceParamsLarge(const SliceContext& ctx, const std::vec... function ConstructSliceParamsSmall (line 128) | void ConstructSliceParamsSmall(const SliceContext& ctx, const std::vec... function SliceParams (line 163) | SliceParams ConstructSliceParams(user_op::KernelComputeContext* ctx, c... function WriteSlice (line 212) | void WriteSlice(user_op::KernelComputeContext* ctx, const user_op::Ten... class SliceKernel (line 285) | class SliceKernel final : public user_op::OpKernel, public user_op::Cu... method SliceKernel (line 287) | SliceKernel() = default; method InitOpKernelCache (line 290) | std::shared_ptr InitOpKernelCache( method Compute (line 328) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 337) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class SliceUpdateKernel (line 341) | class SliceUpdateKernel final : public user_op::OpKernel { method SliceUpdateKernel (line 343) | SliceUpdateKernel() = default; method InitOpKernelCache (line 346) | std::shared_ptr InitOpKernelCache( method Compute (line 384) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 402) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return true; } class SliceGradKernel (line 406) | class SliceGradKernel final : public user_op::OpKernel, public user_op... method SliceGradKernel (line 408) | SliceGradKernel() = default; method Compute (line 412) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 422) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/slice_util.cpp type oneflow (line 20) | namespace oneflow { function SliceParams (line 22) | SliceParams FoldContiguousFullSliceDimensions(const SliceParams& param... type SliceKernelUtil (line 45) | struct SliceKernelUtil { method Forward (line 46) | static void Forward(ep::Stream* stream, const SliceParams& params, c... method Forward (line 51) | static void Forward(ep::Stream* stream, const SliceParams& entire_pa... method Backward (line 57) | static void Backward(ep::Stream* stream, const SliceParams& params, ... method DoForward (line 64) | static void DoForward(ep::Stream* stream, const SliceParams& params,... method SteppedMultiThreadLoop (line 76) | static void SteppedMultiThreadLoop(size_t elem_cnt, size_t step, con... method DoForward (line 84) | static void DoForward(ep::Stream* stream, const SliceParams& entire_... method DoBackward (line 122) | static void DoBackward(ep::Stream* stream, const SliceParams& params... FILE: oneflow/user/kernels/slice_util.h function namespace (line 24) | namespace oneflow { function SliceOffsetToEntireOffset (line 82) | int64_t SliceOffsetToEntireOffset(int64_t offset, const SliceParams& par... FILE: oneflow/user/kernels/smooth_l1_loss_kernel.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 20) | namespace user_op { function ComputeSmoothL1Out (line 27) | void ComputeSmoothL1Out(int64_t elem_cnt, const T* input, const T* t... function ComputeSmoothL1GradOut (line 39) | void ComputeSmoothL1GradOut(int64_t elem_cnt, const T* input, const ... class SmoothL1LossKernel (line 55) | class SmoothL1LossKernel : public SimpleLossKernel (line 23) | struct CrossEntropyKernelUtil { method ComputeEntropy (line 24) | static void ComputeEntropy(ep::Stream* stream, const int64_t num_i... method ComputeDiffWithSoftmax (line 38) | static void ComputeDiffWithSoftmax(ep::Stream* stream, const int64... FILE: oneflow/user/kernels/softmax_cross_entropy_kernel.h function namespace (line 19) | namespace oneflow { FILE: oneflow/user/kernels/softmax_kernel.cpp type oneflow (line 21) | namespace oneflow { function NewSoftmaxPrimitive (line 26) | std::unique_ptr NewSoftmaxPrimitive(Context* c... function SoftmaxPrimitiveExists (line 31) | auto SoftmaxPrimitiveExists() { function NewSoftmaxBackwardPrimitive (line 38) | std::unique_ptr NewSoftmaxBackwardPrim... function SoftmaxBackwardPrimitiveExists (line 44) | auto SoftmaxBackwardPrimitiveExists() { class SoftmaxKernel (line 53) | class SoftmaxKernel final : public user_op::OpKernel, public user_op::... method SoftmaxKernel (line 55) | SoftmaxKernel() = default; method Compute (line 60) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 70) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class SoftmaxGradKernel (line 76) | class SoftmaxGradKernel final : public user_op::OpKernel, public user_... method SoftmaxGradKernel (line 78) | SoftmaxGradKernel() = default; method Compute (line 83) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 96) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/sort_kernel.cpp type oneflow (line 19) | namespace oneflow { class CpuSortKernel (line 22) | class CpuSortKernel final : public user_op::OpKernel { method CpuSortKernel (line 24) | CpuSortKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 50) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/sparse_cross_entropy_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 22) | namespace user_op { class SparseCrossEntropyOpKernelCache (line 26) | class SparseCrossEntropyOpKernelCache final : public user_op::OpKern... method SparseCrossEntropyOpKernelCache (line 28) | SparseCrossEntropyOpKernelCache(int64_t lower, int64_t upper) : lo... method lower (line 31) | int64_t lower() const { return lower_; } method upper (line 32) | int64_t upper() const { return upper_; } class SparseCrossEntropyKernel (line 42) | class SparseCrossEntropyKernel final : public user_op::OpKernel { method SparseCrossEntropyKernel (line 44) | SparseCrossEntropyKernel() = default; method Compute (line 48) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 61) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class SparseCrossEntropyMsKernel (line 65) | class SparseCrossEntropyMsKernel final : public user_op::OpKernel { method SparseCrossEntropyMsKernel (line 67) | SparseCrossEntropyMsKernel() = default; method InitOpKernelCache (line 70) | std::shared_ptr InitOpKernelCache( method Compute (line 88) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 110) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class SparseCrossEntropyGradKernel (line 142) | class SparseCrossEntropyGradKernel final : public user_op::OpKernel { method SparseCrossEntropyGradKernel (line 144) | SparseCrossEntropyGradKernel() = default; method Compute (line 148) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 166) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class SparseCrossEntropyMsGradKernel (line 170) | class SparseCrossEntropyMsGradKernel final : public user_op::OpKernel { method SparseCrossEntropyMsGradKernel (line 172) | SparseCrossEntropyMsGradKernel() = default; method InitOpKernelCache (line 175) | std::shared_ptr InitOpKernelCache( method Compute (line 193) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 218) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/sparse_cross_entropy_kernel_util.cpp type oneflow (line 19) | namespace oneflow { type user_op (line 20) | namespace user_op { type SparseCrossEntropyKernelUtil (line 23) | struct SparseCrossEntropyKernelUtil { method ComputeEntropy (line 24) | static void ComputeEntropy(ep::Stream* stream, const int64_t num_i... method ComputeDiff (line 35) | static void ComputeDiff(ep::Stream* stream, const int64_t num_inst... method ComputeDiffWithSoftmax (line 48) | static void ComputeDiffWithSoftmax(ep::Stream* stream, const int64... FILE: oneflow/user/kernels/sparse_cross_entropy_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/sparse_softmax_cross_entropy_kernel.cpp type oneflow (line 24) | namespace oneflow { type user_op (line 25) | namespace user_op { function NewLogSoftmaxPrimitive (line 30) | std::unique_ptr NewLogSoftmaxPrimitive(Co... function LogSoftmaxPrimitiveExists (line 36) | auto LogSoftmaxPrimitiveExists() { class SparseSoftmaxCrossEntropyOpKernelCache (line 42) | class SparseSoftmaxCrossEntropyOpKernelCache final : public user_op:... method SparseSoftmaxCrossEntropyOpKernelCache (line 44) | SparseSoftmaxCrossEntropyOpKernelCache(int64_t lower, int64_t upper) method lower (line 48) | int64_t lower() const { return lower_; } method upper (line 49) | int64_t upper() const { return upper_; } class SparseSoftmaxCrossEntropyKernel (line 59) | class SparseSoftmaxCrossEntropyKernel final : public user_op::OpKernel, method SparseSoftmaxCrossEntropyKernel (line 62) | SparseSoftmaxCrossEntropyKernel() = default; method Compute (line 66) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 93) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class SparseSoftmaxCrossEntropyMsKernel (line 97) | class SparseSoftmaxCrossEntropyMsKernel final : public user_op::OpKe... method SparseSoftmaxCrossEntropyMsKernel (line 99) | SparseSoftmaxCrossEntropyMsKernel() = default; method Compute (line 103) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 106) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class SparseSoftmaxCrossEntropyGradKernel (line 138) | class SparseSoftmaxCrossEntropyGradKernel final : public user_op::Op... method SparseSoftmaxCrossEntropyGradKernel (line 141) | SparseSoftmaxCrossEntropyGradKernel() = default; method Compute (line 145) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 159) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class SparseSoftmaxCrossEntropyMsGradKernel (line 163) | class SparseSoftmaxCrossEntropyMsGradKernel final : public user_op::... method SparseSoftmaxCrossEntropyMsGradKernel (line 165) | SparseSoftmaxCrossEntropyMsGradKernel() = default; method InitOpKernelCache (line 167) | std::shared_ptr InitOpKernelCache( method Compute (line 184) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 205) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/sparse_softmax_cross_entropy_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 19) | namespace user_op { type SparseSoftmaxCrossEntropyKernelUtil (line 22) | struct SparseSoftmaxCrossEntropyKernelUtil { method ComputeDiff (line 23) | static void ComputeDiff(ep::Stream* stream, const int64_t num_inst... FILE: oneflow/user/kernels/sparse_softmax_cross_entropy_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/split_like_kernel.cpp type oneflow (line 19) | namespace oneflow { function NewCopyNdPrimitive (line 24) | std::unique_ptr NewCopyNdPrimitive(Context* ctx) { class SplitLikeKernel (line 28) | class SplitLikeKernel final : public user_op::OpKernel { method SplitLikeKernel (line 30) | SplitLikeKernel() = default; method InferShape (line 34) | void InferShape(user_op::KernelInferContext* ctx) const override { method Compute (line 65) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 95) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function CopyNdPrimitiveExists (line 98) | auto CopyNdPrimitiveExists() { FILE: oneflow/user/kernels/sqrt_square_sum_kernel.cpp type oneflow (line 22) | namespace oneflow { type user_op (line 24) | namespace user_op { function getThreadNumBlocks (line 26) | int64_t getThreadNumBlocks(int64_t n) { class SqrtSquareSumKernel (line 35) | class SqrtSquareSumKernel final : public user_op::OpKernel, public u... method SqrtSquareSumKernel (line 37) | SqrtSquareSumKernel() = default; method Compute (line 41) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 50) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/sqrt_square_sum_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type SqrtSquareSumKernelUtil (line 21) | struct SqrtSquareSumKernelUtil { method SqrtSquareSum (line 22) | static void SqrtSquareSum(ep::Stream* stream, int64_t n, const T* x,... FILE: oneflow/user/kernels/sqrt_square_sum_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/square_sum_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { class SquareSumKernel (line 26) | class SquareSumKernel final : public user_op::OpKernel, public user_... method SquareSumKernel (line 28) | SquareSumKernel() = default; method Compute (line 32) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 39) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class MultiSquareSumKernel (line 52) | class MultiSquareSumKernel final : public user_op::OpKernel, public ... method MultiSquareSumKernel (line 54) | MultiSquareSumKernel() = default; method Compute (line 58) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 69) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/square_sum_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type SquareSumKernelUtil (line 21) | struct SquareSumKernelUtil { method SquareSum (line 22) | static void SquareSum(ep::Stream* stream, int64_t n, const T* x, T* ... method MultiSquareSum (line 28) | static void MultiSquareSum(ep::Stream* stream, const std::vector NewCopyNdPrimitive(Context* ctx) { class StackKernel (line 28) | class StackKernel final : public user_op::OpKernel { method StackKernel (line 30) | StackKernel() = default; method InferShape (line 34) | void InferShape(user_op::KernelInferContext* ctx) const override { method Compute (line 72) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 105) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function CopyNdPrimitiveExists (line 108) | auto CopyNdPrimitiveExists() { class StackGradKernel (line 120) | class StackGradKernel final : public user_op::OpKernel { method StackGradKernel (line 122) | StackGradKernel() = default; method InferShape (line 126) | void InferShape(user_op::KernelInferContext* ctx) const override { method Compute (line 172) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 203) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/stateful_opkernel.cpp type oneflow (line 31) | namespace oneflow { type one (line 32) | namespace one { class GlobalTensorInferResult (line 34) | class GlobalTensorInferResult class ZeroCopyBaseContextHelper (line 41) | class ZeroCopyBaseContextHelper { method ZeroCopyBaseContextHelper (line 43) | ZeroCopyBaseContextHelper(const std::shared_ptr& i... method GlobalTensorMeta (line 75) | const GlobalTensorMeta* GlobalTensorMeta4ArgNameAndIndex(eager::Ca... method parallel_desc (line 84) | Optional> parallel_desc(eager::CallContext* c... method ParallelContext (line 97) | const ParallelContext& parallel_ctx(eager::CallContext* call_ctx) ... method ArgVec (line 108) | const ArgVec& inputs() const { return input_arg_tuple_->indexed_ar... method ArgVec (line 109) | const ArgVec& outputs() const { return output_arg_tuple_->indexed_... method TryGetTensorTupleIndex (line 112) | static int32_t TryGetTensorTupleIndex(const std::unordered_mapinputs(); } method ArgVec (line 465) | const ArgVec& outputs() const override { return helper_->outputs(); } method JobDesc (line 466) | const JobDesc* job_desc() const override { return helper_->job_des... method ParallelContext (line 467) | const ParallelContext& parallel_ctx() const override { return help... method ParallelDesc (line 468) | const ParallelDesc& parallel_desc() const override { return helper... method SbpParallel (line 469) | const SbpParallel& SbpParallel4ArgNameAndIndex(const std::string& ... method NdSbp (line 473) | const NdSbp& NdSbp4ArgNameAndIndex(const std::string& arg_name, in... method parallel_num (line 477) | int64_t parallel_num() const override { return helper_->parallel_n... method has_input (line 485) | bool has_input(const std::string& arg_name, int32_t index) const o... method has_output (line 488) | bool has_output(const std::string& arg_name, int32_t index) const ... method input_size (line 491) | int32_t input_size(const std::string& arg_name) const override { method output_size (line 494) | int32_t output_size(const std::string& arg_name) const override { class UserKernelComputeContextHelper (line 511) | class UserKernelComputeContextHelper final { method UserKernelComputeContextHelper (line 513) | UserKernelComputeContextHelper(DeviceType device_type, method DeviceType (line 533) | DeviceType device_type() const { return base_ctx_helper_.device_ty... method ParallelContext (line 534) | const ParallelContext& parallel_ctx(eager::CallContext* call_ctx) ... method ArgVec (line 538) | const ArgVec& inputs() const { return base_ctx_helper_.inputs(); } method ArgVec (line 539) | const ArgVec& outputs() const { return base_ctx_helper_.outputs(); } class UserKernelComputeContext (line 552) | class UserKernelComputeContext final : public user_op::KernelCompute... method UserKernelComputeContext (line 554) | UserKernelComputeContext(const UserKernelComputeContextHelper* hel... method DeviceType (line 574) | DeviceType device_type() const override { return helper_->device_t... method ParallelContext (line 576) | const ParallelContext& parallel_ctx() const override { return help... method ArgVec (line 578) | const ArgVec& inputs() const override { return helper_->inputs(); } method ArgVec (line 579) | const ArgVec& outputs() const override { return helper_->outputs(); } class UserKernelRegContextHelper (line 596) | class UserKernelRegContextHelper final { method UserKernelRegContextHelper (line 598) | UserKernelRegContextHelper(DeviceType device_type, const user_op::... method DeviceType (line 605) | DeviceType device_type() const { return base_ctx_helper_.device_ty... method ParallelContext (line 606) | const ParallelContext& parallel_ctx(eager::CallContext* call_ctx) ... method ArgVec (line 614) | const ArgVec& inputs() const { return base_ctx_helper_.inputs(); } method ArgVec (line 615) | const ArgVec& outputs() const { return base_ctx_helper_.outputs(); } class UserKernelRegContext (line 629) | class UserKernelRegContext final : public user_op::KernelRegContext { method UserKernelRegContext (line 631) | UserKernelRegContext(const UserKernelRegContextHelper* helper, eag... method DeviceType (line 635) | DeviceType device_type() const override { return helper_->device_t... method ParallelContext (line 636) | const ParallelContext& parallel_ctx() const override { return help... method ArgVec (line 641) | const ArgVec& inputs() const override { return helper_->inputs(); } method ArgVec (line 642) | const ArgVec& outputs() const override { return helper_->outputs(); } class UserKernelInitAndCacheContextHelper (line 658) | class UserKernelInitAndCacheContextHelper final { method UserKernelInitAndCacheContextHelper (line 660) | UserKernelInitAndCacheContextHelper(DeviceType device_type, method DeviceType (line 669) | DeviceType device_type() const { return base_ctx_helper_.device_ty... method ParallelContext (line 670) | const ParallelContext& parallel_ctx(eager::CallContext* call_ctx) ... method SbpParallel (line 683) | const SbpParallel& SbpParallel4ArgNameAndIndex(eager::CallContext*... method NdSbp (line 690) | const NdSbp& NdSbp4ArgNameAndIndex(eager::CallContext* call_ctx, c... method ArgVec (line 697) | const ArgVec& inputs() const { return base_ctx_helper_.inputs(); } method ArgVec (line 698) | const ArgVec& outputs() const { return base_ctx_helper_.outputs(); } method ParallelDesc (line 699) | const ParallelDesc& parallel_desc(eager::CallContext* call_ctx) co... class UserKernelInitAndCacheContext (line 715) | class UserKernelInitAndCacheContext final : public user_op::KernelIn... method UserKernelInitAndCacheContext (line 718) | UserKernelInitAndCacheContext(const UserKernelInitAndCacheContextH... method DeviceType (line 729) | DeviceType device_type() const override { return helper_->device_t... method ParallelContext (line 730) | const ParallelContext& parallel_ctx() const override { return help... method SbpParallel (line 739) | const SbpParallel& SbpParallel4ArgNameAndIndex(const std::string& ... method NdSbp (line 744) | const NdSbp& NdSbp4ArgNameAndIndex(const std::string& arg_name, in... method ArgVec (line 748) | const ArgVec& inputs() const override { return helper_->inputs(); } method ArgVec (line 749) | const ArgVec& outputs() const override { return helper_->outputs(); } method ParallelDesc (line 750) | const ParallelDesc& parallel_desc() const override { return helper... function InitTensorTupleIndexes4Bns (line 769) | Maybe InitTensorTupleIndexes4Bns(const std::shared_ptr NewTensorFillPrimitive(Co... class TensorConstantKernel (line 32) | class TensorConstantKernel final : public OpKernel { method TensorConstantKernel (line 34) | TensorConstantKernel() = default; method Compute (line 38) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 51) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function TensorFillPrimitiveExists (line 54) | auto TensorFillPrimitiveExists() { FILE: oneflow/user/kernels/tf_pool_cpu_kernel.cpp type oneflow (line 21) | namespace oneflow { type PoolOpKernelCache (line 25) | struct PoolOpKernelCache final : public user_op::OpKernelCache { method PoolOpKernelCache (line 27) | explicit PoolOpKernelCache(const Params3D& params_3d) : params_3d(pa... method Params3D (line 28) | const Params3D& GetParams3D() const { return params_3d; } function InitPoolOpKernelCache (line 31) | std::shared_ptr InitPoolOpKernelCache(user_op::Kern... type PoolCpuKernelUtil (line 48) | struct PoolCpuKernelUtil { method CFirstForward (line 66) | static void CFirstForward(const Params3D& params_3d, const user_op::... method CFirstBackward (line 113) | static void CFirstBackward(const Params3D& params_3d, const user_op:... method CLastForward (line 165) | static void CLastForward(const Params3D& params_3d, const user_op::T... method CLastBackward (line 206) | static void CLastBackward(const Params3D& params_3d, const user_op::... method AvgFWCompute (line 252) | static void AvgFWCompute(user_op::KernelComputeContext* ctx, method AvgBWCompute (line 275) | static void AvgBWCompute(user_op::KernelComputeContext* ctx, method MaxFWCompute (line 299) | static void MaxFWCompute(user_op::KernelComputeContext* ctx, method MaxBWCompute (line 325) | static void MaxBWCompute(user_op::KernelComputeContext* ctx, class AvgPool1DCpuKernel (line 358) | class AvgPool1DCpuKernel final : public user_op::OpKernel { method AvgPool1DCpuKernel (line 360) | AvgPool1DCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 363) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 364) | std::shared_ptr InitOpKernelCache( method Compute (line 370) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool1DGradCpuKernel (line 377) | class AvgPool1DGradCpuKernel final : public user_op::OpKernel { method AvgPool1DGradCpuKernel (line 379) | AvgPool1DGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 382) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 383) | std::shared_ptr InitOpKernelCache( method Compute (line 389) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool2DCpuKernel (line 396) | class AvgPool2DCpuKernel final : public user_op::OpKernel { method AvgPool2DCpuKernel (line 398) | AvgPool2DCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 401) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 402) | std::shared_ptr InitOpKernelCache( method Compute (line 408) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool2DGradCpuKernel (line 415) | class AvgPool2DGradCpuKernel final : public user_op::OpKernel { method AvgPool2DGradCpuKernel (line 417) | AvgPool2DGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 420) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 421) | std::shared_ptr InitOpKernelCache( method Compute (line 427) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool3DCpuKernel (line 434) | class AvgPool3DCpuKernel final : public user_op::OpKernel { method AvgPool3DCpuKernel (line 436) | AvgPool3DCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 439) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 440) | std::shared_ptr InitOpKernelCache( method Compute (line 446) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool3DGradCpuKernel (line 453) | class AvgPool3DGradCpuKernel final : public user_op::OpKernel { method AvgPool3DGradCpuKernel (line 455) | AvgPool3DGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 458) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 459) | std::shared_ptr InitOpKernelCache( method Compute (line 465) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool1DCpuKernel (line 472) | class MaxPool1DCpuKernel final : public user_op::OpKernel { method MaxPool1DCpuKernel (line 474) | MaxPool1DCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 477) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 478) | std::shared_ptr InitOpKernelCache( method Compute (line 484) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool1DGradCpuKernel (line 491) | class MaxPool1DGradCpuKernel final : public user_op::OpKernel { method MaxPool1DGradCpuKernel (line 493) | MaxPool1DGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 496) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 497) | std::shared_ptr InitOpKernelCache( method Compute (line 503) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool2DCpuKernel (line 510) | class MaxPool2DCpuKernel final : public user_op::OpKernel { method MaxPool2DCpuKernel (line 512) | MaxPool2DCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 515) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 516) | std::shared_ptr InitOpKernelCache( method Compute (line 522) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool2DGradCpuKernel (line 529) | class MaxPool2DGradCpuKernel final : public user_op::OpKernel { method MaxPool2DGradCpuKernel (line 531) | MaxPool2DGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 534) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 535) | std::shared_ptr InitOpKernelCache( method Compute (line 541) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool3DCpuKernel (line 548) | class MaxPool3DCpuKernel final : public user_op::OpKernel { method MaxPool3DCpuKernel (line 550) | MaxPool3DCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 553) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 554) | std::shared_ptr InitOpKernelCache( method Compute (line 560) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool3DGradCpuKernel (line 567) | class MaxPool3DGradCpuKernel final : public user_op::OpKernel { method MaxPool3DGradCpuKernel (line 569) | MaxPool3DGradCpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 572) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 573) | std::shared_ptr InitOpKernelCache( method Compute (line 579) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/tf_pool_gpu_kernel.cpp type oneflow (line 24) | namespace oneflow { class CudnnPoolDesc (line 28) | class CudnnPoolDesc final { method CudnnPoolDesc (line 31) | CudnnPoolDesc(cudnnPoolingMode_t pooling_mode, int dims, const int* ... method cudnnPoolingDescriptor_t (line 40) | const cudnnPoolingDescriptor_t& Get() const { return val_; } class GPUPoolOpKernelCache (line 46) | class GPUPoolOpKernelCache final : public user_op::OpKernelCache { method GPUPoolOpKernelCache (line 48) | GPUPoolOpKernelCache(const int32_t dim, const std::string& pooling_t... method Reset (line 56) | void Reset(const int32_t dim, const std::string& pooling_type, const... method FromKernelComputeContext (line 83) | static std::shared_ptr FromKernelComputeContext( method cudnnTensorDescriptor_t (line 102) | const cudnnTensorDescriptor_t& cudnn_x_tensor_desc() const { return ... method cudnnTensorDescriptor_t (line 103) | const cudnnTensorDescriptor_t& cudnn_y_tensor_desc() const { return ... method cudnnPoolingDescriptor_t (line 104) | const cudnnPoolingDescriptor_t& cudnn_pooling_desc() const { return ... type PoolGpuKernelUtil (line 113) | struct PoolGpuKernelUtil { method FWCompute (line 114) | static void FWCompute(user_op::KernelComputeContext* ctx, method BWCompute (line 126) | static void BWCompute(user_op::KernelComputeContext* ctx, class AvgPool1DGpuKernel (line 145) | class AvgPool1DGpuKernel final : public user_op::OpKernel, public user... method AvgPool1DGpuKernel (line 147) | AvgPool1DGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 150) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 151) | std::shared_ptr InitOpKernelCache( method Compute (line 157) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool1DGradGpuKernel (line 163) | class AvgPool1DGradGpuKernel final : public user_op::OpKernel, public ... method AvgPool1DGradGpuKernel (line 165) | AvgPool1DGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 168) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 169) | std::shared_ptr InitOpKernelCache( method Compute (line 173) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool2DGpuKernel (line 179) | class AvgPool2DGpuKernel final : public user_op::OpKernel, public user... method AvgPool2DGpuKernel (line 181) | AvgPool2DGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 184) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 185) | std::shared_ptr InitOpKernelCache( method Compute (line 191) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool2DGradGpuKernel (line 197) | class AvgPool2DGradGpuKernel final : public user_op::OpKernel, public ... method AvgPool2DGradGpuKernel (line 199) | AvgPool2DGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 202) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 203) | std::shared_ptr InitOpKernelCache( method Compute (line 209) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool3DGpuKernel (line 215) | class AvgPool3DGpuKernel final : public user_op::OpKernel, public user... method AvgPool3DGpuKernel (line 217) | AvgPool3DGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 220) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 221) | std::shared_ptr InitOpKernelCache( method Compute (line 227) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class AvgPool3DGradGpuKernel (line 233) | class AvgPool3DGradGpuKernel final : public user_op::OpKernel, public ... method AvgPool3DGradGpuKernel (line 235) | AvgPool3DGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 238) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 239) | std::shared_ptr InitOpKernelCache( method Compute (line 245) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool1DGpuKernel (line 251) | class MaxPool1DGpuKernel final : public user_op::OpKernel, public user... method MaxPool1DGpuKernel (line 253) | MaxPool1DGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 256) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 257) | std::shared_ptr InitOpKernelCache( method Compute (line 263) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool1DGradGpuKernel (line 269) | class MaxPool1DGradGpuKernel final : public user_op::OpKernel, public ... method MaxPool1DGradGpuKernel (line 271) | MaxPool1DGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 274) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 275) | std::shared_ptr InitOpKernelCache( method Compute (line 281) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool2DGpuKernel (line 287) | class MaxPool2DGpuKernel final : public user_op::OpKernel, public user... method MaxPool2DGpuKernel (line 289) | MaxPool2DGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 292) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 293) | std::shared_ptr InitOpKernelCache( method Compute (line 299) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool2DGradGpuKernel (line 305) | class MaxPool2DGradGpuKernel final : public user_op::OpKernel, public ... method MaxPool2DGradGpuKernel (line 307) | MaxPool2DGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 310) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 311) | std::shared_ptr InitOpKernelCache( method Compute (line 317) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool3DGpuKernel (line 323) | class MaxPool3DGpuKernel final : public user_op::OpKernel, public user... method MaxPool3DGpuKernel (line 325) | MaxPool3DGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 328) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 329) | std::shared_ptr InitOpKernelCache( method Compute (line 335) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... class MaxPool3DGradGpuKernel (line 341) | class MaxPool3DGradGpuKernel final : public user_op::OpKernel, public ... method MaxPool3DGradGpuKernel (line 343) | MaxPool3DGradGpuKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 346) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method InitOpKernelCache (line 347) | std::shared_ptr InitOpKernelCache( method Compute (line 353) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... FILE: oneflow/user/kernels/tf_prelu_kernel.cpp type oneflow (line 19) | namespace oneflow { class TfCpuPReluKernel (line 22) | class TfCpuPReluKernel final : public user_op::OpKernel { method TfCpuPReluKernel (line 24) | TfCpuPReluKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 46) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class TfCpuPReluGradKernel (line 63) | class TfCpuPReluGradKernel final : public user_op::OpKernel { method TfCpuPReluGradKernel (line 65) | TfCpuPReluGradKernel() = default; method Compute (line 69) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 99) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/throw_error_kernel.cpp type oneflow (line 18) | namespace oneflow { class ThrowErrorKernel (line 22) | class ThrowErrorKernel final : public user_op::OpKernel { method ThrowErrorKernel (line 24) | ThrowErrorKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 31) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/to_contiguous_kernel.cpp type oneflow (line 23) | namespace oneflow { type ToContiguousUtil (line 26) | struct ToContiguousUtil : ToContiguousUtilBase { class ToContiguousKernel (line 78) | class ToContiguousKernel final : public user_op::OpKernel { method ToContiguousKernel (line 80) | ToContiguousKernel() = default; method Compute (line 84) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 100) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/to_contiguous_kernel.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/top_k_kernel.cpp type oneflow (line 20) | namespace oneflow { function ComputeTopOne (line 25) | void ComputeTopOne(const T* in_ptr, const Range& range, int64_t instan... function ComputeTopK (line 33) | void ComputeTopK(const T* in_ptr, int64_t* indices_ptr, const Range& r... function CpuTopK (line 56) | void CpuTopK(ep::Stream* /*stream*/, const T* in_ptr, int64_t* indices... class TopKCpuKernel (line 79) | class TopKCpuKernel final : public user_op::OpKernel { method TopKCpuKernel (line 81) | TopKCpuKernel() = default; method Compute (line 85) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 98) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/transpose_kernel.cpp type oneflow (line 22) | namespace oneflow { type user_op (line 24) | namespace user_op { function IsIdentity (line 28) | bool IsIdentity(const ShapeView& in_shape, const std::vector NewPermutePrimitive(Context*... class TransposeKernel (line 53) | class TransposeKernel final : public OpKernel, public user_op::CudaG... method TransposeKernel (line 56) | TransposeKernel() = default; method Compute (line 60) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 90) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function PermutePrimitiveExists (line 93) | auto PermutePrimitiveExists() { FILE: oneflow/user/kernels/tril_kernel.cpp type oneflow (line 20) | namespace oneflow { class CpuTrilKernel (line 23) | class CpuTrilKernel final : public user_op::OpKernel { method CpuTrilKernel (line 25) | CpuTrilKernel() = default; method Compute (line 29) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 49) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/triu_kernel.cpp type oneflow (line 19) | namespace oneflow { class CpuTriuKernel (line 22) | class CpuTriuKernel final : public user_op::OpKernel { method CpuTriuKernel (line 24) | CpuTriuKernel() = default; method Compute (line 28) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 45) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/tuple_identity_kernel.cpp type oneflow (line 19) | namespace oneflow { class TupleIdentityKernel (line 24) | class TupleIdentityKernel final : public user_op::OpKernel { method TupleIdentityKernel (line 26) | TupleIdentityKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 44) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/two_stage_reduce_kernel.cpp type oneflow (line 23) | namespace oneflow { type user_op (line 25) | namespace user_op { class ReduceDeviceStageKernel (line 28) | class ReduceDeviceStageKernel final : public OpKernel { method ReduceDeviceStageKernel (line 30) | ReduceDeviceStageKernel() = default; method Compute (line 34) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 69) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function GenDeviceStageInferTmpSizeFn (line 73) | user_op::InferTmpSizeFn GenDeviceStageInferTmpSizeFn() { class ReduceDeviceStageGradKernel (line 98) | class ReduceDeviceStageGradKernel final : public OpKernel { method ReduceDeviceStageGradKernel (line 100) | ReduceDeviceStageGradKernel() = default; method Compute (line 104) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 127) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function GenDeviceStageGradInferTmpSizeFn (line 131) | user_op::InferTmpSizeFn GenDeviceStageGradInferTmpSizeFn() { class ReduceGlobalStageKernel (line 156) | class ReduceGlobalStageKernel final : public OpKernel { method ReduceGlobalStageKernel (line 158) | ReduceGlobalStageKernel() = default; method Compute (line 162) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 182) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... class ReduceGlobalStageGradKernel (line 203) | class ReduceGlobalStageGradKernel final : public OpKernel { method ReduceGlobalStageGradKernel (line 205) | ReduceGlobalStageGradKernel() = default; method Compute (line 209) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 261) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function GenGlobalStageGradInferTmpSizeFn (line 265) | user_op::InferTmpSizeFn GenGlobalStageGradInferTmpSizeFn() { FILE: oneflow/user/kernels/two_stage_reduce_kernel_util.cpp type oneflow (line 19) | namespace oneflow { type TwoStageReduceKernelUtil (line 22) | struct TwoStageReduceKernelUtil { method Divide (line 23) | static void Divide(ep::Stream* stream, const int64_t n, const T* x, ... method Mask (line 27) | static void Mask(ep::Stream* stream, const int64_t n, const T* x, co... method Scale (line 31) | static void Scale(ep::Stream* stream, const int64_t n, const T* x, c... FILE: oneflow/user/kernels/two_stage_reduce_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/unfold_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 23) | namespace user_op { class UnfoldOpKernelState (line 30) | class UnfoldOpKernelState : public OpKernelState { method UnfoldOpKernelState (line 33) | UnfoldOpKernelState(const ShapeView& input_shape, const std::vecto... method ParamType (line 39) | const ParamType& params() const { return params_; } function CreateUnfoldOpKernelState (line 46) | std::shared_ptr> CreateUnfo... class UnfoldKernel (line 57) | class UnfoldKernel final : public OpKernel { method UnfoldKernel (line 59) | UnfoldKernel() = default; method Compute (line 63) | void Compute(KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 79) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... FILE: oneflow/user/kernels/unfold_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 20) | namespace user_op { type UnfoldKernelUtil (line 25) | struct UnfoldKernelUtil { method Forward (line 27) | static void Forward(ep::Stream* stream, const UnfoldParams (line 21) | struct UniqueKernelUtil { method Unique (line 22) | static void Unique(ep::Stream* stream, int64_t n, const KEY* in, IDX... method UniqueWithCounts (line 28) | static void UniqueWithCounts(ep::Stream* stream, int64_t n, const KE... method GetUniqueWorkspaceSizeInBytes (line 57) | static void GetUniqueWorkspaceSizeInBytes(ep::Stream* stream, int64_... method GetUniqueWithCountsWorkspaceSizeInBytes (line 61) | static void GetUniqueWithCountsWorkspaceSizeInBytes(ep::Stream* stre... FILE: oneflow/user/kernels/unique_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/unique_with_counts_kernel.cpp type oneflow (line 19) | namespace oneflow { class UniqueWithCountsKernel (line 24) | class UniqueWithCountsKernel final : public user_op::OpKernel { method UniqueWithCountsKernel (line 26) | UniqueWithCountsKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 44) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... function GenInferTmpSizeFn (line 48) | user_op::InferTmpSizeFn GenInferTmpSizeFn() { FILE: oneflow/user/kernels/unpack_kernel.cpp type oneflow (line 20) | namespace oneflow { class UnpackKernel (line 25) | class UnpackKernel final : public user_op::OpKernel { method UnpackKernel (line 27) | UnpackKernel() = default; method CreateOpKernelState (line 30) | std::shared_ptr CreateOpKernelState( method Compute (line 37) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernelSt... method AlwaysComputeWhenAllOutputsEmpty (line 57) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/unsorted_batch_segment_sum_kernel.cpp type oneflow (line 20) | namespace oneflow { type user_op (line 22) | namespace user_op { function Shape (line 26) | Shape GetFlatShape(const ShapeView& shape, const int64_t axis) { class UnsortedBatchSegmentSumKernel (line 36) | class UnsortedBatchSegmentSumKernel final : public user_op::OpKernel, method UnsortedBatchSegmentSumKernel (line 39) | UnsortedBatchSegmentSumKernel() = default; method Compute (line 43) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 56) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return tr... FILE: oneflow/user/kernels/unsorted_segment_sum_kernel.cpp type oneflow (line 25) | namespace oneflow { type user_op (line 27) | namespace user_op { function CheckNdSbp (line 31) | void CheckNdSbp(const Shape& hierarchy, int64_t sum_axis, const NdSb... class UnsortedSegmentSumOpKernelCache (line 46) | class UnsortedSegmentSumOpKernelCache final : public user_op::OpKern... method UnsortedSegmentSumOpKernelCache (line 48) | UnsortedSegmentSumOpKernelCache(int64_t lower, int64_t upper) : lo... method lower (line 51) | int64_t lower() const { return lower_; } method upper (line 52) | int64_t upper() const { return upper_; } function CreateUnsortedSegmentSumOpKernelCache (line 59) | std::shared_ptr CreateUnsortedSegmentSumOpKe... class UnsortedSegmentSumKernel (line 80) | class UnsortedSegmentSumKernel final : public user_op::OpKernel, pub... method UnsortedSegmentSumKernel (line 82) | UnsortedSegmentSumKernel() = default; method InitOpKernelCache (line 85) | std::shared_ptr InitOpKernelCache( method Compute (line 91) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 118) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return tr... class UnsortedSegmentSumHalfKernel (line 146) | class UnsortedSegmentSumHalfKernel final : public user_op::OpKernel { method UnsortedSegmentSumHalfKernel (line 148) | UnsortedSegmentSumHalfKernel() = default; method InitOpKernelCache (line 151) | std::shared_ptr InitOpKernelCache( method Compute (line 157) | void Compute(user_op::KernelComputeContext* ctx, user_op::OpKernel... method AlwaysComputeWhenAllOutputsEmpty (line 188) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return tr... FILE: oneflow/user/kernels/unsorted_segment_sum_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type UnsortedSegmentSumKernelUtil (line 21) | struct UnsortedSegmentSumKernelUtil final { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, outer_idx, 0, outer_dim_size) { FILE: oneflow/user/kernels/unsorted_segment_sum_kernel_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/kernels/upsample_bicubic_2d_kernel.cpp type oneflow (line 21) | namespace oneflow { class UpsampleBicubic2dCPUKernel (line 24) | class UpsampleBicubic2dCPUKernel final : public user_op::OpKernel { method UpsampleBicubic2dCPUKernel (line 26) | UpsampleBicubic2dCPUKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 100) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleBicubic2dGradCPUKernel (line 104) | class UpsampleBicubic2dGradCPUKernel final : public user_op::OpKernel { method UpsampleBicubic2dGradCPUKernel (line 106) | UpsampleBicubic2dGradCPUKernel() = default; method Compute (line 110) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 177) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/upsample_bilinear_2d_kernel.cpp type oneflow (line 21) | namespace oneflow { function UpsampleBilinear2DForward (line 26) | static void UpsampleBilinear2DForward(const int64_t elem_cnt, const T*... function UpsampleBilinearBackward (line 50) | static void UpsampleBilinearBackward(const int64_t elem_cnt, const T* ... class UpsampleBilinear2DCPUKernel (line 78) | class UpsampleBilinear2DCPUKernel final : public user_op::OpKernel { method UpsampleBilinear2DCPUKernel (line 80) | UpsampleBilinear2DCPUKernel() = default; method Compute (line 84) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 122) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleBilinear2DGradCPUKernel (line 126) | class UpsampleBilinear2DGradCPUKernel final : public user_op::OpKernel { method UpsampleBilinear2DGradCPUKernel (line 128) | UpsampleBilinear2DGradCPUKernel() = default; method Compute (line 132) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 171) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/upsample_kernel.h function OF_DEVICE_FUNC (line 19) | OF_DEVICE_FUNC double GetLinearInputIndex(const int64_t out_dim_idx, con... function OF_DEVICE_FUNC (line 29) | OF_DEVICE_FUNC static int64_t GetNearestInputIndex(const int64_t out_dim... function OF_DEVICE_FUNC (line 36) | OF_DEVICE_FUNC double GetAreaPixelScale(const int64_t input_size, const ... FILE: oneflow/user/kernels/upsample_linear_1d_kernel.cpp type oneflow (line 21) | namespace oneflow { function UpsampleLinear1DForward (line 26) | static void UpsampleLinear1DForward(const int64_t elem_cnt, const T* i... function UpsampleLinear1DBackward (line 44) | static void UpsampleLinear1DBackward(const int64_t elem_cnt, const T* ... class UpsampleLinear1DCPUKernel (line 65) | class UpsampleLinear1DCPUKernel final : public user_op::OpKernel { method UpsampleLinear1DCPUKernel (line 67) | UpsampleLinear1DCPUKernel() = default; method Compute (line 71) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 99) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleLinearGrad1DCPUKernel (line 103) | class UpsampleLinearGrad1DCPUKernel final : public user_op::OpKernel { method UpsampleLinearGrad1DCPUKernel (line 105) | UpsampleLinearGrad1DCPUKernel() = default; method Compute (line 109) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 143) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/upsample_nearest_kernel.cpp type oneflow (line 21) | namespace oneflow { function UpsampleNearest1DForward (line 26) | static void UpsampleNearest1DForward(const int64_t elem_cnt, const T* ... function UpsampleNearest1DBackward (line 40) | static void UpsampleNearest1DBackward(const int64_t elem_cnt, const T*... function UpsampleNearest2DForward (line 54) | static void UpsampleNearest2DForward(const int64_t elem_cnt, const T* ... function UpsampleNearest2DBackward (line 69) | static void UpsampleNearest2DBackward(const int64_t elem_cnt, const T*... function UpsampleNearest3DForward (line 84) | static void UpsampleNearest3DForward(const int64_t elem_cnt, const T* ... function UpsampleNearest3DBackward (line 101) | static void UpsampleNearest3DBackward(const int64_t elem_cnt, const T*... class UpsampleNearest1DCPUKernel (line 120) | class UpsampleNearest1DCPUKernel final : public user_op::OpKernel { method UpsampleNearest1DCPUKernel (line 122) | UpsampleNearest1DCPUKernel() = default; method Compute (line 126) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 153) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleNearestGrad1DCPUKernel (line 157) | class UpsampleNearestGrad1DCPUKernel final : public user_op::OpKernel { method UpsampleNearestGrad1DCPUKernel (line 159) | UpsampleNearestGrad1DCPUKernel() = default; method Compute (line 163) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 194) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleNearest2DCPUKernel (line 211) | class UpsampleNearest2DCPUKernel final : public user_op::OpKernel { method UpsampleNearest2DCPUKernel (line 213) | UpsampleNearest2DCPUKernel() = default; method Compute (line 217) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 250) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleNearest2DGradCPUKernel (line 254) | class UpsampleNearest2DGradCPUKernel final : public user_op::OpKernel { method UpsampleNearest2DGradCPUKernel (line 256) | UpsampleNearest2DGradCPUKernel() = default; method Compute (line 260) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 296) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleNearest3DCPUKernel (line 313) | class UpsampleNearest3DCPUKernel final : public user_op::OpKernel { method UpsampleNearest3DCPUKernel (line 315) | UpsampleNearest3DCPUKernel() = default; method Compute (line 319) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 349) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleNearestGrad3DCPUKernel (line 353) | class UpsampleNearestGrad3DCPUKernel final : public user_op::OpKernel { method UpsampleNearestGrad3DCPUKernel (line 355) | UpsampleNearestGrad3DCPUKernel() = default; method Compute (line 359) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 392) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/upsample_trilinear_3d_kernel.cpp type oneflow (line 21) | namespace oneflow { function UpsampleTrilinear3DForward (line 26) | static void UpsampleTrilinear3DForward(const int64_t elem_cnt, const T... function UpsampleTrilinear3DBackward (line 72) | static void UpsampleTrilinear3DBackward(const int64_t elem_cnt, const ... class UpsampleTrilinear3DCPUKernel (line 118) | class UpsampleTrilinear3DCPUKernel final : public user_op::OpKernel { method UpsampleTrilinear3DCPUKernel (line 120) | UpsampleTrilinear3DCPUKernel() = default; method Compute (line 124) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 163) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... class UpsampleTrilinearGrad3DCPUKernel (line 167) | class UpsampleTrilinearGrad3DCPUKernel final : public user_op::OpKernel { method UpsampleTrilinearGrad3DCPUKernel (line 169) | UpsampleTrilinearGrad3DCPUKernel() = default; method Compute (line 173) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 215) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/kernels/util_ops_kernels.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 19) | namespace user_op { FILE: oneflow/user/kernels/variance_kernel.cpp type oneflow (line 21) | namespace oneflow { type user_op (line 22) | namespace user_op { class VarKernel (line 25) | class VarKernel final : public user_op::OpKernel { method VarKernel (line 27) | VarKernel() = default; method Compute (line 31) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 55) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fa... function InferTmpBufferSize (line 72) | size_t InferTmpBufferSize(user_op::InferContext* ctx) { FILE: oneflow/user/kernels/variance_kernel_util.cpp type oneflow (line 18) | namespace oneflow { type user_op (line 20) | namespace user_op { type VarFunctor (line 23) | struct VarFunctor final { type VarFunctor (line 41) | struct VarFunctor type VarFunctor (line 42) | struct VarFunctor type VarFunctor (line 43) | struct VarFunctor type VarFunctor (line 44) | struct VarFunctor FILE: oneflow/user/kernels/variance_kernel_util.h function namespace (line 24) | namespace oneflow { FILE: oneflow/user/kernels/vector_matrix_product_kernel.cpp type oneflow (line 22) | namespace oneflow { function GetBlasTransposeType (line 26) | ep::primitive::BlasTransposeType GetBlasTransposeType(bool transpose) { function NewMemcpyPrimitive (line 31) | std::unique_ptr NewMemcpyPrimitive(Context* ctx) { function NewMatmulPrimitive (line 36) | std::unique_ptr NewMatmulPrimitive(DeviceType d... function NewVectorMatrixProductPrimitive (line 46) | std::unique_ptr NewVectorMatrixProductPrimitive... function NewVectorMatrixProductGradAPrimitive (line 52) | std::unique_ptr NewVectorMatrixProductGradAPrim... function NewVectorMatrixProductGradBPrimitive (line 58) | std::unique_ptr NewVectorMatrixProductGradBPrim... function VectorMatrixProductPrimitiveExists (line 63) | auto VectorMatrixProductPrimitiveExists() { function VectorMatrixProductGradAPrimitiveExists (line 70) | auto VectorMatrixProductGradAPrimitiveExists() { function VectorMatrixProductGradBPrimitiveExists (line 77) | auto VectorMatrixProductGradBPrimitiveExists() { class VectorMatrixProductKernel (line 84) | class VectorMatrixProductKernel final : public user_op::OpKernel, publ... method VectorMatrixProductKernel (line 86) | VectorMatrixProductKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 89) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 92) | void Compute(user_op::KernelComputeContext* ctx) const override { class VectorMatrixProductGradAKernel (line 120) | class VectorMatrixProductGradAKernel final : public user_op::OpKernel, method VectorMatrixProductGradAKernel (line 123) | VectorMatrixProductGradAKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 126) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 129) | void Compute(user_op::KernelComputeContext* ctx) const override { class VectorMatrixProductGradBKernel (line 152) | class VectorMatrixProductGradBKernel final : public user_op::OpKernel, method VectorMatrixProductGradBKernel (line 155) | VectorMatrixProductGradBKernel() = default; method AlwaysComputeWhenAllOutputsEmpty (line 158) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... method Compute (line 161) | void Compute(user_op::KernelComputeContext* ctx) const override { FILE: oneflow/user/kernels/where_kernel.cpp type oneflow (line 20) | namespace oneflow { function NewPrimitive (line 25) | auto NewPrimitive(Context* ctx) -> std::unique_ptr (line 21) | struct WhereKernelUtil { method Where (line 22) | static void Where(ep::Stream* stream, const int64_t elem_cnt, const ... method WhereXScalar (line 26) | static void WhereXScalar(ep::Stream* stream, const int64_t elem_cnt,... method WhereYScalar (line 30) | static void WhereYScalar(ep::Stream* stream, const int64_t elem_cnt,... method WhereXYScalar (line 34) | static void WhereXYScalar(ep::Stream* stream, const int64_t elem_cnt... FILE: oneflow/user/kernels/where_kernel_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/kernels/zero_like_kernel.cpp type oneflow (line 20) | namespace oneflow { class ZeroLikeKernel (line 24) | class ZeroLikeKernel final : public user_op::OpKernel { method ZeroLikeKernel (line 26) | ZeroLikeKernel() = default; method Compute (line 30) | void Compute(user_op::KernelComputeContext* ctx) const override { method AlwaysComputeWhenAllOutputsEmpty (line 42) | bool AlwaysComputeWhenAllOutputsEmpty() const override { return fals... FILE: oneflow/user/ops/acc_ctrl_tick_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/acc_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/adaptive_max_pool_op.cpp type oneflow (line 20) | namespace oneflow { function InferFWTensorDesc (line 24) | Maybe InferFWTensorDesc(user_op::InferContext* ctx) { function InferBWTensorDesc (line 48) | Maybe InferBWTensorDesc(user_op::InferContext* ctx) { function FwGetSbpFn (line 54) | Maybe FwGetSbpFn(user_op::SbpContext* ctx) { function BwGetSbpFn (line 67) | Maybe BwGetSbpFn(user_op::SbpContext* ctx) { function InferFWDataType (line 80) | Maybe InferFWDataType(user_op::InferContext* ctx) { function InferBWDataType (line 86) | Maybe InferBWDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/adaptive_pool_op.cpp type oneflow (line 20) | namespace oneflow { function InferFWTensorDesc (line 24) | Maybe InferFWTensorDesc(user_op::InferContext* ctx) { function InferBWTensorDesc (line 46) | Maybe InferBWTensorDesc(user_op::InferContext* ctx) { function FwGetSbpFn (line 52) | Maybe FwGetSbpFn(user_op::SbpContext* ctx) { function BwGetSbpFn (line 61) | Maybe BwGetSbpFn(user_op::SbpContext* ctx) { function InferFWDataType (line 73) | Maybe InferFWDataType(user_op::InferContext* ctx) { function InferBWDataType (line 78) | Maybe InferBWDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/add_n_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/affine_grid_op.cpp type oneflow (line 20) | namespace oneflow { function CheckAttr_ (line 24) | Maybe CheckAttr_(const user_op::UserOpDefWrapper& def, FILE: oneflow/user/ops/amp_white_identity_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/arange_op.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/arg_sort_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/arg_where_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc (line 23) | Maybe InferTensorDesc(user_op::InferContext* ctx) { FILE: oneflow/user/ops/argmax_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/as_strided_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/assign_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc (line 23) | Maybe InferTensorDesc(user_op::InferContext* ctx) { function GetSbpSignatures (line 36) | Maybe GetSbpSignatures(user_op::SbpContext* ctx) { function InputArgModifierFn (line 52) | Maybe InputArgModifierFn(const user_op::GetInputArgModifier& Get... function InferDataType_ (line 68) | Maybe InferDataType_(user_op::InferContext* ctx) { FILE: oneflow/user/ops/avg_pool_op.cpp type oneflow (line 20) | namespace oneflow { function TensorDescInferFn (line 26) | TensorDescInferFn AvgPoolMakeForwardTensorDescInferFn(const int32_t di... function AvgPoolForwardGetSbpFn (line 64) | Maybe AvgPoolForwardGetSbpFn(user_op::SbpContext* ctx) { function AvgPoolBackwardGetSbpFn (line 72) | Maybe AvgPoolBackwardGetSbpFn(user_op::SbpContext* ctx) { function GetComputationCost (line 86) | Maybe GetComputationCost(user_op::ComputeComplexityFnContext* ... function BackwardTensorDescInferFn (line 102) | Maybe BackwardTensorDescInferFn(user_op::InferContext* ctx) { function FwInferDataType (line 107) | Maybe FwInferDataType(user_op::InferContext* ctx) { function BwInferDataType (line 112) | Maybe BwInferDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/batch_gather_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 38) | FOR_RANGE(int64_t, i, 0, indices.shape().dim_vec().size() - 1) { function FOR_RANGE (line 71) | FOR_RANGE(int64_t, i, 0, indices_num_axes - 1) { FILE: oneflow/user/ops/batch_norm_backward_elemt_op.cpp type oneflow (line 19) | namespace oneflow { function MakeSetOutTensorDescFn (line 23) | std::function(const std::string&)> MakeSetOutTensorDescFn(... function MakeSetOutDataTypeFn (line 35) | std::function(const std::string&)> MakeSetOutDataTypeFn(us... FILE: oneflow/user/ops/batch_norm_backward_reduce_op.cpp type oneflow (line 19) | namespace oneflow { function MakeSetOutTensorDescFn (line 23) | std::function(const std::string&)> MakeSetOutTensorDescFn(... function MakeSetOutDataTypeFn (line 35) | std::function(const std::string&)> MakeSetOutDataTypeFn(us... FILE: oneflow/user/ops/batch_norm_elemt_op.cpp type oneflow (line 19) | namespace oneflow { function MakeSetOutTensorDescFn (line 23) | std::function(const std::string&)> MakeSetOutTensorDescFn(... function MakeSetOutDataTypeFn (line 35) | std::function(const std::string&)> MakeSetOutDataTypeFn(us... FILE: oneflow/user/ops/batch_norm_gather_stats_with_counts_op.cpp type oneflow (line 19) | namespace oneflow { function MakeSetOutTensorDescFn (line 23) | std::function(const std::string&)> MakeSetOutTensorDescFn(... function MakeSetOutDataTypeFn (line 35) | std::function(const std::string&)> MakeSetOutDataTypeFn(us... FILE: oneflow/user/ops/batch_norm_stats_op.cpp type oneflow (line 19) | namespace oneflow { function MakeSetOutTensorDescFn (line 23) | std::function(const std::string&)> MakeSetOutTensorDescFn(... function MakeSetOutDataTypeFn (line 35) | std::function(const std::string&)> MakeSetOutDataTypeFn(us... FILE: oneflow/user/ops/bernoulli_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/bias_add_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/binary_cross_entropy_op.cpp type oneflow (line 21) | namespace oneflow { function InferTensorDescFn_ (line 25) | Maybe InferTensorDescFn_(user_op::InferContext* ctx) { function InferDataType_ (line 43) | Maybe InferDataType_(user_op::InferContext* ctx) { function InferGradTensorDescFn (line 61) | Maybe InferGradTensorDescFn(user_op::InferContext* ctx) { function InferGradDataType (line 80) | Maybe InferGradDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/binary_cross_entropy_with_logits_op.cpp type oneflow (line 22) | namespace oneflow { function InferTensorDescFn (line 24) | Maybe InferTensorDescFn(user_op::InferContext* ctx) { function InferDataType_ (line 46) | Maybe InferDataType_(user_op::InferContext* ctx) { function InferGradTensorDescFn (line 69) | Maybe InferGradTensorDescFn(user_op::InferContext* ctx) { function InferGradDataType (line 94) | Maybe InferGradDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/binary_cross_entropy_with_logits_reduce_mean_op.cpp type oneflow (line 21) | namespace oneflow { function InferTensorDescFn (line 25) | Maybe InferTensorDescFn(user_op::InferContext* ctx) { function InferFwDataType (line 36) | Maybe InferFwDataType(user_op::InferContext* ctx) { function InferGradTensorDescFn (line 48) | Maybe InferGradTensorDescFn(user_op::InferContext* ctx) { function InferGradDataType (line 59) | Maybe InferGradDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/bincount_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc (line 23) | Maybe InferTensorDesc(user_op::InferContext* ctx) { FILE: oneflow/user/ops/broadcast_div_grad_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, y_shape.NumAxes()) { FILE: oneflow/user/ops/broadcast_like_op.cpp type oneflow (line 20) | namespace oneflow { function GetSbpSignatures (line 24) | Maybe GetSbpSignatures(user_op::SbpContext* ctx) { function IsAxesLegal (line 64) | bool IsAxesLegal(const AxisVector& axis_vec, const Shape& like_shape, ... function InferTensorDesc (line 86) | Maybe InferTensorDesc(user_op::InferContext* ctx) { FILE: oneflow/user/ops/buffer_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/cast_like_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/cast_op.cpp type oneflow (line 21) | namespace oneflow { function MakeCastStream (line 25) | Maybe> MakeCastStream(const Symbol& in_device, ... FILE: oneflow/user/ops/cast_to_static_shape_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, input_desc.shape().NumAxes()) { FILE: oneflow/user/ops/cast_to_tick_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/categorical_ordinal_encode_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/celu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, y_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/clip_by_value_op.cpp type oneflow (line 19) | namespace oneflow { function InferClipTensorDesc (line 23) | Maybe InferClipTensorDesc(user_op::InferContext* ctx) { function GetClipSbpSignature (line 28) | Maybe GetClipSbpSignature(user_op::SbpContext* ctx) { function InferClipGradTensorDesc (line 36) | Maybe InferClipGradTensorDesc(user_op::InferContext* ctx) { function GetClipGradSbpSignature (line 41) | Maybe GetClipGradSbpSignature(user_op::SbpContext* ctx) { function InferClipTensorDataType (line 58) | Maybe InferClipTensorDataType(user_op::InferContext* ctx) { function InferClipGradDataType (line 63) | Maybe InferClipGradDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/coco_reader_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/combined_margin_loss_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/comm_net_device_infer_util.cpp type oneflow (line 19) | namespace oneflow { function RawGetTransportDevice (line 23) | Maybe> RawGetTransportDevice(Symbol device) { function DefaultGetOutputDeivce (line 31) | Maybe> DefaultGetOutputDeivce(user_op::DeviceAndStreamI... FILE: oneflow/user/ops/comm_net_device_infer_util.h function namespace (line 26) | namespace oneflow { FILE: oneflow/user/ops/complex_ops.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/concat_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 46) | FOR_RANGE(int64_t, i, 0, in_desc.shape().NumAxes()) { function FOR_RANGE (line 80) | FOR_RANGE(int64_t, i, 0, first_in_desc.shape().NumAxes()) { FILE: oneflow/user/ops/constant_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/conv_op.cpp type oneflow (line 20) | namespace oneflow { function InferTensorDesc4Conv (line 25) | Maybe InferTensorDesc4Conv(user_op::InferContext* ctx) { function GetSbpSignatures4Conv (line 91) | Maybe GetSbpSignatures4Conv(user_op::SbpContext* ctx) { function ConvComputationCost (line 128) | Maybe ConvComputationCost(user_op::ComputeComplexityFnContext*... function CheckAttr_ (line 152) | Maybe CheckAttr_(const user_op::UserOpDefWrapper& def, FILE: oneflow/user/ops/convert_memory_format_op.cpp type oneflow (line 21) | namespace oneflow { function Shape (line 23) | static Shape ComputeShapeIdentity(const Shape& shape) { return shape; } function Shape (line 25) | Shape ComputeShapeContiguousToChannelsLast(const Shape& shape) { function Shape (line 35) | Shape ComputeShapeChannelsLastToContiguous(const Shape& shape) { function GetSbpIdentity (line 45) | static Maybe GetSbpIdentity(user_op::SbpContext* ctx, const Shap... function GetSbpContiguousToChannelsLast (line 52) | static Maybe GetSbpContiguousToChannelsLast(user_op::SbpContext*... function GetSbpChannelsLastToContiguous (line 63) | static Maybe GetSbpChannelsLastToContiguous(user_op::SbpContext*... function Shape (line 87) | Shape ComputeConvertMemoryFormatShape(const Shape& shape, MemoryFormat... function GetConvertMemoryFormatSbp (line 93) | static Maybe GetConvertMemoryFormatSbp(user_op::SbpContext* ctx,... FILE: oneflow/user/ops/convert_memory_format_op.h function namespace (line 18) | namespace oneflow { FILE: oneflow/user/ops/copy_hd_op.cpp type oneflow (line 22) | namespace oneflow { function InferLogical (line 26) | Maybe InferLogical(user_op::InferContext* ctx) { function InferPhysical (line 30) | Maybe InferPhysical(user_op::InferContext* ctx) { function FwGetSbpFn (line 35) | Maybe FwGetSbpFn(user_op::SbpContext* ctx) { return Maybe:... function InferFWDataType (line 37) | Maybe InferFWDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/copy_op.cpp type oneflow (line 22) | namespace oneflow { function StreamType (line 26) | StreamType GetH2DStreamType() { function MakeCopyStream (line 34) | Maybe> MakeCopyStream(const Symbol& in_device, FILE: oneflow/user/ops/count_not_finite_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, x.shape().NumAxes()) { FILE: oneflow/user/ops/ctc_loss_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/cublas_bias_add_relu_matmul_grad_op.cpp type oneflow (line 22) | namespace oneflow { function InferTensorDesc4FusedMatmulBackward (line 26) | Maybe InferTensorDesc4FusedMatmulBackward(user_op::InferContext*... function InferDataType4MatmulBackward (line 36) | Maybe InferDataType4MatmulBackward(user_op::InferContext* ctx) { FILE: oneflow/user/ops/cublas_fused_matmul_bias_add_grad_op.cpp type oneflow (line 22) | namespace oneflow { function InferTensorDesc4MatmulBiasAddBackward (line 26) | Maybe InferTensorDesc4MatmulBiasAddBackward(user_op::InferContex... function InferDataType4MatmulBiasAddBackward (line 44) | Maybe InferDataType4MatmulBiasAddBackward(user_op::InferContext*... FILE: oneflow/user/ops/cublas_fused_mlp_grad_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc4FusedMatmulBackward (line 23) | Maybe InferTensorDesc4FusedMatmulBackward(user_op::InferContext*... function InferDataType4MatmulBackward (line 35) | Maybe InferDataType4MatmulBackward(user_op::InferContext* ctx) { FILE: oneflow/user/ops/cublas_fused_mlp_op.cpp type oneflow (line 23) | namespace oneflow { function AlignReluAuxLd (line 29) | long AlignReluAuxLd(long aux_ld) { function InferTensorDesc4FusedMatmul (line 39) | Maybe InferTensorDesc4FusedMatmul(user_op::InferContext* ctx) { function InferDataType4Matmul (line 77) | Maybe InferDataType4Matmul(user_op::InferContext* ctx) { FILE: oneflow/user/ops/cum_ops.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/data_shuffle_op.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/deconv_op.cpp type oneflow (line 20) | namespace oneflow { function InferTensorDesc4DeConv (line 25) | Maybe InferTensorDesc4DeConv(user_op::InferContext* ctx) { function InferDataType_ (line 87) | Maybe InferDataType_(user_op::InferContext* ctx) { function GetSbpSignatures4DeConv (line 92) | Maybe GetSbpSignatures4DeConv(user_op::SbpContext* ctx) { function CheckAttr_ (line 103) | Maybe CheckAttr_(const user_op::UserOpDefWrapper& def, FILE: oneflow/user/ops/deform_conv_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/depend_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/det_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, i, 0, x.shape().NumAxes() - 2) { FILE: oneflow/user/ops/diag_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/diagonal_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 29) | FOR_RANGE(int32_t, index, 2, in_dim) { out_dim_vec.push_back(in_shape.... FILE: oneflow/user/ops/dim_gather_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 55) | FOR_RANGE(int64_t, i, 0, index_num_axes) { FILE: oneflow/user/ops/dim_scatter_ops.cpp type oneflow (line 22) | namespace oneflow { function InferTensorDesc (line 25) | Maybe InferTensorDesc(user_op::InferContext* ctx) { function InferScalarTensorDesc (line 81) | Maybe InferScalarTensorDesc(user_op::InferContext* ctx) { function InputArgModifierFn (line 104) | Maybe InputArgModifierFn(const user_op::GetInputArgModifier& Get... function InputScalarArgModifierFn (line 113) | Maybe InputScalarArgModifierFn(const user_op::GetInputArgModifie... function _SetSbp (line 122) | void _SetSbp(user_op::SbpContext* ctx, const char* like_or_input) { function SetSbpLike (line 151) | Maybe SetSbpLike(user_op::SbpContext* ctx) { function SetSbpScatter (line 156) | Maybe SetSbpScatter(user_op::SbpContext* ctx) { function SetSbpScatterScalar (line 161) | Maybe SetSbpScatterScalar(user_op::SbpContext* ctx) { function InferDtype (line 180) | Maybe InferDtype(user_op::InferContext* ctx) { function InferScalarDtype (line 196) | Maybe InferScalarDtype(user_op::InferContext* ctx) { FILE: oneflow/user/ops/distributions/exponential_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/distributions/multinomial_with_replacement_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/distributions/normal_op.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/distributions/uniform_int_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/distributions/uniform_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/dot_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/dropout_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, axis, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 68) | FOR_RANGE(int64_t, axis, 0, dy_tensor.shape().NumAxes()) { function FOR_RANGE (line 104) | FOR_RANGE(int64_t, axis, 0, like_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/dynamic_loss_scale_schedule_op.cpp type oneflow (line 19) | namespace oneflow { function IsScalarTensor (line 23) | bool IsScalarTensor(const user_op::TensorDesc* desc) { function IsTensorWithType (line 27) | bool IsTensorWithType(const user_op::TensorDesc* desc, DataType data_t... FILE: oneflow/user/ops/eager_b_to_s_op.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/user/ops/eager_ccl_ops.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/user/ops/eager_p_to_b_op.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/user/ops/eager_p_to_s_op.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/user/ops/eager_s_to_b_op.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/user/ops/eager_s_to_p_op.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/user/ops/eager_s_to_s_op.cpp type oneflow (line 25) | namespace oneflow { FILE: oneflow/user/ops/eager_symmetric_s_to_p_op.cpp type oneflow (line 22) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/elementwise_maximum_minimum_ops.cpp type oneflow (line 19) | namespace oneflow { function GetSbpSignature_ (line 24) | Maybe GetSbpSignature_(SbpContext* ctx) { function InferTensorDesc_ (line 39) | Maybe InferTensorDesc_(InferContext* ctx) { function InferDataType_ (line 60) | Maybe InferDataType_(InferContext* ctx) { FILE: oneflow/user/ops/elu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/embedding_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 64) | FOR_RANGE(int64_t, i, 0, indices_num_axes) { FILE: oneflow/user/ops/empty_op.cpp type oneflow (line 23) | namespace oneflow { function MakeEmptyStream (line 27) | Maybe> MakeEmptyStream(const Symbol& out_device... FILE: oneflow/user/ops/erfinv_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/expand_dims_op.cpp type oneflow (line 19) | namespace oneflow { function TransformNegativeAxisToPositive (line 23) | int32_t TransformNegativeAxisToPositive(int32_t axis, const int32_t nu... function FOR_RANGE (line 53) | FOR_RANGE(int32_t, in_axis, 0, dim_vec.size()) { FILE: oneflow/user/ops/expand_op.cpp type oneflow (line 20) | namespace oneflow { function InferExpandOutputStride (line 24) | Maybe InferExpandOutputStride(const Shape& input_shape, const St... FILE: oneflow/user/ops/eye_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fake_quantization_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 69) | FOR_RANGE(int64_t, i, 1, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/fft_ops.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/fill_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 58) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/flip_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 37) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/frac_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/fused_attention_ops.cpp type oneflow (line 19) | namespace oneflow { function ParseDims (line 23) | Maybe ParseDims(const Shape& shape, const std::string& layout, function ParseDims (line 165) | Maybe ParseDims(const Shape& shape, const std::string& layout, function LayoutToShape (line 173) | Maybe LayoutToShape(int64_t b, int64_t m, int64_t h, int64_t k,... function ParseSplitAxis (line 197) | Maybe ParseSplitAxis(const std::string& layout, bool can_hk_spli... FILE: oneflow/user/ops/fused_bias_add_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_bias_add_scale_mask_softmax_dropout_op.cpp type oneflow (line 19) | namespace oneflow { function CheckBroadcastable (line 23) | bool CheckBroadcastable(const Shape& shape, const Shape& broadcast_sha... function CheckBroadcastAndSimplifyDims (line 33) | bool CheckBroadcastAndSimplifyDims(const Shape& shape, const Shape& br... function GetBroadcastDims (line 57) | int GetBroadcastDims(const Shape& shape, const Shape& broadcast_shape, FILE: oneflow/user/ops/fused_cast_scale_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_center_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 75) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { function FOR_RANGE (line 199) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { FILE: oneflow/user/ops/fused_clip_grad_ops.cpp type oneflow (line 22) | namespace oneflow { function SetInputArgModifierMutable (line 26) | Maybe SetInputArgModifierMutable(const user_op::GetInputArgModif... function InputArgModifyFn (line 34) | Maybe InputArgModifyFn(const user_op::GetInputArgModifier& GetIn... FILE: oneflow/user/ops/fused_codegeex_qkv_reshape.cpp type oneflow (line 21) | namespace oneflow { function FOR_RANGE (line 75) | FOR_RANGE(int64_t, i, 0, query.shape().NumAxes() - 1) { FILE: oneflow/user/ops/fused_cross_feature_interaction_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/fused_dot_feature_interaction_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_get_boundding_boxes_coord_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 96) | FOR_RANGE(int64_t, i, 0, x1.shape().NumAxes()) { function FOR_RANGE (line 194) | FOR_RANGE(int64_t, i, 0, b1_x1_diff.shape().NumAxes()) { FILE: oneflow/user/ops/fused_get_ciou_diagonal_angle_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 59) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { function FOR_RANGE (line 139) | FOR_RANGE(int64_t, i, 0, w1.shape().NumAxes()) { FILE: oneflow/user/ops/fused_get_ciou_result_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 53) | FOR_RANGE(int64_t, i, 0, v.shape().NumAxes()) { function FOR_RANGE (line 112) | FOR_RANGE(int64_t, i, 0, dy.shape().NumAxes()) { FILE: oneflow/user/ops/fused_get_convex_diagonal_squared_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 45) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { function FOR_RANGE (line 137) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { FILE: oneflow/user/ops/fused_get_intersection_area_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 75) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { function FOR_RANGE (line 199) | FOR_RANGE(int64_t, i, 0, b1_x1.shape().NumAxes()) { FILE: oneflow/user/ops/fused_get_iou_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 46) | FOR_RANGE(int64_t, i, 0, w1.shape().NumAxes()) { function FOR_RANGE (line 98) | FOR_RANGE(int64_t, i, 0, dy.shape().NumAxes()) { FILE: oneflow/user/ops/fused_glu_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_glu_without_linear_grad_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_gru_cell_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_linear_with_groupwise_quantized_weight_op.cpp type oneflow (line 23) | namespace oneflow { function InferTensorDesc4FusedMatmulBias (line 27) | Maybe InferTensorDesc4FusedMatmulBias(user_op::InferContext* ctx) { function InferDataType4MatmulBias (line 81) | Maybe InferDataType4MatmulBias(user_op::InferContext* ctx) { FILE: oneflow/user/ops/fused_lstm_cell_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_matmul_bias_add_relu_dropout_op.cpp type oneflow (line 19) | namespace oneflow { function AlignReluAuxLd (line 25) | long AlignReluAuxLd(long aux_ld) { function InferTensorDesc4FusedMatmul (line 35) | Maybe InferTensorDesc4FusedMatmul(user_op::InferContext* ctx) { function InferDataType4Matmul (line 77) | Maybe InferDataType4Matmul(user_op::InferContext* ctx) { FILE: oneflow/user/ops/fused_matmul_bias_op.cpp type oneflow (line 23) | namespace oneflow { function InferTensorDesc4FusedMatmulBias (line 27) | Maybe InferTensorDesc4FusedMatmulBias(user_op::InferContext* ctx) { function InferDataType4MatmulBias (line 60) | Maybe InferDataType4MatmulBias(user_op::InferContext* ctx) { FILE: oneflow/user/ops/fused_relu_dropout_grad_op.cpp type oneflow (line 23) | namespace oneflow { function InferTensorDesc4FusedReluDropoutGrad (line 27) | Maybe InferTensorDesc4FusedReluDropoutGrad(user_op::InferContext... function InferDataType4FusedReluDropoutGrad (line 32) | Maybe InferDataType4FusedReluDropoutGrad(user_op::InferContext* ... FILE: oneflow/user/ops/fused_scale_mask_bias_softmax_op.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/user/ops/fused_scale_mask_softmax_dropout_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 68) | FOR_RANGE(int64_t, axis, 0, x_tensor.shape().NumAxes() - 2) { function FOR_RANGE (line 131) | FOR_RANGE(int64_t, axis, 0, dy_tensor.shape().NumAxes() - 2) { FILE: oneflow/user/ops/fused_scale_mask_softmax_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 59) | FOR_RANGE(int64_t, axis, 0, x_tensor.shape().NumAxes() - 2) { function FOR_RANGE (line 112) | FOR_RANGE(int64_t, axis, 0, dy_tensor.shape().NumAxes() - 2) { FILE: oneflow/user/ops/fused_scale_tril_softmax_mask_scale_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 51) | FOR_RANGE(int64_t, axis, 0, x_tensor.shape().NumAxes() - 2) { function FOR_RANGE (line 89) | FOR_RANGE(int64_t, axis, 0, dy_tensor.shape().NumAxes() - 2) { FILE: oneflow/user/ops/fused_self_attention_query_mul_key_and_value_ops.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/fused_weighted_sum_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/gather_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 59) | FOR_RANGE(int64_t, i, 0, indices_num_axes) { function FOR_RANGE (line 66) | FOR_RANGE(int64_t, i, 0, in_num_axes) { FILE: oneflow/user/ops/gelu_op.cpp type oneflow (line 19) | namespace oneflow { function InferGeluTensorDesc (line 23) | Maybe InferGeluTensorDesc(user_op::InferContext* ctx) { function InferGeluDataType (line 28) | Maybe InferGeluDataType(user_op::InferContext* ctx) { function GetGeluSbp (line 33) | Maybe GetGeluSbp(user_op::SbpContext* ctx) { function InferGeluGradTensorDesc (line 69) | Maybe InferGeluGradTensorDesc(user_op::InferContext* ctx) { function InferGeluGradDataType (line 79) | Maybe InferGeluGradDataType(user_op::InferContext* ctx) { function GetGeluGradSbp (line 87) | Maybe GetGeluGradSbp(user_op::SbpContext* ctx) { function FOR_RANGE (line 187) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/generate_random_batch_permutation_indices_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/gpt_data_loader_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/greater_inplace_op.cpp type oneflow (line 19) | namespace oneflow { function CheckBroadCastAble (line 23) | bool CheckBroadCastAble(const Shape& shape, const Shape& broadcast_sha... function FOR_RANGE (line 54) | FOR_RANGE(int64_t, i, 0, x.shape().NumAxes()) { function FOR_RANGE (line 82) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/grid_sample_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/group_norm_op.cpp type oneflow (line 20) | namespace oneflow { function InferGnParamDataType (line 26) | oneflow::DataType InferGnParamDataType(const DataType x_data_type) { FILE: oneflow/user/ops/grouped_matmul_bias_op.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/user/ops/groupwise_dequantize_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/hardshrink_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, y_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/hardsigmoid_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/hardswish_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/hardtanh_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 63) | FOR_RANGE(int64_t, i, 0, y_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/hierarchical_parallel_cast_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/identity_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/image_batch_align_op.cpp type oneflow (line 19) | namespace oneflow { function PowerOfTwo (line 24) | bool PowerOfTwo(T x) { function FOR_RANGE (line 38) | FOR_RANGE(int64_t, i, 0, shape_attr.NumAxes()) { dim_vec.at(i + 1) = s... FILE: oneflow/user/ops/image_decode_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/image_object_preprocess_ops.cpp type oneflow (line 19) | namespace oneflow { function ImageObjectGetSbp (line 23) | Maybe ImageObjectGetSbp(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/image_preprocess_ops.cpp type oneflow (line 23) | namespace oneflow { function FOR_RANGE (line 193) | FOR_RANGE(int, i, 0, hierarchy.NumAxes()) { FILE: oneflow/user/ops/image_resize_ops.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/image_target_resize_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/in_top_k_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/index_add_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/indexed_slices_reduce_sum_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 26) | FOR_RANGE(int64_t, i, 0, x_indices.shape().NumAxes()) { FILE: oneflow/user/ops/inv_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, x.shape().NumAxes() - 2) { FILE: oneflow/user/ops/kl_div_op.cpp type oneflow (line 21) | namespace oneflow { function KlInferTensorDescFn (line 23) | Maybe KlInferTensorDescFn(user_op::InferContext* ctx) { function KlInferDataType (line 36) | Maybe KlInferDataType(user_op::InferContext* ctx) { function InferGradTensorDescFn (line 48) | Maybe InferGradTensorDescFn(user_op::InferContext* ctx) { function InferGradDataType (line 63) | Maybe InferGradDataType(user_op::InferContext* ctx) { function FOR_RANGE (line 87) | FOR_RANGE(int64_t, i, 0, input_shape.NumAxes()) { function FOR_RANGE (line 115) | FOR_RANGE(int64_t, i, 0, input_shape.NumAxes()) { FILE: oneflow/user/ops/l1_l2_regularize_gradient_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc (line 23) | Maybe InferTensorDesc(user_op::InferContext* ctx) { function GetSbpSignatures (line 32) | Maybe GetSbpSignatures(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/l2_normalize_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 75) | FOR_RANGE(int32_t, i, 0, dy_shape.NumAxes()) { function FOR_RANGE (line 93) | FOR_RANGE(int64_t, i, 0, y_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/layer_norm_op.cpp type oneflow (line 19) | namespace oneflow { function ShiftNegativeAxisIfNeed (line 25) | int64_t ShiftNegativeAxisIfNeed(const Shape& shape, int64_t axis) { function Shape (line 32) | Shape InferBnParamShape(const Shape& x_shape, const int64_t begin_norm... function InferBnParamDataType (line 45) | oneflow::DataType InferBnParamDataType(const DataType x_data_type) { FILE: oneflow/user/ops/leaky_relu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/lerp_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 65) | FOR_RANGE(int64_t, i, 0, start.shape().NumAxes()) { function FOR_RANGE (line 140) | FOR_RANGE(int64_t, i, 0, start.shape().NumAxes()) { function FOR_RANGE (line 188) | FOR_RANGE(int64_t, i, 0, start.shape().NumAxes()) { function FOR_RANGE (line 247) | FOR_RANGE(int64_t, i, 0, start.shape().NumAxes()) { FILE: oneflow/user/ops/linalg_cross_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, num_axes) { FILE: oneflow/user/ops/log_softmax_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, axis, 0, in_tensor.shape().NumAxes() - 1) { function FOR_RANGE (line 60) | FOR_RANGE(int64_t, axis, 0, y_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/logical_not_op.cpp type oneflow (line 19) | namespace oneflow { function InferDataTypeLogicalNot (line 23) | Maybe InferDataTypeLogicalNot(user_op::InferContext* ctx) { FILE: oneflow/user/ops/loss_op_util.cpp type oneflow (line 19) | namespace oneflow { function GenLossForwardDefaultGetSbpFn (line 21) | user_op::GetSbpFn GenLossForwardDefaultGetSbpFn( function GenLossBackwardDefaultGetSbpFn (line 38) | user_op::GetSbpFn GenLossBackwardDefaultGetSbpFn( FILE: oneflow/user/ops/loss_op_util.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/ops/lu_composition_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, x.shape().NumAxes() - 2) { FILE: oneflow/user/ops/masked_fill_op.cpp type oneflow (line 19) | namespace oneflow { function InferMaskedFillTensorDesc (line 23) | Maybe InferMaskedFillTensorDesc(user_op::InferContext* ctx) { function InferMaskedFillDataType (line 29) | Maybe InferMaskedFillDataType(user_op::InferContext* ctx) { function GetMaskedFillSbpSignatures (line 37) | Maybe GetMaskedFillSbpSignatures(user_op::SbpContext* ctx) { function GetMaskedFillInputArgModify (line 63) | Maybe GetMaskedFillInputArgModify(const user_op::GetInputArgModi... FILE: oneflow/user/ops/math_binary_broadcast_ops.cpp type oneflow (line 21) | namespace oneflow { function IsScalarTensor (line 25) | bool IsScalarTensor(const user_op::TensorDesc* tensor) { function IsZeroDimTensor (line 29) | bool IsZeroDimTensor(const user_op::TensorDesc* tensor) { return tenso... function InferTensorDescBinaryBroadcastNormal (line 31) | Maybe InferTensorDescBinaryBroadcastNormal(user_op::InferContext... function InferTensorDescBinaryBroadcastLogical (line 71) | Maybe InferTensorDescBinaryBroadcastLogical(user_op::InferContex... function InferDataTypeBinaryBroadcastNormal (line 75) | Maybe InferDataTypeBinaryBroadcastNormal(user_op::InferContext* ... function InferDataTypeBinaryBroadcastLogical (line 85) | Maybe InferDataTypeBinaryBroadcastLogical(user_op::InferContext*... function GenPartialSbpSign (line 96) | void GenPartialSbpSign(user_op::SbpContext* ctx) {} function GetBinaryBroadcastSbpSignature (line 149) | Maybe GetBinaryBroadcastSbpSignature(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/math_binary_broadcast_seq.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/ops/math_binary_elementwise_ops.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/math_binary_elementwise_seq.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/ops/math_unary_elementwise_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/math_unary_elementwise_seq.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/ops/matmul_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc4Matmul (line 23) | Maybe InferTensorDesc4Matmul(user_op::InferContext* ctx) { function InferDataType4Matmul (line 67) | Maybe InferDataType4Matmul(user_op::InferContext* ctx) { function GetComputationCost (line 86) | Maybe GetComputationCost(user_op::ComputeComplexityFnContext* ... function FOR_RANGE (line 198) | FOR_RANGE(int64_t, i, 0, num_axes - 2) { function FOR_RANGE (line 280) | FOR_RANGE(int64_t, i, 0, out_dim_vec.size() - 2) { FILE: oneflow/user/ops/matrix_vector_product_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc4MatrixVectorProduct (line 23) | Maybe InferTensorDesc4MatrixVectorProduct(user_op::InferContext*... function InferDataType4MatrixVectorProduct (line 33) | Maybe InferDataType4MatrixVectorProduct(user_op::InferContext* c... function InferTensorDesc4MatrixVectorProductGradA (line 42) | Maybe InferTensorDesc4MatrixVectorProductGradA(user_op::InferCon... function InferTensorDesc4MatrixVectorProductGradB (line 55) | Maybe InferTensorDesc4MatrixVectorProductGradB(user_op::InferCon... function InferDataType4Grad (line 66) | Maybe InferDataType4Grad(user_op::InferContext* ctx) { FILE: oneflow/user/ops/max_pool_op.cpp type oneflow (line 20) | namespace oneflow { function TensorDescInferFn (line 26) | TensorDescInferFn MaxPoolMakeForwardTensorDescInferFn(const int32_t di... function MaxPoolForwardGetSbpFn (line 60) | Maybe MaxPoolForwardGetSbpFn(user_op::SbpContext* ctx) { function MaxPoolBackwardGetSbpFn (line 72) | Maybe MaxPoolBackwardGetSbpFn(user_op::SbpContext* ctx) { function GetComputationCost (line 88) | Maybe GetComputationCost(user_op::ComputeComplexityFnContext* ... function BackwardTensorDescInferFn (line 104) | Maybe BackwardTensorDescInferFn(user_op::InferContext* ctx) { function FwInferDataType (line 109) | Maybe FwInferDataType(user_op::InferContext* ctx) { function BwInferDataType (line 114) | Maybe BwInferDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/max_unpool_op.cpp type oneflow (line 21) | namespace oneflow { function TensorDescInferFn (line 27) | TensorDescInferFn MaxUnpoolMakeForwardTensorDescInferFn(const int32_t ... function MaxUnpoolForwardGetSbpFn (line 49) | Maybe MaxUnpoolForwardGetSbpFn(user_op::SbpContext* ctx) { function MaxUnpoolBackwardGetSbpFn (line 61) | Maybe MaxUnpoolBackwardGetSbpFn(user_op::SbpContext* ctx) { function BackwardTensorDescInferFn (line 74) | Maybe BackwardTensorDescInferFn(user_op::InferContext* ctx) { function FwInferDataType (line 79) | Maybe FwInferDataType(user_op::InferContext* ctx) { function BwInferDataType (line 84) | Maybe BwInferDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/median_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/median_with_indices_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 24) | FOR_RANGE(int64_t, i, 0, num_axes - 1) { FILE: oneflow/user/ops/min_max_observer_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/mish_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/mode_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 24) | FOR_RANGE(int64_t, i, 0, num_axes - 1) { FILE: oneflow/user/ops/model_update_ops.cpp type oneflow (line 22) | namespace oneflow { function CheckShapeLike (line 26) | Maybe CheckShapeLike(const user_op::TensorDesc* tensor_desc, function CheckDataTypeLike (line 31) | Maybe CheckDataTypeLike(const user_op::TensorDesc* tensor_desc, function CheckScalarShape (line 39) | Maybe CheckScalarShape(const user_op::TensorDesc* tensor_desc) { function CheckScalarDataType (line 45) | Maybe CheckScalarDataType(const user_op::TensorDesc* tensor_desc... function CheckLearningRateShape (line 52) | Maybe CheckLearningRateShape(user_op::InferContext* ctx) { function CheckLearningRateDataType (line 59) | Maybe CheckLearningRateDataType(user_op::InferContext* ctx) { function CheckIndexedSlicesModelDiffDesc (line 67) | Maybe CheckIndexedSlicesModelDiffDesc(const user_op::TensorDesc*... function CheckIndexedSlicesModelDiffDataType (line 84) | Maybe CheckIndexedSlicesModelDiffDataType(const user_op::TensorD... function InferSGDUpdateTensorDesc (line 94) | Maybe InferSGDUpdateTensorDesc(user_op::InferContext* ctx) { function InferSGDUpdateDataType (line 112) | Maybe InferSGDUpdateDataType(user_op::InferContext* ctx) { function InferIndexedSlicesSGDUpdateTensorDesc (line 122) | Maybe InferIndexedSlicesSGDUpdateTensorDesc(user_op::InferContex... function InferIndexedSlicesSGDUpdateDataType (line 130) | Maybe InferIndexedSlicesSGDUpdateDataType(user_op::InferContext*... function InferMomentumUpdateTensorDesc (line 139) | Maybe InferMomentumUpdateTensorDesc(user_op::InferContext* ctx) { function InferMomentumUpdateDataType (line 152) | Maybe InferMomentumUpdateDataType(user_op::InferContext* ctx) { function InferIndexedSlicesMomentumUpdateTensorDesc (line 164) | Maybe InferIndexedSlicesMomentumUpdateTensorDesc(user_op::InferC... function InferIndexedSlicesMomentumUpdateDataType (line 174) | Maybe InferIndexedSlicesMomentumUpdateDataType(user_op::InferCon... function InferAdamUpdateTensorDesc (line 185) | Maybe InferAdamUpdateTensorDesc(user_op::InferContext* ctx) { function InferAdamUpdateDataType (line 205) | Maybe InferAdamUpdateDataType(user_op::InferContext* ctx) { function InferAdagradUpdateTensorDesc (line 219) | Maybe InferAdagradUpdateTensorDesc(user_op::InferContext* ctx) { function InferAdagradUpdateDataType (line 230) | Maybe InferAdagradUpdateDataType(user_op::InferContext* ctx) { function InferIndexedSlicesAdamUpdateTensorDesc (line 238) | Maybe InferIndexedSlicesAdamUpdateTensorDesc(user_op::InferConte... function InferIndexedSlicesAdamUpdateDataType (line 246) | Maybe InferIndexedSlicesAdamUpdateDataType(user_op::InferContext... function InferLambUpdateTensorDesc (line 255) | Maybe InferLambUpdateTensorDesc(user_op::InferContext* ctx) { function InferLambUpdateDataType (line 279) | Maybe InferLambUpdateDataType(user_op::InferContext* ctx) { function InferFtrlUpdateTensorDesc (line 293) | Maybe InferFtrlUpdateTensorDesc(user_op::InferContext* ctx) { function InferFtrlUpdateDataType (line 307) | Maybe InferFtrlUpdateDataType(user_op::InferContext* ctx) { function InferAdadeltaUpdateTensorDesc (line 317) | Maybe InferAdadeltaUpdateTensorDesc(user_op::InferContext* ctx) { function InferAdadeltaUpdateDataType (line 329) | Maybe InferAdadeltaUpdateDataType(user_op::InferContext* ctx) { function SetInputArgModifierMutable (line 339) | Maybe SetInputArgModifierMutable(const user_op::GetInputArgModif... function AdamInputArgModifyFn (line 347) | Maybe AdamInputArgModifyFn(const user_op::GetInputArgModifier& G... function AdagradInputArgModifyFn (line 361) | Maybe AdagradInputArgModifyFn(const user_op::GetInputArgModifier... function LambInputArgModifyFn (line 368) | Maybe LambInputArgModifyFn(const user_op::GetInputArgModifier& G... function SgdInputArgModifyFn (line 376) | Maybe SgdInputArgModifyFn(const user_op::GetInputArgModifier& Ge... function IndexedSlicesSgdInputArgModifyFn (line 385) | Maybe IndexedSlicesSgdInputArgModifyFn( function MomentumInputArgModifyFn (line 392) | Maybe MomentumInputArgModifyFn(const user_op::GetInputArgModifie... function IndexedSlicesMomentumInputArgModifyFn (line 399) | Maybe IndexedSlicesMomentumInputArgModifyFn( function RmsPropUpdateInputArgModifyFn (line 407) | Maybe RmsPropUpdateInputArgModifyFn(const user_op::GetInputArgMo... function LarsUpdateInputArgModifyFn (line 417) | Maybe LarsUpdateInputArgModifyFn(const user_op::GetInputArgModif... function FtrlInputArgModifyFn (line 424) | Maybe FtrlInputArgModifyFn(const user_op::GetInputArgModifier& G... function AdadeltaInputArgModifyFn (line 432) | Maybe AdadeltaInputArgModifyFn(const user_op::GetInputArgModifie... function InferRmsPropUpdateTensorDesc (line 440) | Maybe InferRmsPropUpdateTensorDesc(user_op::InferContext* ctx) { function InferRmsPropUpdateDataType (line 463) | Maybe InferRmsPropUpdateDataType(user_op::InferContext* ctx) { function InferLarsUpdateTensorDesc (line 480) | Maybe InferLarsUpdateTensorDesc(user_op::InferContext* ctx) { function InferLarsUpdateDataType (line 495) | Maybe InferLarsUpdateDataType(user_op::InferContext* ctx) { function FOR_RANGE (line 520) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 562) | FOR_RANGE(int64_t, i, 1, model.shape().NumAxes()) { function FOR_RANGE (line 592) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 633) | FOR_RANGE(int64_t, i, 1, model.shape().NumAxes()) { function FOR_RANGE (line 664) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 699) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 749) | FOR_RANGE(int64_t, i, 1, model.shape().NumAxes()) { function FOR_RANGE (line 820) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 860) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 895) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { function FOR_RANGE (line 926) | FOR_RANGE(int64_t, axis, 0, model.shape().NumAxes()) { FILE: oneflow/user/ops/moving_average_min_max_observer_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/multi_reduce_ops.cpp type oneflow (line 20) | namespace oneflow { function InferMultiReduceOpShape (line 24) | Maybe InferMultiReduceOpShape(user_op::InferContext* ctx) { function InferMultiReduceOpDataType (line 30) | Maybe InferMultiReduceOpDataType(user_op::InferContext* ctx) { function GetMultiReduceOpSbp (line 40) | Maybe GetMultiReduceOpSbp(user_op::SbpContext* ctx) { function InferLocalMultiReduceOpLogicalShape (line 53) | Maybe InferLocalMultiReduceOpLogicalShape(user_op::InferContext*... function InferLocalMultiReduceOpPhysicalShape (line 74) | Maybe InferLocalMultiReduceOpPhysicalShape(user_op::InferContext... function GetLocalMultiReduceOpSbp (line 80) | Maybe GetLocalMultiReduceOpSbp(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/multi_tensor_model_update_ops.cpp type oneflow (line 22) | namespace oneflow { function CheckShapeLike (line 26) | Maybe CheckShapeLike(const user_op::TensorDesc* tensor_desc, function CheckDataTypeLike (line 33) | Maybe CheckDataTypeLike(const user_op::TensorDesc* tensor_desc, function CheckScalarShape (line 40) | Maybe CheckScalarShape(const user_op::TensorDesc* tensor_desc) { function CheckScalarDataType (line 47) | Maybe CheckScalarDataType(const user_op::TensorDesc* tensor_desc... function CheckLearningRateShape (line 53) | Maybe CheckLearningRateShape(user_op::InferContext* ctx) { function CheckLearningRateDataType (line 60) | Maybe CheckLearningRateDataType(user_op::InferContext* ctx) { function SetInputArgModifierMutable (line 68) | Maybe SetInputArgModifierMutable(const user_op::GetInputArgModif... function InferSGDUpdateTensorDesc (line 76) | Maybe InferSGDUpdateTensorDesc(user_op::InferContext* ctx) { function InferSGDUpdateDataType (line 92) | Maybe InferSGDUpdateDataType(user_op::InferContext* ctx) { function SgdInputArgModifyFn (line 107) | Maybe SgdInputArgModifyFn(const user_op::GetInputArgModifier& Ge... function InferMomentumUpdateTensorDesc (line 115) | Maybe InferMomentumUpdateTensorDesc(user_op::InferContext* ctx) { function InferMomentumUpdateDataType (line 134) | Maybe InferMomentumUpdateDataType(user_op::InferContext* ctx) { function MomentumInputArgModifyFn (line 152) | Maybe MomentumInputArgModifyFn(const user_op::GetInputArgModifie... function InferAdamUpdateTensorDesc (line 161) | Maybe InferAdamUpdateTensorDesc(user_op::InferContext* ctx) { function InferAdamUpdateDataType (line 182) | Maybe InferAdamUpdateDataType(user_op::InferContext* ctx) { // ... function AdamInputArgModifyFn (line 201) | Maybe AdamInputArgModifyFn(const user_op::GetInputArgModifier& G... function InferSGDUpdateWithCastTensorDesc (line 211) | Maybe InferSGDUpdateWithCastTensorDesc(user_op::InferContext* ct... function SgdWithCastInputArgModifyFn (line 230) | Maybe SgdWithCastInputArgModifyFn(const user_op::GetInputArgModi... function InferMomentumUpdateWithCastTensorDesc (line 239) | Maybe InferMomentumUpdateWithCastTensorDesc(user_op::InferContex... function MomentumWithCastInputArgModifyFn (line 261) | Maybe MomentumWithCastInputArgModifyFn( function InferAdamUpdateWithCastTensorDesc (line 272) | Maybe InferAdamUpdateWithCastTensorDesc(user_op::InferContext* c... function AdamWithCastInputArgModifyFn (line 296) | Maybe AdamWithCastInputArgModifyFn(const user_op::GetInputArgMod... function InferYoloV5WeightUpdateTensorDesc (line 307) | Maybe InferYoloV5WeightUpdateTensorDesc(user_op::InferContext* c... function InferYoloV5WeightUpdateDataType (line 318) | Maybe InferYoloV5WeightUpdateDataType(user_op::InferContext* ctx) { function YoloV5WeightInputArgModifyFn (line 332) | Maybe YoloV5WeightInputArgModifyFn(const user_op::GetInputArgMod... FILE: oneflow/user/ops/mutable_cast_once_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/narrow_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 51) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 99) | FOR_RANGE(int64_t, i, 0, ndim) { FILE: oneflow/user/ops/nccl_logical_2d_sbp_ops.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/user/ops/nccl_logical_fusion_op.cpp type oneflow (line 23) | namespace oneflow { FILE: oneflow/user/ops/nccl_logical_ops.cpp type oneflow (line 22) | namespace oneflow { FILE: oneflow/user/ops/nccl_logical_util.cpp type oneflow (line 19) | namespace oneflow { function GetCommKeyFromNcclType (line 21) | std::string GetCommKeyFromNcclType(const std::string& op_type_name) { FILE: oneflow/user/ops/nccl_logical_util.h function namespace (line 22) | namespace oneflow { function InferNdSbpFnContext (line 56) | struct OpTypeNameFromContext { function KernelInitContext (line 63) | struct OpTypeNameFromContext { function InferContext (line 68) | struct OpTypeNameFromContext { FILE: oneflow/user/ops/nd_index_slice_ops.cpp type oneflow (line 19) | namespace oneflow { function CheckScatterNdShape (line 23) | Maybe CheckScatterNdShape(const Shape& params_shape, const Shape... function InferScatterNdTensorDesc (line 40) | Maybe InferScatterNdTensorDesc(user_op::InferContext* ctx) { function InferScatterNdDataType (line 49) | Maybe InferScatterNdDataType(user_op::InferContext* ctx) { function InferScatterNdLikeTensorDesc (line 54) | Maybe InferScatterNdLikeTensorDesc(user_op::InferContext* ctx) { function InferScatterNdLikeDataType (line 63) | Maybe InferScatterNdLikeDataType(user_op::InferContext* ctx) { function InferTensorScatterNdOptTensorDesc (line 68) | Maybe InferTensorScatterNdOptTensorDesc(user_op::InferContext* c... function InferTensorScatterNdOptDataType (line 78) | Maybe InferTensorScatterNdOptDataType(user_op::InferContext* ctx) { function GetTensorScatterNdOptSbpSignatures (line 83) | Maybe GetTensorScatterNdOptSbpSignatures(user_op::SbpContext* ct... function FOR_RANGE (line 123) | FOR_RANGE(int64_t, i, index_ndims, params_shape.NumAxes()) { function FOR_RANGE (line 144) | FOR_RANGE(int64_t, i, 0, indices_num_axes - 1) { function FOR_RANGE (line 152) | FOR_RANGE(int64_t, i, index_ndims, params_tensor.shape().NumAxes()) { function FOR_RANGE (line 192) | FOR_RANGE(int64_t, i, 0, indices_num_axes - 1) { function FOR_RANGE (line 202) | FOR_RANGE(int64_t, i, 0, slice_ndims) { function FOR_RANGE (line 241) | FOR_RANGE(int64_t, i, 0, indices_num_axes - 1) { function FOR_RANGE (line 252) | FOR_RANGE(int64_t, i, 0, slice_ndims) { FILE: oneflow/user/ops/nll_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/nms_op.cpp type oneflow (line 19) | namespace oneflow { function InferNmsTensorDesc (line 23) | Maybe InferNmsTensorDesc(user_op::InferContext* ctx) { function InferNmsDataType (line 28) | Maybe InferNmsDataType(user_op::InferContext* ctx) { FILE: oneflow/user/ops/nn_util.cpp type oneflow (line 18) | namespace oneflow { function CalcOutAndPadding (line 20) | Maybe CalcOutAndPadding(int64_t input_size, int32_t filter_size,... function CalcSamePadding (line 48) | Maybe CalcSamePadding(int64_t input_size, int32_t filter_size, i... function CalcConvOut (line 62) | Maybe CalcConvOut(int64_t input_size, int32_t filter_size, int32... function IdxOffset (line 75) | const size_t IdxOffset(const std::string& data_format) { function ChannelIdx (line 85) | const int32_t ChannelIdx(const std::string& data_format, int32_t num_a... FILE: oneflow/user/ops/nn_util.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/ops/noncontiguous_binary_op.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/user/ops/normalization_op.cpp type oneflow (line 23) | namespace oneflow { function MakeCheckParamTensorDescFn (line 27) | std::function(const std::string&)> MakeCheckParamTensorDes... function MakeCheckParamDataTypeFn (line 38) | std::function(const std::string&)> MakeCheckParamDataTypeF... function MakeSetParamTensorDescFn (line 51) | std::function(const std::string&)> MakeSetParamTensorDescF... function MakeSetParamDataTypeFn (line 63) | std::function(const std::string&)> MakeSetParamDataTypeFn(... function FwInputArgModifyFn (line 75) | Maybe FwInputArgModifyFn(const user_op::GetInputArgModifier& Get... function FwGetSbpFn (line 99) | Maybe FwGetSbpFn(user_op::SbpContext* ctx) { function MakeFwTensorDescInferFn (line 123) | user_op::TensorDescInferFn MakeFwTensorDescInferFn( function MakeFwDataTypeInferFn (line 172) | user_op::DataTypeInferFn MakeFwDataTypeInferFn( function MakeFwTensorDescInferFn (line 214) | user_op::TensorDescInferFn MakeFwTensorDescInferFn() { function MakeFwDataTypeInferFn (line 220) | user_op::DataTypeInferFn MakeFwDataTypeInferFn() { function InferCudnnReserveSpaceSize (line 308) | void InferCudnnReserveSpaceSize(DataType data_type, cudnnBatchNormOps_... function BwTensorDescInferFn (line 433) | Maybe BwTensorDescInferFn(user_op::InferContext* ctx) { function BwDataTypeInferFn (line 460) | Maybe BwDataTypeInferFn(user_op::InferContext* ctx) { function BwGetSbpFn (line 488) | Maybe BwGetSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/nvtx_range_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 36) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 63) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/ofrecord_decoder_ops.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/ofrecord_image_classification_reader_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/ofrecord_reader_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/one_embedding_ops.cpp type oneflow (line 21) | namespace oneflow { function CheckDataShape (line 221) | Maybe CheckDataShape(user_op::InferContext* ctx) { function CheckDataType (line 244) | Maybe CheckDataType(user_op::InferContext* ctx) { function GetEmbeddingUpdateSbp (line 261) | Maybe GetEmbeddingUpdateSbp(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/one_hot_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 44) | FOR_RANGE(int64_t, i, 0, indices_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/ones_like_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, like_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/p2p_comm_op.cpp type oneflow (line 20) | namespace oneflow { function GetRecvOutputDeivce (line 41) | Maybe> GetRecvOutputDeivce(user_op::DeviceAndStreamInfe... FILE: oneflow/user/ops/pack_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/pad_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 26) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 40) | FOR_RANGE(int64_t, i, 0, x_shape.NumAxes()) { FILE: oneflow/user/ops/parallel_cast_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/partial_fc_sample_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/pinned_identity_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 33) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/prelu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 31) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 79) | FOR_RANGE(int64_t, i, 1, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/quantization_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 49) | FOR_RANGE(int64_t, i, 1, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/quick_gelu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 37) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 67) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/randperm_op.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/raw_reader_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/reduce_like_ops.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 36) | FOR_RANGE(int64_t, i, 0, num_axes) { FILE: oneflow/user/ops/reduce_ops.cpp type oneflow (line 21) | namespace oneflow { function InferTensorDescFn (line 23) | Maybe InferTensorDescFn(user_op::InferContext* ctx) { function InferDataType (line 45) | Maybe InferDataType(user_op::InferContext* ctx) { function InferLogicalDataType (line 50) | Maybe InferLogicalDataType(user_op::InferContext* ctx) { function GeneratePartialSbp (line 56) | void GeneratePartialSbp(user_op::SbpContext* ctx, int64_t axis) { function GetSbpFn (line 67) | Maybe GetSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/reflection_pad_op.cpp type oneflow (line 21) | namespace oneflow { function GetOpSbpSignature (line 26) | Maybe GetOpSbpSignature(user_op::SbpContext* ctx) { function GetOpGradSbpSignature (line 38) | Maybe GetOpGradSbpSignature(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/relu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, y_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/repeat_interleave_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/repeat_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 24) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/replication_pad_op.cpp type oneflow (line 21) | namespace oneflow { function GetOpSbpSignature (line 26) | Maybe GetOpSbpSignature(user_op::SbpContext* ctx) { function GetOpGradSbpSignature (line 38) | Maybe GetOpGradSbpSignature(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/reshape_like_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/reshape_op.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/user/ops/reshape_user_op_util.cpp type oneflow (line 21) | namespace oneflow { function FOR_RANGE (line 26) | FOR_RANGE(int, axis, 0, reshape.NumAxes()) { function FOR_RANGE (line 43) | FOR_RANGE(int, axis, 0, reshape.NumAxes()) { function FOR_RANGE (line 81) | FOR_RANGE(int, axis, 0, origin.NumAxes()) { function FowardRankMesh (line 180) | void FowardRankMesh(size_t depth, size_t max_depth, std::deque& r... function GenRankMeshSubset (line 198) | void GenRankMeshSubset(size_t mesh_depth, std::vector... FILE: oneflow/user/ops/reshape_user_op_util.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/ops/reshape_user_op_util_test.cpp type oneflow (line 20) | namespace oneflow { type test (line 21) | namespace test { function NdSbpIn2OutSignaturesToString (line 25) | std::string NdSbpIn2OutSignaturesToString( function NdSbpSignatureGroupsToString (line 46) | std::string NdSbpSignatureGroupsToString( function TestEnumerateNdSbpIn2OutSignatures (line 68) | void TestEnumerateNdSbpIn2OutSignatures( function TEST (line 111) | TEST(ReshapeUserOpUtil, EnumerateNdSbpIn2OutSignatures) { FILE: oneflow/user/ops/rms_norm_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/roc_auc_score_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/roi_align_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/roll_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 30) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/rrelu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, axis, 0, in_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/same_padding_op.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/scalar_bitwise_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/scalar_by_tensor_op.cpp type oneflow (line 19) | namespace oneflow { function TensorDescInferFn (line 23) | Maybe TensorDescInferFn(user_op::InferContext* ctx) { function DataTypeInferFn (line 34) | Maybe DataTypeInferFn(user_op::InferContext* ctx) { function GetBasicSbpSignature (line 44) | Maybe GetBasicSbpSignature(user_op::SbpContext* ctx) { function GetSbpFn (line 57) | GetSbpFn MakeGetSbpFn(GetSbpFn extra) { FILE: oneflow/user/ops/scalar_logical_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/scalar_math_op.cpp type oneflow (line 19) | namespace oneflow { function GetSbp4ScalarMath (line 23) | Maybe GetSbp4ScalarMath(user_op::SbpContext* ctx) { function GetSbp4ScalarMul (line 31) | Maybe GetSbp4ScalarMul(user_op::SbpContext* ctx) { function FOR_RANGE (line 69) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 90) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/scaled_dot_product_flash_attention_op.cpp type oneflow (line 24) | namespace oneflow { FILE: oneflow/user/ops/search_sorted_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/selu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/sigmoid_cross_entropy_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 24) | FOR_RANGE(int64_t, i, 0, num_out_axes) { function FOR_RANGE (line 61) | FOR_RANGE(int64_t, i, 0, num_dy_axes) { FILE: oneflow/user/ops/silu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/skip_layer_norm_op.cpp type oneflow (line 19) | namespace oneflow { function InferParamDataType (line 23) | oneflow::DataType InferParamDataType(const DataType x_data_type) { FILE: oneflow/user/ops/skip_rms_norm_op.cpp type oneflow (line 19) | namespace oneflow { function InferParamDataType (line 23) | oneflow::DataType InferParamDataType(const DataType x_data_type) { FILE: oneflow/user/ops/slice_op.cpp type oneflow (line 22) | namespace oneflow { function IsFullSlice (line 25) | bool IsFullSlice(int64_t start, int64_t stop, int64_t step, int64_t si... function FOR_RANGE (line 52) | FOR_RANGE(int64_t, axis, 0, ndim) { function FOR_RANGE (line 78) | FOR_RANGE(size_t, i, 0, step_vec.size()) { function FOR_RANGE (line 138) | FOR_RANGE(int64_t, axis, 0, input_desc.shape().NumAxes()) { function FOR_RANGE (line 158) | FOR_RANGE(size_t, i, 0, dim_vec.size()) { function FOR_RANGE (line 188) | FOR_RANGE(size_t, i, 0, dim_vec.size()) { function FOR_RANGE (line 232) | FOR_RANGE(int, i, 0, ndim) { FILE: oneflow/user/ops/smooth_l1_loss_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 24) | FOR_RANGE(int64_t, i, 0, input_shape.NumAxes()) { function FOR_RANGE (line 73) | FOR_RANGE(int64_t, i, 0, input_shape.NumAxes()) { FILE: oneflow/user/ops/softmax_cross_entropy_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 25) | FOR_RANGE(int64_t, i, 0, num_out_axes) { function FOR_RANGE (line 51) | FOR_RANGE(int64_t, i, 0, num_out_axes) { function FOR_RANGE (line 86) | FOR_RANGE(int64_t, i, 0, num_dy_axes) { function FOR_RANGE (line 113) | FOR_RANGE(int64_t, i, 0, dy_desc.shape().NumAxes()) { FILE: oneflow/user/ops/softmax_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, axis, 0, in_tensor.shape().NumAxes() - 1) { function FOR_RANGE (line 60) | FOR_RANGE(int64_t, axis, 0, y_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/softplus_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 58) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/softshrink_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 58) | FOR_RANGE(int64_t, i, 0, y_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/softsign_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 42) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/sort_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 25) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/sparse_cross_entropy_op.cpp type oneflow (line 19) | namespace oneflow { function CheckPredictionLabelDesc (line 23) | Maybe CheckPredictionLabelDesc(const user_op::TensorDesc* predic... function InferTensorDescFn (line 47) | Maybe InferTensorDescFn(user_op::InferContext* ctx) { function InferGradTensorDescFn (line 57) | Maybe InferGradTensorDescFn(user_op::InferContext* ctx) { function InferDataType (line 70) | Maybe InferDataType(user_op::InferContext* ctx) { function InferDataTypeGrad (line 81) | Maybe InferDataTypeGrad(user_op::InferContext* ctx) { FILE: oneflow/user/ops/sparse_softmax_cross_entropy_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDescFn (line 23) | Maybe InferTensorDescFn(user_op::InferContext* ctx) { function InferGradTensorDescFn (line 53) | Maybe InferGradTensorDescFn(user_op::InferContext* ctx) { function InferDataType (line 83) | Maybe InferDataType(user_op::InferContext* ctx) { function InferDataTypeGrad (line 93) | Maybe InferDataTypeGrad(user_op::InferContext* ctx) { function AddSignature (line 107) | Maybe AddSignature(user_op::SbpContext* ctx) { function AddMsSignature (line 117) | Maybe AddMsSignature(user_op::SbpContext* ctx) { function AddGradSignature (line 135) | Maybe AddGradSignature(user_op::SbpContext* ctx) { function AddGradMsSignature (line 145) | Maybe AddGradMsSignature(user_op::SbpContext* ctx) { function GetSbpFn (line 163) | Maybe GetSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/split_like_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 27) | FOR_RANGE(int64_t, i, 0, like_num_axes) { function FOR_RANGE (line 34) | FOR_RANGE(int32_t, i, 0, like_arg_size) { like_arg_vec.emplace_back("l... function FOR_RANGE (line 35) | FOR_RANGE(int64_t, i, like_num_axes, in_num_axes) { function FOR_RANGE (line 77) | FOR_RANGE(int32_t, i, 0, ctx->outputs().size()) { function FOR_RANGE (line 122) | FOR_RANGE(int32_t, i, 0, ctx->outputs().size()) { function FOR_RANGE (line 130) | FOR_RANGE(int32_t, i, 0, user_op_conf.input_size("like")) { FILE: oneflow/user/ops/sqrt_square_sum_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, num_x_axes) { FILE: oneflow/user/ops/square_relu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 37) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 67) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/square_sum_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, num_x_axes) { FILE: oneflow/user/ops/squeeze_op.cpp type oneflow (line 19) | namespace oneflow { function TransformNegativeAxesToPositive (line 23) | Maybe TransformNegativeAxesToPositive(const std::vector... function CheckAndLabelAxesToSqueezeMinusOne (line 34) | Maybe CheckAndLabelAxesToSqueezeMinusOne(const AxisVector& axes,... function FOR_RANGE (line 53) | FOR_RANGE(int32_t, in_axis, 0, dim_vec.size()) { FILE: oneflow/user/ops/ssp_variable_proxy_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, var_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/stack_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 44) | FOR_RANGE(int64_t, i, 0, in_num_axes + 1) { function FOR_RANGE (line 81) | FOR_RANGE(int64_t, i, 0, first_in_desc.shape().NumAxes()) { function FOR_RANGE (line 125) | FOR_RANGE(int32_t, i, 0, like_arg_size) { like_arg_vec.emplace_back("l... function FOR_RANGE (line 126) | FOR_RANGE(int64_t, i, 0, like_num_axes) { function FOR_RANGE (line 165) | FOR_RANGE(int32_t, i, 0, ctx->outputs().size()) { function FOR_RANGE (line 212) | FOR_RANGE(int32_t, i, 0, ctx->outputs().size()) { function FOR_RANGE (line 220) | FOR_RANGE(int32_t, i, 0, user_op_conf.input_size("like")) { FILE: oneflow/user/ops/stft_op.cpp type oneflow (line 20) | namespace oneflow { function Stride (line 22) | const Stride InferOutputStride(const Shape& in_shape, bool onesided = ... function Shape (line 36) | const Shape InferOutputShape(const Shape& in_shape, bool onesided = true, FILE: oneflow/user/ops/summary_ops.cpp type oneflow (line 19) | namespace oneflow { function CheckStepShape (line 23) | Maybe CheckStepShape(const Shape* step) { function CheckStepShapeInCtx (line 28) | Maybe CheckStepShapeInCtx(user_op::InferContext* ctx) { function CheckInAndStepScalar (line 33) | Maybe CheckInAndStepScalar(user_op::InferContext* ctx) { FILE: oneflow/user/ops/tanh_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/tensor_buffer_ops.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { function FOR_RANGE (line 54) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes() - instance_dims) { function FOR_RANGE (line 146) | FOR_RANGE(int64_t, i, 0, conf.output_size("out")) { function FOR_RANGE (line 200) | FOR_RANGE(int64_t, i, 0, conf.output_size("out")) { FILE: oneflow/user/ops/tensor_constant_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/tf_pool_op.cpp type oneflow (line 20) | namespace oneflow { function GetComputationCost (line 27) | Maybe GetComputationCost(user_op::ComputeComplexityFnContext* ... function TensorDescInferFn (line 44) | TensorDescInferFn MakeFwTensorDescInferFn(const int32_t dim) { function BwTensorDescInferFn (line 69) | Maybe BwTensorDescInferFn(user_op::InferContext* ctx) { function FwInferDataType (line 75) | Maybe FwInferDataType(user_op::InferContext* ctx) { function BwInferDataType (line 80) | Maybe BwInferDataType(user_op::InferContext* ctx) { function FwGetSbpFn (line 85) | Maybe FwGetSbpFn(user_op::SbpContext* ctx) { function BwGetSbpFn (line 93) | Maybe BwGetSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/tf_prelu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 29) | FOR_RANGE(int64_t, i, 1, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 45) | FOR_RANGE(int64_t, i, 1, x_desc.shape().NumAxes()) { function FOR_RANGE (line 78) | FOR_RANGE(int64_t, i, 1, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 97) | FOR_RANGE(int64_t, i, 1, x_desc.shape().NumAxes()) { FILE: oneflow/user/ops/threshold_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 32) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes()) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/throw_error_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/to_contiguous_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/top_k_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 25) | FOR_RANGE(int64_t, i, 0, in_tensor.shape().NumAxes() - 1) { FILE: oneflow/user/ops/transpose_ops.cpp type oneflow (line 19) | namespace oneflow { function CheckIsPerm (line 21) | void CheckIsPerm(const std::vector& perm) { function FOR_RANGE (line 35) | FOR_RANGE(int32_t, i, 0, perm.size()) { function FOR_RANGE (line 55) | FOR_RANGE(size_t, i, 0, perm.size()) { out_shape.Set(i, in_shape.At(pe... FILE: oneflow/user/ops/tril_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes() - 2) { function FOR_RANGE (line 57) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes() - 2) { FILE: oneflow/user/ops/triu_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes() - 2) { FILE: oneflow/user/ops/trunc_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/tuple_identity_op.cpp type oneflow (line 20) | namespace oneflow { FILE: oneflow/user/ops/two_stage_reduce_ops.cpp type oneflow (line 21) | namespace oneflow { function InferReduceDeviceStageDtypeFn (line 25) | Maybe InferReduceDeviceStageDtypeFn(user_op::InferContext* ctx) { function InferReduceDeviceStageLogicalTensorDescFn (line 32) | Maybe InferReduceDeviceStageLogicalTensorDescFn(user_op::InferCo... function InferReduceDeviceStagePhysicalTensorDescFn (line 72) | Maybe InferReduceDeviceStagePhysicalTensorDescFn(user_op::InferC... function InferReduceDeviceStageGradDtypeFn (line 92) | Maybe InferReduceDeviceStageGradDtypeFn(user_op::InferContext* c... function InferReduceDeviceStageGradTensorDescFn (line 103) | Maybe InferReduceDeviceStageGradTensorDescFn(user_op::InferConte... function InferReduceGlobalStageDtypeFn (line 109) | Maybe InferReduceGlobalStageDtypeFn(user_op::InferContext* ctx) { function InferReduceGlobalStageTensorDescFn (line 119) | Maybe InferReduceGlobalStageTensorDescFn(user_op::InferContext* ... function InferReduceGlobalStageGradDtypeFn (line 148) | Maybe InferReduceGlobalStageGradDtypeFn(user_op::InferContext* c... function InferReduceGlobalStageGradTensorDescFn (line 161) | Maybe InferReduceGlobalStageGradTensorDescFn(user_op::InferConte... function GetReduceDeviceStageSbpFn (line 169) | Maybe GetReduceDeviceStageSbpFn(user_op::SbpContext* ctx) { function GetReduceDeviceStageGradSbpFn (line 190) | Maybe GetReduceDeviceStageGradSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/unfold_fold_op.cpp type oneflow (line 21) | namespace oneflow { function UnfoldTensorDescInferFn (line 25) | Maybe UnfoldTensorDescInferFn(user_op::InferContext* ctx) { function SetUnfoldDTypeFn (line 65) | Maybe SetUnfoldDTypeFn(user_op::InferContext* ctx) { function GetUnfoldSbpFn (line 70) | Maybe GetUnfoldSbpFn(user_op::SbpContext* ctx) { function FoldTensorDescInferFn (line 77) | Maybe FoldTensorDescInferFn(user_op::InferContext* ctx) { function FoldDTypeFn (line 125) | Maybe FoldDTypeFn(user_op::InferContext* ctx) { function GetFoldSbpFn (line 130) | Maybe GetFoldSbpFn(user_op::SbpContext* ctx) { FILE: oneflow/user/ops/unfold_tensor_op.cpp type oneflow (line 20) | namespace oneflow { function FOR_RANGE (line 25) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { function FOR_RANGE (line 52) | FOR_RANGE(int32_t, d, 0, in_dim) { function FOR_RANGE (line 74) | FOR_RANGE(int64_t, i, 0, x_tensor.shape().NumAxes()) { FILE: oneflow/user/ops/unique_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/unique_with_counts_op.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/unpack_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, in.shape().NumAxes()) { FILE: oneflow/user/ops/unsorted_batch_segment_sum_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 25) | FOR_RANGE(int64_t, i, 0, segment_ids_num_axes - 1) { function FOR_RANGE (line 51) | FOR_RANGE(int64_t, i, 0, segment_ids.shape().NumAxes() - 1) { FILE: oneflow/user/ops/unsorted_segment_sum_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 27) | FOR_RANGE(int64_t, i, 0, segment_ids_num_axes) { function FOR_RANGE (line 34) | FOR_RANGE(int64_t, i, 0, data_num_axes) { function FOR_RANGE (line 88) | FOR_RANGE(int64_t, i, 0, segment_ids_num_axes) { function FOR_RANGE (line 102) | FOR_RANGE(int64_t, i, 0, data_num_axes) { function FOR_RANGE (line 141) | FOR_RANGE(int64_t, i, 0, axis) { CHECK_EQ_OR_RETURN(like_shape.At(i), ... function FOR_RANGE (line 143) | FOR_RANGE(int64_t, i, axis + 1, like_shape.NumAxes()) { FILE: oneflow/user/ops/upsample_op.cpp function UpsamplingInferLogicalDesc (line 22) | typename std::enable_if<(N <= 3), Maybe>::type UpsamplingInferLogi... type oneflow (line 105) | namespace oneflow { FILE: oneflow/user/ops/util_ops.cpp type oneflow (line 19) | namespace oneflow { FILE: oneflow/user/ops/variance_op.cpp type oneflow (line 21) | namespace oneflow { FILE: oneflow/user/ops/vector_matrix_product_op.cpp type oneflow (line 19) | namespace oneflow { function InferTensorDesc4VectorMatrixProduct (line 23) | Maybe InferTensorDesc4VectorMatrixProduct(user_op::InferContext*... function InferDataType4VectorMatrixProduct (line 33) | Maybe InferDataType4VectorMatrixProduct(user_op::InferContext* c... function InferTensorDesc4VectorMatrixProductGradA (line 41) | Maybe InferTensorDesc4VectorMatrixProductGradA(user_op::InferCon... function InferTensorDesc4VectorMatrixProductGradB (line 52) | Maybe InferTensorDesc4VectorMatrixProductGradB(user_op::InferCon... function InferDataType4Grad (line 65) | Maybe InferDataType4Grad(user_op::InferContext* ctx) { FILE: oneflow/user/ops/where_op.cpp type oneflow (line 20) | namespace oneflow { function GetBroadcastShape (line 24) | Maybe GetBroadcastShape(const Shape& cond_shape, const Shape& x... FILE: oneflow/user/ops/zero_like_op.cpp type oneflow (line 19) | namespace oneflow { function FOR_RANGE (line 23) | FOR_RANGE(int64_t, i, 0, like_tensor.shape().NumAxes()) { FILE: oneflow/user/summary/crc32c.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/summary/env_time.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/summary/event_writer_helper.cpp type oneflow (line 30) | namespace oneflow { type summary (line 32) | namespace summary { function SetPluginData (line 38) | void SetPluginData(SummaryMetadata* metadata, const char* name) { function FillScalarInSummary (line 44) | Maybe FillScalarInSummary(const float& value, const std::strin... function FillHistogramInSummary (line 55) | Maybe FillHistogramInSummary(const user_op::Tensor& value, con... function WriteImageDataFn (line 71) | void WriteImageDataFn(png_structp png_ptr, png_bytep data, png_size_... function WriteImageToBuffer (line 76) | bool WriteImageToBuffer(const uint8_t* image, int width, int height,... function FillImageInSummary (line 117) | Maybe FillImageInSummary(const user_op::Tensor& tensor, const ... type EventWriterHelper (line 163) | struct EventWriterHelper { method WritePbToFile (line 164) | static void WritePbToFile(int64_t step, const std::string& value) { method WriteScalarToFile (line 174) | static void WriteScalarToFile(int64_t step, float value, const std... method WriteHistogramToFile (line 182) | static void WriteHistogramToFile(int64_t step, const user_op::Tens... method WriteImageToFile (line 191) | static void WriteImageToFile(int64_t step, const user_op::Tensor& ... FILE: oneflow/user/summary/event_writer_helper.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/summary/events_writer.cpp type oneflow (line 20) | namespace oneflow { type summary (line 22) | namespace summary { FILE: oneflow/user/summary/events_writer.h function namespace (line 27) | namespace oneflow { FILE: oneflow/user/summary/histogram.cpp type oneflow (line 21) | namespace oneflow { type summary (line 23) | namespace summary { FILE: oneflow/user/summary/histogram.h function namespace (line 22) | namespace oneflow { FILE: oneflow/user/summary/plan_to_physical_graph.cpp type oneflow (line 24) | namespace oneflow { type summary (line 26) | namespace summary { function PlanToPhysicalGraphFile (line 28) | void PlanToPhysicalGraphFile(const Plan& plan) { FILE: oneflow/user/summary/plan_to_physical_graph.h function namespace (line 21) | namespace oneflow { FILE: oneflow/user/summary/summary_converter.h function namespace (line 23) | namespace oneflow { FILE: oneflow/user/utils/pool_util.cpp type oneflow (line 19) | namespace oneflow { function Shape (line 54) | Shape Params3D::GetYShape() const { function Shape (line 76) | Shape Params3D::GetXShape5D() const { function Shape (line 80) | Shape Params3D::GetYShape5D() const { FILE: oneflow/user/utils/pool_util.h type small_vector (line 24) | typedef small_vector FixedDimVector; type small_vector (line 25) | typedef small_vector FixedVector; function class (line 27) | class Params3D { function Get3DVecType (line 60) | enum class Get3DVecType { kPad, kNonPad }; FILE: python/oneflow/_C/__init__.py function allclose (line 21) | def allclose(input, other, atol=1e-08, rtol=1e-05, equal_nan=False): function _log_api_usage_once (line 25) | def _log_api_usage_once(event): FILE: python/oneflow/_C/_nn.py function _parse_to (line 27) | def _parse_to( function _parse_to (line 39) | def _parse_to( function _parse_to (line 46) | def _parse_to( function _parse_to (line 52) | def _parse_to(*args, **kwargs): FILE: python/oneflow/__init__.py function oneflow_deprecate (line 79) | def oneflow_deprecate(*api_names, **kwargs): function is_deprecated (line 87) | def is_deprecated(func_or_class): function use_deterministic_algorithms (line 94) | def use_deterministic_algorithms(mode, *, warn_only=False): class ExitHook (line 306) | class ExitHook: method __init__ (line 307) | def __init__(self): method is_normal_exit (line 326) | def is_normal_exit(self): function atexit_hook (line 335) | def atexit_hook(hook): FILE: python/oneflow/__main__.py function main (line 24) | def main(): FILE: python/oneflow/_dynamo/__init__.py function allow_in_graph (line 24) | def allow_in_graph(fn): FILE: python/oneflow/_utils.py class KeyErrorMessage (line 21) | class KeyErrorMessage(str): method __repr__ (line 24) | def __repr__(self): class ExceptionWrapper (line 28) | class ExceptionWrapper(object): method __init__ (line 31) | def __init__(self, exc_info=None, where="in background"): method reraise (line 40) | def reraise(self): function _flatten_dense_tensors (line 59) | def _flatten_dense_tensors(tensors): function _unflatten_dense_tensors (line 84) | def _unflatten_dense_tensors(flat, tensors): FILE: python/oneflow/amp/autocast_mode.py function autocast_decorator (line 27) | def autocast_decorator(autocast_instance, func): class autocast (line 36) | class autocast(object): method __init__ (line 148) | def __init__( method __enter__ (line 183) | def __enter__(self): method __exit__ (line 189) | def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any): method __call__ (line 192) | def __call__(self, func): FILE: python/oneflow/amp/grad_scaler.py class GradScaler (line 18) | class GradScaler(object): method __init__ (line 19) | def __init__( method _generate_conf_for_graph (line 36) | def _generate_conf_for_graph(self, train_conf): class StaticGradScaler (line 42) | class StaticGradScaler(object): method __init__ (line 43) | def __init__(self, scale_factor): method _generate_conf_for_graph (line 49) | def _generate_conf_for_graph(self, train_conf): FILE: python/oneflow/ao/quantization.py class DeQuantStub (line 18) | class DeQuantStub: method __init__ (line 19) | def __init__(self, *args, **kwargs): class QuantStub (line 25) | class QuantStub: method __init__ (line 26) | def __init__(self, *args, **kwargs): FILE: python/oneflow/asyncs/thread.py class thread (line 21) | class thread: method __init__ (line 41) | def __init__(self, worker_thread: Thread): method __enter__ (line 45) | def __enter__(self): method __exit__ (line 48) | def __exit__(self, type, value, traceback): FILE: python/oneflow/autograd/autograd.py function grad (line 25) | def grad( function backward (line 81) | def backward( FILE: python/oneflow/autograd/autograd_function.py class Function (line 21) | class Function(AutogradFunctionBase): method __init__ (line 51) | def __init__(self): method __call__ (line 54) | def __call__(self, *inputs): method apply (line 61) | def apply(cls, *inputs): method forward (line 70) | def forward(ctx, *inputs): method backward (line 79) | def backward(ctx, *out_grads): FILE: python/oneflow/autograd/autograd_mode.py function is_grad_enabled (line 21) | def is_grad_enabled(): class inference_mode (line 28) | class inference_mode: method __init__ (line 59) | def __init__(self, mode=True): method __call__ (line 62) | def __call__(self, func): method __enter__ (line 69) | def __enter__(self): method __exit__ (line 73) | def __exit__(self, exc_type, exc_val, exc_tb): class enable_grad (line 77) | class enable_grad: method __call__ (line 105) | def __call__(self, func): method __enter__ (line 112) | def __enter__(self): method __exit__ (line 116) | def __exit__(self, exc_type, exc_val, exc_tb): class no_grad (line 120) | class no_grad: method __call__ (line 151) | def __call__(self, func): method __enter__ (line 158) | def __enter__(self): method __exit__ (line 162) | def __exit__(self, exc_type, exc_val, exc_tb): class set_grad_enabled (line 166) | class set_grad_enabled: method __init__ (line 197) | def __init__(self, is_train=True): method __call__ (line 202) | def __call__(self, func): method __enter__ (line 212) | def __enter__(self): method __exit__ (line 218) | def __exit__(self, exc_type, exc_val, exc_tb): FILE: python/oneflow/autograd/functional.py function _as_tuple_nocheck (line 27) | def _as_tuple_nocheck(x): function _as_tuple (line 36) | def _as_tuple(inp, arg_name=None, fn_name=None): function _tuple_postprocess (line 63) | def _tuple_postprocess(res, to_unpack): function _grad_preprocess (line 81) | def _grad_preprocess(inputs, create_graph, need_graph): function _grad_postprocess (line 99) | def _grad_postprocess(inputs, create_graph): function _validate_v (line 111) | def _validate_v(v, other, is_other_tuple): function _check_requires_grad (line 132) | def _check_requires_grad(inputs, input_type, strict): function _autograd_grad (line 175) | def _autograd_grad( function _fill_in_zeros (line 207) | def _fill_in_zeros(grads, refs, strict, create_graph, stage): function vjp (line 265) | def vjp(func, inputs, v=None, create_graph=False, strict=False): function jvp (line 361) | def jvp(func, inputs, v=None, create_graph=False, strict=False): function _construct_standard_basis_for (line 474) | def _construct_standard_basis_for(tensors, tensor_numels: Tuple[int, ...]): function _jacfwd (line 510) | def _jacfwd(func, inputs, strict=False, vectorize=False): function jacobian (line 531) | def jacobian( function hessian (line 786) | def hessian( function vhp (line 926) | def vhp(func, inputs, v=None, create_graph=False, strict=False): function hvp (line 1030) | def hvp(func, inputs, v=None, create_graph=False, strict=False): FILE: python/oneflow/autograd/graph.py class saved_tensors_hooks (line 22) | class saved_tensors_hooks: method __init__ (line 83) | def __init__( method __enter__ (line 91) | def __enter__(self): method __exit__ (line 96) | def __exit__(self, *args: Any): FILE: python/oneflow/autograd/profiler.py function record_function (line 19) | def record_function(): class emit_nvtx (line 23) | class emit_nvtx: method __init__ (line 24) | def __init__(self): FILE: python/oneflow/autoprof/util.py function get_sole_value (line 29) | def get_sole_value(x: Iterable[T]) -> T: function get_pytorch_cpu_kernel_time (line 35) | def get_pytorch_cpu_kernel_time(prof) -> Union[str, float]: function get_oneflow_cpu_kernel_time (line 46) | def get_oneflow_cpu_kernel_time(prof) -> Union[str, float]: function get_pytorch_gpu_kernel_time (line 55) | def get_pytorch_gpu_kernel_time(prof) -> Union[str, float]: function get_oneflow_gpu_kernel_time (line 65) | def get_oneflow_gpu_kernel_time(prof) -> Union[str, float]: function get_oneflow_gpu_kernel_bandwidth (line 75) | def get_oneflow_gpu_kernel_bandwidth(prof) -> str: function get_pytorch_cpu_end_to_end_time (line 86) | def get_pytorch_cpu_end_to_end_time(prof) -> float: function get_oneflow_cpu_end_to_end_time (line 94) | def get_oneflow_cpu_end_to_end_time(prof) -> float: function add_row (line 102) | def add_row(profs, writer, f): function print_summary_from_csv (line 144) | def print_summary_from_csv(filename) -> None: FILE: python/oneflow/backends/cuda/__init__.py class cuMatmulMode (line 20) | class cuMatmulMode: method __getattr__ (line 21) | def __getattr__(self, name): method __setattr__ (line 30) | def __setattr__(self, name, value): FILE: python/oneflow/backends/mps/__init__.py function is_available (line 20) | def is_available() -> bool: FILE: python/oneflow/comm/comm_ops.py function all_reduce (line 21) | def all_reduce(tensor): function all_gather (line 58) | def all_gather(tensor_list, tensor): function all_gather_into_tensor (line 116) | def all_gather_into_tensor(output_tensor, input_tensor): function broadcast (line 186) | def broadcast(tensor, src): function scatter (line 223) | def scatter(tensor, scatter_list=None, src=0): function reduce (line 258) | def reduce(tensor, dst): function all_to_all (line 279) | def all_to_all(output_tensor_list, input_tensor_list): function barrier (line 319) | def barrier(): function reduce_scatter (line 327) | def reduce_scatter(output, input_list): function reduce_scatter_tensor (line 354) | def reduce_scatter_tensor(output_tensor, input_tensor): function gather (line 413) | def gather(tensor, gather_list=None, dst=0): FILE: python/oneflow/cuda/__init__.py function is_available (line 26) | def is_available() -> bool: function device_count (line 33) | def device_count() -> int: function current_device (line 38) | def current_device() -> int: function get_device_properties (line 43) | def get_device_properties(device: Union[flow.device, str, int] = None): function get_device_capability (line 59) | def get_device_capability( function get_device_name (line 80) | def get_device_name(device: Optional[Union[flow.device, str, int]] = Non... function manual_seed_all (line 97) | def manual_seed_all(seed) -> None: function manual_seed (line 113) | def manual_seed(seed: int) -> None: function set_device (line 134) | def set_device(device: Union[flow.device, str, int]) -> None: function synchronize (line 159) | def synchronize(device: Union[flow.device, str, int, None] = None) -> None: function empty_cache (line 182) | def empty_cache() -> None: function mem_get_info (line 198) | def mem_get_info(device: Any = None) -> Tuple[int, int]: class Event (line 219) | class Event: method __init__ (line 220) | def __init__(self): FILE: python/oneflow/cuda/_utils.py function _get_current_device_index (line 20) | def _get_current_device_index() -> int: function _get_device_index (line 32) | def _get_device_index( FILE: python/oneflow/cuda/amp/autocast_mode.py class autocast (line 20) | class autocast(flow.amp.autocast_mode.autocast): method __init__ (line 26) | def __init__( method __enter__ (line 36) | def __enter__(self): method __exit__ (line 39) | def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any): # type:... method __call__ (line 42) | def __call__(self, func): FILE: python/oneflow/cuda/random.py function get_rng_state (line 22) | def get_rng_state(device: Union[int, str, flow.device] = "cuda") -> Tensor: function get_rng_state_all (line 42) | def get_rng_state_all() -> List[Tensor]: function set_rng_state (line 51) | def set_rng_state( function set_rng_state_all (line 78) | def set_rng_state_all(new_states: Iterable[Tensor]) -> None: FILE: python/oneflow/distributed/__init__.py class ReduceOp (line 24) | class ReduceOp: function is_initialized (line 34) | def is_initialized() -> bool: function get_rank (line 48) | def get_rank(group=None) -> int: function get_world_size (line 57) | def get_world_size(group=None) -> int: function send (line 66) | def send(tensor: flow.Tensor, dst: int, group=None, tag: int = 0) -> None: function recv (line 76) | def recv(tensor: flow.Tensor, src: int, group=None, tag: int = 0) -> None: function broadcast (line 86) | def broadcast( function barrier (line 98) | def barrier(group=None, async_op=False, device_ids=None) -> None: function all_reduce (line 109) | def all_reduce( function all_gather (line 122) | def all_gather( function reduce (line 137) | def reduce( function all_to_all (line 150) | def all_to_all( function reduce_scatter (line 165) | def reduce_scatter( function gather (line 182) | def gather( function is_available (line 198) | def is_available(): FILE: python/oneflow/distributed/launch.py function parse_args (line 31) | def parse_args(): function main (line 103) | def main(): FILE: python/oneflow/distributions/categorical.py class Categorical (line 26) | class Categorical(Distribution): method __init__ (line 66) | def __init__(self, probs=None, logits=None, validate_args=None): method logits (line 103) | def logits(self): method probs (line 106) | def probs(self): method sample (line 109) | def sample(self, sample_shape=flow.Size()): FILE: python/oneflow/distributions/distribution.py class Distribution (line 26) | class Distribution(object): method set_default_validate_args (line 36) | def set_default_validate_args(value): method __init__ (line 52) | def __init__( method expand (line 60) | def expand(self, batch_shape, _instance=None): method batch_shape (line 82) | def batch_shape(self): method event_shape (line 89) | def event_shape(self): method mean (line 96) | def mean(self): method mode (line 103) | def mode(self): method variance (line 110) | def variance(self): method stddev (line 117) | def stddev(self): method sample (line 123) | def sample(self, sample_shape=oneflow.Size()): method rsample (line 131) | def rsample(self, sample_shape=oneflow.Size()): method sample_n (line 139) | def sample_n(self, n): method log_prob (line 149) | def log_prob(self, value): method cdf (line 159) | def cdf(self, value): method icdf (line 169) | def icdf(self, value): method enumerate_support (line 179) | def enumerate_support(self, expand=True): method entropy (line 203) | def entropy(self): method perplexity (line 212) | def perplexity(self): method _extended_shape (line 221) | def _extended_shape(self, sample_shape=oneflow.Size()): method _validate_sample (line 235) | def _validate_sample(self, value): method _get_checked_instance (line 289) | def _get_checked_instance(self, cls, _instance=None): method __repr__ (line 299) | def __repr__(self): FILE: python/oneflow/distributions/utils.py function logits_to_probs (line 28) | def logits_to_probs(logits, is_binary=False): function clamp_probs (line 40) | def clamp_probs(probs): function probs_to_logits (line 45) | def probs_to_logits(probs, is_binary=False): FILE: python/oneflow/env.py function get_local_rank (line 20) | def get_local_rank(): function get_rank (line 31) | def get_rank(): function get_node_size (line 42) | def get_node_size(): function get_world_size (line 52) | def get_world_size(): function init_rdma (line 62) | def init_rdma(): function rdma_is_initialized (line 80) | def rdma_is_initialized(): function destory_rdma (line 90) | def destory_rdma(): FILE: python/oneflow/experimental/load_mnist.py function get_sha256hash (line 24) | def get_sha256hash(file_path, Bytes=1024): function download_mnist_file (line 37) | def download_mnist_file(out_path, url): function get_mnist_file (line 49) | def get_mnist_file(sha256, url, out_dir): function load_mnist (line 62) | def load_mnist( FILE: python/oneflow/fft/__init__.py function fft (line 20) | def fft(input, n=None, dim=-1, norm=None) -> Tensor: function ifft (line 69) | def ifft(input, n=None, dim=-1, norm=None) -> Tensor: function fft2 (line 104) | def fft2(input, s=None, dim=(-2, -1), norm=None) -> Tensor: function ifft2 (line 146) | def ifft2(input, s=None, dim=(-2, -1), norm=None) -> Tensor: function fftn (line 181) | def fftn(input, s=None, dim=None, norm=None) -> Tensor: function ifftn (line 222) | def ifftn(input, s=None, dim=None, norm=None) -> Tensor: function rfft (line 255) | def rfft(input, n=None, dim=-1, norm=None) -> Tensor: function irfft (line 306) | def irfft(input, n=None, dim=-1, norm=None) -> Tensor: function rfft2 (line 357) | def rfft2(input, s=None, dim=(-2, -1), norm=None) -> Tensor: function irfft2 (line 397) | def irfft2(input, s=None, dim=(-2, -1), norm=None) -> Tensor: function rfftn (line 451) | def rfftn(input, s=None, dim=None, norm=None) -> Tensor: function irfftn (line 491) | def irfftn(input, s=None, dim=None, norm=None) -> Tensor: function hfft (line 543) | def hfft(input, n=None, dim=-1, norm=None) -> Tensor: function ihfft (line 628) | def ihfft(input, n=None, dim=-1, norm=None) -> Tensor: function hfft2 (line 680) | def hfft2(input, s=None, dim=(-2, -1), norm=None) -> Tensor: function ihfft2 (line 743) | def ihfft2(input, s=None, dim=(-2, -1), norm=None) -> Tensor: function hfftn (line 778) | def hfftn(input, s=None, dim=None, norm=None) -> Tensor: function ihfftn (line 840) | def ihfftn(input, s=None, dim=None, norm=None) -> Tensor: FILE: python/oneflow/framework/args_tree.py function _is_raw_type (line 21) | def _is_raw_type(value, raw_type): class NamedArg (line 33) | class NamedArg(object): method __init__ (line 44) | def __init__( method prefix (line 54) | def prefix(self): method name (line 57) | def name(self): method global_index (line 60) | def global_index(self): method value (line 63) | def value(self): method is_leaf (line 67) | def is_leaf(self): method set_value (line 76) | def set_value(self, value): method __repr__ (line 81) | def __repr__(self): class ArgsTree (line 120) | class ArgsTree(object): method __init__ (line 121) | def __init__( method gen_name (line 143) | def gen_name(self): method iter_nodes (line 146) | def iter_nodes(self): method iter_named_nodes (line 187) | def iter_named_nodes(self): method _construct_named_io_args (line 192) | def _construct_named_io_args(self, value, prefix: str, name: str) -> N... method map_tuple_leaf (line 222) | def map_tuple_leaf(self, map_function: Callable): method map_leaf (line 264) | def map_leaf(self, map_function: Callable): method _execute_mapping (line 277) | def _execute_mapping(self, value, map_function): method __repr__ (line 299) | def __repr__(self): FILE: python/oneflow/framework/attr_util.py function get_nested_attribute (line 26) | def get_nested_attribute(owning_object, attrs_chain): function SetProtoAttrValue (line 45) | def SetProtoAttrValue(attr_value, py_value, default_attr_value): FILE: python/oneflow/framework/balanced_splitter.py function BalancedPartNums (line 18) | def BalancedPartNums(total, part_size): function BalancedRanges (line 24) | def BalancedRanges(total, part_size): FILE: python/oneflow/framework/c_api_util.py function CurrentResource (line 35) | def CurrentResource(): function EnvResource (line 40) | def EnvResource(): function GetEnvContext (line 45) | def GetEnvContext(env_proto): function JobBuildAndInferCtx_Open (line 52) | def JobBuildAndInferCtx_Open(job_name): function CurJobBuildAndInferCtx_SetJobConf (line 57) | def CurJobBuildAndInferCtx_SetJobConf(job_config_proto): function InferOpConf (line 63) | def InferOpConf(op_conf_proto, upstream_signature): function IsInterfaceOpConf (line 72) | def IsInterfaceOpConf(op_conf): function GetOpParallelSymbolId (line 80) | def GetOpParallelSymbolId(op_conf_proto): function CheckAndCompleteUserOpConf (line 85) | def CheckAndCompleteUserOpConf(op_conf_proto): function GetFunctionConfigDef (line 93) | def GetFunctionConfigDef(): function GetScopeConfigDef (line 98) | def GetScopeConfigDef(): function GetCurrentJob (line 103) | def GetCurrentJob(): FILE: python/oneflow/framework/check_point_v2.py class _opener (line 66) | class _opener(object): method __init__ (line 67) | def __init__(self, file_like): method __enter__ (line 70) | def __enter__(self): method __exit__ (line 73) | def __exit__(self, *args): class _open_file (line 77) | class _open_file(_opener): method __init__ (line 78) | def __init__(self, path, mode): method __exit__ (line 81) | def __exit__(self, *args): class _open_buffer_reader (line 85) | class _open_buffer_reader(_opener): method __init__ (line 86) | def __init__(self, buffer): class _open_buffer_writer (line 91) | class _open_buffer_writer(_opener): method __exit__ (line 92) | def __exit__(self, *args): function _open_file_like (line 96) | def _open_file_like(path_or_buffer, mode): function _is_path (line 108) | def _is_path(path_or_buffer): function _check_seekable (line 112) | def _check_seekable(f) -> bool: class FileBackendVariableBlob (line 133) | class FileBackendVariableBlob: method __init__ (line 134) | def __init__( method file_path (line 173) | def file_path(self) -> str: method shape (line 177) | def shape(self) -> Tuple[int]: method quant_info (line 181) | def quant_info(self): method dtype (line 185) | def dtype(self) -> oneflow.dtype: method numpy (line 188) | def numpy(self) -> np.ndarray: function _save_tensor_to_disk (line 197) | def _save_tensor_to_disk(tensor: "oneflow.Tensor", dir_name: Union[str, ... function _default_restore_location (line 215) | def _default_restore_location(storage, location=None): function _get_restore_location (line 219) | def _get_restore_location(map_location): function smart_to (line 243) | def smart_to(obj: Any, dest: MAP_LOCATION) -> "oneflow.Tensor": function module_to (line 251) | def module_to(obj: flow.nn.Module, dest: MAP_LOCATION) -> "oneflow.nn.Mo... function _map_location (line 260) | def _map_location(obj: Any, map_location: MAP_LOCATION): function _LoadSingleVariable (line 268) | def _LoadSingleVariable( function _broadcast_py_object (line 288) | def _broadcast_py_object(obj, src: int = 0): function tensor_getstate (line 300) | def tensor_getstate(self): function tensor_setstate (line 358) | def tensor_setstate(self, pickle_dict): function placement_getstate (line 389) | def placement_getstate(self): function placement_setstate (line 396) | def placement_setstate(self, state): function RegisterMethods (line 400) | def RegisterMethods(): function load_if (line 410) | def load_if(condition): function is_dir_and_no_pickle_file (line 427) | def is_dir_and_no_pickle_file(path: FILE_LIKE, support_pytorch_format: b... function legacy_load (line 435) | def legacy_load( function tensor_pickling_context (line 462) | def tensor_pickling_context( function is_oneflow_pickle_file (line 476) | def is_oneflow_pickle_file(path: FILE_LIKE, support_pytorch_format: bool... function load_from_oneflow_single_file (line 494) | def load_from_oneflow_single_file( function is_file_and_support_pytorch_format (line 515) | def is_file_and_support_pytorch_format( function load_from_pytorch_file (line 537) | def load_from_pytorch_file( function is_dir_and_has_pickle_file (line 564) | def is_dir_and_has_pickle_file(path: FILE_LIKE, support_pytorch_format: ... function load_from_oneflow_pickle_dir (line 572) | def load_from_oneflow_pickle_dir( function load (line 596) | def load( function save_one_embedding_info (line 646) | def save_one_embedding_info(state_dict: Any, path: Union[str, Path]) -> ... function save (line 683) | def save( function _save_graph (line 743) | def _save_graph(obj: graph_util.Graph, path: Union[str, Path]): function frombuffer (line 763) | def frombuffer( class ContextData (line 782) | class ContextData: method __init__ (line 783) | def __init__( FILE: python/oneflow/framework/config_util.py function _set_resource_attr (line 27) | def _set_resource_attr(attrs_chain: Union[List[str], str], attr_value, t... function api_load_library (line 78) | def api_load_library(val: str) -> None: function api_numa_aware_cuda_malloc_host (line 87) | def api_numa_aware_cuda_malloc_host(val: bool = True) -> None: function api_reserved_device_mem_mbyte (line 98) | def api_reserved_device_mem_mbyte(val: int) -> None: function api_enable_cudnn_fused_normalization_add_relu (line 108) | def api_enable_cudnn_fused_normalization_add_relu(val: bool) -> None: function api_enable_cudnn_conv_heuristic_search_algo (line 119) | def api_enable_cudnn_conv_heuristic_search_algo(val: bool) -> None: function api_enable_fusion (line 130) | def api_enable_fusion(val: bool = True) -> None: function api_nccl_use_compute_stream (line 141) | def api_nccl_use_compute_stream(val: bool = False) -> None: function api_disable_group_boxing_by_dst_parallel (line 152) | def api_disable_group_boxing_by_dst_parallel(val: bool = False) -> None: function api_nccl_num_streams (line 163) | def api_nccl_num_streams(val: int) -> None: function api_nccl_fusion_threshold_mb (line 174) | def api_nccl_fusion_threshold_mb(val: int) -> None: function api_nccl_fusion_all_reduce_use_buffer (line 185) | def api_nccl_fusion_all_reduce_use_buffer(val: bool) -> None: function api_nccl_fusion_all_reduce (line 196) | def api_nccl_fusion_all_reduce(val: bool) -> None: function api_nccl_fusion_reduce_scatter (line 207) | def api_nccl_fusion_reduce_scatter(val: bool) -> None: function api_nccl_fusion_all_gather (line 218) | def api_nccl_fusion_all_gather(val: bool) -> None: function api_nccl_fusion_reduce (line 229) | def api_nccl_fusion_reduce(val: bool) -> None: function api_nccl_fusion_broadcast (line 240) | def api_nccl_fusion_broadcast(val: bool) -> None: function api_nccl_fusion_max_ops (line 251) | def api_nccl_fusion_max_ops(val: int) -> None: function api_nccl_enable_all_to_all (line 262) | def api_nccl_enable_all_to_all(val: bool) -> None: function api_nccl_enable_mixed_fusion (line 273) | def api_nccl_enable_mixed_fusion(val: bool) -> None: FILE: python/oneflow/framework/distribute.py function split_sbp (line 23) | def split_sbp(dim=None, **kwargs) -> oneflow._oneflow_internal.sbp.sbp: FILE: python/oneflow/framework/docstr/utils.py function _test_docstr (line 21) | def _test_docstr(docstr, verbose=True, optionflags=0, raise_on_error=True): function add_docstr (line 31) | def add_docstr(fun, docstr: str): function reset_docstr (line 35) | def reset_docstr(o, docstr): FILE: python/oneflow/framework/dtype.py function dtypes (line 48) | def dtypes(): function convert_proto_dtype_to_oneflow_dtype (line 52) | def convert_proto_dtype_to_oneflow_dtype(proto_dtype): function convert_oneflow_dtype_to_numpy_dtype (line 79) | def convert_oneflow_dtype_to_numpy_dtype(oneflow_dtype: oneflow.dtype): function convert_numpy_dtype_to_oneflow_dtype (line 85) | def convert_numpy_dtype_to_oneflow_dtype(numpy_dtype: np.dtype): function set_default_tensor_type (line 96) | def set_default_tensor_type(tensor_type): function is_floating_point (line 138) | def is_floating_point(input): FILE: python/oneflow/framework/env_util.py function api_all_device_placement (line 29) | def api_all_device_placement(device_type: str) -> oneflow._oneflow_inter... function check_non_localhost_proxy_and_print_warning (line 55) | def check_non_localhost_proxy_and_print_warning(): function create_env (line 71) | def create_env(): function CompleteEnvProto (line 85) | def CompleteEnvProto(env_proto): function _MakeMachine (line 96) | def _MakeMachine(machines): function _MakeBootstrapConf (line 123) | def _MakeBootstrapConf(bootstrap_info: dict): function _DefaultEnvProto (line 144) | def _DefaultEnvProto(): function _FindFreePort (line 152) | def _FindFreePort(): function CheckAndWarnAbnormalEnvVars (line 159) | def CheckAndWarnAbnormalEnvVars(): function _UpdateDefaultEnvProtoByMultiClientEnvVars (line 169) | def _UpdateDefaultEnvProtoByMultiClientEnvVars(env_proto): class EnvHolder (line 193) | class EnvHolder(object): method __init__ (line 194) | def __init__(self): method is_shutting_down (line 199) | def is_shutting_down(self): method switch_to_shutting_down (line 211) | def switch_to_shutting_down(self, is_normal_exit=True): function GetEnv (line 216) | def GetEnv(): FILE: python/oneflow/framework/function_desc.py class FunctionAttribute (line 23) | class FunctionAttribute(object): method __init__ (line 24) | def __init__(self): class FunctionDesc (line 30) | class FunctionDesc(object): method __init__ (line 31) | def __init__(self, job_func=None, job_config_proto=None, function_attr... method IsTrainable (line 41) | def IsTrainable(self): method HasAttr (line 48) | def HasAttr(self, attr_name): method __getattr__ (line 56) | def __getattr__(self, attr_name): FILE: python/oneflow/framework/function_util.py class FunctionConfig (line 33) | class FunctionConfig(object): method __init__ (line 37) | def __init__(self) -> None: method __getattr__ (line 40) | def __getattr__( function _CloneFunctionDesc (line 74) | def _CloneFunctionDesc(func_desc, job_func): function oneflow_function_config (line 81) | def oneflow_function_config(*field_paths): function _MakeInnerJobConfigClassProperty (line 110) | def _MakeInnerJobConfigClassProperty(return_obj_class): function _MakeLeafJobConfigCall (line 114) | def _MakeLeafJobConfigCall(method): function set_default_data_type (line 119) | def set_default_data_type(func_desc, value): function set_default_initializer_conf (line 132) | def set_default_initializer_conf(func_desc, value): function set_exp_run_conf (line 146) | def set_exp_run_conf(value): function static_mem_alloc_policy_white_list_has_policy (line 157) | def static_mem_alloc_policy_white_list_has_policy(func_desc, policy): function static_mem_alloc_policy_white_list_add_policy (line 171) | def static_mem_alloc_policy_white_list_add_policy(func_desc, policy): function static_mem_alloc_policy_white_list_remove_policy (line 182) | def static_mem_alloc_policy_white_list_remove_policy(func_desc, policy): function policy_mem_size_first (line 193) | def policy_mem_size_first(func_desc): function policy_mutual_exclusion_first (line 208) | def policy_mutual_exclusion_first(func_desc): function policy_time_line (line 221) | def policy_time_line(func_desc): function show_static_mem_alloc_algo_white_list (line 234) | def show_static_mem_alloc_algo_white_list(func_desc): function set_enable_cudnn (line 252) | def set_enable_cudnn(func_desc, value=True): function set_cudnn_buf_limit_mbyte (line 263) | def set_cudnn_buf_limit_mbyte(func_desc, value): function set_cudnn_conv_force_fwd_algo (line 274) | def set_cudnn_conv_force_fwd_algo(func_desc, value): function set_cudnn_conv_force_bwd_data_algo (line 285) | def set_cudnn_conv_force_bwd_data_algo(func_desc, value): function set_cudnn_conv_force_bwd_filter_algo (line 296) | def set_cudnn_conv_force_bwd_filter_algo(func_desc, value): function set_cudnn_conv_heuristic_search_algo (line 307) | def set_cudnn_conv_heuristic_search_algo(func_desc, value): function set_enable_cudnn_fused_normalization_add_relu (line 318) | def set_enable_cudnn_fused_normalization_add_relu(func_desc, value): function set_enable_fuse_add_to_output (line 329) | def set_enable_fuse_add_to_output(func_desc, value): function set_enable_fuse_cast_scale (line 341) | def set_enable_fuse_cast_scale(func_desc, value=True): function set_cudnn_conv_use_deterministic_algo_only (line 353) | def set_cudnn_conv_use_deterministic_algo_only(func_desc, value): function set_enable_reused_mem (line 364) | def set_enable_reused_mem(func_desc, value=True): function set_enable_inplace (line 375) | def set_enable_inplace(func_desc, value=True): function set_enable_inplace_in_reduce_struct (line 386) | def set_enable_inplace_in_reduce_struct(func_desc, value=True): function set_enable_nccl (line 393) | def set_enable_nccl(func_desc, value=True): function set_use_nccl_inter_node_communication (line 400) | def set_use_nccl_inter_node_communication(func_desc, value=True): function set_use_boxing_v2 (line 407) | def set_use_boxing_v2(func_desc, value=True): function set_do_parallel_cast_before_widening_type_cast (line 414) | def set_do_parallel_cast_before_widening_type_cast(func_desc, value=True): function set_enable_all_reduce_group (line 419) | def set_enable_all_reduce_group(func_desc, value=True): function set_all_reduce_group_num (line 426) | def set_all_reduce_group_num(func_desc, value): function set_all_reduce_lazy_ratio (line 433) | def set_all_reduce_lazy_ratio(func_desc, value): function set_all_reduce_group_min_mbyte (line 440) | def set_all_reduce_group_min_mbyte(func_desc, value): function set_all_reduce_group_size_warmup (line 447) | def set_all_reduce_group_size_warmup(func_desc, value): function set_all_reduce_fp16 (line 454) | def set_all_reduce_fp16(func_desc, value=True): function set_optimizer_placement_optimization_mode (line 464) | def set_optimizer_placement_optimization_mode(func_desc, mode): function set_optimizer_placement_optimization_threshold (line 479) | def set_optimizer_placement_optimization_threshold(func_desc, value): function set_enable_non_distributed_optimizer (line 484) | def set_enable_non_distributed_optimizer(func_desc, value=True): function set_disable_all_reduce_sequence (line 496) | def set_disable_all_reduce_sequence(func_desc, value=True): function set_prune_parallel_cast_ops (line 503) | def set_prune_parallel_cast_ops(func_desc, value=True): function set_prune_cast_to_static_shape_ops (line 514) | def set_prune_cast_to_static_shape_ops(func_desc, value=True): function set_prune_amp_white_identity_ops (line 525) | def set_prune_amp_white_identity_ops(func_desc, value=True): function set_prune_depend_ops (line 536) | def set_prune_depend_ops(func_desc, value=True): function set_non_distributed_optimizer_group_size_mbyte (line 547) | def set_non_distributed_optimizer_group_size_mbyte(func_desc, value): function set_cudnn_conv_enable_true_half (line 556) | def set_cudnn_conv_enable_true_half(func_desc, value=True): function set_cudnn_conv_enable_pseudo_half (line 569) | def set_cudnn_conv_enable_pseudo_half(func_desc, value): function set_enable_float_compute_for_half_gemm (line 580) | def set_enable_float_compute_for_half_gemm(func_desc, value=True): function set_enable_quantization_aware_training (line 596) | def set_enable_quantization_aware_training(func_desc, value=True): function set_qat_per_channel (line 607) | def set_qat_per_channel(func_desc, value=True): function set_qat_symmetric (line 612) | def set_qat_symmetric(func_desc, value=True): function set_qat_moving_min_max_momentum (line 617) | def set_qat_moving_min_max_momentum(func_desc, value: float): function set_qat_moving_min_max_momentum (line 622) | def set_qat_moving_min_max_momentum(func_desc, value: float): function set_qat_symmetric (line 627) | def set_qat_symmetric(func_desc, value: str): function set_enable_auto_mixed_precision (line 632) | def set_enable_auto_mixed_precision(func_desc, value=True): function set_enable_keep_header_only (line 643) | def set_enable_keep_header_only(func_desc, value=True): function set_concurrency_width (line 654) | def set_concurrency_width(func_desc, value): function set_model_update_conf (line 665) | def set_model_update_conf(func_desc, value): function set_indexed_slices_optimizer_conf (line 684) | def set_indexed_slices_optimizer_conf(func_desc, value): function set_enable_fuse_model_update_ops (line 699) | def set_enable_fuse_model_update_ops(func_desc, value=True): function set_enable_gradients_stats_aggregation (line 711) | def set_enable_gradients_stats_aggregation(func_desc, value=True): function set_loss_scale_factor (line 723) | def set_loss_scale_factor(func_desc, value): function set_primary_lr (line 738) | def set_primary_lr(func_desc, value): function set_secondary_lr (line 753) | def set_secondary_lr(func_desc, value): function set_num_gradient_accumulation_steps (line 768) | def set_num_gradient_accumulation_steps(func_desc, value): function set_default_distribute_strategy (line 773) | def set_default_distribute_strategy(func_desc, value): function allow_cpu_return_op (line 785) | def allow_cpu_return_op(func_desc, value): function deprecated_set_default_distribute_strategy (line 797) | def deprecated_set_default_distribute_strategy(*args, **kwargs): FILE: python/oneflow/framework/generator.py function create_generator (line 20) | def create_generator(device=None): function seed (line 26) | def seed() -> int: function manual_seed (line 39) | def manual_seed(seed): function initial_seed (line 57) | def initial_seed() -> int: function _getstate (line 69) | def _getstate(self): function _setstate (line 73) | def _setstate(self, state_dict): function get_rng_state (line 78) | def get_rng_state(): function set_rng_state (line 94) | def set_rng_state(state): FILE: python/oneflow/framework/graph_build_util.py function graph_build_context (line 37) | def graph_build_context(config_proto, session): class JobBuildAndInferCtx (line 54) | class JobBuildAndInferCtx(object): method __init__ (line 55) | def __init__(self, config_proto): method __enter__ (line 58) | def __enter__(self): method __exit__ (line 62) | def __exit__(self, exc_type, exc_val, exc_tb): class BlockScopeContext (line 71) | class BlockScopeContext(object): method __init__ (line 72) | def __init__(self, prev_scope, new_scope): method __enter__ (line 78) | def __enter__(self): method __exit__ (line 82) | def __exit__(self, exc_type, exc_val, exc_tb): class DebugScopeContext (line 92) | class DebugScopeContext(object): method __init__ (line 93) | def __init__( method __enter__ (line 118) | def __enter__(self): method __exit__ (line 126) | def __exit__(self, exc_type, exc_val, exc_tb): function _make_new_scope (line 139) | def _make_new_scope(prev_scope, scope_proto_str_setter): function _make_new_graph_scope (line 154) | def _make_new_graph_scope(prev_scope, graph_name): function make_new_blockgraph_scope (line 169) | def make_new_blockgraph_scope(prev_scope, graph_block): function make_new_name_scope (line 206) | def make_new_name_scope(prev_scope, name): function scope_to_proto (line 222) | def scope_to_proto(scope): function build_graph_input_arg (line 226) | def build_graph_input_arg(op_name, arg): function build_graph_state (line 240) | def build_graph_state(op_name, state_tensor, state_config): function build_graph_output (line 260) | def build_graph_output(op_name, out): FILE: python/oneflow/framework/id_util.py function UniqueStr (line 19) | def UniqueStr(prefix): FILE: python/oneflow/framework/infer_compiler/import_tools/format_utils.py class MockEntityNameFormatter (line 21) | class MockEntityNameFormatter: method __init__ (line 22) | def __init__(self, prefix: str = "mock_", suffix: str = "_oflow"): method _format_pkg_name (line 26) | def _format_pkg_name(self, pkg_name: str) -> str: method _reverse_pkg_name (line 31) | def _reverse_pkg_name(self, pkg_name: str) -> str: method _format_full_class_name (line 37) | def _format_full_class_name(self, obj: Union[str, type, FunctionType]): method format (line 53) | def format(self, entity: Union[str, type, FunctionType]) -> str: method unformat (line 56) | def unformat(self, mock_entity_name: str) -> str: FILE: python/oneflow/framework/infer_compiler/import_tools/importer.py function is_need_mock (line 41) | def is_need_mock(cls) -> bool: function import_module_from_path (line 57) | def import_module_from_path(module_path: Union[str, Path]) -> ModuleType: class LazyMocker (line 80) | class LazyMocker: method __init__ (line 81) | def __init__(self, prefix: str, suffix: str, tmp_dir: Optional[Union[s... method mock_package (line 88) | def mock_package(self, package: str): method cleanup (line 91) | def cleanup(self): method get_mock_entity_name (line 94) | def get_mock_entity_name(self, entity: Union[str, type, FunctionType]): method mock_entity (line 99) | def mock_entity(self, entity: Union[str, type, FunctionType]): method add_mocked_package (line 112) | def add_mocked_package(self, package: str): method load_entity_with_mock (line 126) | def load_entity_with_mock(self, entity: Union[str, type, FunctionType]): FILE: python/oneflow/framework/infer_compiler/transform/builtin_transform.py function singledispatch_proxy (line 43) | def singledispatch_proxy(func): function proxy_class (line 66) | def proxy_class(cls: type): class ProxySubmodule (line 77) | class ProxySubmodule: method __init__ (line 78) | def __init__(self, submod): method __getitem__ (line 83) | def __getitem__(self, index): # __getitem__ method __repr__ (line 90) | def __repr__(self) -> str: method __getattribute__ (line 93) | def __getattribute__(self, attribute): method __call__ (line 150) | def __call__(self, *args: Any, **kwargs: Any) -> Any: function torch2oflow (line 162) | def torch2oflow(mod, *args, **kwargs): function default_converter (line 166) | def default_converter(obj, verbose=False, *, proxy_cls=None): function _ (line 190) | def _(mod: torch.nn.Module, verbose=False): function _ (line 256) | def _(mod: torch.nn.BatchNorm1d, verbose=False): function _ (line 264) | def _(mod: torch.nn.BatchNorm2d, verbose=False): function _ (line 275) | def _(mod: torch.nn.BatchNorm3d, verbose=False): function _ (line 283) | def _(mod: torch.nn.MaxPool1d, verbose=False): function _ (line 291) | def _(mod: torch.nn.MaxPool2d, verbose=False): function _ (line 302) | def _(mod: torch.nn.MaxPool3d, verbose=False): function _ (line 310) | def _(mod: torch.nn.AvgPool1d, verbose=False): function _ (line 318) | def _(mod: torch.nn.AvgPool2d, verbose=False): function _ (line 329) | def _(mod: torch.nn.AvgPool3d, verbose=False): function _ (line 337) | def _(mod: torch.nn.AdaptiveAvgPool2d, verbose=False): function _ (line 351) | def _(mod: Conv2dNormActivation, verbose=False): function _ (line 360) | def _(mod: torch.nn.ModuleList, verbose=False): function _ (line 370) | def _(mod: torch.nn.Sequential, verbose=False): function _ (line 381) | def _(mod: torch.nn.parameter.Parameter, verbose=False) -> flow.nn.Param... function _ (line 390) | def _(mod: torch.Tensor, verbose=False) -> flow.Tensor: function _ (line 406) | def _(mod: torch.dtype, verbose=False) -> flow.dtype: function _ (line 411) | def _(mod: list, verbose=False) -> list: function _ (line 416) | def _(mod: tuple, verbose=False) -> tuple: function _ (line 421) | def _(mod: OrderedDict, verbose=False) -> OrderedDict: function _ (line 429) | def _(mod: set, verbose=False) -> set: function _ (line 437) | def _(mod, verbose=False) -> Union[int, float, str, bool]: function _ (line 442) | def _(mod: None, verbose=False): function _ (line 447) | def _(mod: types.BuiltinFunctionType, verbose=False): function _ (line 473) | def _(mod: torch.device, verbose=False): function _ (line 479) | def _(mod: dict, verbose=False) -> dict: function _ (line 484) | def _(func: types.FunctionType, verbose=False): function _ (line 489) | def _(mod: partial, verbose=False): function map_args (line 500) | def map_args(args, kwargs): function get_attr (line 506) | def get_attr(gm, node, torch2flow={}): FILE: python/oneflow/framework/infer_compiler/transform/custom_transform.py function register_torch2oflow_class (line 29) | def register_torch2oflow_class(cls: type, replacement: type, verbose=True): function register_torch2oflow_func (line 38) | def register_torch2oflow_func(func, first_param_type=None, verbose=False): function ensure_list (line 53) | def ensure_list(obj): function register (line 59) | def register( FILE: python/oneflow/framework/infer_compiler/transform/manager.py class TransformManager (line 28) | class TransformManager: method __init__ (line 36) | def __init__(self, debug_mode=False, tmp_dir="./output"): method _setup_logger (line 42) | def _setup_logger(self): method get_mocked_packages (line 48) | def get_mocked_packages(self): method load_class_proxies_from_packages (line 51) | def load_class_proxies_from_packages(self, package_names: List[Union[P... method update_class_proxies (line 57) | def update_class_proxies(self, class_proxy_dict: Dict[str, type], verb... method _transform_entity (line 70) | def _transform_entity(self, entity): method get_transformed_entity_name (line 76) | def get_transformed_entity_name(self, entity): method transform_cls (line 79) | def transform_cls(self, full_cls_name: str): method transform_func (line 91) | def transform_func(self, func: types.FunctionType): method transform_package (line 95) | def transform_package(self, package_name): FILE: python/oneflow/framework/infer_compiler/utils/args_tree_util.py function input_output_processor (line 21) | def input_output_processor(func): FILE: python/oneflow/framework/infer_compiler/utils/cost_util.py class cost_cnt (line 25) | class cost_cnt: method __init__ (line 26) | def __init__(self, debug=False, message="\t"): method __enter__ (line 30) | def __enter__(self): method __exit__ (line 43) | def __exit__(self, exc_type, exc_val, exc_tb): method __call__ (line 59) | def __call__(self, func): FILE: python/oneflow/framework/infer_compiler/utils/log_utils.py class ColorFormatter (line 22) | class ColorFormatter(logging.Formatter): method format (line 31) | def format(self, record): class ConfigurableLogger (line 37) | class ConfigurableLogger: method __init__ (line 38) | def __init__(self) -> None: method __getattr__ (line 41) | def __getattr__(self, name): method configure_logging (line 44) | def configure_logging(self, name, level, log_dir=None, file_name=None): FILE: python/oneflow/framework/infer_compiler/utils/oneflow_exec_mode.py class oneflow_exec_mode (line 21) | class oneflow_exec_mode(object): method __init__ (line 22) | def __init__(self, enabled=None): method __enter__ (line 28) | def __enter__(self): method __exit__ (line 35) | def __exit__(self, exc_type, exc_val, exc_tb): function oneflow_exec_mode_enabled (line 41) | def oneflow_exec_mode_enabled(): FILE: python/oneflow/framework/infer_compiler/utils/param_utils.py function parse_device (line 22) | def parse_device(args: List[Any], kwargs: Dict[str, Any]): function check_device (line 33) | def check_device(current_device, target_device) -> bool: FILE: python/oneflow/framework/infer_compiler/utils/patch_for_compiler.py class FakeCuda (line 22) | class FakeCuda: method current_device (line 24) | def current_device(): method mem_get_info (line 28) | def mem_get_info(dev): method _scaled_dot_product_attention_math (line 32) | def _scaled_dot_product_attention_math( method scaled_dot_product_attention (line 67) | def scaled_dot_product_attention( FILE: python/oneflow/framework/infer_compiler/utils/patch_for_diffusers.py class InstanceChecker (line 31) | class InstanceChecker(ABC): method is_attention_instance (line 33) | def is_attention_instance(self, instance): class DiffusersChecker (line 37) | class DiffusersChecker(InstanceChecker): method is_attention_instance (line 38) | def is_attention_instance(self, instance): FILE: python/oneflow/framework/infer_compiler/with_fx_graph.py function fx_node_tranform (line 26) | def fx_node_tranform(gm): function to_of_transform (line 81) | def to_of_transform( function replace_node (line 137) | def replace_node(node, name2node): function node_replace_args (line 144) | def node_replace_args(args, name2node): FILE: python/oneflow/framework/infer_compiler/with_fx_interpreter.py class OneFlowInterpreter (line 24) | class OneFlowInterpreter(torch.fx.Interpreter): method call_function (line 27) | def call_function(self, target: Target, args: Tuple, kwargs: Dict) -> ... method call_method (line 32) | def call_method(self, target: Target, args: Tuple, kwargs: Dict) -> Any: method call_module (line 36) | def call_module( FILE: python/oneflow/framework/infer_compiler/with_oneflow_backend.py function oneflow_backend (line 26) | def oneflow_backend(gm, example_inputs, *args, **kwargs): FILE: python/oneflow/framework/infer_compiler/with_oneflow_compile.py class DualModule (line 35) | class DualModule(torch.nn.Module): method __init__ (line 36) | def __init__(self, torch_module, oneflow_module): method oneflow_module (line 42) | def oneflow_module(self): method oneflow_module (line 52) | def oneflow_module(self): method to (line 57) | def to(self, *args, **kwargs): method _torch_module_to_with_check (line 69) | def _torch_module_to_with_check(self, *args, **kwargs): method __getattr__ (line 94) | def __getattr__(self, name): method __setattr__ (line 116) | def __setattr__(self, name: str, value: Any) -> None: class DualModuleList (line 132) | class DualModuleList(torch.nn.ModuleList): method __init__ (line 133) | def __init__(self, torch_modules, oneflow_modules): method __setitem__ (line 151) | def __setitem__(self, idx: int, module: DualModule): method __setattr__ (line 157) | def __setattr__(self, name, value): function get_mixed_dual_module (line 172) | def get_mixed_dual_module(module_cls): function graph_file_management (line 180) | def graph_file_management(func): function handle_deployable_exception (line 226) | def handle_deployable_exception(func): class DeployableModule (line 244) | class DeployableModule(torch.nn.Module): method __init__ (line 245) | def __init__( method from_existing (line 264) | def from_existing(cls, existing_module, use_graph=None, options=None): method get_graph (line 273) | def get_graph(self): method apply_model (line 296) | def apply_model(self, *args, **kwargs): method __call__ (line 311) | def __call__(self, *args, **kwargs): method to (line 321) | def to(self, *args, **kwargs): method decode (line 344) | def decode(self, *args, **kwargs): method __getattr__ (line 361) | def __getattr__(self, name): method load_graph (line 366) | def load_graph(self, file_path, device=None, run_warmup=True): method warmup_with_load (line 369) | def warmup_with_load(self, file_path, device=None, run_warmup=True): method save_graph (line 372) | def save_graph(self, file_path): class OneflowGraph (line 376) | class OneflowGraph(flow.nn.Graph): method __init__ (line 378) | def __init__(self, model): method build (line 408) | def build(self, *args, **kwargs): method warmup_with_load (line 412) | def warmup_with_load(self, file_path, device=None, run_warmup=True): method save_graph (line 419) | def save_graph(self, file_path): function get_oneflow_graph (line 424) | def get_oneflow_graph(model, size=9, dynamic=True): function state_dict_hook (line 431) | def state_dict_hook(module, state_dict, prefix, local_metadata): function get_mixed_deployable_module (line 445) | def get_mixed_deployable_module(module_cls): function compile_from_torch (line 480) | def compile_from_torch( FILE: python/oneflow/framework/model.py class DataModule (line 40) | class DataModule(Module): method __init__ (line 41) | def __init__(self, *args, **kwargs): method forward (line 44) | def forward(self, step_idx: int = 0, optimizer_idx: int = 0): method infer_oneflow_data_placeholder (line 47) | def infer_oneflow_data_placeholder( class NumpyDataModule (line 53) | class NumpyDataModule(DataModule): method __init__ (line 54) | def __init__(self, *args, **kwargs): method forward (line 57) | def forward(self, step_idx: int = 0, optimizer_idx: int = 0): method __call__ (line 60) | def __call__(self, *args): method infer_oneflow_data_placeholder (line 64) | def infer_oneflow_data_placeholder( class TrainingConfig (line 81) | class TrainingConfig: method __init__ (line 82) | def __init__(self): method config_execution (line 88) | def config_execution(self, exe_cfg: ExecutionConfig = None): method config_data (line 91) | def config_data(self, data: DataModule = None): method check_valid (line 94) | def check_valid(self): class ValidationConfig (line 109) | class ValidationConfig: method __init__ (line 110) | def __init__(self): method config_execution (line 117) | def config_execution(self, exe_cfg: ExecutionConfig = None): method config_data (line 120) | def config_data(self, data: DataModule = None): method config_step_interval (line 123) | def config_step_interval(self, step_interval: int = 1): method check_valid (line 126) | def check_valid(self): class CheckpointConfig (line 143) | class CheckpointConfig(object): method __init__ (line 144) | def __init__(self): method config_load (line 152) | def config_load(self, dirpath: str = None): method config_save (line 157) | def config_save(self, dirpath: str = None, step_interval: int = 1): method check_valid (line 165) | def check_valid(self): class Callback (line 171) | class Callback(ABC): method on_training_step_end (line 175) | def on_training_step_end( method on_validation_step_end (line 183) | def on_validation_step_end( class Model (line 189) | class Model(ABC, Module): method __init__ (line 193) | def __init__(self, *args, **kwargs): method forward (line 201) | def forward(self, *args, **kwargs): method training_step (line 206) | def training_step(self, *args, **kwargs): method validation_step (line 211) | def validation_step(self, *args, **kwargs): method configure_optimizers (line 216) | def configure_optimizers(self): method fit (line 222) | def fit( method method_overrided (line 250) | def method_overrided(self, method_name: str = None) -> bool: method _get_and_check_sub_models (line 253) | def _get_and_check_sub_models( class SubModel (line 307) | class SubModel(ABC): method __init__ (line 308) | def __init__(self, name, cfg, model, callbacks): method step (line 323) | def step(self, step_idx: int = 0): method _get_and_check_cfg (line 326) | def _get_and_check_cfg(self): method _get_and_check_cbs (line 336) | def _get_and_check_cbs(self): method _method_callback (line 353) | def _method_callback(self, method_name: str = None, *args, **kwargs): class TrainModel (line 359) | class TrainModel(SubModel): method __init__ (line 360) | def __init__( method step (line 374) | def step(self, step_idx: int = 0): method _get_and_check_step (line 394) | def _get_and_check_step(self): method _get_and_check_opts (line 401) | def _get_and_check_opts(self): method _get_and_check_jobs (line 421) | def _get_and_check_jobs(self): method _construct_job (line 439) | def _construct_job(self, optimizer_idx: int = 0): method _construct_numpy_job (line 459) | def _construct_numpy_job(self, batch, optimizer_idx): class ValidateModel (line 482) | class ValidateModel(SubModel): method __init__ (line 483) | def __init__( method step (line 495) | def step(self, step_idx: int = 0): method _get_and_check_step (line 512) | def _get_and_check_step(self): method _get_and_check_job (line 519) | def _get_and_check_job(self): method _construct_job (line 532) | def _construct_job(self): method _construct_numpy_job (line 541) | def _construct_numpy_job(self, batch: Tuple[np.ndarray, ...] = None): class CheckpointModel (line 551) | class CheckpointModel(SubModel): method __init__ (line 552) | def __init__( method load (line 560) | def load(self): method step (line 565) | def step(self, step_idx: int = 0): method _load_checkpoint (line 573) | def _load_checkpoint(self, dirpath: str): method _save_checkpoint (line 579) | def _save_checkpoint(self, dirpath: str): class TrainModelOOPStyle (line 586) | class TrainModelOOPStyle(SubModel): method __init__ (line 587) | def __init__( method step (line 599) | def step(self, step_idx: int = 0): method _get_and_check_step (line 622) | def _get_and_check_step(self): method _get_and_check_opts (line 629) | def _get_and_check_opts(self): class ValidateModelOOPStyle (line 650) | class ValidateModelOOPStyle(SubModel): method __init__ (line 651) | def __init__( method step (line 661) | def step(self, step_idx: int = 0): method _get_and_check_step (line 675) | def _get_and_check_step(self): class CheckpointModelOOPStyle (line 683) | class CheckpointModelOOPStyle(SubModel): method __init__ (line 684) | def __init__( method load (line 692) | def load(self): method step (line 697) | def step(self, step_idx: int = 0): method _load_checkpoint (line 705) | def _load_checkpoint(self, dirpath: str): method _save_checkpoint (line 711) | def _save_checkpoint(self, dirpath: str): function _infer_job_signature (line 718) | def _infer_job_signature(data_module, batch, optimizer_idx, job): FILE: python/oneflow/framework/multi_client_session.py class MultiClientSession (line 28) | class MultiClientSession(object): class Status (line 29) | class Status(enum.Enum): method __init__ (line 34) | def __init__(self, env, sess_id): method __del__ (line 47) | def __del__(self): method TryInit (line 53) | def TryInit(self): method _TryClose (line 60) | def _TryClose(self): method status (line 66) | def status(self): method id (line 70) | def id(self): method config_proto (line 74) | def config_proto(self): method resource (line 78) | def resource(self): method function_flag_name2default_val (line 83) | def function_flag_name2default_val(self): method scope_attr_name2default_val (line 87) | def scope_attr_name2default_val(self): method is_running (line 91) | def is_running(self): method _check_status (line 94) | def _check_status(self, *status): method _make_config_proto (line 109) | def _make_config_proto(self): method _update_function_flag_name2defaultVal (line 115) | def _update_function_flag_name2defaultVal(self): method _update_scope_attr_name2defaultVal (line 119) | def _update_scope_attr_name2defaultVal(self): method update_resource_eagerly (line 123) | def update_resource_eagerly(self, resource_config): FILE: python/oneflow/framework/register_class_method_util.py function RegisterMethod4Class (line 21) | def RegisterMethod4Class(): FILE: python/oneflow/framework/scope_util.py function api_scope_config (line 28) | def api_scope_config(**kwargs): function current_scope (line 53) | def current_scope(): function MakeScope (line 62) | def MakeScope(build_func): function ScopeContext (line 77) | def ScopeContext(scope): FILE: python/oneflow/framework/session_context.py class SessionStatus (line 23) | class SessionStatus: function GetDefaultSession (line 29) | def GetDefaultSession(): function NewDefaultSession (line 36) | def NewDefaultSession(env): function TryCloseDefaultSession (line 45) | def TryCloseDefaultSession(): function try_init_default_session (line 55) | def try_init_default_session(func): FILE: python/oneflow/framework/sysconfig.py function get_include (line 25) | def get_include() -> str: function get_lib (line 29) | def get_lib() -> str: function get_compile_flags (line 33) | def get_compile_flags() -> List[str]: function get_liboneflow_link_flags (line 46) | def get_liboneflow_link_flags() -> List[str]: function get_link_flags (line 62) | def get_link_flags() -> List[str]: function with_cuda (line 74) | def with_cuda() -> bool: function get_cuda_version (line 78) | def get_cuda_version() -> int: function has_rpc_backend_grpc (line 82) | def has_rpc_backend_grpc() -> bool: function has_rpc_backend_local (line 86) | def has_rpc_backend_local() -> bool: function cmake_build_type (line 90) | def cmake_build_type() -> str: function with_rdma (line 94) | def with_rdma() -> bool: FILE: python/oneflow/framework/tensor.py function _ndim (line 28) | def _ndim(self): function _backward (line 32) | def _backward(self, gradient=None, retain_graph=False, create_graph=False): function _str (line 47) | def _str(self): function _repr (line 51) | def _repr(self): function _meta_repr (line 55) | def _meta_repr(self): function _eq (line 59) | def _eq(self, other): function _cuda (line 68) | def _cuda(self, device: Union[int, str, flow.device] = None): function _norm (line 76) | def _norm(self, p=None, dim=None, keepdim=False, dtype=None): function is_nonzero (line 82) | def is_nonzero(input): function _add (line 113) | def _add(self, other, *, alpha=1): function _addmm (line 117) | def _addmm(self, mat1, mat2, alpha=1, beta=1): function _add_inplace (line 121) | def _add_inplace(self, other, *, alpha=1): function _iadd (line 125) | def _iadd(self, other): function _sub_inplace (line 129) | def _sub_inplace(self, other): function _expand (line 133) | def _expand(self, *size): function _expand_as (line 137) | def _expand_as(input, other): function _argwhere (line 141) | def _argwhere(self): function _index (line 145) | def _index(self): function _scalar_float (line 156) | def _scalar_float(self): function _scalar_int (line 163) | def _scalar_int(self): function _new_empty (line 170) | def _new_empty( function _new_ones (line 176) | def _new_ones( function _new_zeros (line 182) | def _new_zeros( function _squeeze_inplace (line 188) | def _squeeze_inplace(self, dim=None): function _unsqueeze_inplace (line 192) | def _unsqueeze_inplace(self, dim=None): function _new_full (line 196) | def _new_full( function _argsort (line 211) | def _argsort(self, dim=-1, descending=None): function _uniform (line 215) | def _uniform(self, a=0, b=1): function _exponential (line 219) | def _exponential(self, lambd=1.0, generator=None): function _trunc_normal_ (line 223) | def _trunc_normal_( function _kaiming_uniform (line 229) | def _kaiming_uniform( function _kaiming_normal (line 237) | def _kaiming_normal( function _xavier_normal (line 245) | def _xavier_normal(self, gain=1.0): function _xavier_uniform (line 249) | def _xavier_uniform(self, gain=1.0): function _orthogonal (line 253) | def _orthogonal(self, gain=1.0): function _normal (line 272) | def _normal(self, mean=0, std=1): function _copy_from_numpy_to_eager_local_tensor (line 276) | def _copy_from_numpy_to_eager_local_tensor(eager_local_tensor, np_arr): function _copy (line 284) | def _copy(self, other: Union[Tensor, np.ndarray]): function _format (line 309) | def _format(self, format_spec): function _to (line 315) | def _to(self, *args, **kwargs): function _tolist (line 329) | def _tolist(self): function _repeat (line 335) | def _repeat(self, *sizes): function _tile (line 345) | def _tile(self, *dims): function _T (line 355) | def _T(self): function _nms (line 359) | def _nms(boxes, scores, iou_threshold: float): function _nonzero (line 363) | def _nonzero(self, as_tuple=False): function _prod (line 367) | def _prod(self, dim=[], keepdim=False): function _masked_select (line 371) | def _masked_select(self, mask): function _sort (line 375) | def _sort(self, dim: int = -1, descending: bool = False): function _where (line 379) | def _where(self, x=None, y=None): function _numpy (line 383) | def _numpy(self, dtype=None): function zero_ (line 417) | def zero_(self): function _is_consistent (line 422) | def _is_consistent(self): function _to_consistent (line 426) | def _to_consistent(self, *args, **kwargs): function _new_tensor (line 430) | def _new_tensor( function _cumsum (line 455) | def _cumsum(self, dim, dtype=None): function _cumprod (line 459) | def _cumprod(self, dim, dtype=None): function _cross (line 463) | def _cross(self, other, dim=None): function _scatter (line 467) | def _scatter(self, dim, index, src, *, reduce=None): function _scatter_inplace (line 471) | def _scatter_inplace(self, dim, index, src, *, reduce=None): function _scatter_add_inplace (line 475) | def _scatter_add_inplace(self, dim, index, src): function _contains (line 479) | def _contains(self, element): function _allclose (line 496) | def _allclose(self, other, atol=1e-08, rtol=1e-05, equal_nan=False): function _index_add (line 500) | def _index_add(self, dim, index, source, alpha=1): function _index_add_inplace (line 504) | def _index_add_inplace(self, dim, index, source, alpha=1): function _as_strided (line 508) | def _as_strided(self, size, stride, storage_offset=0): function _as_strided_inplace (line 512) | def _as_strided_inplace(self, size, stride, storage_offset=0): function _logaddexp (line 516) | def _logaddexp(self, other): function _real (line 520) | def _real(self): function _imag (line 524) | def _imag(self): function _conj (line 528) | def _conj(self): function _conj_physical (line 532) | def _conj_physical(self): function _storage (line 536) | def _storage(self): function _layout (line 541) | def _layout(self): function RegisterMethods (line 545) | def RegisterMethods(): function register_tensor_op (line 620) | def register_tensor_op(op_name): FILE: python/oneflow/framework/tensor_str.py class __PrinterOptions (line 29) | class __PrinterOptions(object): method linewidth (line 39) | def linewidth(self): method linewidth (line 45) | def linewidth(self, value): function set_printoptions (line 52) | def set_printoptions( class _Formatter (line 110) | class _Formatter(object): method __init__ (line 111) | def __init__(self, tensor): method width (line 180) | def width(self): method format (line 183) | def format(self, value): function _scalar_str (line 202) | def _scalar_str(self, formatter1): function _vector_str (line 206) | def _vector_str(self, indent, summarize, formatter1): function _tensor_str_with_formatter (line 239) | def _tensor_str_with_formatter(self, indent, summarize, formatter1): function _tensor_str (line 270) | def _tensor_str(self, indent): function _add_suffixes (line 280) | def _add_suffixes(tensor_str, suffixes, indent): function get_summarized_data (line 295) | def get_summarized_data(self): function _format_tensor_on_cpu (line 316) | def _format_tensor_on_cpu(tensor): function _gen_tensor_str_template (line 324) | def _gen_tensor_str_template(tensor, is_meta): function _gen_tensor_str (line 369) | def _gen_tensor_str(tensor): function _gen_tensor_meta_str (line 373) | def _gen_tensor_meta_str(tensor): FILE: python/oneflow/framework/tensor_str_util.py function _autoset_linewidth (line 21) | def _autoset_linewidth(): function _try_convert_to_local_tensor (line 32) | def _try_convert_to_local_tensor(tensor): FILE: python/oneflow/framework/tensor_tuple_util.py function convert_to_tensor_tuple (line 23) | def convert_to_tensor_tuple(args: Optional[Union[Tensor, Sequence[Tensor... FILE: python/oneflow/framework/unittest.py function register_test_cases (line 37) | def register_test_cases( function num_nodes_required (line 69) | def num_nodes_required(num_nodes: int) -> Callable[[Callable], Callable]: function _GetNumOfNodes (line 77) | def _GetNumOfNodes(func): function eager_execution_enabled (line 83) | def eager_execution_enabled(): function typing_check_enabled (line 87) | def typing_check_enabled(): function node_list (line 91) | def node_list(): function has_node_list (line 97) | def has_node_list(): function node_size (line 104) | def node_size(): function has_world_size (line 115) | def has_world_size(): function world_size (line 119) | def world_size(): function device_num (line 123) | def device_num(): function enable_init_by_host_list (line 131) | def enable_init_by_host_list(): function enable_multi_process (line 135) | def enable_multi_process(): function find_free_port (line 139) | def find_free_port(): function worker_agent_port (line 149) | def worker_agent_port(): function worker_agent_authkey (line 157) | def worker_agent_authkey(): function use_worker_agent (line 163) | def use_worker_agent(): function cast (line 167) | def cast(conn=None, cmd=None, msg=None): function call (line 174) | def call(conn=None, cmd=None, msg=None): function skip_unless (line 188) | def skip_unless(n, d): function skip_unless_1n1d (line 201) | def skip_unless_1n1d(): function skip_unless_1n2d (line 205) | def skip_unless_1n2d(): function skip_unless_1n4d (line 209) | def skip_unless_1n4d(): function skip_unless_2n1d (line 213) | def skip_unless_2n1d(): function skip_unless_2n2d (line 217) | def skip_unless_2n2d(): function skip_unless_2n4d (line 221) | def skip_unless_2n4d(): class CondSkipChecker (line 225) | class CondSkipChecker(doctest.OutputChecker): method __init__ (line 226) | def __init__(self, check_flags): method check_output (line 229) | def check_output(self, want, got, optionflags): function check_multi_rank_docstr (line 243) | def check_multi_rank_docstr(module): FILE: python/oneflow/fx/__init__.py class Proxy (line 21) | class Proxy: method __init__ (line 22) | def __init__(self): FILE: python/oneflow/hub.py class tqdm (line 44) | class tqdm(object): # type: ignore[no-redef] method __init__ (line 45) | def __init__( method update (line 58) | def update(self, n): method close (line 71) | def close(self): method __enter__ (line 74) | def __enter__(self): method __exit__ (line 77) | def __exit__(self, exc_type, exc_val, exc_tb): function _import_module (line 109) | def _import_module(name, path): function _remove_if_exists (line 121) | def _remove_if_exists(path): function _git_archive_link (line 129) | def _git_archive_link(repo_owner, repo_name, ref): function _load_attr_from_module (line 134) | def _load_attr_from_module(module, func_name): function _get_oneflow_home (line 141) | def _get_oneflow_home(): function _parse_repo_info (line 154) | def _parse_repo_info(github): function _read_url (line 176) | def _read_url(url): function _validate_not_a_forked_repo (line 181) | def _validate_not_a_forked_repo(repo_owner, repo_name, ref): function _get_cache_or_reload (line 209) | def _get_cache_or_reload( function _check_repo_is_trusted (line 289) | def _check_repo_is_trusted( function _check_module_exists (line 343) | def _check_module_exists(name): function _check_dependencies (line 349) | def _check_dependencies(m): function _load_entry_from_hubconf (line 360) | def _load_entry_from_hubconf(m, model): function get_dir (line 378) | def get_dir(): function set_dir (line 396) | def set_dir(d): function list (line 407) | def list(github, force_reload=False, skip_validation=False, trust_repo=N... function help (line 474) | def help(github, model, force_reload=False, skip_validation=False, trust... function load (line 535) | def load( function _load_local (line 634) | def _load_local(hubconf_dir, model, *args, **kwargs): function download_url_to_file (line 667) | def download_url_to_file(url, dst, hash_prefix=None, progress=True): function _is_legacy_zip_format (line 737) | def _is_legacy_zip_format(filename): function _legacy_zip_load (line 744) | def _legacy_zip_load(filename, model_dir, map_location): function load_state_dict_from_url (line 756) | def load_state_dict_from_url( FILE: python/oneflow/ir/ast_gen_transformer.py class ASTTransformer (line 20) | class ASTTransformer(ast.NodeTransformer): method visit_arg (line 21) | def visit_arg(self, node: ast.arg): method visit_arguments (line 25) | def visit_arguments(self, node: ast.arguments): method visit_FunctionDef (line 33) | def visit_FunctionDef(self, node: ast.FunctionDef): method visit_Return (line 44) | def visit_Return(self, node: ast.Return): method visit_Assign (line 50) | def visit_Assign(self, node: ast.Assign): method visit_If (line 59) | def visit_If(self, node: ast.If): method visit_Raise (line 74) | def visit_Raise(self, node: ast.Raise): method visit_Assert (line 78) | def visit_Assert(self, node: ast.Assert): method visit_Expr (line 82) | def visit_Expr(self, node: ast.Expr): method visit_BoolOp (line 86) | def visit_BoolOp(self, node: ast.BoolOp): method visit_BinOp (line 90) | def visit_BinOp(self, node: ast.BinOp): method visit_Lambda (line 110) | def visit_Lambda(self, node: ast.Lambda): method visit_Compare (line 114) | def visit_Compare(self, node: ast.Compare): method visit_Call (line 136) | def visit_Call(self, node: ast.Call): method visit_Constant (line 148) | def visit_Constant(self, node: ast.Constant): method visit_Num (line 152) | def visit_Num(self, node: ast.Num): method visit_Attribute (line 156) | def visit_Attribute(self, node: ast.Attribute): method visit_Name (line 163) | def visit_Name(self, node: ast.Name): FILE: python/oneflow/ir/bisect_transformer.py class BisectTransformer (line 20) | class BisectTransformer(ast.NodeTransformer): method visit_FunctionDef (line 21) | def visit_FunctionDef(self, node: ast.FunctionDef): method visit_Call (line 29) | def visit_Call(self, node: ast.Call): FILE: python/oneflow/ir/lr_jit.py function lr_jit_register (line 30) | def lr_jit_register(lr_obj, is_dump=False): function _test_current_lr_jit (line 56) | def _test_current_lr_jit(test_case): class TestCurrentLRJIT (line 105) | class TestCurrentLRJIT(oneflow.unittest.MLIRTestCase): method test_current_lr_jit (line 106) | def test_current_lr_jit(test_case): FILE: python/oneflow/ir/math_params_transformer.py class MathParamsTransformer (line 19) | class MathParamsTransformer(ast.NodeTransformer): method visit_Attribute (line 20) | def visit_Attribute(self, node): FILE: python/oneflow/ir/self_params_transformer.py class SelfParamsTransformer (line 19) | class SelfParamsTransformer(ast.NodeTransformer): method __init__ (line 20) | def __init__(self, lr_obj): method visit_Attribute (line 24) | def visit_Attribute(self, node): method visit_arguments (line 34) | def visit_arguments(self, node: ast.arguments): FILE: python/oneflow/jit/__init__.py function script (line 20) | def script( function ignore (line 33) | def ignore(drop=False, **kwargs): function unused (line 44) | def unused(fn): function _script_if_tracing (line 52) | def _script_if_tracing(fn): function _overload_method (line 60) | def _overload_method(fn): function is_scripting (line 68) | def is_scripting(): function is_tracing (line 72) | def is_tracing(): class _Final (line 76) | class _Final: method __init_subclass__ (line 81) | def __init_subclass__(self, *args, **kwds): class _SpecialForm (line 86) | class _SpecialForm(_Final, _root=True): method __init__ (line 89) | def __init__(self, getitem): method __getattr__ (line 94) | def __getattr__(self, item): method __mro_entries__ (line 100) | def __mro_entries__(self, bases): method __repr__ (line 103) | def __repr__(self): method __reduce__ (line 106) | def __reduce__(self): method __call__ (line 109) | def __call__(self, *args, **kwds): method __or__ (line 112) | def __or__(self, other): method __ror__ (line 115) | def __ror__(self, other): method __instancecheck__ (line 118) | def __instancecheck__(self, obj): method __subclasscheck__ (line 121) | def __subclasscheck__(self, cls): method __getitem__ (line 124) | def __getitem__(self, parameters): function Final (line 129) | def Final(*args, **kwargs): function interface (line 135) | def interface(fn): FILE: python/oneflow/linalg.py function norm (line 19) | def norm(self, ord=None, dim=None, keepdim=False, dtype=None): function vector_norm (line 23) | def vector_norm(self, ord=2, dim=None, keepdim=False, dtype=None): function matrix_norm (line 27) | def matrix_norm(self, ord="fro", dim=(-2, -1), keepdim=False, dtype=None): function inv (line 31) | def inv(self): function diagonal (line 35) | def diagonal(self, input, offset=0, dim1=-2, dim2=-1): function cross (line 42) | def cross(input, other, dim=-1): function det (line 46) | def det(A): function solve (line 70) | def solve(): FILE: python/oneflow/mock_torch/__main__.py function main (line 45) | def main(): FILE: python/oneflow/mock_torch/dyn_mock_mod.py class DynamicMockModule (line 25) | class DynamicMockModule(ModuleType): method __init__ (line 26) | def __init__( method __repr__ (line 34) | def __repr__(self) -> str: method hijack (line 37) | def hijack(self, module_name: str, obj: object): method from_package (line 41) | def from_package( method _get_module (line 68) | def _get_module(self, _name: str): method __getattr__ (line 80) | def __getattr__(self, name: str): method __all__ (line 95) | def __all__(self): FILE: python/oneflow/mock_torch/mock_importer.py function hasattr (line 47) | def hasattr(obj, name): function probably_called_from_hasattr (line 54) | def probably_called_from_hasattr(): class ModuleWrapper (line 60) | class ModuleWrapper(ModuleType): method __init__ (line 62) | def __init__(self, module): method __setattr__ (line 65) | def __setattr__(self, name, value): method __getattr__ (line 70) | def __getattr__(self, name: str) -> Any: class OneflowImporter (line 94) | class OneflowImporter(MockEnableDisableMixin, MetaPathFinder, Loader): method __init__ (line 95) | def __init__(self): method find_spec (line 105) | def find_spec(self, fullname, path, target=None): method find_module (line 116) | def find_module(self, fullname, path=None): method create_module (line 120) | def create_module(self, spec): method exec_module (line 166) | def exec_module(self, module): method _enable (line 176) | def _enable( method _disable (line 221) | def _disable(self, globals, *, verbose=False): class BaseMockConfig (line 256) | class BaseMockConfig: method __init__ (line 257) | def __init__( class enable (line 301) | class enable(BaseMockConfig): method __init__ (line 304) | def __init__( method __enter__ (line 351) | def __enter__(self): method __exit__ (line 354) | def __exit__(self, exception_type, exception_value, traceback): class disable (line 368) | class disable: method __init__ (line 369) | def __init__(self): method __enter__ (line 380) | def __enter__(self): method __exit__ (line 383) | def __exit__(self, exception_type, exception_value, traceback): function is_enabled (line 391) | def is_enabled(): FILE: python/oneflow/mock_torch/mock_modules.py class MockModuleDict (line 21) | class MockModuleDict: method __init__ (line 22) | def __init__(self, mapping=None): method add (line 31) | def add(self, key, value): method remove (line 38) | def remove(self, key=None, value=None): method in_forward_dict (line 48) | def in_forward_dict(self, s): method in_inverse_dict (line 51) | def in_inverse_dict(self, s): method inverse_name (line 54) | def inverse_name(self, s: str): # s: spec.name method forward_name (line 57) | def forward_name(self, s: str): class DummyModule (line 61) | class DummyModule(ModuleType): method __init__ (line 62) | def __init__(self, name, verbose=False): method __getattr__ (line 66) | def __getattr__(self, name): method __getitem__ (line 82) | def __getitem__(self, name): method __call__ (line 94) | def __call__(self, *args, **kwargs): method __bool__ (line 100) | def __bool__(self): method __enter__ (line 107) | def __enter__(self): method __exit__ (line 112) | def __exit__(self, exception_type, exception_value, traceback): method __subclasscheck__ (line 117) | def __subclasscheck__(self, subclass): method __instancecheck__ (line 120) | def __instancecheck__(self, instance): FILE: python/oneflow/mock_torch/mock_utils.py class PackageDependencyMixin (line 43) | class PackageDependencyMixin: method find_matching_dependencies (line 57) | def find_matching_dependencies( method varify_input (line 158) | def varify_input(main_pkg, dependencies, callback, verbose=False): method has_dependencies (line 179) | def has_dependencies( class VersionMixin (line 206) | class VersionMixin: method mock_version (line 209) | def mock_version(self, module_a: ModuleType, module_b: ModuleType): method restore_version (line 221) | def restore_version(self): class MockEnableDisableMixin (line 227) | class MockEnableDisableMixin(PackageDependencyMixin, VersionMixin): method is_safe_module (line 239) | def is_safe_module(self, module_key): method mock_enable (line 248) | def mock_enable( method mock_disable (line 332) | def mock_disable(self, globals, module_dict, delete_list, from_cli=Fal... FILE: python/oneflow/mock_torch/torch/__init__.py function __getattr__ (line 21) | def __getattr__(name: str): FILE: python/oneflow/multiprocessing/__init__.py function set_sharing_strategy (line 75) | def set_sharing_strategy(new_strategy): function get_sharing_strategy (line 87) | def get_sharing_strategy(): function get_all_sharing_strategies (line 92) | def get_all_sharing_strategies(): function unlink_all_shared_memory (line 97) | def unlink_all_shared_memory(): FILE: python/oneflow/multiprocessing/_atfork.py function _register (line 23) | def _register(func): function _register (line 33) | def _register(func): function register_after_fork (line 37) | def register_after_fork(func): FILE: python/oneflow/multiprocessing/pool.py function clean_worker (line 22) | def clean_worker(*args, **kwargs): class Pool (line 32) | class Pool(multiprocessing.pool.Pool): method _setup_queues (line 37) | def _setup_queues(self): method _repopulate_pool (line 43) | def _repopulate_pool(self): FILE: python/oneflow/multiprocessing/queue.py class ConnectionWrapper (line 22) | class ConnectionWrapper(object): method __init__ (line 26) | def __init__(self, conn): method send (line 29) | def send(self, obj): method recv (line 34) | def recv(self): method __getattr__ (line 38) | def __getattr__(self, name): class Queue (line 46) | class Queue(multiprocessing.queues.Queue): method __init__ (line 47) | def __init__(self, *args, **kwargs): class SimpleQueue (line 55) | class SimpleQueue(multiprocessing.queues.SimpleQueue): method _make_methods (line 56) | def _make_methods(self): FILE: python/oneflow/multiprocessing/reductions.py function rebuild_empty_tensor (line 37) | def rebuild_empty_tensor(shape, dtype, requires_grad): function rebuild_shm_tensor (line 43) | def rebuild_shm_tensor(shm, shape, dtype, requires_grad): function rebuild_empty_parameter (line 63) | def rebuild_empty_parameter(shape, dtype, requires_grad): function rebuild_shm_parameter (line 69) | def rebuild_shm_parameter(shm, shape, dtype, requires_grad): function reduce_tensor (line 80) | def reduce_tensor(tensor): function reduce_parameter (line 98) | def reduce_parameter(tensor): function init_reductions (line 116) | def init_reductions(): FILE: python/oneflow/multiprocessing/shared_memory/__init__.py class SharedMemory (line 21) | class SharedMemory: method __init__ (line 22) | def __init__(self, name=None, create=False, size=0): method __del__ (line 32) | def __del__(self): method __reduce__ (line 39) | def __reduce__(self): method __repr__ (line 45) | def __repr__(self): method buf (line 49) | def buf(self): method name (line 54) | def name(self): method size (line 59) | def size(self): method close (line 63) | def close(self): method unlink (line 68) | def unlink(self): FILE: python/oneflow/multiprocessing/spawn.py class ProcessException (line 27) | class ProcessException(Exception): method __init__ (line 30) | def __init__(self, msg: str, error_index: int, pid: int): class ProcessRaisedException (line 36) | class ProcessRaisedException(ProcessException): method __init__ (line 42) | def __init__( class ProcessExitedException (line 48) | class ProcessExitedException(ProcessException): method __init__ (line 56) | def __init__( function _wrap (line 69) | def _wrap(fn, i, args, error_queue): class ProcessContext (line 88) | class ProcessContext: method __init__ (line 89) | def __init__(self, processes, error_queues): method pids (line 96) | def pids(self): method join (line 99) | def join(self, timeout=None): class SpawnContext (line 166) | class SpawnContext(ProcessContext): method __init__ (line 167) | def __init__(self, processes, error_queues): function start_processes (line 182) | def start_processes( function spawn (line 206) | def spawn(fn, args=(), nprocs=1, join=True, daemon=False, start_method="... FILE: python/oneflow/nn/__init__.py class DataParallel (line 184) | class DataParallel(Module): method __init__ (line 185) | def __init__(self): FILE: python/oneflow/nn/functional/batch_norm.py function batch_norm (line 23) | def batch_norm( FILE: python/oneflow/nn/functional/ctc_loss.py function ctc_loss (line 20) | def ctc_loss( FILE: python/oneflow/nn/functional/deform_conv.py function deform_conv2d (line 22) | def deform_conv2d( FILE: python/oneflow/nn/functional/depend.py function depend (line 22) | def depend(input: Tensor, depend: Union[Tensor, List[Tensor]]) -> Tensor: FILE: python/oneflow/nn/functional/maxpool.py function max_pool1d (line 20) | def max_pool1d( function max_pool2d (line 69) | def max_pool2d( function max_pool3d (line 118) | def max_pool3d( function adaptive_max_pool1d (line 167) | def adaptive_max_pool1d(input, output_size, return_indices: bool = False): function adaptive_max_pool2d (line 189) | def adaptive_max_pool2d( function adaptive_max_pool3d (line 213) | def adaptive_max_pool3d(input, output_size, return_indices: bool = False): FILE: python/oneflow/nn/functional/pad.py function pad (line 21) | def pad( FILE: python/oneflow/nn/functional/softmax.py function softmax (line 23) | def softmax(input: Tensor, dim: Optional[int] = None, dtype=None) -> Ten... FILE: python/oneflow/nn/graph/cache.py class LRUCache (line 26) | class LRUCache(object): method __init__ (line 29) | def __init__(self, cache_size, keep_the_1st=True): method is_empty (line 36) | def is_empty(self): method is_full (line 39) | def is_full(self): method pop (line 42) | def pop(self): method set (line 50) | def set(self, key, value): method get (line 70) | def get(self, key): method items (line 79) | def items(self): class AvoidRecursiveCacheCall (line 84) | class AvoidRecursiveCacheCall(object): method __init__ (line 85) | def __init__(self, graph) -> None: method __enter__ (line 89) | def __enter__(self): method __exit__ (line 92) | def __exit__(self, exc_type, exc_val, exc_tb): class GraphCache (line 96) | class GraphCache(object): method __init__ (line 97) | def __init__(self, base_graph, cache_size=10, enable_graph_shared=True): method set_cache_size (line 106) | def set_cache_size(self, cache_size): method enable_shared (line 109) | def enable_shared(self, enabled=True): method __call__ (line 112) | def __call__(self, *args, **kwargs): method _compile (line 117) | def _compile(self, *args, **kwargs): method runtime_state_dict (line 122) | def runtime_state_dict( method runtime_state_dict_to (line 138) | def runtime_state_dict_to( method _init_and_get_a_graph_in_cache (line 159) | def _init_and_get_a_graph_in_cache(self, cache_key): method load_runtime_state_dict (line 199) | def load_runtime_state_dict( method gen_key (line 222) | def gen_key(self, *args, **kwargs): method get_graph (line 230) | def get_graph(self, *args, **kwargs): FILE: python/oneflow/nn/graph/graph.py class Graph (line 65) | class Graph(object): method __init__ (line 110) | def __init__( method build (line 211) | def build(self, *args, **kwargs): method __call__ (line 257) | def __call__(self, *args, **kwargs): method add_optimizer (line 288) | def add_optimizer( method set_grad_scaler (line 377) | def set_grad_scaler(self, grad_scaler: GradScaler = None): method state_dict (line 382) | def state_dict( method load_state_dict (line 433) | def load_state_dict( method name (line 466) | def name(self): method is_compiled (line 471) | def is_compiled(self): method training (line 477) | def training(self): method debug (line 481) | def debug( method __repr__ (line 565) | def __repr__(self): method _shallow_repr (line 611) | def _shallow_repr(self): method _ops_repr (line 615) | def _ops_repr(self): method __print (line 630) | def __print(self, s_level=2, v_level=0, msg=None): method _print (line 642) | def _print(self, s_level=2, v_level=0, msg=None): method _config_proto (line 646) | def _config_proto(self): method _optimization_conf_proto (line 650) | def _optimization_conf_proto(self): method _graph_proto (line 654) | def _graph_proto(self): method _full_graph_proto (line 665) | def _full_graph_proto(self): method _full_graph_proto (line 676) | def _full_graph_proto(self, full_job_proto): method _compiled_graph_proto (line 684) | def _compiled_graph_proto(self): method _generate_name (line 694) | def _generate_name(self): method _state (line 701) | def _state(self): method __ensure_state_tensors_contiguous (line 710) | def __ensure_state_tensors_contiguous(self): method _filter_states (line 716) | def _filter_states(self): method _generate_config_proto (line 743) | def _generate_config_proto(self): method _create_states_builder (line 756) | def _create_states_builder(self): method _mark_variable_gradients (line 785) | def _mark_variable_gradients(self): method trace (line 799) | def trace(func): method _compile (line 846) | def _compile(self, *args, **kwargs): method _compile_new (line 861) | def _compile_new(self, *args, **kwargs): method enable_shared (line 879) | def enable_shared(self, mode: bool = True): method share_from (line 889) | def share_from(self, shared_graph: "Graph") -> None: method _compile_from_shared (line 901) | def _compile_from_shared(self, *args, **kwargs): method enable_save_runtime_state_dict (line 1033) | def enable_save_runtime_state_dict(self, mode: bool = True): method runtime_state_dict (line 1043) | def runtime_state_dict( method load_runtime_state_dict (line 1146) | def load_runtime_state_dict( method runtime_state_dict_to (line 1313) | def runtime_state_dict_to( method build_graph (line 1359) | def build_graph(self, *args, **kwargs): method finish_compile_and_init_runtime (line 1392) | def finish_compile_and_init_runtime(self): method __build_graph (line 1451) | def __build_graph(self, *args, **kwargs): method __prepare_for_share_or_runtime_save (line 1627) | def __prepare_for_share_or_runtime_save( method __rebuild_outputs (line 1649) | def __rebuild_outputs( method __build_outputs_buffer (line 1686) | def __build_outputs_buffer(self): method __check_outputs_buffer (line 1718) | def __check_outputs_buffer(self): method __run (line 1740) | def __run(self, *args, **kwargs): method __build_io (line 1797) | def __build_io(self, io_type, build_func, *args, **kwargs): method __io_item_check_and_gen_repr (line 1839) | def __io_item_check_and_gen_repr(self, item, expect_type, io_type, name): method __map_io (line 1878) | def __map_io(self, io_type, func, *args, **kwargs): method __map_io_lite (line 1907) | def __map_io_lite(self, func, *args, **kwargs): method __flatten_io (line 1914) | def __flatten_io(self, io_type, *args, **kwargs): method __io_item_check (line 1925) | def __io_item_check(self, item, expect_type, io_type, name): method __empty_like_io (line 1940) | def __empty_like_io(self, io_type, *args, **kwargs): method __copy_io (line 1957) | def __copy_io(self, io_type, *args, **kwargs): method _add_module (line 1965) | def _add_module(self, name: str, module: Module = None) -> None: method __setattr__ (line 2028) | def __setattr__(self, name: str, value=None): method __getattr__ (line 2048) | def __getattr__(self, name: str): method __del__ (line 2058) | def __del__(self): method __ensure_input_tensors_contiguous (line 2077) | def __ensure_input_tensors_contiguous(self, *args, **kwargs): method __ensure_input_tensors_contiguous_and_flatten (line 2092) | def __ensure_input_tensors_contiguous_and_flatten(self, *args, **kwargs): method with_dynamic_input_shape (line 2125) | def with_dynamic_input_shape(*, size: int = 10, enable_shared: bool = ... FILE: python/oneflow/nn/graph/graph_block.py class GraphBlockType (line 30) | class GraphBlockType: class GraphBlock (line 38) | class GraphBlock(object): method __init__ (line 39) | def __init__( method name (line 58) | def name(self): method name_prefix (line 62) | def name_prefix(self): method type (line 66) | def type(self): method prev_scope (line 70) | def prev_scope(self): method scope (line 76) | def scope(self): method scope_context (line 83) | def scope_context(self): class GraphModule (line 87) | class GraphModule(GraphBlock): method __init__ (line 94) | def __init__( method set_stage (line 119) | def set_stage(self, stage_id: int = None, placement=None): method stage_id (line 155) | def stage_id(self): method stage_id (line 164) | def stage_id(self, value: int = None): method stage_placement (line 177) | def stage_placement(self): method activation_checkpointing (line 182) | def activation_checkpointing(self): method activation_checkpointing (line 211) | def activation_checkpointing(self, mode: bool = False): method _config_repr (line 217) | def _config_repr(self): method debug (line 240) | def debug( method _ops_repr (line 291) | def _ops_repr(self): method _shallow_repr (line 317) | def _shallow_repr(self): method _repr_lines (line 343) | def _repr_lines(self): method __repr__ (line 349) | def __repr__(self): class GraphTensor (line 362) | class GraphTensor(GraphBlock): method __init__ (line 366) | def __init__( method set_stage (line 380) | def set_stage(self, stage_id: int = None, placement=None): method stage_id (line 385) | def stage_id(self): method stage_id (line 389) | def stage_id(self, value: int = None): method stage_placement (line 398) | def stage_placement(self): FILE: python/oneflow/nn/graph/graph_config.py class GraphConfig (line 26) | class GraphConfig(object): method __init__ (line 30) | def __init__(self): method _train (line 36) | def _train(self, mode: bool = True): method training (line 43) | def training(self): method enable_amp (line 50) | def enable_amp(self, mode: bool = True, *, dtype: flow.dtype = flow.fl... method set_zero_redundancy_optimizer_mode (line 81) | def set_zero_redundancy_optimizer_mode(self, mode: str = "distributed_... method enable_zero (line 86) | def enable_zero( method allow_fuse_model_update_ops (line 142) | def allow_fuse_model_update_ops(self, mode: bool = True): method allow_fuse_add_to_output (line 166) | def allow_fuse_add_to_output(self, mode: bool = True): method allow_fuse_cast_scale (line 192) | def allow_fuse_cast_scale(self, mode: bool = True): method set_gradient_accumulation_steps (line 219) | def set_gradient_accumulation_steps(self, value): method set_outputs_buffer_size (line 249) | def set_outputs_buffer_size(self, value: int = 2): method enable_cudnn_conv_heuristic_search_algo (line 267) | def enable_cudnn_conv_heuristic_search_algo(self, mode: bool = True): method enable_straighten_algorithm (line 295) | def enable_straighten_algorithm(self, mode: str = "MemoryFirst"): method enable_compress_memory (line 343) | def enable_compress_memory(self, mode: bool = True): method enable_choose_best_memory_allocation (line 354) | def enable_choose_best_memory_allocation(self, mode: bool = True): method enable_auto_parallel (line 376) | def enable_auto_parallel(self, mode: bool = True): method enable_auto_parallel_ignore_user_sbp_config (line 384) | def enable_auto_parallel_ignore_user_sbp_config(self, mode: bool = True): method set_auto_parallel_computation_cost_ratio (line 392) | def set_auto_parallel_computation_cost_ratio(self, ratio): method set_auto_parallel_wait_time (line 398) | def set_auto_parallel_wait_time(self, cost): method enable_auto_parallel_trunk_algo (line 408) | def enable_auto_parallel_trunk_algo(self, mode: bool = True): method enable_auto_parallel_sbp_collector (line 414) | def enable_auto_parallel_sbp_collector(self, mode: bool = True): method enable_auto_memory (line 420) | def enable_auto_memory(self, mode: str = "AdaptiveMemory"): method enable_multi_tensor_update (line 460) | def enable_multi_tensor_update(self, mode: bool = True): method enable_fused_model_update_cast (line 466) | def enable_fused_model_update_cast(self, mode: bool = True): method _generate_optimizer_and_variable_configs (line 472) | def _generate_optimizer_and_variable_configs( method __repr__ (line 477) | def __repr__(self): FILE: python/oneflow/nn/graph/optimizer.py class OptDict (line 20) | class OptDict(object): method __init__ (line 21) | def __init__(self, opt_dict): method generate_optimizer_and_variable_configs (line 50) | def generate_optimizer_and_variable_configs(self, job_conf, vars_conf): class VariableConfig (line 72) | class VariableConfig(object): method __init__ (line 73) | def __init__(self, name: str): method name (line 80) | def name(self): method l2 (line 84) | def l2(self): method l2 (line 88) | def l2(self, l2: float = 0.0): method bound_optimizer (line 92) | def bound_optimizer(self): method bound_optimizer (line 96) | def bound_optimizer(self, opt): method __repr__ (line 99) | def __repr__(self): FILE: python/oneflow/nn/graph/proxy.py function get_proxy_cls (line 41) | def get_proxy_cls(item): class Proxy (line 60) | class Proxy(object): method __init__ (line 61) | def __init__(self): method to (line 71) | def to(self, *args, **kwargs): class ProxyModule (line 82) | class ProxyModule(Proxy): method __init__ (line 83) | def __init__( method _oneflow_internal_graphblock__set_origin (line 101) | def _oneflow_internal_graphblock__set_origin(self, origin): method __call__ (line 141) | def __call__(self, *args, **kwargs): method __class__ (line 221) | def __class__(self): method __block_forward (line 227) | def __block_forward(self, *args, **kwargs): method __pre_forward_map (line 240) | def __pre_forward_map(self, *args, **kwargs): method __get_or_create_global (line 272) | def __get_or_create_global(self, input_tensor: Tensor = None, placemen... method __get_or_create_identity (line 287) | def __get_or_create_identity(self, input_tensor: Tensor = None): method add_module (line 310) | def add_module(self, name: str, module: Optional[Module]) -> None: method register_parameter (line 326) | def register_parameter(self, name: str, param: Optional[Parameter]) ->... method modules (line 336) | def modules(self, memo: Optional[Set["Proxy"]] = None) -> Iterator["Pr... method __map_io (line 349) | def __map_io(self, io_type, func, func_desc, *args, **kwargs): method __io_tensor_check_and_gen (line 393) | def __io_tensor_check_and_gen(self, item, io_type, name): method __members (line 412) | def __members(self, get_members_fn, recurse=True) -> Iterator["Proxy"]: method parameters (line 424) | def parameters(self, recurse: bool = True) -> Iterator["Proxy"]: method buffers (line 430) | def buffers(self, recurse: bool = True) -> Iterator["Proxy"]: method __setattr__ (line 436) | def __setattr__(self, name: str, value=None) -> None: method __getattr__ (line 466) | def __getattr__(self, name: str): method _get_from_states (line 507) | def _get_from_states(self, name, states_name): method __repr__ (line 529) | def __repr__(self): method _shallow_repr (line 563) | def _shallow_repr(self): method __print (line 576) | def __print(self, s_level=2, v_level=0, msg: str = ""): class LazyBuilder (line 589) | class LazyBuilder(object): method __init__ (line 590) | def __init__(self, name: str = None, method=None): method try_build (line 596) | def try_build(self, block=None): class ProxyTensor (line 606) | class ProxyTensor(Proxy): method __init__ (line 607) | def __init__( method _oneflow_internal_graphblock__set_origin (line 633) | def _oneflow_internal_graphblock__set_origin(self, origin): method lazy_origin (line 637) | def lazy_origin(self): method lazy_origin_builder (line 644) | def lazy_origin_builder(self): method set_lazy_origin_builder (line 651) | def set_lazy_origin_builder(self, builder=None): method try_build (line 658) | def try_build(self): method __repr__ (line 663) | def __repr__(self): method _shallow_repr (line 671) | def _shallow_repr(self): class ProxySequential (line 685) | class ProxySequential(get_seq(ProxyModule)): method __init__ (line 686) | def __init__( class ProxyModuleList (line 701) | class ProxyModuleList(get_list(ProxyModule)): method __init__ (line 702) | def __init__( class ProxyModuleDict (line 734) | class ProxyModuleDict(get_dict(ProxyModule)): method __init__ (line 735) | def __init__( class ProxyParameterList (line 750) | class ProxyParameterList(get_para_list(ProxyModule)): method __init__ (line 751) | def __init__( method __getitem__ (line 766) | def __getitem__(self, idx): class ProxyParameterDict (line 777) | class ProxyParameterDict(get_para_dict(ProxyModule)): method __init__ (line 778) | def __init__( method __getitem__ (line 793) | def __getitem__(self, key: str): FILE: python/oneflow/nn/graph/util.py function _nd_sbp2repr (line 32) | def _nd_sbp2repr(nd_sbp): function _blob_desc_repr (line 49) | def _blob_desc_repr(blob_desc): function _get_args_repr (line 62) | def _get_args_repr(ordered_bn, bn2lbn, bn2nd_sbp, lbn2blob_desc): function _get_user_op_io_repr (line 102) | def _get_user_op_io_repr(op_conf, bn2nd_sbp, lbn2blob_desc): function _get_var_op_io_repr (line 117) | def _get_var_op_io_repr(op_conf, bn2nd_sbp, lbn2blob_desc): function _get_iden_op_io_repr (line 129) | def _get_iden_op_io_repr(op_conf, bn2nd_sbp, lbn2blob_desc): function _get_input_op_io_repr (line 150) | def _get_input_op_io_repr(op_conf, bn2nd_sbp, lbn2blob_desc): function _get_output_op_io_repr (line 164) | def _get_output_op_io_repr(op_conf, bn2nd_sbp, lbn2blob_desc): class GraphIR (line 188) | class GraphIR(object): method __init__ (line 189) | def __init__(self, g_proto: job_pb.Job): method get_op_conf (line 195) | def get_op_conf(self, op_name: str) -> Optional[op_conf_util.OperatorC... method get_op_placement (line 204) | def get_op_placement(self, op_name: str) -> Optional[oneflow.placement]: function _op_signature (line 218) | def _op_signature( function operators_repr (line 273) | def operators_repr(ops: Message, graph_ir: GraphIR, show_op_loc: bool,) ... function add_indent (line 289) | def add_indent(in_s, num_spaces): function sys_exc_error_msg (line 300) | def sys_exc_error_msg(): function seq_to_func_return (line 310) | def seq_to_func_return(seq, need_unpack=False): function _rsd_sub_destination_to (line 316) | def _rsd_sub_destination_to(origin_dict, dest_device_str): function _parallel_conf_to (line 327) | def _parallel_conf_to(parallel_conf, dest_device): function _mem_case_to (line 333) | def _mem_case_to(mem_case, dest_device): function _job_to (line 343) | def _job_to(job, dest_device): function _modify_bits (line 350) | def _modify_bits(original_num, k, j, new_num): function _get_bits (line 359) | def _get_bits(original_num, k, j): function _task_id_to (line 366) | def _task_id_to(task_id, dest_device): function _thrd_id_to (line 375) | def _thrd_id_to(thrd_id, dest_device): function _plan_to (line 383) | def _plan_to(plan_str, dest_device): FILE: python/oneflow/nn/init.py function uniform_ (line 32) | def uniform_(tensor, a=0.0, b=1.0): function normal_ (line 55) | def normal_(tensor, mean=0.0, std=1.0): function xavier_uniform_ (line 87) | def xavier_uniform_(tensor, gain=1.0, *, data_format="NCHW"): function xavier_normal_ (line 117) | def xavier_normal_(tensor, gain=1.0, *, data_format="NCHW"): function orthogonal_ (line 148) | def orthogonal_(tensor, gain=1.0): function kaiming_uniform_ (line 171) | def kaiming_uniform_( function kaiming_normal_ (line 213) | def kaiming_normal_( function trunc_normal_ (line 256) | def trunc_normal_(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0): function constant_ (line 293) | def constant_(tensor, val): function ones_ (line 314) | def ones_(tensor): function zeros_ (line 333) | def zeros_(tensor): function eye_ (line 352) | def eye_(tensor): function _calculate_fan_in_and_fan_out (line 375) | def _calculate_fan_in_and_fan_out(tensor): FILE: python/oneflow/nn/modules/_functions.py class BackwardHookFunction (line 19) | class BackwardHookFunction(flow.autograd.Function): method forward (line 21) | def forward(ctx, *args): method backward (line 26) | def backward(ctx, *args): FILE: python/oneflow/nn/modules/activation.py class PReLU (line 23) | class PReLU(Module): method __init__ (line 68) | def __init__( method forward (line 77) | def forward(self, x): method extra_repr (line 80) | def extra_repr(self) -> str: class ReLU (line 84) | class ReLU(Module): method __init__ (line 111) | def __init__(self, inplace: bool = False): method forward (line 115) | def forward(self, x): method extra_repr (line 118) | def extra_repr(self): class ReLU6 (line 123) | class ReLU6(Module): method __init__ (line 159) | def __init__(self, inplace: bool = False): method forward (line 163) | def forward(self, x): method extra_repr (line 168) | def extra_repr(self): function relu6 (line 173) | def relu6(input, inplace=False): class Tanh (line 185) | class Tanh(Module): method __init__ (line 216) | def __init__(self): method forward (line 219) | def forward(self, input): class ELU (line 223) | class ELU(Module): method __init__ (line 254) | def __init__(self, alpha: float = 1.0, inplace: bool = False): method forward (line 259) | def forward(self, x): method extra_repr (line 264) | def extra_repr(self): class CELU (line 270) | class CELU(Module): method __init__ (line 307) | def __init__(self, alpha: float = 1.0, inplace: bool = False): method forward (line 312) | def forward(self, x): method extra_repr (line 315) | def extra_repr(self): class GELU (line 321) | class GELU(Module): method __init__ (line 362) | def __init__(self, approximate: str = "none"): method forward (line 366) | def forward(self, input): class QuickGELU (line 373) | class QuickGELU(Module): method __init__ (line 403) | def __init__(self): method forward (line 406) | def forward(self, x): class SquareReLU (line 410) | class SquareReLU(Module): method __init__ (line 442) | def __init__(self): method forward (line 445) | def forward(self, x): class Sigmoid (line 449) | class Sigmoid(Module): method __init__ (line 474) | def __init__(self): method forward (line 477) | def forward(self, x): class Hardsigmoid (line 481) | class Hardsigmoid(Module): method __init__ (line 517) | def __init__(self, inplace: bool = False): method forward (line 521) | def forward(self, x): method extra_repr (line 526) | def extra_repr(self): class Hardshrink (line 531) | class Hardshrink(Module): method __init__ (line 568) | def __init__(self, lambd: float = 0.5, inplace: bool = False): method forward (line 573) | def forward(self, x): method extra_repr (line 576) | def extra_repr(self) -> str: class Softmax (line 582) | class Softmax(Module): method __init__ (line 628) | def __init__(self, dim: Optional[int] = None): method forward (line 632) | def forward(self, x): method extra_repr (line 635) | def extra_repr(self): class LogSoftmax (line 639) | class LogSoftmax(Module): method __init__ (line 675) | def __init__(self, dim: Optional[int] = None): method forward (line 679) | def forward(self, x): method extra_repr (line 682) | def extra_repr(self): class LogSigmoid (line 686) | class LogSigmoid(Module): method __init__ (line 715) | def __init__(self): method forward (line 718) | def forward(self, x): class Softplus (line 722) | class Softplus(Module): method __init__ (line 759) | def __init__(self, beta: int = 1, threshold: int = 20): method forward (line 764) | def forward(self, x): method extra_repr (line 767) | def extra_repr(self): class Hardswish (line 771) | class Hardswish(Module): method __init__ (line 805) | def __init__(self, inplace: bool = False): method forward (line 809) | def forward(self, x): method extra_repr (line 814) | def extra_repr(self): class Hardtanh (line 819) | class Hardtanh(Module): method __init__ (line 865) | def __init__( method forward (line 888) | def forward(self, x): method extra_repr (line 893) | def extra_repr(self): class LeakyReLU (line 899) | class LeakyReLU(Module): method __init__ (line 932) | def __init__(self, negative_slope: float = 0.01, inplace: bool = False): method forward (line 937) | def forward(self, x): method extra_repr (line 940) | def extra_repr(self): class RReLU (line 946) | class RReLU(Module): method __init__ (line 986) | def __init__( method forward (line 994) | def forward(self, x): method extra_repr (line 997) | def extra_repr(self): class Mish (line 1004) | class Mish(Module): method __init__ (line 1034) | def __init__(self, inplace: bool = False): method forward (line 1038) | def forward(self, x): class SiLU (line 1042) | class SiLU(Module): method __init__ (line 1078) | def __init__(self, inplace: bool = False): method forward (line 1082) | def forward(self, x): class SELU (line 1086) | class SELU(Module): method __init__ (line 1127) | def __init__(self, inplace: bool = False): method forward (line 1131) | def forward(self, x): class Softshrink (line 1135) | class Softshrink(Module): method __init__ (line 1176) | def __init__(self, lambd: float = 0.5, inplace: bool = False): method forward (line 1181) | def forward(self, x): method extra_repr (line 1184) | def extra_repr(self) -> str: class Softsign (line 1190) | class Softsign(Module): method __init__ (line 1218) | def __init__(self, inplace: bool = False): method forward (line 1222) | def forward(self, x): class GLU (line 1226) | class GLU(Module): method __init__ (line 1262) | def __init__(self, dim: Optional[int] = -1): method forward (line 1266) | def forward(self, input): class Threshold (line 1270) | class Threshold(Module): method __init__ (line 1312) | def __init__(self, threshold: float, value: float): method forward (line 1317) | def forward(self, input): FILE: python/oneflow/nn/modules/affine_grid.py function affine_grid (line 21) | def affine_grid(theta, size: List[int], align_corners: bool = False): FILE: python/oneflow/nn/modules/all_reduce.py class AllReduce (line 22) | class AllReduce(Module): method __init__ (line 23) | def __init__(self, parallel_conf_str: str): method forward (line 30) | def forward(self, x): FILE: python/oneflow/nn/modules/arange.py function arange_op (line 20) | def arange_op( FILE: python/oneflow/nn/modules/argsort.py function argsort_op (line 25) | def argsort_op(input, dim: int = -1, descending: bool = False): FILE: python/oneflow/nn/modules/argwhere.py function argwhere_op (line 25) | def argwhere_op(input, dtype: Optional[flow.dtype] = flow.int32): FILE: python/oneflow/nn/modules/as_tensor.py function as_tensor (line 20) | def as_tensor(data, dtype=None, device=None): FILE: python/oneflow/nn/modules/batchnorm.py class _NormBase (line 24) | class _NormBase(Module): method __init__ (line 27) | def __init__( method reset_running_stats (line 58) | def reset_running_stats(self) -> None: method reset_parameters (line 64) | def reset_parameters(self) -> None: method _check_input_dim (line 70) | def _check_input_dim(self, input): method _load_from_state_dict (line 73) | def _load_from_state_dict( method extra_repr (line 106) | def extra_repr(self): class _BatchNorm (line 112) | class _BatchNorm(_NormBase): method __init__ (line 113) | def __init__( method forward (line 124) | def forward(self, x): class BatchNorm1d (line 150) | class BatchNorm1d(_BatchNorm): method _check_input_dim (line 220) | def _check_input_dim(self, input): class BatchNorm2d (line 227) | class BatchNorm2d(_BatchNorm): method __init__ (line 297) | def __init__( method to_memory_format (line 309) | def to_memory_format(self, memory_format) -> None: method _check_input_dim (line 315) | def _check_input_dim(self, input): class BatchNorm3d (line 320) | class BatchNorm3d(_BatchNorm): method _check_input_dim (line 392) | def _check_input_dim(self, input): class SyncBatchNormFunction (line 403) | class SyncBatchNormFunction(flow.autograd.Function): method forward (line 405) | def forward(self, input, weight, bias, running_mean, running_var): method backward (line 489) | def backward(self, grad_output): class SyncBatchNorm (line 533) | class SyncBatchNorm(_BatchNorm): method __init__ (line 614) | def __init__( method _check_input_dim (line 624) | def _check_input_dim(self, input): method _check_non_zero_input_channels (line 637) | def _check_non_zero_input_channels(self, input): method forward (line 643) | def forward(self, input): method convert_sync_batchnorm (line 709) | def convert_sync_batchnorm(cls, module): FILE: python/oneflow/nn/modules/batchnorm_fused.py class _FusedNormBase (line 23) | class _FusedNormBase(Module): method __init__ (line 26) | def __init__( method reset_running_stats (line 54) | def reset_running_stats(self) -> None: method reset_parameters (line 59) | def reset_parameters(self) -> None: method _check_input_dim (line 65) | def _check_input_dim(self, input): method extra_repr (line 68) | def extra_repr(self): class _FusedBatchNorm (line 74) | class _FusedBatchNorm(_FusedNormBase): method __init__ (line 75) | def __init__( method forward (line 86) | def forward(self, x, addend=None): class FusedBatchNorm1d (line 109) | class FusedBatchNorm1d(_FusedBatchNorm): method _check_input_dim (line 182) | def _check_input_dim(self, input): class FusedBatchNorm2d (line 189) | class FusedBatchNorm2d(_FusedBatchNorm): method __init__ (line 262) | def __init__( method _check_input_dim (line 274) | def _check_input_dim(self, input): class FusedBatchNorm3d (line 279) | class FusedBatchNorm3d(_FusedBatchNorm): method _check_input_dim (line 352) | def _check_input_dim(self, input): FILE: python/oneflow/nn/modules/broadcast_ops.py function broadcast_shapes (line 20) | def broadcast_shapes(*shapes): function broadcast_tensors (line 53) | def broadcast_tensors(*tensors): function broadcast_to (line 88) | def broadcast_to(input, shape): FILE: python/oneflow/nn/modules/constant.py class _ConstantBase (line 25) | class _ConstantBase: method __init__ (line 26) | def __init__( method forward (line 70) | def forward(self): function _handle_meta_args (line 107) | def _handle_meta_args( class Ones (line 181) | class Ones(_ConstantBase): method __init__ (line 182) | def __init__( function ones_op (line 194) | def ones_op( function ones_like_op (line 240) | def ones_like_op( class Zeros (line 263) | class Zeros(_ConstantBase): method __init__ (line 264) | def __init__( function zeros_op (line 276) | def zeros_op( function zeros_like_op (line 317) | def zeros_like_op( class Full (line 340) | class Full(_ConstantBase): method __init__ (line 341) | def __init__( function full_op (line 354) | def full_op( function full_like_op (line 410) | def full_like_op( function new_ones_op (line 479) | def new_ones_op( function new_zeros_op (line 500) | def new_zeros_op( function new_full_op (line 521) | def new_full_op( FILE: python/oneflow/nn/modules/container.py class Sequential (line 20) | class Sequential(get_seq(Module)): class ModuleList (line 60) | class ModuleList(get_list(Module)): class ModuleDict (line 108) | class ModuleDict(get_dict(Module)): class ParameterList (line 167) | class ParameterList(get_para_list(Module)): class ParameterDict (line 215) | class ParameterDict(get_para_dict(Module)): FILE: python/oneflow/nn/modules/conv.py function slice (line 28) | def slice(x, begin, size): class ConvUtil (line 61) | class ConvUtil(object): method split (line 63) | def split(cls, x, axis, split_num): function get_padding (line 76) | def get_padding(padding, kernel_size, dilation, stride): class Conv1d (line 97) | class Conv1d(Module): method __init__ (line 188) | def __init__( method reset_parameters (line 236) | def reset_parameters(self) -> None: method _conv_forward (line 243) | def _conv_forward(self, x, weight, bias): method forward (line 255) | def forward(self, x): method extra_repr (line 258) | def extra_repr(self): class Conv2d (line 273) | class Conv2d(Module): method __init__ (line 389) | def __init__( method reset_parameters (line 456) | def reset_parameters(self) -> None: method to_memory_format (line 463) | def to_memory_format(self, memory_format) -> None: method _conv_forward (line 476) | def _conv_forward(self, x, weight, bias): method forward (line 488) | def forward(self, x): method extra_repr (line 491) | def extra_repr(self): class Conv3d (line 506) | class Conv3d(Module): method __init__ (line 601) | def __init__( method reset_parameters (line 650) | def reset_parameters(self) -> None: method to_memory_format (line 657) | def to_memory_format(self, memory_format) -> None: method _conv_forward (line 670) | def _conv_forward(self, x, weight, bias): method forward (line 682) | def forward(self, x): method extra_repr (line 685) | def extra_repr(self): class ConvTranspose1d (line 700) | class ConvTranspose1d(Module): method __init__ (line 783) | def __init__( method reset_parameters (line 820) | def reset_parameters(self) -> None: method forward (line 827) | def forward(self, x): class ConvTranspose2d (line 841) | class ConvTranspose2d(Module): method __init__ (line 910) | def __init__( method reset_parameters (line 947) | def reset_parameters(self) -> None: method forward (line 954) | def forward(self, x): class ConvTranspose3d (line 969) | class ConvTranspose3d(Module): method __init__ (line 1075) | def __init__( method reset_parameters (line 1110) | def reset_parameters(self) -> None: method forward (line 1117) | def forward(self, x): FILE: python/oneflow/nn/modules/dataset.py function local_gen_random_seed (line 33) | def local_gen_random_seed(seed=None): class OFRecordReader (line 42) | class OFRecordReader(Module): method __init__ (line 43) | def __init__( method forward (line 94) | def forward(self): class OFRecordRawDecoder (line 127) | class OFRecordRawDecoder(Module): method __init__ (line 128) | def __init__( method forward (line 155) | def forward(self, input): class CoinFlip (line 168) | class CoinFlip(Module): method __init__ (line 199) | def __init__( method forward (line 241) | def forward(self): class CropMirrorNormalize (line 264) | class CropMirrorNormalize(Module): method __init__ (line 300) | def __init__( method forward (line 360) | def forward(self, input, mirror=None): class OFRecordImageDecoderRandomCrop (line 428) | class OFRecordImageDecoderRandomCrop(Module): method __init__ (line 429) | def __init__( method forward (line 452) | def forward(self, input): class OFRecordImageDecoder (line 467) | class OFRecordImageDecoder(Module): method __init__ (line 468) | def __init__(self, blob_name: str, color_space: str = "BGR"): method forward (line 476) | def forward(self, input): class OFRecordImageGpuDecoderRandomCropResize (line 483) | class OFRecordImageGpuDecoderRandomCropResize(Module): method __init__ (line 484) | def __init__( method forward (line 523) | def forward(self, input): class TensorBufferToListOfTensors (line 553) | class TensorBufferToListOfTensors(Module): method __init__ (line 554) | def __init__( method forward (line 568) | def forward(self, input): function tensor_buffer_to_list_of_tensors (line 578) | def tensor_buffer_to_list_of_tensors(tensor, out_shapes, out_dtypes): class ImageResize (line 584) | class ImageResize(Module): method __init__ (line 585) | def __init__( method forward (line 703) | def forward(self, input): function raw_decoder (line 736) | def raw_decoder( function get_ofrecord_handle (line 760) | def get_ofrecord_handle( class ImageFlip (line 784) | class ImageFlip(Module): method __init__ (line 835) | def __init__(self): method forward (line 838) | def forward(self, images, flip_code): class ImageDecode (line 842) | class ImageDecode(Module): method __init__ (line 843) | def __init__(self, dtype: flow.dtype = flow.uint8, color_space: str = ... method forward (line 849) | def forward(self, input): class ImageNormalize (line 855) | class ImageNormalize(Module): method __init__ (line 856) | def __init__(self, std: Sequence[float], mean: Sequence[float]): method forward (line 862) | def forward(self, input): class COCOReader (line 868) | class COCOReader(Module): method __init__ (line 869) | def __init__( method forward (line 907) | def forward(self): class ImageBatchAlign (line 942) | class ImageBatchAlign(Module): method __init__ (line 943) | def __init__(self, shape: Sequence[int], dtype: flow.dtype, alignment:... method forward (line 952) | def forward(self, input): class OFRecordBytesDecoder (line 963) | class OFRecordBytesDecoder(Module): method __init__ (line 1010) | def __init__(self, blob_name: str, name: Optional[str] = None): method forward (line 1019) | def forward(self, input): class GPTIndexedBinDataReader (line 1023) | class GPTIndexedBinDataReader(Module): method __init__ (line 1024) | def __init__( method forward (line 1069) | def forward(self): class RawReader (line 1104) | class RawReader(Module): method __init__ (line 1105) | def __init__( method forward (line 1131) | def forward(self): function _handle_distributed_args (line 1161) | def _handle_distributed_args(module, device, placement, sbp): function _handle_shuffle_args (line 1191) | def _handle_shuffle_args(module, shuffle, random_seed): FILE: python/oneflow/nn/modules/distance.py class CosineSimilarity (line 23) | class CosineSimilarity(Module): method __init__ (line 55) | def __init__(self, dim: Optional[int] = 1, eps: Optional[float] = 1e-0... method forward (line 60) | def forward(self, x1: Tensor, x2: Tensor) -> Tensor: class PairwiseDistance (line 64) | class PairwiseDistance(Module): method __init__ (line 98) | def __init__( method forward (line 109) | def forward(self, x1: Tensor, x2: Tensor) -> Tensor: FILE: python/oneflow/nn/modules/distributed_partial_fc_sample.py class DistributedPariticalFCSample (line 22) | class DistributedPariticalFCSample(Module): method __init__ (line 23) | def __init__(self, num_sample): method forward (line 36) | def forward(self, weight, label): function distributed_partial_fc_sample_op (line 43) | def distributed_partial_fc_sample_op(weight, label, num_sample): FILE: python/oneflow/nn/modules/dropout.py class _DropoutNd (line 24) | class _DropoutNd(Module): method __init__ (line 29) | def __init__(self, p: float = 0.5, inplace: bool = False) -> None: method extra_repr (line 38) | def extra_repr(self) -> str: class Dropout (line 42) | class Dropout(_DropoutNd): method __init__ (line 43) | def __init__(self, p: float = 0.5, inplace: bool = False, generator=No... method forward (line 48) | def forward(self, x, addend=None): class Dropout1d (line 59) | class Dropout1d(Dropout): method forward (line 60) | def forward(self, x, addend=None): class Dropout2d (line 64) | class Dropout2d(Dropout): method forward (line 65) | def forward(self, x, addend=None): class Dropout3d (line 69) | class Dropout3d(Dropout): method forward (line 70) | def forward(self, x, addend=None): FILE: python/oneflow/nn/modules/einsum.py function einsum_op (line 19) | def einsum_op(equation, *operands): FILE: python/oneflow/nn/modules/empty.py function empty_op (line 24) | def empty_op( function empty_like_op (line 79) | def empty_like_op( function new_empty_op (line 106) | def new_empty_op( FILE: python/oneflow/nn/modules/expand.py function expand_op (line 20) | def expand_op(input, *sizes): FILE: python/oneflow/nn/modules/fake_quantization.py class FakeQuantization (line 20) | class FakeQuantization(Module): method __init__ (line 93) | def __init__( method forward (line 104) | def forward(self, input, scale, zero_point): FILE: python/oneflow/nn/modules/flatten.py class Flatten (line 21) | class Flatten(Module): method __init__ (line 42) | def __init__(self, start_dim: int = 1, end_dim: int = -1) -> None: method forward (line 47) | def forward(self, input): method extra_repr (line 50) | def extra_repr(self) -> str: FILE: python/oneflow/nn/modules/fold.py class Fold (line 21) | class Fold(Module): method __init__ (line 140) | def __init__( method forward (line 155) | def forward(self, input): method extra_repr (line 166) | def extra_repr(self) -> str: class Unfold (line 175) | class Unfold(Module): method __init__ (line 289) | def __init__( method forward (line 302) | def forward(self, input): method extra_repr (line 312) | def extra_repr(self) -> str: FILE: python/oneflow/nn/modules/fused_mlp.py class FusedMLP (line 25) | class FusedMLP(Module): method __init__ (line 65) | def __init__( method add_parameters (line 95) | def add_parameters(self) -> None: method weight (line 150) | def weight(self, i): method weights (line 156) | def weights(self): method bias (line 162) | def bias(self, i): method biases (line 168) | def biases(self): method reset_parameters (line 174) | def reset_parameters(self) -> None: method forward (line 184) | def forward(self, x): method extra_repr (line 198) | def extra_repr(self) -> str: FILE: python/oneflow/nn/modules/global_cast.py function _check_sbp (line 21) | def _check_sbp(sbp): function local_to_global_op (line 37) | def local_to_global_op(input, placement=None, sbp=None, *, check_meta=Tr... function global_to_global_op (line 58) | def global_to_global_op( function to_global_op (line 81) | def to_global_op(input, placement=None, sbp=None, **kwargs): function to_local_op (line 92) | def to_local_op(input, *, copy=False): FILE: python/oneflow/nn/modules/grid_sample.py function grid_sample (line 20) | def grid_sample( FILE: python/oneflow/nn/modules/instancenorm.py class _InstanceNorm (line 20) | class _InstanceNorm(_NormBase): method __init__ (line 21) | def __init__( method _forward (line 31) | def _forward(self, x): method forward (line 51) | def forward(self, x): class InstanceNorm1d (line 59) | class InstanceNorm1d(_InstanceNorm): method _check_input_dim (line 137) | def _check_input_dim(self, input): class InstanceNorm2d (line 146) | class InstanceNorm2d(_InstanceNorm): method _check_input_dim (line 224) | def _check_input_dim(self, input): class InstanceNorm3d (line 229) | class InstanceNorm3d(_InstanceNorm): method _check_input_dim (line 306) | def _check_input_dim(self, input): FILE: python/oneflow/nn/modules/interpolate.py class Interpolate (line 25) | class Interpolate: method __init__ (line 26) | def __init__( method forward (line 72) | def forward(self, x): function interpolate (line 215) | def interpolate( function interpolate_like (line 313) | def interpolate_like( FILE: python/oneflow/nn/modules/is_tensor.py function is_tensor_op (line 20) | def is_tensor_op(obj): FILE: python/oneflow/nn/modules/linear.py class Identity (line 25) | class Identity(Module): method __init__ (line 48) | def __init__(self, *args, **kwargs): method forward (line 51) | def forward(self, input: Tensor) -> Tensor: class Linear (line 55) | class Linear(Module): method __init__ (line 95) | def __init__( method reset_parameters (line 120) | def reset_parameters(self) -> None: method forward (line 129) | def forward(self, x): method extra_repr (line 138) | def extra_repr(self) -> str: function linear (line 144) | def linear(input, weight, bias=None): FILE: python/oneflow/nn/modules/linspace.py function linspace_op (line 21) | def linspace_op( FILE: python/oneflow/nn/modules/logspace.py function logspace_op (line 22) | def logspace_op( FILE: python/oneflow/nn/modules/loss.py class _Loss (line 24) | class _Loss(Module): method __init__ (line 25) | def __init__(self, reduction: str = "mean") -> None: class _WeightedLoss (line 31) | class _WeightedLoss(_Loss): method __init__ (line 32) | def __init__( class L1Loss (line 39) | class L1Loss(_Loss): method __init__ (line 94) | def __init__(self, reduction: str = "mean") -> None: method forward (line 97) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class CrossEntropyLoss (line 101) | class CrossEntropyLoss(_WeightedLoss): method __init__ (line 237) | def __init__( method forward (line 252) | def forward(self, input, target): class BCELoss (line 263) | class BCELoss(_WeightedLoss): method __init__ (line 327) | def __init__( method forward (line 332) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class NLLLoss (line 338) | class NLLLoss(_WeightedLoss): method __init__ (line 414) | def __init__( method forward (line 423) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class KLDivLoss (line 429) | class KLDivLoss(_Loss): method __init__ (line 517) | def __init__(self, reduction: str = "mean", log_target: bool = False) ... method forward (line 526) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class MSELoss (line 530) | class MSELoss(_Loss): method __init__ (line 601) | def __init__(self, reduction: str = "mean") -> None: method forward (line 604) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class MarginRankingLoss (line 608) | class MarginRankingLoss(_Loss): method __init__ (line 663) | def __init__(self, margin: float = 0.0, reduction: str = "mean") -> None: method forward (line 667) | def forward(self, input1: Tensor, input2: Tensor, target: Tensor) -> T... class CTCLoss (line 673) | class CTCLoss(_Loss): method __init__ (line 760) | def __init__( method forward (line 767) | def forward( class BCEWithLogitsLoss (line 791) | class BCEWithLogitsLoss(_WeightedLoss): method __init__ (line 860) | def __init__( method forward (line 870) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class SmoothL1Loss (line 876) | class SmoothL1Loss(_Loss): method __init__ (line 974) | def __init__(self, reduction: str = "mean", beta: float = 1.0) -> None: method forward (line 978) | def forward(self, input: Tensor, target: Tensor) -> Tensor: class CombinedMarginLoss (line 982) | class CombinedMarginLoss(Module): method __init__ (line 1036) | def __init__(self, m1: float = 1.0, m2: float = 0.0, m3: float = 0.0) ... method forward (line 1042) | def forward(self, x: Tensor, label: Tensor) -> Tensor: class TripletMarginLoss (line 1048) | class TripletMarginLoss(Module): method __init__ (line 1105) | def __init__( method forward (line 1122) | def forward(self, anchor, positive, negative): FILE: python/oneflow/nn/modules/masked_select.py function masked_select_op (line 19) | def masked_select_op(input, mask): FILE: python/oneflow/nn/modules/math_ops.py function asin_op (line 29) | def asin_op(input): function arcsin_op (line 62) | def arcsin_op(input): function asinh_op (line 70) | def asinh_op(input): function arcsinh_op (line 105) | def arcsinh_op(input): function asinh_op_tensor (line 113) | def asinh_op_tensor(input): function inplace_sin_op_tensor (line 121) | def inplace_sin_op_tensor(input): function atan_op (line 129) | def atan_op(input): function arctan_op (line 154) | def arctan_op(input): function fmod_op (line 162) | def fmod_op(input, other): function addmm (line 195) | def addmm(x, mat1, mat2, alpha=1, beta=1): function addmm_op (line 202) | def addmm_op(input, mat1, mat2, alpha=1, beta=1): FILE: python/oneflow/nn/modules/meshgrid.py function meshgrid_op (line 19) | def meshgrid_op(*tensors, indexing="ij"): FILE: python/oneflow/nn/modules/min_max_observer.py class MinMaxObserver (line 21) | class MinMaxObserver(Module): method __init__ (line 94) | def __init__( method forward (line 107) | def forward(self, input): FILE: python/oneflow/nn/modules/module.py class _WrappedHook (line 43) | class _WrappedHook(object): method __init__ (line 44) | def __init__(self, hook: Callable, module: Optional["Module"] = None): method __call__ (line 54) | def __call__(self, *args: Any, **kwargs: Any) -> Any: method __getstate__ (line 62) | def __getstate__(self) -> Dict: method __setstate__ (line 69) | def __setstate__(self, state: Dict): class _IncompatibleKeys (line 81) | class _IncompatibleKeys( method __repr__ (line 84) | def __repr__(self): function _addindent (line 92) | def _addindent(s_, numSpaces): class Module (line 106) | class Module(object): method __init__ (line 143) | def __init__(self): method __getstate__ (line 165) | def __getstate__(self): method __setstate__ (line 186) | def __setstate__(self, state): method forward (line 199) | def forward(self, *args, **kwargs): method __call__ (line 202) | def __call__(self, *args, **kwargs): method add_module (line 252) | def add_module(self, name: str, module: Optional["Module"]) -> None: method register_buffer (line 277) | def register_buffer( method register_parameter (line 332) | def register_parameter(self, name: str, param: Optional[Parameter]) ->... method _register_state_dict_hook (line 373) | def _register_state_dict_hook(self, hook): method _register_load_state_dict_pre_hook (line 387) | def _register_load_state_dict_pre_hook( method register_state_dict_pre_hook (line 410) | def register_state_dict_pre_hook(self, hook): method __getattr__ (line 420) | def __getattr__(self, name: str) -> Union[Tensor, "Module"]: method __setattr__ (line 437) | def __setattr__(self, name: str, value: Union[Tensor, "Module"]) -> None: method __delattr__ (line 502) | def __delattr__(self, name): method _named_members (line 513) | def _named_members(self, get_members_fn, prefix="", recurse=True): method parameters (line 525) | def parameters(self, recurse: bool = True) -> Iterator[Parameter]: method named_parameters (line 551) | def named_parameters( method buffers (line 582) | def buffers(self, recurse: bool = True) -> Iterator[Tensor]: method named_buffers (line 606) | def named_buffers( method children (line 637) | def children(self) -> Iterator["Module"]: method named_children (line 661) | def named_children(self) -> Iterator[Tuple[str, "Module"]]: method modules (line 684) | def modules(self) -> Iterator["Module"]: method named_modules (line 714) | def named_modules(self, memo: Optional[Set["Module"]] = None, prefix: ... method train (line 758) | def train(self: T, mode: bool = True) -> T: method eval (line 781) | def eval(self: T) -> T: method requires_grad_ (line 799) | def requires_grad_(self: T, requires_grad: bool = True) -> T: method zero_grad (line 822) | def zero_grad(self, set_to_none: bool = False) -> None: method _save_to_state_dict (line 852) | def _save_to_state_dict(self, destination, prefix, keep_vars): method _load_from_state_dict (line 860) | def _load_from_state_dict( method load_state_dict (line 946) | def load_state_dict( method state_dict (line 1024) | def state_dict( method register_backward_hook (line 1082) | def register_backward_hook( method register_full_backward_hook (line 1108) | def register_full_backward_hook( method _get_backward_hooks (line 1152) | def _get_backward_hooks(self): method _maybe_warn_non_full_backward_hook (line 1167) | def _maybe_warn_non_full_backward_hook(self, args, res, grad_fn): method register_forward_pre_hook (line 1226) | def register_forward_pre_hook(self, hook: Callable[..., None]): method register_forward_hook (line 1248) | def register_forward_hook(self, hook: Callable[..., None]): method _apply (line 1270) | def _apply(self, fn): method apply (line 1356) | def apply(self: T, fn: Callable[["Module"], None]) -> T: method to_empty (line 1408) | def to_empty(self: T, *, device: Union[str, flow.device]) -> T: method _to_memory_format (line 1420) | def _to_memory_format(self, memory_format): method to_memory_format (line 1442) | def to_memory_format(self, memory_format) -> None: method to (line 1446) | def to( method to (line 1454) | def to(self: T, dtype: flow.dtype) -> T: method to (line 1458) | def to(self: T, tensor: Tensor) -> T: method to (line 1461) | def to(self, *args, **kwargs): method to_consistent (line 1581) | def to_consistent(self, *args, **kwargs): method to_global (line 1586) | def to_global(self, placement=None, sbp=None): method to_local (line 1592) | def to_local(self): method cpu (line 1598) | def cpu(self: T) -> T: method cuda (line 1612) | def cuda(self: T, device: Optional[Union[int, flow.device]] = None) -> T: method float (line 1634) | def float(self: T) -> T: method double (line 1648) | def double(self: T) -> T: method half (line 1662) | def half(self: T) -> T: method _get_name (line 1676) | def _get_name(self): method get_submodule (line 1679) | def get_submodule(self, target: str): method get_parameter (line 1741) | def get_parameter(self, target: str): method extra_repr (line 1795) | def extra_repr(self) -> str: method make_contiguous_params_group (line 1804) | def make_contiguous_params_group(self): method __repr__ (line 1825) | def __repr__(self): method _shallow_repr (line 1845) | def _shallow_repr(self): FILE: python/oneflow/nn/modules/moving_average_min_max_observer.py class MovingAverageMinMaxObserver (line 22) | class MovingAverageMinMaxObserver(Module): method __init__ (line 119) | def __init__( method reset_running_stats (line 137) | def reset_running_stats(self) -> None: method forward (line 141) | def forward(self, input, current_train_step): FILE: python/oneflow/nn/modules/nms.py function nms_op (line 21) | def nms_op(boxes, scores, iou_threshold: float): FILE: python/oneflow/nn/modules/nonzero.py function nonzero_op (line 25) | def nonzero_op(input, as_tuple=False): FILE: python/oneflow/nn/modules/norm.py function norm (line 19) | def norm(input, p="fro", dim=None, keepdim=False, dtype=None): FILE: python/oneflow/nn/modules/normalization.py function group_norm (line 27) | def group_norm( class GroupNorm (line 64) | class GroupNorm(Module): method __init__ (line 118) | def __init__( method reset_parameters (line 151) | def reset_parameters(self) -> None: method forward (line 156) | def forward(self, input: Tensor) -> Tensor: method extra_repr (line 161) | def extra_repr(self) -> str: function layer_norm (line 167) | def layer_norm(input, normalized_shape, weight=None, bias=None, eps=1e-05): class LayerNorm (line 230) | class LayerNorm(Module): method __init__ (line 317) | def __init__( method reset_parameters (line 337) | def reset_parameters(self) -> None: method forward (line 342) | def forward(self, x): method extra_repr (line 345) | def extra_repr(self) -> str: class RMSLayerNorm (line 351) | class RMSLayerNorm(Module): method __init__ (line 385) | def __init__(self, hidden_size, eps=1e-6): method forward (line 394) | def forward(self, hidden_states): class RMSNorm (line 398) | class RMSNorm(Module): method __init__ (line 485) | def __init__( method forward (line 507) | def forward(self, x): method extra_repr (line 510) | def extra_repr(self) -> str: FILE: python/oneflow/nn/modules/numel.py function numel_op (line 19) | def numel_op(input): FILE: python/oneflow/nn/modules/padding.py class ReplicationPad1d (line 24) | class ReplicationPad1d(Module): method __init__ (line 67) | def __init__(self, padding: _size_4_t): method forward (line 78) | def forward(self, x): method extra_repr (line 81) | def extra_repr(self) -> str: class ReplicationPad2d (line 85) | class ReplicationPad2d(Module): method __init__ (line 133) | def __init__(self, padding: _size_4_t): method forward (line 144) | def forward(self, x): method extra_repr (line 147) | def extra_repr(self) -> str: class ReflectionPad1d (line 151) | class ReflectionPad1d(Module): method __init__ (line 193) | def __init__(self, padding: _size_2_t) -> None: method forward (line 204) | def forward(self, x): method extra_repr (line 207) | def extra_repr(self) -> str: class ReflectionPad2d (line 211) | class ReflectionPad2d(Module): method __init__ (line 259) | def __init__(self, padding: _size_4_t) -> None: method forward (line 270) | def forward(self, x): method extra_repr (line 273) | def extra_repr(self) -> str: class ConstantPad1d (line 277) | class ConstantPad1d(Module): method __init__ (line 320) | def __init__(self, padding: Union[int, tuple, list], value: Union[int,... method forward (line 331) | def forward(self, x): class ConstantPad2d (line 335) | class ConstantPad2d(Module): method __init__ (line 388) | def __init__(self, padding: Union[int, tuple, list], value: Union[int,... method forward (line 399) | def forward(self, x): class ConstantPad3d (line 403) | class ConstantPad3d(Module): method __init__ (line 462) | def __init__(self, padding: Union[int, tuple, list], value: Union[int,... method forward (line 473) | def forward(self, x): class ZeroPad2d (line 477) | class ZeroPad2d(Module): method __init__ (line 541) | def __init__(self, padding: Union[int, tuple, list]): method forward (line 552) | def forward(self, x): FILE: python/oneflow/nn/modules/pixelshuffle.py class PixelShufflev2 (line 23) | class PixelShufflev2(Module): method __init__ (line 88) | def __init__( method forward (line 111) | def forward(self, input: Tensor) -> Tensor: method extra_repr (line 116) | def extra_repr(self) -> str: FILE: python/oneflow/nn/modules/pooling.py class MaxPool1d (line 31) | class MaxPool1d(Module): method __init__ (line 84) | def __init__( method forward (line 103) | def forward(self, x): method extra_repr (line 119) | def extra_repr(self) -> str: function get_dhw_offset (line 125) | def get_dhw_offset(channel_pos): function get_ndim_pads_list (line 132) | def get_ndim_pads_list(padding, dhw_offset, ndims): function calc_pool_padding (line 150) | def calc_pool_padding(padding, dhw_offset, ndims): class MaxPool2d (line 164) | class MaxPool2d(Module): method __init__ (line 228) | def __init__( method to_memory_format (line 249) | def to_memory_format(self, memory_format) -> None: method forward (line 255) | def forward(self, x): method extra_repr (line 279) | def extra_repr(self) -> str: class MaxPool3d (line 285) | class MaxPool3d(Module): method __init__ (line 356) | def __init__( method forward (line 377) | def forward(self, x): method extra_repr (line 394) | def extra_repr(self) -> str: class AvgPool1d (line 400) | class AvgPool1d(Module): method __init__ (line 439) | def __init__( method forward (line 458) | def forward(self, x): method extra_repr (line 470) | def extra_repr(self) -> str: class AvgPool2d (line 477) | class AvgPool2d(Module): method __init__ (line 510) | def __init__( method to_memory_format (line 530) | def to_memory_format(self, memory_format) -> None: method forward (line 536) | def forward(self, x): method extra_repr (line 548) | def extra_repr(self) -> str: class AvgPool3d (line 555) | class AvgPool3d(Module): method __init__ (line 608) | def __init__( method forward (line 629) | def forward(self, x): method extra_repr (line 641) | def extra_repr(self) -> str: class AdaptiveAvgPool1d (line 648) | class AdaptiveAvgPool1d(Module): method __init__ (line 673) | def __init__(self, output_size: _size_1_t) -> None: method forward (line 678) | def forward(self, x): function adaptive_avg_pool1d (line 688) | def adaptive_avg_pool1d(input, output_size): class AdaptiveAvgPool2d (line 700) | class AdaptiveAvgPool2d(Module): method __init__ (line 740) | def __init__(self, output_size, data_format=None) -> None: method to_memory_format (line 755) | def to_memory_format(self, memory_format) -> None: method forward (line 761) | def forward(self, x): function adaptive_avg_pool2d (line 771) | def adaptive_avg_pool2d(input, output_size, data_format=None): class AdaptiveAvgPool3d (line 783) | class AdaptiveAvgPool3d(Module): method __init__ (line 823) | def __init__(self, output_size) -> None: method forward (line 828) | def forward(self, x): function adaptive_avg_pool3d (line 836) | def adaptive_avg_pool3d(input, output_size): class _AdaptiveMaxPoolNd (line 848) | class _AdaptiveMaxPoolNd(Module): method __init__ (line 849) | def __init__(self, output_size, return_indices: bool = False) -> None: method extra_repr (line 854) | def extra_repr(self) -> str: class AdaptiveMaxPool1d (line 858) | class AdaptiveMaxPool1d(_AdaptiveMaxPoolNd): method forward (line 890) | def forward(self, input): class AdaptiveMaxPool2d (line 901) | class AdaptiveMaxPool2d(_AdaptiveMaxPoolNd): method __init__ (line 943) | def __init__(self, output_size, return_indices=False, data_format=None... method to_memory_format (line 956) | def to_memory_format(self, memory_format) -> None: method forward (line 962) | def forward(self, input): class AdaptiveMaxPool3d (line 973) | class AdaptiveMaxPool3d(_AdaptiveMaxPoolNd): method forward (line 1017) | def forward(self, input): class MaxUnpool1d (line 1028) | class MaxUnpool1d(Module): method __init__ (line 1091) | def __init__( method forward (line 1102) | def forward(self, x, indices, output_size=None): class MaxUnpool2d (line 1108) | class MaxUnpool2d(Module): method __init__ (line 1179) | def __init__( method forward (line 1190) | def forward(self, x, indices, output_size=None): class MaxUnpool3d (line 1196) | class MaxUnpool3d(Module): method __init__ (line 1257) | def __init__( method forward (line 1268) | def forward(self, x, indices, output_size=None): FILE: python/oneflow/nn/modules/quantization.py class Quantization (line 20) | class Quantization(Module): method __init__ (line 90) | def __init__( method forward (line 101) | def forward(self, input, scale, zero_point): FILE: python/oneflow/nn/modules/reshape.py function _input_args_is_int (line 23) | def _input_args_is_int(args): function _input_args_is_flow_size (line 27) | def _input_args_is_flow_size(args): function reshape_op (line 31) | def reshape_op(input, shape: Sequence[int] = None): function view_op (line 61) | def view_op(input, *shape): FILE: python/oneflow/nn/modules/rnn.py function apply_permutation (line 29) | def apply_permutation(tensor: Tensor, permutation: Tensor, dim: int = 1)... class RNNBase (line 33) | class RNNBase(nn.Module): method __init__ (line 34) | def __init__( method __setattr__ (line 147) | def __setattr__(self, attr, value): method to_global (line 154) | def to_global(self, placement=None, sbp=None): method reset_parameters (line 165) | def reset_parameters(self) -> None: method check_input (line 170) | def check_input(self, input: Tensor, batch_sizes: Optional[Tensor]) ->... method get_expected_hidden_size (line 185) | def get_expected_hidden_size( method check_hidden_size (line 207) | def check_hidden_size( method check_forward_args (line 216) | def check_forward_args( method permute_hidden (line 224) | def permute_hidden(self, hx: Tensor, permutation: Optional[Tensor]): method extra_repr (line 229) | def extra_repr(self) -> str: method all_weights (line 246) | def all_weights(self) -> List[List[nn.Parameter]]: class RNN (line 253) | class RNN(RNNBase): method __init__ (line 351) | def __init__(self, *args, **kwargs): method forward (line 365) | def forward(self, input, hx=None): # noqa: F811 class LSTM (line 491) | class LSTM(RNNBase): method __init__ (line 623) | def __init__(self, *args, **kwargs): method get_expected_cell_size (line 626) | def get_expected_cell_size( method check_forward_args (line 641) | def check_forward_args( method permute_hidden (line 659) | def permute_hidden( method forward (line 669) | def forward(self, input, hx=None): class GRU (line 792) | class GRU(RNNBase): method __init__ (line 898) | def __init__(self, *args, **kwargs): method forward (line 905) | def forward(self, input, hx=None): class RNNCellBase (line 1003) | class RNNCellBase(nn.Module): method __init__ (line 1004) | def __init__( method extra_repr (line 1037) | def extra_repr(self) -> str: method reset_parameters (line 1045) | def reset_parameters(self) -> None: class RNNCell (line 1051) | class RNNCell(RNNCellBase): method __init__ (line 1114) | def __init__( method forward (line 1129) | def forward(self, input: Tensor, hx: Optional[Tensor] = None) -> Tensor: class LSTMCell (line 1174) | class LSTMCell(RNNCellBase): method __init__ (line 1239) | def __init__( method forward (line 1252) | def forward( class GRUCell (line 1292) | class GRUCell(RNNCellBase): method __init__ (line 1361) | def __init__( method forward (line 1372) | def forward(self, input: Tensor, hx: Optional[Tensor] = None) -> Tensor: FILE: python/oneflow/nn/modules/roll.py function roll_op (line 20) | def roll_op(input, shifts, dims=None): FILE: python/oneflow/nn/modules/scatter.py function scatter (line 24) | def scatter(input, dim, index, src, *, reduce=None): function scatter_add (line 95) | def scatter_add(input, dim, index, src): function scatter_nd (line 139) | def scatter_nd(index, update, shape): function tensor_scatter_nd_update (line 164) | def tensor_scatter_nd_update(tensor, indices, updates): FILE: python/oneflow/nn/modules/slice.py function slice_op (line 22) | def slice_op(input, slice_tup_list: Sequence[Tuple[int, int, int]]): function slice_update_op (line 47) | def slice_update_op(input, update, slice_tup_list: Sequence[Tuple[int, i... FILE: python/oneflow/nn/modules/sparse.py class Embedding (line 24) | class Embedding(Module): method __init__ (line 109) | def __init__( method reset_parameters (line 154) | def reset_parameters(self) -> None: method _fill_padding_idx_with_zero (line 160) | def _fill_padding_idx_with_zero(self) -> None: method extra_repr (line 165) | def extra_repr(self) -> str: method forward (line 179) | def forward(self, indices): function embedding (line 193) | def embedding( FILE: python/oneflow/nn/modules/sparse_softmax_cross_entropy.py function sparse_softmax_cross_entropy (line 19) | def sparse_softmax_cross_entropy(labels, logits): FILE: python/oneflow/nn/modules/tensor_buffer.py function tensor_buffer_to_tensor_op (line 21) | def tensor_buffer_to_tensor_op(x, dtype: flow.dtype, instance_shape: Seq... function tensor_to_tensor_buffer (line 56) | def tensor_to_tensor_buffer(x, instance_dims: int): function gen_tensor_buffer (line 86) | def gen_tensor_buffer( FILE: python/oneflow/nn/modules/tensordot.py function tensordot (line 21) | def tensordot( FILE: python/oneflow/nn/modules/trigonometric_ops.py function sign_op (line 21) | def sign_op(input): function sinh_op (line 50) | def sinh_op(input): function tan_op (line 81) | def tan_op(input): function acosh_op (line 106) | def acosh_op(input): function arccosh_op (line 135) | def arccosh_op(input): FILE: python/oneflow/nn/modules/unique.py function unique_op (line 19) | def unique_op( FILE: python/oneflow/nn/modules/upsampling.py class Upsample (line 22) | class Upsample(Module): method __init__ (line 98) | def __init__( method forward (line 111) | def forward(self, x): method extra_repr (line 120) | def extra_repr(self) -> str: class UpsamplingNearest2d (line 129) | class UpsamplingNearest2d(Upsample): method __init__ (line 174) | def __init__( class UpsamplingBilinear2d (line 182) | class UpsamplingBilinear2d(Upsample): method __init__ (line 229) | def __init__( FILE: python/oneflow/nn/modules/utils.py function _ntuple (line 24) | def _ntuple(n): function _getint (line 33) | def _getint(): function _handle_size_arg (line 49) | def _handle_size_arg(size): function _reverse_repeat_tuple (line 61) | def _reverse_repeat_tuple(t, n): function _list_with_default (line 69) | def _list_with_default(out_size, defaults): function _check_axis (line 82) | def _check_axis(axis, shape): function _generate_output_size (line 101) | def _generate_output_size(input_size, output_size): FILE: python/oneflow/nn/modules/where.py function where_op (line 20) | def where_op(condition, x=None, y=None): FILE: python/oneflow/nn/optimizer/adadelta.py class Adadelta (line 25) | class Adadelta(Optimizer): method __init__ (line 97) | def __init__( method step (line 144) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 185) | def _generate_conf_for_graph(self, train_conf, vars_conf): FILE: python/oneflow/nn/optimizer/adagrad.py class Adagrad (line 25) | class Adagrad(Optimizer): method __init__ (line 96) | def __init__( method step (line 143) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 181) | def _generate_conf_for_graph(self, train_conf, vars_conf): FILE: python/oneflow/nn/optimizer/adam.py class Adam (line 25) | class Adam(Optimizer): method __init__ (line 116) | def __init__( method _single_tensor_update (line 187) | def _single_tensor_update(self, param_group): method _fused_update (line 233) | def _fused_update(self, param_group): method step (line 273) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 304) | def _generate_conf_for_graph(self, train_conf, vars_conf): method support_sparse (line 346) | def support_sparse(self): FILE: python/oneflow/nn/optimizer/adamw.py class AdamW (line 25) | class AdamW(Optimizer): method __init__ (line 118) | def __init__( method _single_tensor_update (line 188) | def _single_tensor_update(self, param_group): method _fused_update (line 234) | def _fused_update(self, param_group): method step (line 274) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 304) | def _generate_conf_for_graph(self, train_conf, vars_conf): method support_sparse (line 345) | def support_sparse(self): FILE: python/oneflow/nn/optimizer/chained_scheduler.py class ChainedScheduler (line 19) | class ChainedScheduler(LRScheduler): method __init__ (line 43) | def __init__(self, schedulers): method step (line 64) | def step(self): method state_dict (line 75) | def state_dict(self): method load_state_dict (line 93) | def load_state_dict(self, state_dict): method _generate_conf_for_graph (line 108) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/constant_lr.py class ConstantLR (line 20) | class ConstantLR(LRScheduler): method __init__ (line 47) | def __init__( method get_lr (line 66) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 72) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/cosine_annealing_lr.py class CosineAnnealingLR (line 22) | class CosineAnnealingLR(LRScheduler): method __init__ (line 65) | def __init__( method get_lr (line 77) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 81) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/cosine_annealing_warm_restarts.py class CosineAnnealingWarmRestarts (line 21) | class CosineAnnealingWarmRestarts(LRScheduler): method __init__ (line 52) | def __init__( method get_lr (line 78) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 101) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/cosine_decay_lr.py class CosineDecayLR (line 22) | class CosineDecayLR(LRScheduler): method __init__ (line 68) | def __init__( method get_lr (line 83) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 92) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/exponential_lr.py class ExponentialLR (line 20) | class ExponentialLR(LRScheduler): method __init__ (line 33) | def __init__( method get_lr (line 47) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 50) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/lamb.py class LAMB (line 24) | class LAMB(Optimizer): method __init__ (line 109) | def __init__( method step (line 167) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 228) | def _generate_conf_for_graph(self, train_conf, vars_conf): FILE: python/oneflow/nn/optimizer/lambda_lr.py class LambdaLR (line 21) | class LambdaLR(LRScheduler): method __init__ (line 53) | def __init__(self, optimizer, lr_lambda, last_step=-1, verbose=False): method state_dict (line 63) | def state_dict(self): method load_state_dict (line 82) | def load_state_dict(self, state_dict): method step (line 96) | def step(self): FILE: python/oneflow/nn/optimizer/lbfgs.py function _quadratic_interpolate (line 23) | def _quadratic_interpolate(x1, f1, g1, x2, f2, g2, bounds=None): function _strong_wolfe (line 37) | def _strong_wolfe( class LBFGS (line 137) | class LBFGS(Optimizer): method __init__ (line 188) | def __init__( method _gather_flat_grad (line 225) | def _gather_flat_grad(self): method _numel (line 235) | def _numel(self): method _update (line 243) | def _update(self, step_size, direction): method _try_direction (line 252) | def _try_direction(self, closure, x, t, d): method step (line 261) | def step(self, closure: Callable = None): FILE: python/oneflow/nn/optimizer/linear_lr.py class LinearLR (line 20) | class LinearLR(LRScheduler): method __init__ (line 51) | def __init__( method get_lr (line 77) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 87) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/lr_scheduler.py class LRScheduler (line 19) | class LRScheduler(object): method __init__ (line 20) | def __init__( method state_dict (line 32) | def state_dict(self): method load_state_dict (line 42) | def load_state_dict(self, state_dict): method get_lr (line 51) | def get_lr(self, base_lr, step): method get_last_lr (line 55) | def get_last_lr(self): method print_lr (line 59) | def print_lr(self, group, lr): method step (line 66) | def step(self): method update_lrs (line 71) | def update_lrs(self, lrs): method _init_base_lrs (line 79) | def _init_base_lrs(self): FILE: python/oneflow/nn/optimizer/multiplicative_lr.py class MultiplicativeLR (line 22) | class MultiplicativeLR(LRScheduler): method __init__ (line 51) | def __init__(self, optimizer, lr_lambda, last_step=-1, verbose=False): method state_dict (line 66) | def state_dict(self): method load_state_dict (line 87) | def load_state_dict(self, state_dict): method step (line 102) | def step(self): FILE: python/oneflow/nn/optimizer/multistep_lr.py class MultiStepLR (line 22) | class MultiStepLR(LRScheduler): method __init__ (line 49) | def __init__( method get_lr (line 66) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 71) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/polynomial_lr.py class PolynomialLR (line 22) | class PolynomialLR(LRScheduler): method __init__ (line 66) | def __init__( method get_lr (line 85) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 98) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/reduce_lr_on_plateau.py class ReduceLROnPlateau (line 20) | class ReduceLROnPlateau(object): method __init__ (line 72) | def __init__( method step (line 123) | def step(self, metrics): method in_cooldown (line 151) | def in_cooldown(self): method is_better (line 157) | def is_better(self, a, best): method state_dict (line 175) | def state_dict(self): method load_state_dict (line 185) | def load_state_dict(self, state_dict): method _reduce_lr (line 197) | def _reduce_lr(self, epoch): method _reset (line 209) | def _reset(self): method _init_is_better (line 215) | def _init_is_better(self, mode, threshold, threshold_mode): FILE: python/oneflow/nn/optimizer/rmsprop.py class RMSprop (line 24) | class RMSprop(Optimizer): method __init__ (line 132) | def __init__( method step (line 183) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 234) | def _generate_conf_for_graph(self, train_conf, vars_conf): FILE: python/oneflow/nn/optimizer/sequential_lr.py class SequentialLR (line 22) | class SequentialLR(LRScheduler): method __init__ (line 54) | def __init__( method step (line 102) | def step(self): method state_dict (line 114) | def state_dict(self): method load_state_dict (line 127) | def load_state_dict(self, state_dict): method _generate_conf_for_graph (line 136) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/sgd.py class SGD (line 25) | class SGD(Optimizer): method __init__ (line 103) | def __init__( method _single_tensor_update (line 158) | def _single_tensor_update(self, param_group): method _fused_update (line 194) | def _fused_update(self, param_group): method step (line 234) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 255) | def _generate_conf_for_graph(self, train_conf, vars_conf): method support_sparse (line 297) | def support_sparse(self): FILE: python/oneflow/nn/optimizer/step_lr.py class StepLR (line 22) | class StepLR(LRScheduler): method __init__ (line 49) | def __init__( method get_lr (line 63) | def get_lr(self, base_lr, step): method _generate_conf_for_graph (line 68) | def _generate_conf_for_graph(self, lr_conf): FILE: python/oneflow/nn/optimizer/swa_utils.py class AveragedModel (line 33) | class AveragedModel(Module): method __init__ (line 135) | def __init__(self, model, device=None, avg_fn=None, use_buffers=False): method forward (line 153) | def forward(self, *args, **kwargs): method update_parameters (line 156) | def update_parameters(self, model): function update_bn (line 184) | def update_bn(loader, model, device=None): class SWALR (line 247) | class SWALR(LRScheduler): method __init__ (line 298) | def __init__( method _format_param (line 322) | def _format_param(optimizer, swa_lrs): method _linear_anneal (line 335) | def _linear_anneal(t): method _cosine_anneal (line 339) | def _cosine_anneal(t): method _get_initial_lr (line 343) | def _get_initial_lr(lr, swa_lr, alpha): method get_lr (line 348) | def get_lr(self, base_lr, step): FILE: python/oneflow/nn/optimizer/warmup_lr.py class WarmupLR (line 25) | class WarmupLR(SequentialLR): method __init__ (line 98) | def __init__( method _init_warmup_scheduler (line 143) | def _init_warmup_scheduler(self, scheduler): method _init_seq_scheduler (line 181) | def _init_seq_scheduler(self, scheduler, warmup): FILE: python/oneflow/nn/parallel/distributed.py function allreduce_fn (line 26) | def allreduce_fn(module, param, use_bucket): function DistributedDataParallel (line 70) | def DistributedDataParallel( FILE: python/oneflow/nn/qat/conv.py function get_conv_fake_quantized (line 22) | def get_conv_fake_quantized( function init_conv_fake_quants (line 32) | def init_conv_fake_quants( class QatConv1d (line 61) | class QatConv1d(nn.Conv1d): method __init__ (line 123) | def __init__( method forward (line 161) | def forward(self, x): class QatConv2d (line 173) | class QatConv2d(nn.Conv2d): method __init__ (line 241) | def __init__( method forward (line 279) | def forward(self, x): class QatConv3d (line 291) | class QatConv3d(nn.Conv3d): method __init__ (line 359) | def __init__( method forward (line 397) | def forward(self, x): FILE: python/oneflow/nn/utils/clip_grad.py function clip_grad_norm_ (line 31) | def clip_grad_norm_( function clip_grad_value_ (line 185) | def clip_grad_value_(parameters: _tensor_or_tensors, clip_value: float) ... FILE: python/oneflow/nn/utils/container.py function get_seq (line 40) | def get_seq(T): function get_list (line 105) | def get_list(T): function get_dict (line 196) | def get_dict(T): function get_para_list (line 282) | def get_para_list(T): function get_para_dict (line 388) | def get_para_dict(T): FILE: python/oneflow/nn/utils/convert_parameters.py function parameters_to_vector (line 21) | def parameters_to_vector(parameters: Iterable[Tensor]) -> Tensor: function vector_to_parameters (line 46) | def vector_to_parameters(vec: Tensor, parameters: Iterable[Tensor]) -> N... function _check_param_device (line 79) | def _check_param_device(param: Tensor, old_param_device: Optional[int]) ... FILE: python/oneflow/nn/utils/parameters_grouping.py function numel_in_bucket (line 27) | def numel_in_bucket(tensor: Tensor): class ContiguousParamsGroup (line 42) | class ContiguousParamsGroup(object): method __init__ (line 60) | def __init__( method _make_valid_params_group_list (line 81) | def _make_valid_params_group_list(self): method _remove_no_grad_tensors (line 99) | def _remove_no_grad_tensors(self): method _check_tensor_position_consistency (line 104) | def _check_tensor_position_consistency(self): method _check_current_buffer (line 118) | def _check_current_buffer(self): method _make_buffer_params_mapping (line 132) | def _make_buffer_params_mapping(self): method _parameters_grouping_on_new_buffer (line 148) | def _parameters_grouping_on_new_buffer(self): method _parameters_grouping_on_current_buffer (line 215) | def _parameters_grouping_on_current_buffer(self): method grouped_parameters (line 292) | def grouped_parameters(self): method grouped_parameters_grad (line 298) | def grouped_parameters_grad(self): FILE: python/oneflow/nn/utils/prune.py class BasePruningMethod (line 29) | class BasePruningMethod(ABC): method __init__ (line 36) | def __init__(self): method __call__ (line 39) | def __call__(self, module, inputs): method compute_mask (line 51) | def compute_mask(self, t, default_mask): method apply_mask (line 68) | def apply_mask(self, module): method apply (line 89) | def apply(cls, module, name, *args, importance_scores=None, **kwargs): method prune (line 215) | def prune(self, t, default_mask=None, importance_scores=None): method remove (line 242) | def remove(self, module): class PruningContainer (line 271) | class PruningContainer(BasePruningMethod): method __init__ (line 279) | def __init__(self, *args): method add_pruning_method (line 291) | def add_pruning_method(self, method): method __len__ (line 313) | def __len__(self): method __iter__ (line 316) | def __iter__(self): method __getitem__ (line 319) | def __getitem__(self, idx): method compute_mask (line 322) | def compute_mask(self, t, default_mask): class Identity (line 415) | class Identity(BasePruningMethod): method compute_mask (line 422) | def compute_mask(self, t, default_mask): method apply (line 427) | def apply(cls, module, name): class RandomUnstructured (line 439) | class RandomUnstructured(BasePruningMethod): method __init__ (line 452) | def __init__(self, amount): method compute_mask (line 457) | def compute_mask(self, t, default_mask): method apply (line 479) | def apply(cls, module, name, amount): class L1Unstructured (line 495) | class L1Unstructured(BasePruningMethod): method __init__ (line 507) | def __init__(self, amount): method compute_mask (line 512) | def compute_mask(self, t, default_mask): method apply (line 536) | def apply(cls, module, name, amount, importance_scores=None): class RandomStructured (line 559) | class RandomStructured(BasePruningMethod): method __init__ (line 572) | def __init__(self, amount, dim=-1): method compute_mask (line 578) | def compute_mask(self, t, default_mask): method apply (line 647) | def apply(cls, module, name, amount, dim=-1): class LnStructured (line 665) | class LnStructured(BasePruningMethod): method __init__ (line 681) | def __init__(self, amount, n, dim=-1): method compute_mask (line 688) | def compute_mask(self, t, default_mask): method apply (line 757) | def apply(cls, module, name, amount, n, dim, importance_scores=None): class CustomFromMask (line 789) | class CustomFromMask(BasePruningMethod): method __init__ (line 793) | def __init__(self, mask): method compute_mask (line 796) | def compute_mask(self, t, default_mask): method apply (line 802) | def apply(cls, module, name, mask): function identity (line 814) | def identity(module, name): function random_unstructured (line 842) | def random_unstructured(module, name, amount): function l1_unstructured (line 872) | def l1_unstructured(module, name, amount, importance_scores=None): function random_structured (line 910) | def random_structured(module, name, amount, dim): function ln_structured (line 945) | def ln_structured(module, name, amount, n, dim, importance_scores=None): function global_unstructured (line 986) | def global_unstructured(parameters, pruning_method, importance_scores=No... function custom_from_mask (line 1101) | def custom_from_mask(module, name, mask): function remove (line 1130) | def remove(module, name): function is_pruned (line 1158) | def is_pruned(module): function _validate_pruning_amount_init (line 1182) | def _validate_pruning_amount_init(amount): function _validate_pruning_amount (line 1213) | def _validate_pruning_amount(amount, tensor_size): function _validate_structured_pruning (line 1235) | def _validate_structured_pruning(t): function _compute_nparams_toprune (line 1252) | def _compute_nparams_toprune(amount, tensor_size): function _validate_pruning_dim (line 1274) | def _validate_pruning_dim(t, dim): function _compute_norm (line 1284) | def _compute_norm(t, n, dim): FILE: python/oneflow/nn/utils/rnn.py function bind (line 27) | def bind(optional, fn): function invert_permutation (line 33) | def invert_permutation(permutation: Optional[Tensor]) -> Optional[Tensor]: class PackedSequence (line 46) | class PackedSequence(object): method __init__ (line 86) | def __init__( method pin_memory (line 113) | def pin_memory(self): method cuda (line 121) | def cuda(self, *args, **kwargs): method cpu (line 129) | def cpu(self, *args, **kwargs): method double (line 138) | def double(self): method float (line 141) | def float(self): method half (line 144) | def half(self): method long (line 147) | def long(self): method int (line 150) | def int(self): method short (line 153) | def short(self): method char (line 156) | def char(self): method byte (line 159) | def byte(self): method to (line 162) | def to(self, *args, **kwargs): method is_cuda (line 196) | def is_cuda(self): method is_pinned (line 200) | def is_pinned(self): function pack_padded_sequence (line 205) | def pack_padded_sequence( function pad_packed_sequence (line 260) | def pad_packed_sequence( function pad_sequence (line 411) | def pad_sequence( function unpad_sequence (line 493) | def unpad_sequence( function pack_sequence (line 547) | def pack_sequence( function unpack_sequence (line 594) | def unpack_sequence(packed_sequences: PackedSequence) -> List[Tensor]: FILE: python/oneflow/nn/utils/skip_init.py function skip_init (line 20) | def skip_init(module_cls, *args, **kwargs): FILE: python/oneflow/nn/utils/weight_norm.py function _norm_except_dim_0 (line 24) | def _norm_except_dim_0(v: Tensor): function _norm_except_dim (line 30) | def _norm_except_dim(v: Tensor, dim: int): class WeightNorm (line 47) | class WeightNorm(object): method __init__ (line 51) | def __init__(self, name: str, dim: int) -> None: method compute_weight (line 57) | def compute_weight(self, module: Module) -> Any: method apply (line 63) | def apply(module, name: str, dim: int) -> "WeightNorm": method remove (line 91) | def remove(self, module: Module) -> None: method __call__ (line 98) | def __call__(self, module: Module, inputs: Any) -> None: function weight_norm (line 105) | def weight_norm(module: T_module, name: str = "weight", dim: int = 0) ->... function remove_weight_norm (line 154) | def remove_weight_norm(module: T_module, name: str = "weight") -> T_module: FILE: python/oneflow/one_embedding.py function _check_initializer (line 32) | def _check_initializer(initializer): function _check_cache (line 54) | def _check_cache(cache): function _init (line 69) | def _init( class Embedding (line 177) | class Embedding(Module): method __init__ (line 178) | def __init__( method _save_to_state_dict (line 224) | def _save_to_state_dict(self, destination, prefix, keep_vars): method _load_from_state_dict (line 244) | def _load_from_state_dict( method save_snapshot (line 266) | def save_snapshot(self, snapshot_name): method load_snapshot (line 284) | def load_snapshot(self, snapshot_name): method forward (line 301) | def forward(self, ids, table_ids=None): method shuffle_and_lookup (line 338) | def shuffle_and_lookup(self, state_initializer): method sgd_update (line 407) | def sgd_update(self, param_group, step): method adam_update (line 437) | def adam_update(self, param_group, step): method adagrad_update (line 483) | def adagrad_update(self, param_group, step): method ftrl_update (line 518) | def ftrl_update(self, param_group, step): function make_device_mem_store_options (line 560) | def make_device_mem_store_options( function make_cached_ssd_store_options (line 601) | def make_cached_ssd_store_options( function make_cached_host_mem_store_options (line 675) | def make_cached_host_mem_store_options( function make_uniform_initializer (line 727) | def make_uniform_initializer(low=0.0, high=1.0): function make_normal_initializer (line 749) | def make_normal_initializer(mean=0.0, std=1.0): function make_constant_initializer (line 771) | def make_constant_initializer(value): function make_trunc_normal_initializer (line 792) | def make_trunc_normal_initializer(mean=0.0, std=1.0, a=-2.0, b=2.0): function make_table_options (line 816) | def make_table_options(param): function make_column_options (line 847) | def make_column_options(initializer): function make_table (line 851) | def make_table(param): class MultiTableEmbedding (line 859) | class MultiTableEmbedding(Embedding): method __init__ (line 937) | def __init__( class MultiTableMultiColumnEmbedding (line 963) | class MultiTableMultiColumnEmbedding(Embedding): method __init__ (line 1045) | def __init__( class Ftrl (line 1076) | class Ftrl(Optimizer): method __init__ (line 1119) | def __init__( method step (line 1158) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 1196) | def _generate_conf_for_graph(self, train_conf, vars_conf): method support_sparse (line 1236) | def support_sparse(self): function make_persistent_table_reader (line 1240) | def make_persistent_table_reader( function make_persistent_table_writer (line 1264) | def make_persistent_table_writer( class SmartDecayAdam (line 1288) | class SmartDecayAdam(flow.nn.optimizer.adam.Adam): method _generate_conf_for_graph (line 1295) | def _generate_conf_for_graph(self, train_conf, vars_conf): class Optimizer (line 1301) | class Optimizer(Optimizer): method __init__ (line 1302) | def __init__( method step (line 1323) | def step(self, closure: Callable = None): method _generate_conf_for_graph (line 1339) | def _generate_conf_for_graph(self, train_conf, vars_conf): FILE: python/oneflow/onnx/__init__.py function symbolic_opset11 (line 19) | def symbolic_opset11(): function register_custom_op_symbolic (line 25) | def register_custom_op_symbolic(*args, **kwargs): FILE: python/oneflow/onnx/symbolic_helper.py function parse_args (line 20) | def parse_args(*args, **kwargs): FILE: python/oneflow/ops/__init__.py function load_library (line 18) | def load_library(path): FILE: python/oneflow/ops/array_ops.py function parse_slice_tuple_list (line 18) | def parse_slice_tuple_list(slice_tup_list, shape): FILE: python/oneflow/ops/stateful_ops.py class StatefulOp (line 21) | class StatefulOp(object): method __init__ (line 22) | def __init__(self, op_type_name, op_name=None): method op (line 30) | def op(self): method Input (line 40) | def Input(self, input_name, num=1): method Output (line 54) | def Output(self, output_name, num=1): method Build (line 68) | def Build(self): FILE: python/oneflow/ops/transpose_util.py function is_perm (line 19) | def is_perm(perm: Sequence[int]) -> bool: function get_perm_when_transpose_axis_to_last_dim (line 23) | def get_perm_when_transpose_axis_to_last_dim(num_axes: int, axis: int) -... function get_inversed_perm (line 31) | def get_inversed_perm(perm: Sequence[int]) -> tuple: FILE: python/oneflow/ops/util/initializer_util.py function get_random_distribution (line 28) | def get_random_distribution(distribution): function get_data_format (line 39) | def get_data_format(data_format): function calc_fan (line 52) | def calc_fan(shape, mode, data_format): function calc_gain (line 84) | def calc_gain(nonlinearity, param=None): FILE: python/oneflow/optim/optimizer.py class ParamGroup (line 31) | class ParamGroup(dict): method __init__ (line 32) | def __init__( method _make_options_valid (line 76) | def _make_options_valid(self): method parameters (line 90) | def parameters(self): method contiguous_parameters (line 94) | def contiguous_parameters(self): class _SourceOpOnlyResourceDependenceMode (line 100) | class _SourceOpOnlyResourceDependenceMode: method __init__ (line 101) | def __init__(self): method __enter__ (line 104) | def __enter__(self): method __exit__ (line 109) | def __exit__(self, *args, **kwargs): function _decorate_step (line 113) | def _decorate_step(step): class _RequiredParameter (line 121) | class _RequiredParameter(object): method __repr__ (line 124) | def __repr__(self): class Optimizer (line 131) | class Optimizer(object): method __init__ (line 132) | def __init__(self, parameters, options): method add_param_group (line 168) | def add_param_group(self, param_group) -> None: method load_state_dict (line 250) | def load_state_dict(self, state_dict) -> None: method state_dict (line 348) | def state_dict(self): method step (line 395) | def step(self, closure: Union[Callable, None] = None) -> Union[Tensor,... method clip_grad (line 406) | def clip_grad(self, error_if_nonfinite: bool = False): method zero_grad (line 436) | def zero_grad(self, set_to_none: bool = False): method _parse_input_parameters (line 464) | def _parse_input_parameters(self, parameters): method _generate_grad_clip_conf_for_optim_conf (line 492) | def _generate_grad_clip_conf_for_optim_conf(self, param_group, optimiz... method _generate_lr_scale_for_optim_conf (line 504) | def _generate_lr_scale_for_optim_conf(self, param_group, optimizer_conf): method support_sparse (line 512) | def support_sparse(self): method _check_variables_in_graph (line 518) | def _check_variables_in_graph(self, vars_conf): method _check_variables_optimizer_bound (line 531) | def _check_variables_optimizer_bound(self, vars_conf): method _generate_indexed_slices_optimizer_conf (line 544) | def _generate_indexed_slices_optimizer_conf(self, job_conf, vars_conf): FILE: python/oneflow/profiler/__init__.py function range_push (line 40) | def range_push(range_name): function range_pop (line 44) | def range_pop(): function profiler_start (line 48) | def profiler_start(): function profiler_stop (line 52) | def profiler_stop(): function kineto_available (line 56) | def kineto_available(): FILE: python/oneflow/profiler/events.py class EventType (line 27) | class EventType(Enum): class CustomEventType (line 32) | class CustomEventType(Enum): class EventBase (line 38) | class EventBase: method __init__ (line 41) | def __init__(self, name: str, time_total: float, event_type: EventType... method update (line 47) | def update(self, event) -> None: method name (line 53) | def name(self): method cpu_time_total (line 59) | def cpu_time_total(self): method cpu_time_total (line 63) | def cpu_time_total(self, new_time): method cpu_time (line 67) | def cpu_time(self): method cuda_time_total (line 71) | def cuda_time_total(self): method cuda_time_total (line 75) | def cuda_time_total(self, new_time): method cuda_time (line 79) | def cuda_time(self): method has_cuda_time (line 84) | def has_cuda_time(self) -> bool: method __eq__ (line 87) | def __eq__(self, __o: object) -> bool: class CustomEvent (line 96) | class CustomEvent(EventBase): method __init__ (line 97) | def __init__( method from_dict (line 104) | def from_dict(cls, d: dict): method key (line 108) | def key(self): method cuda_time_total (line 112) | def cuda_time_total(self): method to_dict (line 117) | def to_dict(self): method __eq__ (line 128) | def __eq__(self, __o: object) -> bool: class KernelEvent (line 136) | class KernelEvent(EventBase): method __init__ (line 137) | def __init__( method add_child (line 152) | def add_child(self, event: CustomEvent): method from_dict (line 158) | def from_dict(cls, d: dict): method key (line 170) | def key(self): method cuda_time_total (line 188) | def cuda_time_total(self): method cuda_time_total (line 194) | def cuda_time_total(self, new_time): method input_shapes (line 198) | def input_shapes(self): method attributes (line 204) | def attributes(self): method bandwidth (line 210) | def bandwidth(self): method to_dict (line 216) | def to_dict(self): method update (line 235) | def update(self, event): method make_children_average (line 249) | def make_children_average(self): method __eq__ (line 259) | def __eq__(self, __o: object) -> bool: class Events (line 270) | class Events(list): method __init__ (line 271) | def __init__(self, events: str = "") -> None: method __init_events (line 276) | def __init_events(self, events: str): method __str__ (line 282) | def __str__(self): method key_averages (line 285) | def key_averages(self, group_by_input_shape=False, group_by_attributes... method table (line 311) | def table(self): FILE: python/oneflow/profiler/profiler.py class ProfilerActivity (line 22) | class ProfilerActivity(Enum): class ProfilerAction (line 27) | class ProfilerAction(Enum): function tensorboard_trace_handler (line 38) | def tensorboard_trace_handler(): function supported_activities (line 42) | def supported_activities() -> Set[ProfilerActivity]: class profile (line 49) | class profile: method __init__ (line 50) | def __init__( method __enter__ (line 74) | def __enter__(self): method __exit__ (line 84) | def __exit__(self, exc_type, exc_val, exc_tb): method __check_finish (line 89) | def __check_finish(self): method key_averages (line 93) | def key_averages(self, group_by_input_shape=False, group_by_attributes... method events (line 100) | def events(self): class record_function (line 105) | class record_function: method __init__ (line 106) | def __init__(self, name: str) -> None: method __enter__ (line 110) | def __enter__(self): method __exit__ (line 116) | def __exit__(self, exc_type, exc_val, exc_tb): FILE: python/oneflow/profiler/util.py function format_time (line 20) | def format_time(time_us): FILE: python/oneflow/remat/__init__.py function parse_size (line 21) | def parse_size(size): function set_budget (line 40) | def set_budget(budget: str): function get_budget (line 45) | def get_budget(): FILE: python/oneflow/special/special_ops.py function erf (line 24) | def erf(x: Tensor): function erfc (line 28) | def erfc(x: Tensor): function erfinv (line 32) | def erfinv(x: Tensor): function exp2 (line 36) | def exp2(x: Tensor): function expm1 (line 40) | def expm1(x: Tensor): function log1p (line 44) | def log1p(x: Tensor): function log_softmax (line 48) | def log_softmax(x: Tensor, dim: int): function logsumexp (line 52) | def logsumexp(x: Tensor, dim: int, keepdim=False): function round (line 56) | def round(x: Tensor): function softmax (line 60) | def softmax(x: Tensor, dim: int): function digamma (line 64) | def digamma(x: Tensor): function psi (line 68) | def psi(x: Tensor): function zeta (line 72) | def zeta(input, other): FILE: python/oneflow/support/async_util.py function Await (line 19) | def Await(counter, func): FILE: python/oneflow/support/box.py class Box (line 18) | class Box(object): method __init__ (line 19) | def __init__(self, *arg): method value (line 27) | def value(self): method value_setter (line 32) | def value_setter(self): method set_value (line 35) | def set_value(self, val): method has_value (line 39) | def has_value(self): FILE: python/oneflow/support/enable_if.py function condition (line 21) | def condition(hob_expr): function get_condition_hob (line 29) | def get_condition_hob(func): function set_condition_hob (line 34) | def set_condition_hob(func, hob): function unique (line 38) | def unique(arg_funcs, context=None, default=None): function GetMatchedFunction (line 65) | def GetMatchedFunction(default, conditional_functions, context=None): function MakeDefaultFunction (line 78) | def MakeDefaultFunction(default, conditional_functions, context=None): function _MultiMatchedErrorFunction (line 92) | def _MultiMatchedErrorFunction(default, matched_functions, context=None): FILE: python/oneflow/support/env_var_util.py function string_to_bool (line 19) | def string_to_bool(env_str): function parse_boolean_from_env (line 25) | def parse_boolean_from_env(env_var, defalut_value): FILE: python/oneflow/support/func_inspect_util.py function GetArgNameAndDefaultTuple (line 21) | def GetArgNameAndDefaultTuple(func): function GetArgNameAndDefaultTuple (line 34) | def GetArgNameAndDefaultTuple(func): function GetArgDefaults (line 48) | def GetArgDefaults(func): FILE: python/oneflow/support/high_order_bool.py function bool_functor (line 20) | def bool_functor(verbose_debug_str): function hob_context_attr (line 27) | def hob_context_attr(attr_name): class BoolFunctor (line 34) | class BoolFunctor(object): method debug_str (line 35) | def debug_str(self, ctx, display_result=True): method verbose_debug_str (line 43) | def verbose_debug_str(self, ctx, display_result=True): method __call__ (line 46) | def __call__(self, ctx): method __and__ (line 49) | def __and__(self, rhs): method __or__ (line 52) | def __or__(self, rhs): method __invert__ (line 55) | def __invert__(self): class HighOrderBool (line 59) | class HighOrderBool(BoolFunctor): method __init__ (line 60) | def __init__(self, verbose_debug_str, function): method verbose_debug_str (line 64) | def verbose_debug_str(self, ctx, display_result=True): method __call__ (line 70) | def __call__(self, ctx): class _AndBoolFunctor (line 78) | class _AndBoolFunctor(BoolFunctor): method __init__ (line 79) | def __init__(self, lhs, rhs): method verbose_debug_str (line 85) | def verbose_debug_str(self, ctx, display_result=True): method __call__ (line 91) | def __call__(self, ctx): class _OrBoolFunctor (line 95) | class _OrBoolFunctor(BoolFunctor): method __init__ (line 96) | def __init__(self, lhs, rhs): method verbose_debug_str (line 102) | def verbose_debug_str(self, ctx, display_result=True): method __call__ (line 108) | def __call__(self, ctx): class _NotBoolFunctor (line 112) | class _NotBoolFunctor(BoolFunctor): method __init__ (line 113) | def __init__(self, x): method verbose_debug_str (line 117) | def verbose_debug_str(self, ctx, display_result=True): method __call__ (line 120) | def __call__(self, ctx): class HobContextGetter (line 124) | class HobContextGetter(object): method __init__ (line 125) | def __init__(self, attr_name, attr_getter): method attr_name (line 130) | def attr_name(self): method attr_getter (line 134) | def attr_getter(self): method __eq__ (line 137) | def __eq__(self, other): method __ne__ (line 142) | def __ne__(self, other): method __gt__ (line 147) | def __gt__(self, other): method __ge__ (line 152) | def __ge__(self, other): method __lt__ (line 157) | def __lt__(self, other): method __le__ (line 162) | def __le__(self, other): method _MakeHob (line 167) | def _MakeHob(self, other, cmp_str, cmp_func): class HobContextConstant (line 175) | class HobContextConstant(HobContextGetter): method __init__ (line 176) | def __init__(self, value): class HobContextAttr (line 180) | class HobContextAttr(HobContextGetter): method __init__ (line 181) | def __init__(self, attr_name, attr_getter): method __getattr__ (line 184) | def __getattr__(self, attr_name): method HasField (line 192) | def HasField(self, attr_name): FILE: python/oneflow/support/lazy.py class Lazy (line 18) | class Lazy(object): method __init__ (line 19) | def __init__(self, get_value): method value (line 25) | def value(self): FILE: python/oneflow/support/pb_util.py function PythonDict2PbMessage (line 18) | def PythonDict2PbMessage(value, msg): function MergePbMessage (line 44) | def MergePbMessage(dst, src): function _MergePbMessageField (line 64) | def _MergePbMessageField(dst, src, field): FILE: python/oneflow/support/scope_stack.py class ScopeStack (line 19) | class ScopeStack(object): method __init__ (line 20) | def __init__(self, init=[]): method Current (line 26) | def Current(self): method NewScope (line 31) | def NewScope(self, scope): FILE: python/oneflow/support/traceinfo.py function GetFrameLocationStr (line 20) | def GetFrameLocationStr(depth=-1): function GetStackInfoExcludeOneflowPythonFile (line 26) | def GetStackInfoExcludeOneflowPythonFile(): FILE: python/oneflow/test/dataloader/data_utils.py function load_data_cifar10 (line 22) | def load_data_cifar10( function load_data_mnist (line 54) | def load_data_mnist( function get_fashion_mnist_dataset (line 88) | def get_fashion_mnist_dataset( function load_data_fashion_mnist (line 116) | def load_data_fashion_mnist( FILE: python/oneflow/test/dataloader/test_cifar_dataset_multiprocess.py class Net (line 42) | class Net(nn.Module): method __init__ (line 43) | def __init__(self): method forward (line 52) | def forward(self, x): function _test (line 62) | def _test(test_case): class TestCifarDataset (line 124) | class TestCifarDataset(flow.unittest.TestCase): method test_cifar_dataset (line 125) | def test_cifar_dataset(test_case): FILE: python/oneflow/test/dataloader/test_cifar_dataset_singleprocess.py class Net (line 43) | class Net(nn.Module): method __init__ (line 44) | def __init__(self): method forward (line 53) | def forward(self, x): function _test (line 63) | def _test(test_case): class TestCifarDataset (line 126) | class TestCifarDataset(flow.unittest.TestCase): method test_cifar_dataset (line 127) | def test_cifar_dataset(test_case): FILE: python/oneflow/test/dataloader/test_fashion_mnist_dataset.py function get_fashion_mnist_labels (line 26) | def get_fashion_mnist_labels(labels): class FlattenLayer (line 43) | class FlattenLayer(nn.Module): method __init__ (line 44) | def __init__(self): method forward (line 47) | def forward(self, x): # x shape: (batch, *, *, ...) function evaluate_accuracy (line 52) | def evaluate_accuracy(data_iter, net, device=None): function _test (line 70) | def _test(test_case): class TestFashionMnistDataset (line 135) | class TestFashionMnistDataset(flow.unittest.TestCase): method test_fashion_mnist_dataset (line 136) | def test_fashion_mnist_dataset(test_case): FILE: python/oneflow/test/dataloader/test_lenet.py class LeNet (line 28) | class LeNet(nn.Module): method __init__ (line 29) | def __init__(self): method forward (line 47) | def forward(self, img): function evaluate_accuracy (line 54) | def evaluate_accuracy(data_iter, net, device=None): function _test_train_and_eval (line 69) | def _test_train_and_eval(test_case): class TestLenet (line 134) | class TestLenet(flow.unittest.TestCase): method test_lenet (line 135) | def test_lenet(test_case): FILE: python/oneflow/test/dataloader/test_mnist_dataset.py function evaluate_accuracy (line 39) | def evaluate_accuracy(data_iter, net, device=None): class Net (line 54) | class Net(nn.Module): method __init__ (line 55) | def __init__( method forward (line 65) | def forward(self, x): function _test_train_and_eval (line 74) | def _test_train_and_eval(test_case): class TestMnistDataset (line 115) | class TestMnistDataset(flow.unittest.TestCase): method test_mnist_dataset (line 116) | def test_mnist_dataset(test_case): FILE: python/oneflow/test/dataloader/test_numpy_dataset.py class ScpDataset (line 25) | class ScpDataset(flow.utils.data.Dataset): method __init__ (line 26) | def __init__(self, chunksize=200, dim=81, length=2000): method __getitem__ (line 31) | def __getitem__(self, index): method __len__ (line 35) | def __len__(self): class TestNumpyDataset (line 40) | class TestNumpyDataset(flow.unittest.TestCase): method test_numpy_dataset (line 41) | def test_numpy_dataset(test_case): FILE: python/oneflow/test/dataloader/test_tensor_dataset.py class LinearNet (line 27) | class LinearNet(nn.Module): method __init__ (line 28) | def __init__(self, n_feature): method forward (line 32) | def forward(self, x): class TestTensorDataset (line 38) | class TestTensorDataset(flow.unittest.TestCase): method test_tensor_dataset (line 39) | def test_tensor_dataset(test_case): FILE: python/oneflow/test/dataloader/test_transforms.py class Net (line 29) | class Net(nn.Module): method __init__ (line 30) | def __init__(self): method forward (line 39) | def forward(self, x): function _test (line 49) | def _test(test_case): class TestCifarDataset (line 119) | class TestCifarDataset(flow.unittest.TestCase): method test_cifar_dataset (line 120) | def test_cifar_dataset(test_case): FILE: python/oneflow/test/exceptions/test_activation.py class TestActivationError (line 28) | class TestActivationError(flow.unittest.TestCase): method test_relu_inplace_runtime_error (line 29) | def test_relu_inplace_runtime_error(test_case): method test_prelu_runtime_error (line 38) | def test_prelu_runtime_error(test_case): method test_celu_inplace_runtime_error (line 47) | def test_celu_inplace_runtime_error(test_case): method test_glu_scalar_tensor_runtime_error (line 57) | def test_glu_scalar_tensor_runtime_error(test_case): method test_glu_dim_index_error (line 67) | def test_glu_dim_index_error(test_case): method test_glu_dim_even_runtime_error (line 77) | def test_glu_dim_even_runtime_error(test_case): method test_hard_sigmoid_inplace_runtime_error (line 87) | def test_hard_sigmoid_inplace_runtime_error(test_case): method test_hard_shrink_inplace_runtime_error (line 98) | def test_hard_shrink_inplace_runtime_error(test_case): method test_softmax_index_error (line 109) | def test_softmax_index_error(test_case): method test_soft_shrink_inplace_runtime_error (line 119) | def test_soft_shrink_inplace_runtime_error(test_case): method test_soft_shrink_alpha_runtime_error (line 130) | def test_soft_shrink_alpha_runtime_error(test_case): FILE: python/oneflow/test/exceptions/test_add_n_op.py class TestAddN (line 21) | class TestAddN(flow.unittest.TestCase): method test_add_n_shape_error_msg (line 22) | def test_add_n_shape_error_msg(test_case): method test_add_n_dtype_error_msg (line 33) | def test_add_n_dtype_error_msg(test_case): FILE: python/oneflow/test/exceptions/test_arg_sort_op.py class TestArgSort (line 22) | class TestArgSort(flow.unittest.TestCase): method test_direction_parameter_err (line 23) | def test_direction_parameter_err(test_case): FILE: python/oneflow/test/exceptions/test_array_functor.py class TestArrayError (line 21) | class TestArrayError(flow.unittest.TestCase): method test_argmax_index_error (line 22) | def test_argmax_index_error(test_case): method test_broadcast_like_runtime_error (line 28) | def test_broadcast_like_runtime_error(test_case): method test_broadcast_like_numaxes_runtime_error (line 37) | def test_broadcast_like_numaxes_runtime_error(test_case): method test_concat_index_error (line 45) | def test_concat_index_error(test_case): method test_concat_dim_equal_runtime_error (line 52) | def test_concat_dim_equal_runtime_error(test_case): method test_concat_match_size_runtime_error (line 61) | def test_concat_match_size_runtime_error(test_case): method test_stack_index_error (line 70) | def test_stack_index_error(test_case): method test_stack_runtime_error (line 77) | def test_stack_runtime_error(test_case): method test_expand_dim_runtime_error (line 86) | def test_expand_dim_runtime_error(test_case): method test_expand_g_shape_runtime_error (line 96) | def test_expand_g_shape_runtime_error(test_case): method test_expand_l_shape_runtime_error (line 105) | def test_expand_l_shape_runtime_error(test_case): method test_squeeze_index_error (line 114) | def test_squeeze_index_error(test_case): method test_roll_runtime_error (line 120) | def test_roll_runtime_error(test_case): method test_gather_index_type_runtime_error (line 128) | def test_gather_index_type_runtime_error(test_case): method test_gather_dim_value_runtime_error (line 138) | def test_gather_dim_value_runtime_error(test_case): method test_gather_dim_equal_runtime_error (line 145) | def test_gather_dim_equal_runtime_error(test_case): method test_gather_size_runtime_error (line 155) | def test_gather_size_runtime_error(test_case): method test_tensor_scatter_nd_update_runtime_error (line 164) | def test_tensor_scatter_nd_update_runtime_error(test_case): method test_view_runtime_error (line 174) | def test_view_runtime_error(test_case): method test_narrow_dim_index_error (line 186) | def test_narrow_dim_index_error(test_case): method test_narrow_0_dim_runtime_error (line 192) | def test_narrow_0_dim_runtime_error(test_case): method test_narrow_start_index_error (line 200) | def test_narrow_start_index_error(test_case): method test_narrow_length_exceed_runtime_error (line 206) | def test_narrow_length_exceed_runtime_error(test_case): method test_diagonal_index_error1 (line 212) | def test_diagonal_index_error1(test_case): method test_diagonal_index_error2 (line 218) | def test_diagonal_index_error2(test_case): method test_diagonal_runtime_error (line 224) | def test_diagonal_runtime_error(test_case): method test_split_index_error (line 232) | def test_split_index_error(test_case): method test_split_runtime_error (line 238) | def test_split_runtime_error(test_case): method test_splitwithsize_runtime_error (line 247) | def test_splitwithsize_runtime_error(test_case): method test_unbind_index_error (line 256) | def test_unbind_index_error(test_case): method test_chunk_index_error (line 262) | def test_chunk_index_error(test_case): method test_chunk_tensor_dim_runtime_error (line 268) | def test_chunk_tensor_dim_runtime_error(test_case): method test_chunk_value_runtime_error (line 276) | def test_chunk_value_runtime_error(test_case): method test_meshgrid_tensors_scalar_runtime_error (line 284) | def test_meshgrid_tensors_scalar_runtime_error(test_case): method test_meshgrid_tensors_size_runtime_error (line 293) | def test_meshgrid_tensors_size_runtime_error(test_case): method test_meshgrid_tensors_dtype_runtime_error (line 300) | def test_meshgrid_tensors_dtype_runtime_error(test_case): method test_meshgrid_tensors_placement_runtime_error (line 310) | def test_meshgrid_tensors_placement_runtime_error(test_case): method test_meshgrid_indexing_runtime_error (line 329) | def test_meshgrid_indexing_runtime_error(test_case): method test_index_select_runtime_error (line 338) | def test_index_select_runtime_error(test_case): method test_index_select_index_num_error (line 349) | def test_index_select_index_num_error(test_case): method test_index_select_index_error (line 360) | def test_index_select_index_error(test_case): method test_to_device_runtime_error (line 369) | def test_to_device_runtime_error(test_case): method test_to_other_runtime_error (line 382) | def test_to_other_runtime_error(test_case): method test_in_top_k_num_equal_runtime_error (line 397) | def test_in_top_k_num_equal_runtime_error(test_case): method test_in_top_k_targets_dim_runtime_error (line 409) | def test_in_top_k_targets_dim_runtime_error(test_case): method test_in_top_k_pre_dim_runtime_error (line 420) | def test_in_top_k_pre_dim_runtime_error(test_case): method test_repeat_runtime_error (line 431) | def test_repeat_runtime_error(test_case): method test_tile_runtime_error (line 439) | def test_tile_runtime_error(test_case): method test_t_runtime_error (line 447) | def test_t_runtime_error(test_case): FILE: python/oneflow/test/exceptions/test_autograd.py class TestAutograd (line 23) | class TestAutograd(flow.unittest.TestCase): method test_non_requires_grad_tensor_backward (line 24) | def test_non_requires_grad_tensor_backward(test_case): method test_allow_unused (line 35) | def test_allow_unused(test_case): FILE: python/oneflow/test/exceptions/test_batch_gather_op.py class TestBatchGather (line 23) | class TestBatchGather(flow.unittest.TestCase): method test_input_tensor_dimesion_error_msg (line 24) | def test_input_tensor_dimesion_error_msg(test_case): method test_indices_dimesion_error_msg (line 34) | def test_indices_dimesion_error_msg(test_case): method test_legal_dimension_error_msg (line 44) | def test_legal_dimension_error_msg(test_case): method test_indice_type_error_msg (line 55) | def test_indice_type_error_msg(test_case): method test_tensor_shape_size_error_msg (line 66) | def test_tensor_shape_size_error_msg(test_case): FILE: python/oneflow/test/exceptions/test_bias_add_op.py class TestBiasAdd (line 22) | class TestBiasAdd(flow.unittest.TestCase): method test_b_tensor_numaxes_err (line 23) | def test_b_tensor_numaxes_err(test_case): FILE: python/oneflow/test/exceptions/test_binary_functor_exception.py class TestBinaryFunctorError (line 26) | class TestBinaryFunctorError(flow.unittest.TestCase): method test_add_inplace_runtime_error (line 27) | def test_add_inplace_runtime_error(test_case): method test_add_broad_cast_runtime_error (line 37) | def test_add_broad_cast_runtime_error(test_case): method test_div_inplace_runtime_error (line 81) | def test_div_inplace_runtime_error(test_case): FILE: python/oneflow/test/exceptions/test_bmm.py class TestBmm (line 24) | class TestBmm(flow.unittest.TestCase): method test_bmm_exception_dim_not_right (line 25) | def test_bmm_exception_dim_not_right(test_case): FILE: python/oneflow/test/exceptions/test_broadcast_ops.py class TestBroadcastOps (line 46) | class TestBroadcastOps(flow.unittest.TestCase): method test_broadcast_binary_ops (line 47) | def test_broadcast_binary_ops(test_case): method test_broadcast_shapes (line 58) | def test_broadcast_shapes(test_case): method test_broadcast_tensors (line 69) | def test_broadcast_tensors(test_case): method test_broadcast_to (line 80) | def test_broadcast_to(test_case): FILE: python/oneflow/test/exceptions/test_chunk.py class TestModule (line 24) | class TestModule(flow.unittest.TestCase): method test_chunk_0_dim_input_exception (line 25) | def test_chunk_0_dim_input_exception(test_case): method test_chunk_0_chunks_param_exception (line 37) | def test_chunk_0_chunks_param_exception(test_case): method test_chunk_dim_param_exception (line 49) | def test_chunk_dim_param_exception(test_case): FILE: python/oneflow/test/exceptions/test_cosine_similarity.py class TestCosineSimilarity (line 23) | class TestCosineSimilarity(flow.unittest.TestCase): method test_cosine_similarity_not_floating_type (line 24) | def test_cosine_similarity_not_floating_type(test_case): method test_cosine_similarity_broadcast (line 34) | def test_cosine_similarity_broadcast(test_case): FILE: python/oneflow/test/exceptions/test_deform_conv2d_op.py class TestDeformConv (line 22) | class TestDeformConv(flow.unittest.TestCase): method test_deform_conv2d_invalid_input_sizes (line 23) | def test_deform_conv2d_invalid_input_sizes(test_case): method test_deform_conv2d_invalid_offset_sizes (line 33) | def test_deform_conv2d_invalid_offset_sizes(test_case): method test_deform_conv2d_invalid_weight_sizes (line 43) | def test_deform_conv2d_invalid_weight_sizes(test_case): method test_deform_conv2d_invalid_mask_sizes (line 53) | def test_deform_conv2d_invalid_mask_sizes(test_case): method test_deform_conv2d_invalid_dilation_parm (line 64) | def test_deform_conv2d_invalid_dilation_parm(test_case): method test_deform_conv2d_invalid_pad_parm (line 74) | def test_deform_conv2d_invalid_pad_parm(test_case): method test_deform_conv2d_invalid_stride_parm (line 84) | def test_deform_conv2d_invalid_stride_parm(test_case): method test_deform_conv2d_invalid_offset_shape (line 94) | def test_deform_conv2d_invalid_offset_shape(test_case): method test_deform_conv2d_invalid_batch_size (line 105) | def test_deform_conv2d_invalid_batch_size(test_case): method test_deform_conv2d_invalid_mask_shape (line 113) | def test_deform_conv2d_invalid_mask_shape(test_case): method test_deform_conv2d_invalid_output_size (line 122) | def test_deform_conv2d_invalid_output_size(test_case): method test_deform_conv2d_invalid_offset_output_dims (line 132) | def test_deform_conv2d_invalid_offset_output_dims(test_case): FILE: python/oneflow/test/exceptions/test_device.py class TestDevice (line 24) | class TestDevice(flow.unittest.TestCase): method test_device_type (line 25) | def test_device_type(test_case): method test_device_index (line 36) | def test_device_index(test_case): FILE: python/oneflow/test/exceptions/test_dot.py class TestDot (line 23) | class TestDot(flow.unittest.TestCase): method test_dot_shape_error_msg (line 24) | def test_dot_shape_error_msg(test_case): method test_dot_dims_error_msg (line 31) | def test_dot_dims_error_msg(test_case): method test_dot_dtype_error_msg (line 37) | def test_dot_dtype_error_msg(test_case): FILE: python/oneflow/test/exceptions/test_error_reported_in_thread.py function test_error_reported_in_thread (line 26) | def test_error_reported_in_thread(): function test_python_stack_getter_disabled (line 52) | def test_python_stack_getter_disabled(): FILE: python/oneflow/test/exceptions/test_gird_sample_op.py function _test_dimention_error_msg_impl (line 35) | def _test_dimention_error_msg_impl(test_case, N, C, H_in, H_out): function _test_4d_gird_shape_error_msg_impl (line 45) | def _test_4d_gird_shape_error_msg_impl(test_case, N, C, H_in, W_in, H_ou... function _test_4d_grid_input_not_same_shape_error_msg_impl (line 55) | def _test_4d_grid_input_not_same_shape_error_msg_impl( function _test_5d_gird_shape_error_msg_impl (line 69) | def _test_5d_gird_shape_error_msg_impl( function _test_5d_grid_input_not_same_shape_error_msg_impl (line 81) | def _test_5d_grid_input_not_same_shape_error_msg_impl( class TestGridSample (line 95) | class TestGridSample(flow.unittest.TestCase): method test_dimention_error_msg (line 96) | def test_dimention_error_msg(test_case): method test_4d_gird_shape_error_msg (line 100) | def test_4d_gird_shape_error_msg(test_case): method test_4d_grid_input_not_same_shape_error_msg (line 106) | def test_4d_grid_input_not_same_shape_error_msg(test_case): method test_5d_gird_shape_error_msg (line 112) | def test_5d_gird_shape_error_msg(test_case): method test_5d_grid_input_not_same_shape_error_msg (line 116) | def test_5d_grid_input_not_same_shape_error_msg(test_case): FILE: python/oneflow/test/exceptions/test_global_branch_error_local_to_global_with_broadcast_sbp_1n2d.py class TestLocalToGlobalBranchError (line 30) | class TestLocalToGlobalBranchError(flow.unittest.TestCase): method test_global_branch_error_with_local_to_global (line 32) | def test_global_branch_error_with_local_to_global(test_case): FILE: python/oneflow/test/exceptions/test_global_branch_error_local_to_global_with_broadcast_sbp_1n4d.py class TestLocalToGlobalBranchError (line 30) | class TestLocalToGlobalBranchError(flow.unittest.TestCase): method test_global_branch_error_with_local_to_global (line 32) | def test_global_branch_error_with_local_to_global(test_case): FILE: python/oneflow/test/exceptions/test_global_branch_error_local_to_global_with_split_sbp.py class TestLocalToGlobalBranchError (line 30) | class TestLocalToGlobalBranchError(flow.unittest.TestCase): method test_global_branch_error_with_local_to_global (line 32) | def test_global_branch_error_with_local_to_global(test_case): FILE: python/oneflow/test/exceptions/test_global_branch_error_with_global_mean.py class TestGlobalMeanBranchError (line 30) | class TestGlobalMeanBranchError(flow.unittest.TestCase): method test_global_branch_error_global_data_mean (line 32) | def test_global_branch_error_global_data_mean(test_case): FILE: python/oneflow/test/exceptions/test_hann_window.py class TestHannWindow (line 24) | class TestHannWindow(flow.unittest.TestCase): method test_hann_window_dtype_not_support (line 25) | def test_hann_window_dtype_not_support(test_case): FILE: python/oneflow/test/exceptions/test_in_top_k.py class TestInTopK (line 23) | class TestInTopK(flow.unittest.TestCase): method test_in_top_k_error_msg (line 24) | def test_in_top_k_error_msg(test_case): FILE: python/oneflow/test/exceptions/test_inv.py class TestInv (line 24) | class TestInv(flow.unittest.TestCase): method test_inv_exception_dim_short (line 25) | def test_inv_exception_dim_short(test_case): method test_inv_exception_not_square_matrix (line 34) | def test_inv_exception_not_square_matrix(test_case): FILE: python/oneflow/test/exceptions/test_layernorm.py class TestLayerNormModule (line 24) | class TestLayerNormModule(flow.unittest.TestCase): method test_layernorm_exception_input_shape_not_match (line 25) | def test_layernorm_exception_input_shape_not_match(test_case): FILE: python/oneflow/test/exceptions/test_linalg.py class TestLinalgCross (line 22) | class TestLinalgCross(flow.unittest.TestCase): method test_cross_has_no_3_error (line 23) | def test_cross_has_no_3_error(test_case): method test_linalg_cross_has_no_3_error (line 32) | def test_linalg_cross_has_no_3_error(test_case): method test_linalg_cross_broadcast_error (line 42) | def test_linalg_cross_broadcast_error(test_case): FILE: python/oneflow/test/exceptions/test_local_global_convert_error.py class TestModule (line 22) | class TestModule(flow.unittest.TestCase): method test_get_sbp_with_invalid_axis (line 24) | def test_get_sbp_with_invalid_axis(test_case): method test_local_to_global_with_invalid_split_axis (line 39) | def test_local_to_global_with_invalid_split_axis(test_case): method test_global_to_global_with_invalid_split_axis (line 49) | def test_global_to_global_with_invalid_split_axis(test_case): method test_call_to_local_for_local_tensor (line 61) | def test_call_to_local_for_local_tensor(test_case): method test_local_to_global_with_invalid_size (line 71) | def test_local_to_global_with_invalid_size(test_case): FILE: python/oneflow/test/exceptions/test_median.py class TestMedian (line 24) | class TestMedian(flow.unittest.TestCase): method test_median_exception_dim_out_of_range (line 25) | def test_median_exception_dim_out_of_range(test_case): method test_median_exception_reduce_0dim (line 34) | def test_median_exception_reduce_0dim(test_case): FILE: python/oneflow/test/exceptions/test_mm.py class TestMm (line 23) | class TestMm(flow.unittest.TestCase): method test_mm_not_2dim (line 24) | def test_mm_not_2dim(test_case): method test_mm_dim_not_match (line 36) | def test_mm_dim_not_match(test_case): FILE: python/oneflow/test/exceptions/test_mode.py class TestMode (line 24) | class TestMode(flow.unittest.TestCase): method test_mode_exception_dim_out_of_range (line 25) | def test_mode_exception_dim_out_of_range(test_case): method test_mode_exception_reduce_0dim (line 34) | def test_mode_exception_reduce_0dim(test_case): FILE: python/oneflow/test/exceptions/test_multi_input_with_diff_device_or_placement.py class TestModule (line 25) | class TestModule(flow.unittest.TestCase): method test_multi_input_with_diff_device (line 27) | def test_multi_input_with_diff_device(test_case): method test_multi_input_with_diff_placement (line 42) | def test_multi_input_with_diff_placement(test_case): FILE: python/oneflow/test/exceptions/test_mv.py class TestMv (line 22) | class TestMv(flow.unittest.TestCase): method test_mv_not_matrix (line 23) | def test_mv_not_matrix(test_case): method test_mv_not_vector (line 32) | def test_mv_not_vector(test_case): method test_mv_size_mismatch (line 41) | def test_mv_size_mismatch(test_case): FILE: python/oneflow/test/exceptions/test_nn_functor.py class TestBiasAddError (line 25) | class TestBiasAddError(flow.unittest.TestCase): method test_bias_add_dimension_match_error (line 26) | def test_bias_add_dimension_match_error(test_case): method test_bias_add_index_error (line 37) | def test_bias_add_index_error(test_case): class TestCrossEntropyError (line 49) | class TestCrossEntropyError(flow.unittest.TestCase): method test_cross_entropy_reduction_type_error (line 50) | def test_cross_entropy_reduction_type_error(test_case): class TestCTCLossError (line 61) | class TestCTCLossError(flow.unittest.TestCase): method test_ctcloss_reduction_type_error (line 62) | def test_ctcloss_reduction_type_error(test_case): class TestPadError (line 88) | class TestPadError(flow.unittest.TestCase): method test_pad_size_attribute_error (line 89) | def test_pad_size_attribute_error(test_case): method test_pad_size_mod2_error (line 98) | def test_pad_size_mod2_error(test_case): method test_reflect_pad_size_error (line 107) | def test_reflect_pad_size_error(test_case): method test_pad_mode_error (line 117) | def test_pad_mode_error(test_case): class TestFusedMLPError (line 128) | class TestFusedMLPError(flow.unittest.TestCase): method test_fuse_mlp_weight_size_error (line 129) | def test_fuse_mlp_weight_size_error(test_case): method test_fuse_mlp_weight_bias_size_error (line 139) | def test_fuse_mlp_weight_bias_size_error(test_case): method test_fuse_mlp_weight_numaxes_error (line 151) | def test_fuse_mlp_weight_numaxes_error(test_case): method test_fuse_mlp_bias_numaxes_error (line 159) | def test_fuse_mlp_bias_numaxes_error(test_case): method test_fuse_mlp_bias_first_dim_error (line 167) | def test_fuse_mlp_bias_first_dim_error(test_case): method test_fuse_mlp_weight_second_dim_error (line 178) | def test_fuse_mlp_weight_second_dim_error(test_case): class TestL2NormalizeError (line 191) | class TestL2NormalizeError(flow.unittest.TestCase): method test_l2normalize_axis_error1 (line 192) | def test_l2normalize_axis_error1(test_case): method test_l2normalize_axis_error2 (line 198) | def test_l2normalize_axis_error2(test_case): class TestLossBaseFunctorError (line 207) | class TestLossBaseFunctorError(flow.unittest.TestCase): method test_loss_base_reduction_type_error (line 208) | def test_loss_base_reduction_type_error(test_case): class TestMatmulError (line 219) | class TestMatmulError(flow.unittest.TestCase): method test_matmul_dimension_error1 (line 220) | def test_matmul_dimension_error1(test_case): method test_matmul_dimension_error2 (line 227) | def test_matmul_dimension_error2(test_case): class TestPixelShuffleError (line 235) | class TestPixelShuffleError(flow.unittest.TestCase): method test_pixel_shuffle_4D_input_error (line 236) | def test_pixel_shuffle_4D_input_error(test_case): method test_pixel_shuffle_channel_divisble_error (line 243) | def test_pixel_shuffle_channel_divisble_error(test_case): class TestTripletMarginLossError (line 254) | class TestTripletMarginLossError(flow.unittest.TestCase): method test_triplet_margin_loss_reduce_type_error (line 255) | def test_triplet_margin_loss_reduce_type_error(test_case): class TestNormalError (line 277) | class TestNormalError(flow.unittest.TestCase): method test_normal_data_type_error (line 278) | def test_normal_data_type_error(test_case): method test_normal_out_tensor_data_type_error (line 286) | def test_normal_out_tensor_data_type_error(test_case): method test_normal_out_tensor_device_type_error (line 299) | def test_normal_out_tensor_device_type_error(test_case): class TestNormalizationError (line 316) | class TestNormalizationError(flow.unittest.TestCase): method test_normalization_moving_mean_error (line 317) | def test_normalization_moving_mean_error(test_case): method test_normalization_x_input_axes_error (line 333) | def test_normalization_x_input_axes_error(test_case): method test_normalization_eval_need_moving_statistic_error (line 347) | def test_normalization_eval_need_moving_statistic_error(test_case): class TestOnehotError (line 363) | class TestOnehotError(flow.unittest.TestCase): method test_onehot_error (line 364) | def test_onehot_error(test_case): FILE: python/oneflow/test/exceptions/test_optim_add_param_group.py class TestSgdAddParamGroup (line 22) | class TestSgdAddParamGroup(flow.unittest.TestCase): method test_sgd_add_param_group_not_unique (line 23) | def test_sgd_add_param_group_not_unique(test_case): FILE: python/oneflow/test/exceptions/test_pad.py class TestPad (line 24) | class TestPad(flow.unittest.TestCase): method test_torch_type (line 25) | def test_torch_type(test_case): method test_numpy_type (line 32) | def test_numpy_type(test_case): FILE: python/oneflow/test/exceptions/test_placement.py class TestPlacement (line 22) | class TestPlacement(flow.unittest.TestCase): method test_inconsistent_placement (line 24) | def test_inconsistent_placement(test_case): FILE: python/oneflow/test/exceptions/test_randperm_op.py class TestRandpermOp (line 23) | class TestRandpermOp(flow.unittest.TestCase): method test_randperm_n_value_err_mes (line 24) | def test_randperm_n_value_err_mes(test_case): FILE: python/oneflow/test/exceptions/test_reduce_like_ops.py class TestReduceSumLikeOps (line 23) | class TestReduceSumLikeOps(flow.unittest.TestCase): method test_reduce_sum_like_empty_axis_case_err (line 24) | def test_reduce_sum_like_empty_axis_case_err(test_case): method test_reduce_sum_like_type_err (line 34) | def test_reduce_sum_like_type_err(test_case): FILE: python/oneflow/test/exceptions/test_reduce_ops.py class TestReduceOps (line 24) | class TestReduceOps(flow.unittest.TestCase): method test_exception_dim_out_of_int_range (line 25) | def test_exception_dim_out_of_int_range(test_case): method test_exception_dim_out_of_list_range (line 31) | def test_exception_dim_out_of_list_range(test_case): FILE: python/oneflow/test/exceptions/test_repeat_interleave.py class TestRepeatInterleave (line 24) | class TestRepeatInterleave(flow.unittest.TestCase): method test_repeat_interleave_index_error (line 25) | def test_repeat_interleave_index_error(test_case): method test_repeat_interleave_tensor_shape_error (line 34) | def test_repeat_interleave_tensor_shape_error(test_case): method test_repeat_interleave_dtype_error (line 44) | def test_repeat_interleave_dtype_error(test_case): method test_repeat_interleave_negative_tensor_error (line 51) | def test_repeat_interleave_negative_tensor_error(test_case): method test_repeat_interleave_negative_tensor_error (line 58) | def test_repeat_interleave_negative_tensor_error(test_case): method test_repeat_interleave_dim_not_match_error (line 68) | def test_repeat_interleave_dim_not_match_error(test_case): FILE: python/oneflow/test/exceptions/test_reshape.py class TestModule (line 24) | class TestModule(flow.unittest.TestCase): method test_reshape_exception_invalid_dim (line 25) | def test_reshape_exception_invalid_dim(test_case): method test_reshape_exception_invalid_size (line 35) | def test_reshape_exception_invalid_size(test_case): method test_reshape_exception_only_one_dim_infered (line 45) | def test_reshape_exception_only_one_dim_infered(test_case): FILE: python/oneflow/test/exceptions/test_reshape_like_op.py class TestReshapeLikeOp (line 23) | class TestReshapeLikeOp(flow.unittest.TestCase): method test_reshape_like_size_match_err (line 24) | def test_reshape_like_size_match_err(test_case): FILE: python/oneflow/test/exceptions/test_roi_align_op.py class TestRoiAlignOp (line 23) | class TestRoiAlignOp(flow.unittest.TestCase): method test_rol_align_x_tensor_dimension_err (line 24) | def test_rol_align_x_tensor_dimension_err(test_case): method test_rol_align_rois_tensor_dimension_err (line 34) | def test_rol_align_rois_tensor_dimension_err(test_case): method test_rol_align_rois_tensor_size_err (line 44) | def test_rol_align_rois_tensor_size_err(test_case): FILE: python/oneflow/test/exceptions/test_save_load.py class TestSaveLoad (line 25) | class TestSaveLoad(flow.unittest.TestCase): method test_support_pytorch_with_global_src_rank (line 26) | def test_support_pytorch_with_global_src_rank(test_case): method test_load_invalid_file (line 37) | def test_load_invalid_file(test_case): FILE: python/oneflow/test/exceptions/test_saved_tensor_hooks.py class TestSavedTensorHooks (line 25) | class TestSavedTensorHooks(flow.unittest.TestCase): method test_unpack_returns_non_tensor (line 26) | def test_unpack_returns_non_tensor(test_case): FILE: python/oneflow/test/exceptions/test_slice_op.py class TestSlice (line 22) | class TestSlice(flow.unittest.TestCase): method test_slice_update_start_list_err (line 23) | def test_slice_update_start_list_err(test_case): method test_slice_update_stop_list_err (line 36) | def test_slice_update_stop_list_err(test_case): method test_slice_update_step_list_err (line 49) | def test_slice_update_step_list_err(test_case): method test_slice_update_start_and_stop_compare_value_err (line 62) | def test_slice_update_start_and_stop_compare_value_err(test_case): method test_slice_update_turple_size_match_err (line 75) | def test_slice_update_turple_size_match_err(test_case): method test_slice_update_type_err (line 88) | def test_slice_update_type_err(test_case): method test_slice_start_list_err (line 100) | def test_slice_start_list_err(test_case): method test_slice_stop_list_err (line 112) | def test_slice_stop_list_err(test_case): method test_slice_step_list_err (line 124) | def test_slice_step_list_err(test_case): method test_slice_start_and_stop_compare_value_err (line 136) | def test_slice_start_and_stop_compare_value_err(test_case): FILE: python/oneflow/test/exceptions/test_smooth_l1_loss_op.py class TestSmoothL1LossError (line 22) | class TestSmoothL1LossError(flow.unittest.TestCase): method test_smooth_l1_loss_shape_err (line 23) | def test_smooth_l1_loss_shape_err(test_case): method test_smooth_l1_loss_beta_err (line 32) | def test_smooth_l1_loss_beta_err(test_case): method test_smooth_l1_loss_dtype_err (line 43) | def test_smooth_l1_loss_dtype_err(test_case): FILE: python/oneflow/test/exceptions/test_softmax_cross_entropy_op.py class TestSoftmaxCrossEntropyError (line 22) | class TestSoftmaxCrossEntropyError(flow.unittest.TestCase): method test_softmax_cross_entropy_prediction_numaxes_err (line 23) | def test_softmax_cross_entropy_prediction_numaxes_err(test_case): method test_softmax_cross_entropy_prediction_shape_err (line 33) | def test_softmax_cross_entropy_prediction_shape_err(test_case): method test_softmax_cross_entropy_dtype_err (line 42) | def test_softmax_cross_entropy_dtype_err(test_case): method test_softmax_cross_entropy_grad_prob_numaxes_err (line 52) | def test_softmax_cross_entropy_grad_prob_numaxes_err(test_case): method test_softmax_cross_entropy_grad_dy_numaxes_err (line 63) | def test_softmax_cross_entropy_grad_dy_numaxes_err(test_case): method test_softmax_cross_entropy_grad_dy_i_shape_err (line 74) | def test_softmax_cross_entropy_grad_dy_i_shape_err(test_case): method test_softmax_cross_entropy_grad_prob_shape_err (line 82) | def test_softmax_cross_entropy_grad_prob_shape_err(test_case): method test_softmax_cross_entropy_grad_label_dtype_err (line 90) | def test_softmax_cross_entropy_grad_label_dtype_err(test_case): method test_softmax_cross_entropy_grad_dy_dtype_err (line 101) | def test_softmax_cross_entropy_grad_dy_dtype_err(test_case): FILE: python/oneflow/test/exceptions/test_sparse_cross_entropy_op.py class TestSparseCrossEntropyError (line 22) | class TestSparseCrossEntropyError(flow.unittest.TestCase): method test_sparse_cross_entropy_prediction_numaxes_err (line 23) | def test_sparse_cross_entropy_prediction_numaxes_err(test_case): method test_sparse_cross_entropy_label_numaxes_err (line 34) | def test_sparse_cross_entropy_label_numaxes_err(test_case): method test_sparse_cross_entropy_prediction_i_shape_err (line 45) | def test_sparse_cross_entropy_prediction_i_shape_err(test_case): method test_sparse_cross_entropy_label_dtype_err (line 53) | def test_sparse_cross_entropy_label_dtype_err(test_case): FILE: python/oneflow/test/exceptions/test_sparse_softmax_cross_entropy_op.py class TestSparseSoftmaxCrossEntropyError (line 22) | class TestSparseSoftmaxCrossEntropyError(flow.unittest.TestCase): method test_sparse_softmax_cross_entropy_prediction_numaxes_err (line 23) | def test_sparse_softmax_cross_entropy_prediction_numaxes_err(test_case): method test_sparse_softmax_cross_entropy_label_numaxes_err (line 33) | def test_sparse_softmax_cross_entropy_label_numaxes_err(test_case): method test_sparse_softmax_cross_entropy_prediction_i_shape_err (line 43) | def test_sparse_softmax_cross_entropy_prediction_i_shape_err(test_case): method test_sparse_softmax_cross_entropy_label_dtype_err (line 50) | def test_sparse_softmax_cross_entropy_label_dtype_err(test_case): FILE: python/oneflow/test/exceptions/test_split_like_op.py class TestSplitLikeError (line 22) | class TestSplitLikeError(flow.unittest.TestCase): method test_split_like_like_axes_err (line 23) | def test_split_like_like_axes_err(test_case): method test_split_like_split_axes_err (line 33) | def test_split_like_split_axes_err(test_case): method test_split_like_like_i_axes_err (line 43) | def test_split_like_like_i_axes_err(test_case): method test_split_like_x_i_shape_err (line 53) | def test_split_like_x_i_shape_err(test_case): method test_split_like_non_dynamic_static_dim_err (line 61) | def test_split_like_non_dynamic_static_dim_err(test_case): FILE: python/oneflow/test/exceptions/test_stft_op.py class TestModule (line 23) | class TestModule(flow.unittest.TestCase): method test_stft_illegal_input_dim (line 24) | def test_stft_illegal_input_dim(test_case): method test_stft_illegal_nfft (line 39) | def test_stft_illegal_nfft(test_case): method test_stft_illegal_hop_length (line 58) | def test_stft_illegal_hop_length(test_case): method test_stft_illegal_win_length (line 75) | def test_stft_illegal_win_length(test_case): method test_stft_illegal_window (line 94) | def test_stft_illegal_window(test_case): FILE: python/oneflow/test/exceptions/test_tensor_index.py class TestTensorIndexError (line 21) | class TestTensorIndexError(flow.unittest.TestCase): method test_PrepareSliceIndices_indices_amount_index_error (line 22) | def test_PrepareSliceIndices_indices_amount_index_error(test_case): method test_PrepareSliceIndices_slice_step_runtime_error (line 30) | def test_PrepareSliceIndices_slice_step_runtime_error(test_case): method test_ApplySelectIndexing_input_dim_runtime_error (line 37) | def test_ApplySelectIndexing_input_dim_runtime_error(test_case): method test_ApplySelectIndexing_index_error (line 45) | def test_ApplySelectIndexing_index_error(test_case): method test_ApplyAdvancedIndexing_index_error (line 53) | def test_ApplyAdvancedIndexing_index_error(test_case): FILE: python/oneflow/test/exceptions/test_tensordot.py class TestTensordotError (line 24) | class TestTensordotError(flow.unittest.TestCase): method test_tensordot_neg_dims_runtime_error (line 25) | def test_tensordot_neg_dims_runtime_error(test_case): method test_tensordot_too_large_int_dims_runtime_error (line 35) | def test_tensordot_too_large_int_dims_runtime_error(test_case): method test_tensordot_out_of_range_dims_runtime_error (line 45) | def test_tensordot_out_of_range_dims_runtime_error(test_case): method test_tensordot_unmatch_dims_runtime_error (line 55) | def test_tensordot_unmatch_dims_runtime_error(test_case): method test_tensordot_recurring_dim_runtime_error (line 65) | def test_tensordot_recurring_dim_runtime_error(test_case): method test_tensordot_dims_different_length_runtime_error (line 74) | def test_tensordot_dims_different_length_runtime_error(test_case): FILE: python/oneflow/test/exceptions/test_to_global_error.py class TestToGlobalError (line 30) | class TestToGlobalError(flow.unittest.TestCase): method test_tensor_to_consistent (line 32) | def test_tensor_to_consistent(self): method test_tensor_is_global (line 45) | def test_tensor_is_global(self): method test_module_to_consistent (line 56) | def test_module_to_consistent(self): FILE: python/oneflow/test/exceptions/test_view.py class TestModule (line 24) | class TestModule(flow.unittest.TestCase): method test_view_exception (line 25) | def test_view_exception(test_case): FILE: python/oneflow/test/exceptions/throw_error.py function g (line 23) | def g(x): function f (line 27) | def f(x): FILE: python/oneflow/test/expensive/_internally_replaced_utils.py function _download_file_from_remote_location (line 20) | def _download_file_from_remote_location(fpath: str, url: str) -> None: function _is_remote_location_available (line 24) | def _is_remote_location_available() -> bool: function _get_extension_path (line 34) | def _get_extension_path(lib_name): FILE: python/oneflow/test/expensive/_test_remat.py function evict (line 33) | def evict(tensor): function is_in_memory (line 37) | def is_in_memory(tensor): function allocated_memory (line 44) | def allocated_memory(device, include_test_placeholder=False): function display (line 52) | def display(device): function only_fbip (line 56) | def only_fbip(): function only_copy_on_write (line 63) | def only_copy_on_write(): function loss_test (line 70) | def loss_test(): function generate_placeholder (line 80) | def generate_placeholder(size_mb, device): function memory_budget (line 92) | def memory_budget(budget_mb, device): class TestRemat (line 110) | class TestRemat(flow.unittest.TestCase): method setUpClass (line 112) | def setUpClass(cls): method setUp (line 116) | def setUp(self): method tearDown (line 134) | def tearDown(self): method test_remat_work_on_fbip_1 (line 143) | def test_remat_work_on_fbip_1(self, device): method test_remat_work_on_fbip_2 (line 159) | def test_remat_work_on_fbip_2(self, device): method test_remat_work_on_fbip_3 (line 175) | def test_remat_work_on_fbip_3(self, device): method test_remat_work_on_fbip_4 (line 186) | def test_remat_work_on_fbip_4(self, device): method test_remat_work_on_simple_case_1 (line 200) | def test_remat_work_on_simple_case_1(self, device): method test_remat_work_on_simple_case_2 (line 229) | def test_remat_work_on_simple_case_2(self, device): method test_remat_full_and_init_constant (line 256) | def test_remat_full_and_init_constant(self, device): method test_remat_lifecycle_of_view_tensor (line 272) | def test_remat_lifecycle_of_view_tensor(self, device): method test_remat_init_constant_and_scalar (line 288) | def test_remat_init_constant_and_scalar(self, device): method test_copy (line 303) | def test_copy(self, device): method test_simple_network (line 314) | def test_simple_network(self, device): method _test_resnet18 (line 343) | def _test_resnet18(self, optimizer_fn, ddp, expected_loss): method test_resnet18_naive_sgd (line 399) | def test_resnet18_naive_sgd(self, _): method test_resnet18_naive_sgd_ddp_1n2d (line 411) | def test_resnet18_naive_sgd_ddp_1n2d(self, _): method test_resnet18_momentum_sgd (line 424) | def test_resnet18_momentum_sgd(self, _): method test_resnet18_adam (line 434) | def test_resnet18_adam(self, _): method test_copy_on_write (line 441) | def test_copy_on_write(self, _): FILE: python/oneflow/test/expensive/pytorch_alexnet.py class AlexNet (line 24) | class AlexNet(nn.Module): method __init__ (line 25) | def __init__(self, num_classes: int = 1000) -> None: method forward (line 53) | def forward(self, x: torch.Tensor) -> torch.Tensor: function alexnet (line 61) | def alexnet(pretrained: bool = False, progress: bool = True, **kwargs: A... FILE: python/oneflow/test/expensive/pytorch_convmixer.py class Residual (line 21) | class Residual(nn.Module): method __init__ (line 22) | def __init__(self, fn): method forward (line 26) | def forward(self, x): function ConvMixer (line 30) | def ConvMixer(dim, depth, kernel_size=9, patch_size=7, n_classes=1000): function convmixer_768_32_relu (line 56) | def convmixer_768_32_relu(pretrained: bool = False, progress: bool = Tru... FILE: python/oneflow/test/expensive/pytorch_convnext.py class Block (line 24) | class Block(nn.Module): method __init__ (line 36) | def __init__(self, dim, drop_path=0.0, layer_scale_init_value=1e-6): method forward (line 54) | def forward(self, x): class ConvNeXt (line 70) | class ConvNeXt(nn.Module): method __init__ (line 84) | def __init__( method _init_weights (line 137) | def _init_weights(self, m): method forward_features (line 142) | def forward_features(self, x): method forward (line 150) | def forward(self, x): class LayerNorm (line 156) | class LayerNorm(nn.Module): method __init__ (line 163) | def __init__(self, normalized_shape, eps=1e-6, data_format="channels_l... method forward (line 173) | def forward(self, x): function convnext_tiny (line 186) | def convnext_tiny(pretrained=False, in_22k=False, **kwargs): FILE: python/oneflow/test/expensive/pytorch_crossformer.py class Mlp (line 24) | class Mlp(nn.Module): method __init__ (line 25) | def __init__( method forward (line 41) | def forward(self, x): class DynamicPosBias (line 50) | class DynamicPosBias(nn.Module): method __init__ (line 51) | def __init__(self, dim, num_heads, residual): method forward (line 73) | def forward(self, biases): method flops (line 83) | def flops(self, N): class Attention (line 91) | class Attention(nn.Module): method __init__ (line 103) | def __init__( method forward (line 159) | def forward(self, x, mask=None): method extra_repr (line 210) | def extra_repr(self) -> str: method flops (line 215) | def flops(self, N): class CrossFormerBlock (line 231) | class CrossFormerBlock(nn.Module): method __init__ (line 249) | def __init__( method forward (line 305) | def forward(self, x): method extra_repr (line 339) | def extra_repr(self) -> str: method flops (line 345) | def flops(self): class PatchMerging (line 360) | class PatchMerging(nn.Module): method __init__ (line 368) | def __init__( method forward (line 394) | def forward(self, x): method extra_repr (line 413) | def extra_repr(self) -> str: method flops (line 416) | def flops(self): class Stage (line 428) | class Stage(nn.Module): method __init__ (line 447) | def __init__( method forward (line 509) | def forward(self, x): method extra_repr (line 519) | def extra_repr(self) -> str: method flops (line 522) | def flops(self): class PatchEmbed (line 531) | class PatchEmbed(nn.Module): method __init__ (line 541) | def __init__( method forward (line 575) | def forward(self, x): method flops (line 590) | def flops(self): class CrossFormer (line 610) | class CrossFormer(nn.Module): method __init__ (line 634) | def __init__( method _init_weights (line 735) | def _init_weights(self, m): method no_weight_decay (line 744) | def no_weight_decay(self): method no_weight_decay_keywords (line 747) | def no_weight_decay_keywords(self): method forward_features (line 750) | def forward_features(self, x): method forward (line 764) | def forward(self, x): method flops (line 769) | def flops(self): function _create_cross_former (line 784) | def _create_cross_former(arch, pretrained=False, progress=True, **model_... function crossformer_tiny_patch4_group7_224 (line 789) | def crossformer_tiny_patch4_group7_224(pretrained=False, progress=True, ... FILE: python/oneflow/test/expensive/pytorch_densenet.py class _DenseLayer (line 30) | class _DenseLayer(nn.Module): method __init__ (line 31) | def __init__( method bn_function (line 74) | def bn_function(self, inputs: List[Tensor]) -> Tensor: method any_requires_grad (line 82) | def any_requires_grad(self, input: List[Tensor]) -> bool: method forward (line 90) | def forward(self, input: Tensor) -> Tensor: # noqa: F811 class _DenseBlock (line 112) | class _DenseBlock(nn.ModuleDict): method __init__ (line 115) | def __init__( method forward (line 135) | def forward(self, init_features: Tensor) -> Tensor: class _Transition (line 143) | class _Transition(nn.Sequential): method __init__ (line 144) | def __init__(self, num_input_features: int, num_output_features: int) ... class DenseNet (line 161) | class DenseNet(nn.Module): method __init__ (line 176) | def __init__( method forward (line 248) | def forward(self, x: Tensor) -> Tensor: function _densenet (line 257) | def _densenet( function densenet121 (line 268) | def densenet121(progress: bool = True, **kwargs: Any) -> DenseNet: FILE: python/oneflow/test/expensive/pytorch_efficientnet.py class SqueezeExcitation (line 33) | class SqueezeExcitation(torch.nn.Module): method __init__ (line 44) | def __init__( method _scale (line 58) | def _scale(self, input: Tensor) -> Tensor: method forward (line 65) | def forward(self, input: Tensor) -> Tensor: class ConvNormActivation (line 70) | class ConvNormActivation(torch.nn.Sequential): method __init__ (line 71) | def __init__( class Conv2dNormActivation (line 120) | class Conv2dNormActivation(ConvNormActivation): method __init__ (line 137) | def __init__( function _make_divisible (line 168) | def _make_divisible(v, divisor=8, min_value=None, round_limit=0.9): class _MBConvConfig (line 178) | class _MBConvConfig: method adjust_channels (line 188) | def adjust_channels( class MBConvConfig (line 194) | class MBConvConfig(_MBConvConfig): method __init__ (line 196) | def __init__( method adjust_depth (line 224) | def adjust_depth(num_layers: int, depth_mult: float): class FusedMBConvConfig (line 228) | class FusedMBConvConfig(_MBConvConfig): method __init__ (line 230) | def __init__( class MBConv (line 253) | class MBConv(nn.Module): method __init__ (line 254) | def __init__( method forward (line 324) | def forward(self, input: Tensor) -> Tensor: class FusedMBConv (line 332) | class FusedMBConv(nn.Module): method __init__ (line 333) | def __init__( method forward (line 391) | def forward(self, input: Tensor) -> Tensor: class EfficientNet (line 399) | class EfficientNet(nn.Module): method __init__ (line 400) | def __init__( method _forward_impl (line 518) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 528) | def forward(self, x: Tensor) -> Tensor: function _efficientnet (line 532) | def _efficientnet( function _efficientnet_conf (line 545) | def _efficientnet_conf( function efficientnet_b0 (line 603) | def efficientnet_b0(progress: bool = True, **kwargs: Any) -> EfficientNet: FILE: python/oneflow/test/expensive/pytorch_ghostnet.py function _make_divisible (line 24) | def _make_divisible(v, divisor, min_value=None): class SELayer (line 40) | class SELayer(nn.Module): method __init__ (line 41) | def __init__(self, channel, reduction=4): method forward (line 50) | def forward(self, x): function depthwise_conv (line 58) | def depthwise_conv(inp, oup, kernel_size=3, stride=1, relu=False): class GhostModule (line 68) | class GhostModule(nn.Module): method __init__ (line 69) | def __init__( method forward (line 99) | def forward(self, x): class GhostBottleneck (line 106) | class GhostBottleneck(nn.Module): method __init__ (line 107) | def __init__(self, inp, hidden_dim, oup, kernel_size, stride, use_se): method forward (line 133) | def forward(self, x): class GhostNet (line 137) | class GhostNet(nn.Module): method __init__ (line 138) | def __init__(self, cfgs, num_classes=1000, width_mult=1.0): method forward (line 186) | def forward(self, x): method _initialize_weights (line 193) | def _initialize_weights(self): function ghost_net (line 202) | def ghost_net(**kwargs): FILE: python/oneflow/test/expensive/pytorch_googlenet.py class GoogLeNet (line 27) | class GoogLeNet(nn.Module): method __init__ (line 30) | def __init__( method _transform_input (line 99) | def _transform_input(self, x: Tensor) -> Tensor: method _forward (line 107) | def _forward(self, x: Tensor) -> Tuple[Tensor, Optional[Tensor], Optio... method forward (line 162) | def forward(self, x: Tensor): class Inception (line 168) | class Inception(nn.Module): method __init__ (line 169) | def __init__( method _forward (line 202) | def _forward(self, x: Tensor) -> List[Tensor]: method forward (line 211) | def forward(self, x: Tensor) -> Tensor: class InceptionAux (line 216) | class InceptionAux(nn.Module): method __init__ (line 217) | def __init__( method forward (line 233) | def forward(self, x: Tensor) -> Tensor: class BasicConv2d (line 251) | class BasicConv2d(nn.Module): method __init__ (line 252) | def __init__(self, in_channels: int, out_channels: int, **kwargs: Any)... method forward (line 257) | def forward(self, x: Tensor) -> Tensor: function googlenet (line 263) | def googlenet(progress: bool = True, **kwargs: Any) -> GoogLeNet: FILE: python/oneflow/test/expensive/pytorch_inception_v3.py class Inception3 (line 26) | class Inception3(nn.Module): method __init__ (line 27) | def __init__( method _transform_input (line 104) | def _transform_input(self, x: Tensor) -> Tensor: method _forward (line 112) | def _forward(self, x: Tensor) -> Tuple[Tensor, Optional[Tensor]]: method forward (line 166) | def forward(self, x: Tensor): class InceptionA (line 172) | class InceptionA(nn.Module): method __init__ (line 173) | def __init__( method _forward (line 193) | def _forward(self, x: Tensor) -> List[Tensor]: method forward (line 209) | def forward(self, x: Tensor) -> Tensor: class InceptionB (line 214) | class InceptionB(nn.Module): method __init__ (line 215) | def __init__( method _forward (line 227) | def _forward(self, x: Tensor) -> List[Tensor]: method forward (line 239) | def forward(self, x: Tensor) -> Tensor: class InceptionC (line 244) | class InceptionC(nn.Module): method __init__ (line 245) | def __init__( method _forward (line 269) | def _forward(self, x: Tensor) -> List[Tensor]: method forward (line 288) | def forward(self, x: Tensor) -> Tensor: class InceptionD (line 293) | class InceptionD(nn.Module): method __init__ (line 294) | def __init__( method _forward (line 308) | def _forward(self, x: Tensor) -> List[Tensor]: method forward (line 321) | def forward(self, x: Tensor) -> Tensor: class InceptionE (line 326) | class InceptionE(nn.Module): method __init__ (line 327) | def __init__( method _forward (line 346) | def _forward(self, x: Tensor) -> List[Tensor]: method forward (line 370) | def forward(self, x: Tensor) -> Tensor: class InceptionAux (line 375) | class InceptionAux(nn.Module): method __init__ (line 376) | def __init__( method forward (line 391) | def forward(self, x: Tensor) -> Tensor: class BasicConv2d (line 409) | class BasicConv2d(nn.Module): method __init__ (line 410) | def __init__(self, in_channels: int, out_channels: int, **kwargs: Any)... method forward (line 415) | def forward(self, x: Tensor) -> Tensor: function inception_v3 (line 421) | def inception_v3(progress: bool = True, **kwargs: Any) -> Inception3: FILE: python/oneflow/test/expensive/pytorch_levit.py function LeViT_128S (line 34) | def LeViT_128S(num_classes=1000, distillation=False, pretrained=False, f... class Conv2d_BN (line 47) | class Conv2d_BN(torch.nn.Sequential): method __init__ (line 48) | def __init__( method fuse (line 76) | def fuse(self): class Linear_BN (line 95) | class Linear_BN(torch.nn.Sequential): method __init__ (line 96) | def __init__(self, a, b, bn_weight_init=1, resolution=-100000): method fuse (line 109) | def fuse(self): method forward (line 119) | def forward(self, x): class BN_Linear (line 125) | class BN_Linear(torch.nn.Sequential): method __init__ (line 126) | def __init__(self, a, b, bias=True, std=0.02): method fuse (line 138) | def fuse(self): function b16 (line 156) | def b16(n, activation, resolution=224): class Residual (line 168) | class Residual(torch.nn.Module): method __init__ (line 169) | def __init__(self, m, drop): method forward (line 174) | def forward(self, x): class Attention (line 188) | class Attention(torch.nn.Module): method __init__ (line 189) | def __init__( method train (line 231) | def train(self, mode=True): method forward (line 238) | def forward(self, x): # x (B,N,C) class Subsample (line 259) | class Subsample(torch.nn.Module): method __init__ (line 260) | def __init__(self, stride, resolution): method forward (line 265) | def forward(self, x): class AttentionSubsample (line 273) | class AttentionSubsample(torch.nn.Module): method __init__ (line 274) | def __init__( method train (line 339) | def train(self, mode=True): method forward (line 346) | def forward(self, x): class LeViT (line 373) | class LeViT(torch.nn.Module): method __init__ (line 377) | def __init__( method no_weight_decay (line 491) | def no_weight_decay(self): method forward (line 494) | def forward(self, x): function model_factory (line 508) | def model_factory( FILE: python/oneflow/test/expensive/pytorch_mnasnet.py class _InvertedResidual (line 34) | class _InvertedResidual(nn.Module): method __init__ (line 35) | def __init__( method forward (line 73) | def forward(self, input: Tensor) -> Tensor: function _stack (line 80) | def _stack( function _round_to_multiple_of (line 106) | def _round_to_multiple_of(val: float, divisor: int, round_up_bias: float... function _get_depths (line 118) | def _get_depths(alpha: float) -> List[int]: class MNASNet (line 125) | class MNASNet(torch.nn.Module): method __init__ (line 140) | def __init__( method forward (line 199) | def forward(self, x: Tensor) -> Tensor: function _mnasnet (line 206) | def _mnasnet(alpha: float, progress: bool, **kwargs: Any) -> MNASNet: function mnasnet1_0 (line 211) | def mnasnet1_0(progress: bool = True, **kwargs: Any) -> MNASNet: FILE: python/oneflow/test/expensive/pytorch_poolformer.py class PatchEmbed (line 25) | class PatchEmbed(nn.Module): method __init__ (line 32) | def __init__( method forward (line 50) | def forward(self, x): class LayerNormChannel (line 56) | class LayerNormChannel(nn.Module): method __init__ (line 62) | def __init__(self, num_channels, eps=1e-05): method forward (line 68) | def forward(self, x): class GroupNorm (line 78) | class GroupNorm(nn.GroupNorm): method __init__ (line 84) | def __init__(self, num_channels, **kwargs): class Pooling (line 88) | class Pooling(nn.Module): method __init__ (line 94) | def __init__(self, pool_size=3): method forward (line 100) | def forward(self, x): class Mlp (line 104) | class Mlp(nn.Module): method __init__ (line 110) | def __init__( method _init_weights (line 127) | def _init_weights(self, m): method forward (line 133) | def forward(self, x): class PoolFormerBlock (line 142) | class PoolFormerBlock(nn.Module): method __init__ (line 157) | def __init__( method forward (line 194) | def forward(self, x): function basic_blocks (line 209) | def basic_blocks( class PoolFormer (line 249) | class PoolFormer(nn.Module): method __init__ (line 267) | def __init__( method cls_init_weights (line 371) | def cls_init_weights(self, m): method get_classifier (line 377) | def get_classifier(self): method reset_classifier (line 380) | def reset_classifier(self, num_classes): method forward_embeddings (line 386) | def forward_embeddings(self, x): method forward_tokens (line 390) | def forward_tokens(self, x): method forward (line 404) | def forward(self, x): function poolformer_s12 (line 418) | def poolformer_s12(pretrained=False, **kwargs): FILE: python/oneflow/test/expensive/pytorch_pvt.py class Mlp (line 26) | class Mlp(nn.Module): method __init__ (line 27) | def __init__( method forward (line 43) | def forward(self, x): class Attention (line 52) | class Attention(nn.Module): method __init__ (line 53) | def __init__( method forward (line 84) | def forward(self, x, H, W): class Block (line 120) | class Block(nn.Module): method __init__ (line 121) | def __init__( method forward (line 157) | def forward(self, x, H, W): class PatchEmbed (line 164) | class PatchEmbed(nn.Module): method __init__ (line 168) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=... method forward (line 184) | def forward(self, x): class PyramidVisionTransformer (line 194) | class PyramidVisionTransformer(nn.Module): method __init__ (line 195) | def __init__( method _init_weights (line 280) | def _init_weights(self, m): method no_weight_decay (line 289) | def no_weight_decay(self): method get_classifier (line 293) | def get_classifier(self): method reset_classifier (line 296) | def reset_classifier(self, num_classes, global_pool=""): method _get_pos_embed (line 302) | def _get_pos_embed(self, pos_embed, patch_embed, H, W): method forward_features (line 318) | def forward_features(self, x): method forward (line 346) | def forward(self, x): function pvt_tiny (line 353) | def pvt_tiny(pretrained=False, **kwargs): FILE: python/oneflow/test/expensive/pytorch_res2net.py class Bottle2neck (line 24) | class Bottle2neck(nn.Module): method __init__ (line 27) | def __init__( method forward (line 82) | def forward(self, x): class Res2Net (line 118) | class Res2Net(nn.Module): method __init__ (line 119) | def __init__(self, block, layers, baseWidth=26, scale=4, num_classes=1... method _make_layer (line 142) | def _make_layer(self, block, planes, blocks, stride=1): method forward (line 176) | def forward(self, x): function res2net50 (line 194) | def res2net50(pretrained=False, **kwargs): FILE: python/oneflow/test/expensive/pytorch_resmlp.py class Mlp (line 24) | class Mlp(nn.Module): method __init__ (line 28) | def __init__( method forward (line 47) | def forward(self, x): class PatchEmbed (line 56) | class PatchEmbed(nn.Module): method __init__ (line 60) | def __init__( method forward (line 83) | def forward(self, x): class Affine (line 100) | class Affine(nn.Module): method __init__ (line 101) | def __init__(self, dim): method forward (line 106) | def forward(self, x): class layers_scale_mlp_blocks (line 110) | class layers_scale_mlp_blocks(nn.Module): method __init__ (line 111) | def __init__( method forward (line 134) | def forward(self, x): class resmlp_models (line 142) | class resmlp_models(nn.Module): method __init__ (line 143) | def __init__( method _init_weights (line 193) | def _init_weights(self, m): method get_classifier (line 202) | def get_classifier(self): method reset_classifier (line 205) | def reset_classifier(self, num_classes, global_pool=""): method forward_features (line 211) | def forward_features(self, x): method forward (line 224) | def forward(self, x): function resmlp_12 (line 230) | def resmlp_12(pretrained=False, dist=False, **kwargs): FILE: python/oneflow/test/expensive/pytorch_resnet.py function conv3x3 (line 36) | def conv3x3( function conv1x1 (line 52) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: class BasicBlock (line 57) | class BasicBlock(nn.Module): method __init__ (line 60) | def __init__( method forward (line 87) | def forward(self, x: Tensor) -> Tensor: class Bottleneck (line 106) | class Bottleneck(nn.Module): method __init__ (line 115) | def __init__( method forward (line 141) | def forward(self, x: Tensor) -> Tensor: class ResNet (line 164) | class ResNet(nn.Module): method __init__ (line 165) | def __init__( method _make_layer (line 230) | def _make_layer( method _forward_impl (line 278) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 296) | def forward(self, x: Tensor) -> Tensor: function _resnet (line 300) | def _resnet( function resnet18 (line 312) | def resnet18(pretrained: bool = False, progress: bool = True, **kwargs: ... function resnet34 (line 322) | def resnet34(pretrained: bool = False, progress: bool = True, **kwargs: ... function resnet50 (line 332) | def resnet50(pretrained: bool = False, progress: bool = True, **kwargs: ... function resnet101 (line 342) | def resnet101(pretrained: bool = False, progress: bool = True, **kwargs:... function resnet152 (line 354) | def resnet152(pretrained: bool = False, progress: bool = True, **kwargs:... function resnext50_32x4d (line 366) | def resnext50_32x4d( function resnext101_32x8d (line 382) | def resnext101_32x8d( function wide_resnet50_2 (line 398) | def wide_resnet50_2( function wide_resnet101_2 (line 417) | def wide_resnet101_2( FILE: python/oneflow/test/expensive/pytorch_rexnet.py function silu (line 27) | def silu(x, inplace=False): class SiLU (line 31) | class SiLU(nn.Module): method __init__ (line 32) | def __init__(self, inplace=True): method forward (line 36) | def forward(self, x): function ConvBNAct (line 40) | def ConvBNAct( function ConvBNSiLU (line 61) | def ConvBNSiLU(out, in_channels, channels, kernel=1, stride=1, pad=0, nu... class SE (line 71) | class SE(nn.Module): method __init__ (line 72) | def __init__(self, in_channels, channels, se_ratio=12): method forward (line 83) | def forward(self, x): class LinearBottleneck (line 89) | class LinearBottleneck(nn.Module): method __init__ (line 90) | def __init__( method forward (line 125) | def forward(self, x): class ReXNetV1 (line 133) | class ReXNetV1(nn.Module): method __init__ (line 134) | def __init__( method extract_features (line 218) | def extract_features(self, x): method forward (line 221) | def forward(self, x): function _create_rexnetv1 (line 227) | def _create_rexnetv1(arch, pretrained=False, progress=True, **model_kwar... function rexnetv1_1_0 (line 232) | def rexnetv1_1_0(pretrained=False, progress=True, **kwargs): FILE: python/oneflow/test/expensive/pytorch_rexnetv1_lite.py function _make_divisible (line 26) | def _make_divisible(channel_size, divisor=None, min_value=None): function _add_conv (line 47) | def _add_conv( class LinearBottleneck (line 70) | class LinearBottleneck(nn.Module): method __init__ (line 71) | def __init__( method forward (line 123) | def forward(self, x): class ReXNetV1_lite (line 131) | class ReXNetV1_lite(nn.Module): method __init__ (line 132) | def __init__( method forward (line 231) | def forward(self, x): function _create_rexnet_lite (line 238) | def _create_rexnet_lite(arch, pretrained=False, progress=True, **model_k... function rexnet_lite_1_0 (line 243) | def rexnet_lite_1_0(pretrained=False, progress=True, **kwargs): FILE: python/oneflow/test/expensive/pytorch_senet.py class SEModule (line 24) | class SEModule(nn.Module): method __init__ (line 25) | def __init__(self, channels, reduction): method forward (line 33) | def forward(self, x): class Bottleneck (line 43) | class Bottleneck(nn.Module): method forward (line 48) | def forward(self, x): class SEBottleneck (line 71) | class SEBottleneck(Bottleneck): method __init__ (line 78) | def __init__(self, inplanes, planes, groups, reduction, stride=1, down... class SEResNetBottleneck (line 100) | class SEResNetBottleneck(Bottleneck): method __init__ (line 109) | def __init__(self, inplanes, planes, groups, reduction, stride=1, down... class SEResNeXtBottleneck (line 127) | class SEResNeXtBottleneck(Bottleneck): method __init__ (line 134) | def __init__( class SENet (line 166) | class SENet(nn.Module): method __init__ (line 167) | def __init__( method _make_layer (line 295) | def _make_layer( method features (line 330) | def features(self, x): method logits (line 338) | def logits(self, x): method forward (line 346) | def forward(self, x): function senet154 (line 352) | def senet154(num_classes=1000, pretrained="imagenet"): FILE: python/oneflow/test/expensive/pytorch_shufflenetv2.py function channel_shuffle (line 28) | def channel_shuffle(x: Tensor, groups: int) -> Tensor: class InvertedResidual (line 43) | class InvertedResidual(nn.Module): method __init__ (line 44) | def __init__(self, inp: int, oup: int, stride: int) -> None: method depthwise_conv (line 104) | def depthwise_conv( method forward (line 114) | def forward(self, x: Tensor) -> Tensor: class ShuffleNetV2 (line 126) | class ShuffleNetV2(nn.Module): method __init__ (line 127) | def __init__( method _forward_impl (line 175) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 187) | def forward(self, x: Tensor) -> Tensor: function _shufflenetv2 (line 191) | def _shufflenetv2(progress: bool, *args: Any, **kwargs: Any,) -> Shuffle... function shufflenet_v2_x2_0 (line 196) | def shufflenet_v2_x2_0(progress: bool = True, **kwargs: Any) -> ShuffleN... FILE: python/oneflow/test/expensive/pytorch_squeezenet.py class Fire (line 25) | class Fire(nn.Module): method __init__ (line 26) | def __init__( method forward (line 44) | def forward(self, x: torch.Tensor) -> torch.Tensor: class SqueezeNet (line 55) | class SqueezeNet(nn.Module): method __init__ (line 56) | def __init__( method forward (line 119) | def forward(self, x: torch.Tensor) -> torch.Tensor: function _squeezenet (line 125) | def _squeezenet(version: str, progress: bool, **kwargs: Any,) -> Squeeze... function squeezenet1_1 (line 130) | def squeezenet1_1(progress: bool = True, **kwargs: Any) -> SqueezeNet: FILE: python/oneflow/test/expensive/pytorch_swin_transformer.py class Mlp (line 21) | class Mlp(nn.Module): method __init__ (line 22) | def __init__( method forward (line 38) | def forward(self, x): function window_partition (line 47) | def window_partition(x, window_size): function window_reverse (line 63) | def window_reverse(windows, window_size, H, W): class WindowAttention (line 81) | class WindowAttention(nn.Module): method __init__ (line 94) | def __init__( method forward (line 142) | def forward(self, x, mask=None): method extra_repr (line 192) | def extra_repr(self) -> str: method flops (line 195) | def flops(self, N): class SwinTransformerBlock (line 209) | class SwinTransformerBlock(nn.Module): method __init__ (line 227) | def __init__( method forward (line 312) | def forward(self, x): method extra_repr (line 361) | def extra_repr(self) -> str: method flops (line 367) | def flops(self): class PatchMerging (line 382) | class PatchMerging(nn.Module): method __init__ (line 390) | def __init__(self, input_resolution, dim, norm_layer=nn.LayerNorm): method forward (line 397) | def forward(self, x): method extra_repr (line 420) | def extra_repr(self) -> str: method flops (line 423) | def flops(self): class BasicLayer (line 430) | class BasicLayer(nn.Module): method __init__ (line 449) | def __init__( method forward (line 504) | def forward(self, x): method extra_repr (line 511) | def extra_repr(self) -> str: method flops (line 514) | def flops(self): class PatchEmbed (line 523) | class PatchEmbed(nn.Module): method __init__ (line 533) | def __init__( method forward (line 559) | def forward(self, x): method flops (line 570) | def flops(self): class SwinTransformer (line 584) | class SwinTransformer(nn.Module): method __init__ (line 609) | def __init__( method _init_weights (line 701) | def _init_weights(self, m): method no_weight_decay (line 710) | def no_weight_decay(self): method no_weight_decay_keywords (line 713) | def no_weight_decay_keywords(self): method forward_features (line 716) | def forward_features(self, x): method forward (line 730) | def forward(self, x): method flops (line 735) | def flops(self): function _create_swin_transformer (line 750) | def _create_swin_transformer(arch, pretrained=False, progress=True, **mo... function swin_tiny_patch4_window7_224 (line 755) | def swin_tiny_patch4_window7_224(pretrained=False, progress=True, **kwar... FILE: python/oneflow/test/expensive/pytorch_uniformer.py class Mlp (line 26) | class Mlp(nn.Module): method __init__ (line 27) | def __init__( method forward (line 43) | def forward(self, x): class CMlp (line 52) | class CMlp(nn.Module): method __init__ (line 53) | def __init__( method forward (line 69) | def forward(self, x): class Attention (line 78) | class Attention(nn.Module): method __init__ (line 79) | def __init__( method forward (line 99) | def forward(self, x): class CBlock (line 122) | class CBlock(nn.Module): method __init__ (line 123) | def __init__( method forward (line 153) | def forward(self, x): class SABlock (line 160) | class SABlock(nn.Module): method __init__ (line 161) | def __init__( method forward (line 207) | def forward(self, x): class head_embedding (line 221) | class head_embedding(nn.Module): method __init__ (line 222) | def __init__(self, in_channels, out_channels): method forward (line 245) | def forward(self, x): class middle_embedding (line 250) | class middle_embedding(nn.Module): method __init__ (line 251) | def __init__(self, in_channels, out_channels): method forward (line 265) | def forward(self, x): class PatchEmbed (line 270) | class PatchEmbed(nn.Module): method __init__ (line 274) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=... method forward (line 287) | def forward(self, x): class UniFormer (line 301) | class UniFormer(nn.Module): method __init__ (line 307) | def __init__( method _init_weights (line 480) | def _init_weights(self, m): method no_weight_decay (line 489) | def no_weight_decay(self): method get_classifier (line 492) | def get_classifier(self): method reset_classifier (line 495) | def reset_classifier(self, num_classes, global_pool=""): method forward_features (line 501) | def forward_features(self, x): method forward (line 519) | def forward(self, x): function uniformer_small (line 526) | def uniformer_small(pretrained=True, **kwargs): FILE: python/oneflow/test/expensive/pytroch_mlp_mixer.py class Mlp (line 25) | class Mlp(nn.Module): method __init__ (line 29) | def __init__( method forward (line 48) | def forward(self, x): class PatchEmbed (line 57) | class PatchEmbed(nn.Module): method __init__ (line 61) | def __init__( method forward (line 84) | def forward(self, x): function named_apply (line 101) | def named_apply( class GatedMlp (line 120) | class GatedMlp(nn.Module): method __init__ (line 124) | def __init__( method forward (line 152) | def forward(self, x): class MixerBlock (line 162) | class MixerBlock(nn.Module): method __init__ (line 167) | def __init__( method forward (line 186) | def forward(self, x): class Affine (line 194) | class Affine(nn.Module): method __init__ (line 195) | def __init__(self, dim): method forward (line 200) | def forward(self, x): class ResBlock (line 204) | class ResBlock(nn.Module): method __init__ (line 209) | def __init__( method forward (line 231) | def forward(self, x): class SpatialGatingUnit (line 239) | class SpatialGatingUnit(nn.Module): method __init__ (line 244) | def __init__(self, dim, seq_len, norm_layer=nn.LayerNorm): method init_weights (line 250) | def init_weights(self): method forward (line 255) | def forward(self, x): class SpatialGatingBlock (line 262) | class SpatialGatingBlock(nn.Module): method __init__ (line 267) | def __init__( method forward (line 287) | def forward(self, x): class MlpMixer (line 292) | class MlpMixer(nn.Module): method __init__ (line 293) | def __init__( method init_weights (line 350) | def init_weights(self, nlhb=False): method group_matcher (line 356) | def group_matcher(self, coarse=False): method set_grad_checkpointing (line 362) | def set_grad_checkpointing(self, enable=True): method get_classifier (line 365) | def get_classifier(self): method reset_classifier (line 368) | def reset_classifier(self, num_classes, global_pool=None): method forward_features (line 377) | def forward_features(self, x): method forward (line 383) | def forward(self, x): function _init_weights (line 391) | def _init_weights(module: nn.Module, name: str, head_bias: float = 0.0, ... function mixer_s32_224 (line 425) | def mixer_s32_224(pretrained=False, **kwargs): FILE: python/oneflow/test/expensive/resnet50_model.py class FakeBN (line 23) | class FakeBN(nn.Module): method __init__ (line 26) | def __init__( method forward (line 53) | def forward(self, input): function conv3x3 (line 57) | def conv3x3( function conv1x1 (line 73) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1) -> nn.Conv2d: class BasicBlock (line 78) | class BasicBlock(nn.Module): method __init__ (line 81) | def __init__( method forward (line 107) | def forward(self, x: Tensor) -> Tensor: class Bottleneck (line 121) | class Bottleneck(nn.Module): method __init__ (line 124) | def __init__( method forward (line 149) | def forward(self, x: Tensor) -> Tensor: class ResNet (line 166) | class ResNet(nn.Module): method __init__ (line 167) | def __init__( method _make_layer (line 225) | def _make_layer( method _forward_impl (line 271) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 285) | def forward(self, x: Tensor) -> Tensor: function _resnet (line 289) | def _resnet( function resnet50 (line 299) | def resnet50(**kwargs: Any) -> ResNet: FILE: python/oneflow/test/expensive/test_compatibility.py class TestApiCompatibility (line 22) | class TestApiCompatibility(flow.unittest.TestCase): method test_alexnet_compatibility (line 23) | def test_alexnet_compatibility(test_case): method test_resnet50_compatibility (line 28) | def test_resnet50_compatibility(test_case): method test_convmixer_compatibility (line 37) | def test_convmixer_compatibility(test_case): method test_densenet_compatibility (line 42) | def test_densenet_compatibility(test_case): method test_ghostnet_compatibility (line 47) | def test_ghostnet_compatibility(test_case): method test_googlenet_compatibility (line 52) | def test_googlenet_compatibility(test_case): method test_inception_v3_compatibility (line 57) | def test_inception_v3_compatibility(test_case): method test_mnasnet_compatibility (line 62) | def test_mnasnet_compatibility(test_case): method test_shufflenetv2_compatibility (line 83) | def test_shufflenetv2_compatibility(test_case): method test_squeezenet_compatibility (line 88) | def test_squeezenet_compatibility(test_case): method test_convnext_compatibility (line 97) | def test_convnext_compatibility(test_case): method test_levit_compatibility (line 117) | def test_levit_compatibility(test_case): method test_poolformer_compatibility (line 127) | def test_poolformer_compatibility(test_case): method test_pvt_compatibility (line 132) | def test_pvt_compatibility(test_case): method test_resmlp_compatibility (line 137) | def test_resmlp_compatibility(test_case): method test_uniformer_compatibility (line 142) | def test_uniformer_compatibility(test_case): method test_senet_compatibility (line 158) | def test_senet_compatibility(test_case): FILE: python/oneflow/test/expensive/test_conv3d.py class TestConv3DModule (line 24) | class TestConv3DModule(flow.unittest.TestCase): method test_nn_functional_conv3d (line 26) | def test_nn_functional_conv3d(test_case): method test_conv3d_with_random_data (line 35) | def test_conv3d_with_random_data(test_case): method test_conv3d_group_with_random_data (line 57) | def test_conv3d_group_with_random_data(test_case): FILE: python/oneflow/test/expensive/test_convtranspose.py function _test_convtranspose1d_bias_false (line 29) | def _test_convtranspose1d_bias_false(test_case, device): function _test_convtranspose1d_bias_true (line 57) | def _test_convtranspose1d_bias_true(test_case, device): function _test_convtranspose1d_group_bias_false (line 87) | def _test_convtranspose1d_group_bias_false(test_case, device): function _test_convtranspose1d_group_bias_true (line 116) | def _test_convtranspose1d_group_bias_true(test_case, device): function _test_convtranspose1d_group_large_out_channel (line 162) | def _test_convtranspose1d_group_large_out_channel(test_case, device): function _test_convtranspose1d_group_large_in_channel (line 214) | def _test_convtranspose1d_group_large_in_channel(test_case, device): class TestConvTranspose (line 266) | class TestConvTranspose(flow.unittest.TestCase): method test_ConvTranspose1d (line 267) | def test_ConvTranspose1d(test_case): method test_ConvTranspose1d_ (line 282) | def test_ConvTranspose1d_(test_case): method test_deconv1d_group_with_random_data (line 303) | def test_deconv1d_group_with_random_data(test_case): method test_ConvTranspose3d_ (line 326) | def test_ConvTranspose3d_(test_case): method test_deconv3d_group_with_random_data (line 347) | def test_deconv3d_group_with_random_data(test_case): method test_functional_conv_transpose1d (line 371) | def test_functional_conv_transpose1d(test_case): method test_functional_conv_transpose2d (line 381) | def test_functional_conv_transpose2d(test_case): method test_functional_conv_transpose3d (line 391) | def test_functional_conv_transpose3d(test_case): FILE: python/oneflow/test/expensive/test_dynamic_allocation_gradient_shuffle.py function round_half_away_from_zero (line 28) | def round_half_away_from_zero(x): function _test_embedding_gradient_shuffle (line 37) | def _test_embedding_gradient_shuffle(test_case, enable_quantize, fp16, e... function _test_unique_key_value (line 155) | def _test_unique_key_value(test_case, has_table_id, num_tables): class DataShuffleTestCase (line 203) | class DataShuffleTestCase(flow.unittest.TestCase): method test_embedding_gradient_shuffle (line 204) | def test_embedding_gradient_shuffle(test_case): FILE: python/oneflow/test/expensive/test_einsum.py class TestEinsum (line 24) | class TestEinsum(flow.unittest.TestCase): method test_einsum_matrix_transpose (line 26) | def test_einsum_matrix_transpose(test_case): method test_einsum_eltwise_multiply (line 33) | def test_einsum_eltwise_multiply(test_case): method test_einsum_get_diagonal (line 43) | def test_einsum_get_diagonal(test_case): method test_einsum_batch_permute (line 51) | def test_einsum_batch_permute(test_case): method test_einsum_reduce_sum (line 65) | def test_einsum_reduce_sum(test_case): method test_einsum_matrix_column_sum (line 72) | def test_einsum_matrix_column_sum(test_case): method test_einsum_matrix_vector_multiply (line 79) | def test_einsum_matrix_vector_multiply(test_case): method test_einsum_matmul (line 90) | def test_einsum_matmul(test_case): method test_einsum_vector_inner_product (line 102) | def test_einsum_vector_inner_product(test_case): method test_einsum_eltwise_mul_then_reduce_sum (line 112) | def test_einsum_eltwise_mul_then_reduce_sum(test_case): method test_einsum_vector_outer_product (line 123) | def test_einsum_vector_outer_product(test_case): method test_einsum_batch_matmul (line 132) | def test_einsum_batch_matmul(test_case): method test_einsum_tensor_contraction (line 142) | def test_einsum_tensor_contraction(test_case): method test_einsum_bilinear_transformation (line 161) | def test_einsum_bilinear_transformation(test_case): method test_einsum_0_size_tensor (line 173) | def test_einsum_0_size_tensor(test_case): method test_einsum_tensor_contraction2 (line 183) | def test_einsum_tensor_contraction2(test_case): method test_einsum_eltwise_mul_sum_row (line 194) | def test_einsum_eltwise_mul_sum_row(test_case): method test_einsum_matmul2 (line 205) | def test_einsum_matmul2(test_case): method test_einsum_attention (line 214) | def test_einsum_attention(test_case): method test_einsum_batch_matmul2 (line 229) | def test_einsum_batch_matmul2(test_case): method test_einsum_batch_matrix_vector_multiply (line 245) | def test_einsum_batch_matrix_vector_multiply(test_case): method test_einsum_batch_matmul3 (line 258) | def test_einsum_batch_matmul3(test_case): method test_einsum_batch_matmul4 (line 270) | def test_einsum_batch_matmul4(test_case): method test_einsum_alphaflod_usecase1 (line 282) | def test_einsum_alphaflod_usecase1(test_case): method test_einsum_alphaflod_usecase2 (line 292) | def test_einsum_alphaflod_usecase2(test_case): method test_einsum_alphaflod_usecase3 (line 302) | def test_einsum_alphaflod_usecase3(test_case): method test_einsum_alphaflod_usecase4 (line 312) | def test_einsum_alphaflod_usecase4(test_case): method test_einsum_alphaflod_usecase5 (line 322) | def test_einsum_alphaflod_usecase5(test_case): method test_einsum_alphaflod_usecase6 (line 333) | def test_einsum_alphaflod_usecase6(test_case): method test_einsum_alphaflod_usecase7 (line 343) | def test_einsum_alphaflod_usecase7(test_case): method test_einsum_alphaflod_usecase8 (line 355) | def test_einsum_alphaflod_usecase8(test_case): method test_einsum_alphaflod_usecase9 (line 368) | def test_einsum_alphaflod_usecase9(test_case): method test_einsum_alphaflod_usecase10 (line 379) | def test_einsum_alphaflod_usecase10(test_case): method test_einsum_alphaflod_usecase11 (line 394) | def test_einsum_alphaflod_usecase11(test_case): method test_einsum_ellipsis_usecase1 (line 407) | def test_einsum_ellipsis_usecase1(test_case): method test_einsum_ellipsis_usecase2 (line 420) | def test_einsum_ellipsis_usecase2(test_case): method test_einsum_ellipsis_usecase3 (line 430) | def test_einsum_ellipsis_usecase3(test_case): method test_einsum_ellipsis_usecase4 (line 443) | def test_einsum_ellipsis_usecase4(test_case): method test_einsum_ellipsis_usecase5 (line 455) | def test_einsum_ellipsis_usecase5(test_case): method test_einsum_ellipsis_usecase6 (line 468) | def test_einsum_ellipsis_usecase6(test_case): method test_einsum_ellipsis_usecase7 (line 481) | def test_einsum_ellipsis_usecase7(test_case): method test_einsum_other_usecase1 (line 494) | def test_einsum_other_usecase1(test_case): method test_einsum_other_usecase2 (line 506) | def test_einsum_other_usecase2(test_case): method test_einsum_other_usecase3 (line 520) | def test_einsum_other_usecase3(test_case): method test_einsum_fastfold_usecase1 (line 532) | def test_einsum_fastfold_usecase1(test_case): method test_einsum_fastfold_usecase2 (line 547) | def test_einsum_fastfold_usecase2(test_case): method test_einsum_openfold_usecase1 (line 561) | def test_einsum_openfold_usecase1(test_case): method test_einsum_openfold_usecase2 (line 574) | def test_einsum_openfold_usecase2(test_case): method test_einsum_openfold_usecase3 (line 588) | def test_einsum_openfold_usecase3(test_case): method test_einsum_openfold_usecase4 (line 602) | def test_einsum_openfold_usecase4(test_case): method test_einsum_openfold_usecase5 (line 616) | def test_einsum_openfold_usecase5(test_case): FILE: python/oneflow/test/expensive/test_global_tensor_offload.py function _test_global_tensor_offload_d2h (line 35) | def _test_global_tensor_offload_d2h(test_case, input, tensor_mem): function _test_global_tensor_load_h2d (line 80) | def _test_global_tensor_load_h2d(test_case, input, tensor_mem): function _get_specific_global_tensor_mem (line 124) | def _get_specific_global_tensor_mem(placement, sbp, tensor): class TestGlobalTensorOffload (line 131) | class TestGlobalTensorOffload(flow.unittest.TestCase): method test_global_tensor_offload_and_load_2d (line 134) | def test_global_tensor_offload_and_load_2d(test_case): method test_global_tensor_offload_and_load_4d (line 151) | def test_global_tensor_offload_and_load_4d(test_case): method test_global_tensor_offload_and_load_2d_cpu_mem (line 168) | def test_global_tensor_offload_and_load_2d_cpu_mem(test_case): method test_global_param_offload_and_load (line 201) | def test_global_param_offload_and_load(test_case): FILE: python/oneflow/test/expensive/test_graph_multi_graph_v2.py function _reset_session (line 27) | def _reset_session(): function _with_new_session (line 34) | def _with_new_session(fn): function _test_linear_multi_graph_share (line 51) | def _test_linear_multi_graph_share(test_case, device, with_reshape): function _get_state_dict_tensor_size (line 133) | def _get_state_dict_tensor_size(sd): function _test_linear_multi_graph_save (line 154) | def _test_linear_multi_graph_save(return_dict, device, with_reshape, wit... function _test_linear_multi_graph_load (line 251) | def _test_linear_multi_graph_load( function _graph_save (line 340) | def _graph_save(return_dict, filename, with_eager): function _graph_load (line 352) | def _graph_load(return_dict, filename): function _graph_load_to_another_device (line 361) | def _graph_load_to_another_device(return_dict, filename): function _test_linear_multi_graph_save_load_gpu (line 373) | def _test_linear_multi_graph_save_load_gpu(test_case, with_eager): function _test_load_to_another_device (line 398) | def _test_load_to_another_device(test_case, with_eager): class TestLinearMultiGraph (line 427) | class TestLinearMultiGraph(oneflow.unittest.TestCase): method test_linear_multi_graph_share_gpu (line 428) | def test_linear_multi_graph_share_gpu(test_case): method test_linear_reshape_multi_graph_share_gpu (line 431) | def test_linear_reshape_multi_graph_share_gpu(test_case): method test_linear_multi_graph_save_load_gpu_with_share (line 434) | def test_linear_multi_graph_save_load_gpu_with_share(test_case): method test_linear_multi_graph_save_load_gpu_with_share_without_eager (line 437) | def test_linear_multi_graph_save_load_gpu_with_share_without_eager(tes... method test_load_to_another_device (line 440) | def test_load_to_another_device(test_case): FILE: python/oneflow/test/expensive/test_id_shuffle.py function _test_id_shuffle (line 28) | def _test_id_shuffle(test_case, has_table_id, num_tables): function round_half_away_from_zero (line 89) | def round_half_away_from_zero(x): function embedding_shuffle_quantize (line 98) | def embedding_shuffle_quantize(np_data, np_dtype): function _test_embedding_shuffle (line 122) | def _test_embedding_shuffle(test_case, dtype, enable_quantize): function _test_embedding_gradient_shuffle (line 184) | def _test_embedding_gradient_shuffle(test_case, enable_quantize, fp16, e... function _test_unique_key_value (line 302) | def _test_unique_key_value(test_case, has_table_id, num_tables): class DataShuffleTestCase (line 352) | class DataShuffleTestCase(flow.unittest.TestCase): method test_id_shuffle (line 353) | def test_id_shuffle(test_case): method test_embedding_shuffle (line 360) | def test_embedding_shuffle(test_case): method test_embedding_gradient_shuffle (line 368) | def test_embedding_gradient_shuffle(test_case): method test_unique_key_value (line 376) | def test_unique_key_value(test_case): FILE: python/oneflow/test/expensive/test_id_shuffle_global.py function get_tensors (line 31) | def get_tensors(batch_size, num_tables): function _test_id_shuffle (line 46) | def _test_id_shuffle(test_case, has_table_id, num_tables): function round_half_away_from_zero (line 131) | def round_half_away_from_zero(x): function embedding_shuffle_quantize (line 140) | def embedding_shuffle_quantize(np_data, np_dtype): function _test_embedding_shuffle (line 165) | def _test_embedding_shuffle(test_case, dtype, enable_quantize): function _test_embedding_gradient_shuffle (line 224) | def _test_embedding_gradient_shuffle(test_case, enable_quantize, fp16, e... class DataShuffleTestCase (line 349) | class DataShuffleTestCase(flow.unittest.TestCase): method test_id_shuffle (line 350) | def test_id_shuffle(test_case): method test_embedding_shuffle (line 357) | def test_embedding_shuffle(test_case): method test_embedding_gradient_shuffle (line 365) | def test_embedding_gradient_shuffle(test_case): FILE: python/oneflow/test/expensive/test_layernorm.py function _test_layernorm (line 43) | def _test_layernorm(test_case, device): function _test_layernorm_v2 (line 63) | def _test_layernorm_v2(test_case, device): function _test_layernorm_v3 (line 83) | def _test_layernorm_v3(test_case, device): function _test_layernorm_backward (line 103) | def _test_layernorm_backward(test_case, device): class TestLayerNorm (line 131) | class TestLayerNorm(flow.unittest.TestCase): method test_layernorm (line 132) | def test_layernorm(test_case): method test_layernorm_with_random_data_warp (line 145) | def test_layernorm_with_random_data_warp(test_case): method test_layernorm_with_random_data_shared_mem (line 164) | def test_layernorm_with_random_data_shared_mem(test_case): method test_layernorm_with_random_data_uncached (line 183) | def test_layernorm_with_random_data_uncached(test_case): method test_layernorm_without_affine (line 202) | def test_layernorm_without_affine(test_case): FILE: python/oneflow/test/expensive/test_oneembedding.py class OneEmbedding (line 29) | class OneEmbedding(nn.Module): method __init__ (line 30) | def __init__( method forward (line 64) | def forward(self, ids): class TestModule (line 68) | class TestModule(nn.Module): method __init__ (line 69) | def __init__( method forward (line 83) | def forward(self, inputs) -> flow.Tensor: class TrainGraph (line 89) | class TrainGraph(flow.nn.Graph): method __init__ (line 90) | def __init__( method build (line 100) | def build(self, labels, features): function _test_one_embedding (line 108) | def _test_one_embedding( class OneEmbeddingTestCase (line 149) | class OneEmbeddingTestCase(flow.unittest.TestCase): method test_one_embedding (line 150) | def test_one_embedding(test_case): FILE: python/oneflow/test/expensive/test_oneembedding_padding_idx.py class OneEmbedding (line 30) | class OneEmbedding(nn.Module): method __init__ (line 31) | def __init__( method forward (line 67) | def forward(self, ids): class TestModule (line 71) | class TestModule(nn.Module): method __init__ (line 72) | def __init__( method forward (line 91) | def forward(self, inputs) -> flow.Tensor: class TrainGraph (line 96) | class TrainGraph(flow.nn.Graph): method __init__ (line 97) | def __init__( method build (line 107) | def build(self, labels, features): function _test_one_embedding_padding_idx (line 114) | def _test_one_embedding_padding_idx( class OneEmbeddingWithPaddingIdxTestCase (line 189) | class OneEmbeddingWithPaddingIdxTestCase(flow.unittest.TestCase): method test_one_embedding_padding_idx (line 190) | def test_one_embedding_padding_idx(test_case): FILE: python/oneflow/test/expensive/test_permute.py function _test_permute_impl (line 29) | def _test_permute_impl(test_case, device): function _test_tensor_permute_impl (line 45) | def _test_tensor_permute_impl(test_case, device): class TestPermute (line 68) | class TestPermute(flow.unittest.TestCase): method test_permute (line 69) | def test_permute(test_case): method test_torch_permute4d_with_random_data (line 77) | def test_torch_permute4d_with_random_data(test_case): method test_torch_permute4d_with_random_0dim_data (line 88) | def test_torch_permute4d_with_random_0dim_data(test_case): method test_permute5d_tensor_with_random_data (line 97) | def test_permute5d_tensor_with_random_data(test_case): method test_permute4d_tensor_with_random_data (line 114) | def test_permute4d_tensor_with_random_data(test_case): method test_permute4d_tensor_with_stride (line 130) | def test_permute4d_tensor_with_stride(test_case): method test_permute3d_tensor_with_random_data (line 149) | def test_permute3d_tensor_with_random_data(test_case): method test_permute4d_tensor_bool_with_random_data (line 164) | def test_permute4d_tensor_bool_with_random_data(test_case): FILE: python/oneflow/test/expensive/test_remat.py class TestRemat (line 24) | class TestRemat(flow.unittest.TestCase): method test_remat_in_single_threaded_vm (line 25) | def test_remat_in_single_threaded_vm(test_case): FILE: python/oneflow/test/expensive/test_resnet50_with_bn.py class TestResNet50 (line 28) | class TestResNet50(flow.unittest.TestCase): method test_resnet50_with_batchnorm (line 29) | def test_resnet50_with_batchnorm(test_case): FILE: python/oneflow/test/expensive/test_resnet50_without_bn.py class TestResNet50 (line 29) | class TestResNet50(flow.unittest.TestCase): method test_resnet50_without_batchnorm (line 30) | def test_resnet50_without_batchnorm(test_case): FILE: python/oneflow/test/expensive/test_rnn.py class TestRNNModules (line 24) | class TestRNNModules(flow.unittest.TestCase): method test_rnn (line 26) | def test_rnn(test_case): method test_lstm (line 50) | def test_lstm(test_case): method test_gru (line 75) | def test_gru(test_case): FILE: python/oneflow/test/expensive/test_rnn_cell.py class TestRNN (line 24) | class TestRNN(flow.unittest.TestCase): method test_rnn_tanh_cell (line 26) | def test_rnn_tanh_cell(test_case): method test_rnn_relu_cell (line 47) | def test_rnn_relu_cell(test_case): method test_lstm_cell (line 69) | def test_lstm_cell(test_case): method test_gru_cell (line 97) | def test_gru_cell(test_case): FILE: python/oneflow/test/expensive/test_rnn_pack_sequence.py function _test_rnn_pack_sequence (line 29) | def _test_rnn_pack_sequence(test_case, device): function _test_lstm_pack_sequence (line 131) | def _test_lstm_pack_sequence(test_case, device): function _test_gru_pack_sequence (line 237) | def _test_gru_pack_sequence(test_case, device): class TestRNNModules (line 334) | class TestRNNModules(flow.unittest.TestCase): method test_rnn (line 335) | def test_rnn(test_case): FILE: python/oneflow/test/expensive/test_rnn_utils.py function _test_rnn_utils_pack_padded_sequence (line 30) | def _test_rnn_utils_pack_padded_sequence(test_case, device): function _test_rnn_utils_pad_sequence (line 102) | def _test_rnn_utils_pad_sequence(test_case, device): function _test_rnn_utils_pack_sequence (line 130) | def _test_rnn_utils_pack_sequence(test_case, device): class TestRNNUtils (line 167) | class TestRNNUtils(flow.unittest.TestCase): method test_rnn_utils_pack_padded_sequence (line 168) | def test_rnn_utils_pack_padded_sequence(test_case): method test_rnn_utils_pad_sequence (line 175) | def test_rnn_utils_pad_sequence(test_case): method test_rnn_utils_pack_sequence (line 182) | def test_rnn_utils_pack_sequence(test_case): FILE: python/oneflow/test/expensive/test_sqrt_square_sum.py class TestLinalgVectorNorm2D (line 26) | class TestLinalgVectorNorm2D(flow.unittest.TestCase): method test_sqrt_sum_with_cpu_random_data (line 28) | def test_sqrt_sum_with_cpu_random_data(test_case): method test_sqrt_sum_with_cuda_random_data (line 38) | def test_sqrt_sum_with_cuda_random_data(test_case): method test_scalar_print_random_data (line 47) | def test_scalar_print_random_data(test_case): FILE: python/oneflow/test/expensive/test_tensor_offload.py function _test_tensor_offload_d2h (line 32) | def _test_tensor_offload_d2h(test_case, input, tensor_mem): function _test_tensor_load_h2d (line 59) | def _test_tensor_load_h2d(test_case, input, tensor_mem): function _get_tensor_mem (line 86) | def _get_tensor_mem(input): class TestTensorOffload (line 95) | class TestTensorOffload(flow.unittest.TestCase): method test_tensor_offload_and_load_float32 (line 96) | def test_tensor_offload_and_load_float32(test_case): method test_tensor_offload_and_load_float16 (line 118) | def test_tensor_offload_and_load_float16(test_case): method test_tensor_offload_and_load_int64 (line 140) | def test_tensor_offload_and_load_int64(test_case): method test_tensor_offload_and_load_0dim (line 163) | def test_tensor_offload_and_load_0dim(test_case): method test_tensor_offload_and_load_0size (line 183) | def test_tensor_offload_and_load_0size(test_case): method test_tensor_offload_and_load_cpu_mem (line 205) | def test_tensor_offload_and_load_cpu_mem(test_case): method test_param_offload (line 234) | def test_param_offload(test_case): FILE: python/oneflow/test/expensive/test_tensor_str.py function _test_local_tensor_str (line 30) | def _test_local_tensor_str(test_case, device): function _test_global_tensor_str (line 95) | def _test_global_tensor_str(test_case, device): function _test_global_tensor_str_2d (line 124) | def _test_global_tensor_str_2d(test_case, device): function _test_nd_sbp_tensor_str (line 163) | def _test_nd_sbp_tensor_str(test_case, device, sbp0, sbp1): class TestTensorStrModule (line 172) | class TestTensorStrModule(flow.unittest.TestCase): method test_local_tensor_str_1n1d (line 175) | def test_local_tensor_str_1n1d(test_case): method test_global_tensor_str_1n1d (line 187) | def test_global_tensor_str_1n1d(test_case): method test_tensor_str_1n2d (line 198) | def test_tensor_str_1n2d(test_case): method test_nd_sbp_tensor_str (line 209) | def test_nd_sbp_tensor_str(test_case): FILE: python/oneflow/test/expensive/test_util.py function GenCartesianProduct (line 28) | def GenCartesianProduct(sets): function GenArgList (line 38) | def GenArgList(arg_dict): function GenArgDict (line 45) | def GenArgDict(arg_dict): class Args (line 49) | class Args: method __init__ (line 50) | def __init__(self, flow_args, tf_args=None): method __str__ (line 57) | def __str__(self): method __repr__ (line 60) | def __repr__(self): function FlattenArray (line 84) | def FlattenArray(input_array): function Array2Numpy (line 91) | def Array2Numpy(input_array, target_shape): function Index2Coordinate (line 95) | def Index2Coordinate(idx, tensor_shape): function Coordinate2Index (line 106) | def Coordinate2Index(coordinate, tensor_shape): function generate_graph (line 118) | def generate_graph(func): FILE: python/oneflow/test/gen_ops_process.py function get_api (line 22) | def get_api(rst_dir): function get_profile_func (line 83) | def get_profile_func(path): function get_test_func (line 109) | def get_test_func(path): function pure_match (line 157) | def pure_match(x, y): function match_test_func (line 187) | def match_test_func(func, func_list): FILE: python/oneflow/test/graph/alexnet_model.py class AlexNet (line 25) | class AlexNet(nn.Module): method __init__ (line 26) | def __init__(self, num_classes: int = 1000) -> None: method forward (line 54) | def forward(self, x: flow.Tensor) -> flow.Tensor: function alexnet (line 62) | def alexnet(pretrained: bool = False, progress: bool = True, **kwargs: A... FILE: python/oneflow/test/graph/ofrecord_data_utils.py class OFRecordDataLoader (line 21) | class OFRecordDataLoader(flow.nn.Module): method __init__ (line 22) | def __init__( method __len__ (line 91) | def __len__(self): method forward (line 94) | def forward(self): FILE: python/oneflow/test/graph/optimizer_test_util.py function clip_grad_norm_np (line 19) | def clip_grad_norm_np(np_grad, max_norm, norm_type): FILE: python/oneflow/test/graph/test_alexnet_auto_parallel.py function load_data_fashion_mnist (line 30) | def load_data_fashion_mnist( function _parse_args (line 70) | def _parse_args(): function _test_alexnet_graph (line 92) | def _test_alexnet_graph(test_case, args, placement, sbp): class TestAlexnetAutoParallel (line 216) | class TestAlexnetAutoParallel(oneflow.unittest.TestCase): method test_alexnet_auto_parallel_1d_sbp (line 217) | def test_alexnet_auto_parallel_1d_sbp(test_case): FILE: python/oneflow/test/graph/test_alexnet_graph.py function _parse_args (line 29) | def _parse_args(): function _test_alexnet_graph_repr (line 67) | def _test_alexnet_graph_repr(test_case, args): function _test_alexnet_graph (line 114) | def _test_alexnet_graph(test_case, args): class TestAlexnetGraph (line 220) | class TestAlexnetGraph(oneflow.unittest.TestCase): method test_alexnet_graph_repr (line 221) | def test_alexnet_graph_repr(test_case): method test_alexnet_graph_gpu (line 227) | def test_alexnet_graph_gpu(test_case): method test_alexnet_graph_cpu (line 232) | def test_alexnet_graph_cpu(test_case): FILE: python/oneflow/test/graph/test_comb1to2d.py class _TestModuleDiffHierarchy (line 27) | class _TestModuleDiffHierarchy(nn.Module): method forward (line 28) | def forward(self, x): class _TestModuleDiffPlacement (line 57) | class _TestModuleDiffPlacement(nn.Module): method forward (line 58) | def forward(self, x): class _TestGraph (line 86) | class _TestGraph(nn.Graph): method __init__ (line 87) | def __init__(self, model): method build (line 91) | def build(self, x): class TestLazyAllSbpCombinationTesting (line 98) | class TestLazyAllSbpCombinationTesting(flow.unittest.TestCase): method test_lazy_boxing_2d_all_combination (line 99) | def test_lazy_boxing_2d_all_combination(test_case): FILE: python/oneflow/test/graph/test_comb2d.py class _TestModule (line 27) | class _TestModule(nn.Module): method forward (line 28) | def forward(self, x): class _TestGraph (line 53) | class _TestGraph(nn.Graph): method __init__ (line 54) | def __init__(self, model): method build (line 58) | def build(self, x): class TestLazyAllSbpCombinationTesting (line 65) | class TestLazyAllSbpCombinationTesting(flow.unittest.TestCase): method test_lazy_boxing_2d_all_combination (line 66) | def test_lazy_boxing_2d_all_combination(test_case): FILE: python/oneflow/test/graph/test_forward_graph.py class TestForwardGraph (line 27) | class TestForwardGraph(flow.unittest.TestCase): method test_forward_graph (line 29) | def test_forward_graph(test_case): method test_add_backward (line 80) | def test_add_backward(test_case): FILE: python/oneflow/test/graph/test_free_tensor_not_in_job.py function get_bn_graph (line 24) | def get_bn_graph(): class TestFreeTensorNotInJob (line 42) | class TestFreeTensorNotInJob(flow.unittest.TestCase): method test_free_tensor_not_in_job (line 43) | def test_free_tensor_not_in_job(test_case): FILE: python/oneflow/test/graph/test_fx_fuse.py function _fuse_conv_bn_eval (line 26) | def _fuse_conv_bn_eval(conv, bn): function _fuse_conv_bn_weights (line 47) | def _fuse_conv_bn_weights(conv_w, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn... function _parent_name (line 64) | def _parent_name(target: str) -> Tuple[str, str]: function _replace_node_module (line 73) | def _replace_node_module( function _fx_fuse (line 81) | def _fx_fuse(model: flow.nn.Module) -> flow.nn.Module: class TestConvBnFuse (line 132) | class TestConvBnFuse(flow.unittest.TestCase): method test_fuse (line 133) | def test_fuse(test_case): FILE: python/oneflow/test/graph/test_fx_replace_ops.py class M (line 22) | class M(flow.nn.Module): method __init__ (line 23) | def __init__(self): method forward (line 26) | def forward(self, x, w1, w2): class TestReplaceOps (line 35) | class TestReplaceOps(flow.unittest.TestCase): method test_pattern (line 36) | def test_pattern(test_case): FILE: python/oneflow/test/graph/test_fx_symbolic_trace_module.py class AlexNet (line 23) | class AlexNet(nn.Module): method __init__ (line 24) | def __init__(self, num_classes: int = 1000) -> None: method forward (line 52) | def forward(self, x: flow.Tensor) -> flow.Tensor: class TestAlexNet (line 61) | class TestAlexNet(flow.unittest.TestCase): method test_alexnet (line 62) | def test_alexnet(test_case): FILE: python/oneflow/test/graph/test_gbc1to2d.py function _test_general_basic_communication_1d_to_2d (line 29) | def _test_general_basic_communication_1d_to_2d(test_case, src_nd_sbp, ds... function gen_nd_sbp_1d (line 62) | def gen_nd_sbp_1d(): function gen_nd_sbp_2d (line 72) | def gen_nd_sbp_2d(): class TestGeneralBasicCommunication (line 82) | class TestGeneralBasicCommunication(flow.unittest.TestCase): method test_general_basic_communication (line 83) | def test_general_basic_communication(test_case): FILE: python/oneflow/test/graph/test_gbc2d.py function _test_general_basic_communication_same_placement (line 29) | def _test_general_basic_communication_same_placement(test_case, src_nd_s... function gen_nd_sbp (line 77) | def gen_nd_sbp(): class TestGeneralBasicCommunication (line 93) | class TestGeneralBasicCommunication(flow.unittest.TestCase): method test_general_basic_communication (line 94) | def test_general_basic_communication(test_case): FILE: python/oneflow/test/graph/test_gbc2to1d.py function _test_general_basic_communication_2d_to_1d (line 29) | def _test_general_basic_communication_2d_to_1d(test_case, src_nd_sbp, ds... function gen_nd_sbp_1d (line 62) | def gen_nd_sbp_1d(): function gen_nd_sbp_2d (line 72) | def gen_nd_sbp_2d(): class TestGeneralBasicCommunication (line 82) | class TestGeneralBasicCommunication(flow.unittest.TestCase): method test_general_basic_communication (line 83) | def test_general_basic_communication(test_case): FILE: python/oneflow/test/graph/test_gbc2to2d.py function _test_general_basic_communication_2d_to_2d (line 29) | def _test_general_basic_communication_2d_to_2d(test_case, src_nd_sbp, ds... function gen_nd_sbp (line 65) | def gen_nd_sbp(): class TestGeneralBasicCommunication (line 81) | class TestGeneralBasicCommunication(flow.unittest.TestCase): method test_general_basic_communication (line 82) | def test_general_basic_communication(test_case): FILE: python/oneflow/test/graph/test_graph.py class SubModule (line 31) | class SubModule(flow.nn.Module): method __init__ (line 32) | def __init__(self): method forward (line 37) | def forward(self, x): class CustomModule (line 43) | class CustomModule(flow.nn.Module): method __init__ (line 44) | def __init__(self): method forward (line 50) | def forward(self, x): class TestGraph (line 59) | class TestGraph(flow.unittest.TestCase): method test_add_nested_module (line 60) | def test_add_nested_module(test_case): method test_graph_name (line 92) | def test_graph_name(test_case): method test_graph_build_ctx (line 126) | def test_graph_build_ctx(test_case): method test_block_scope (line 165) | def test_block_scope(test_case): method test_create_optimizer_in_graph (line 259) | def test_create_optimizer_in_graph(test_case): method test_graph_in_subthread (line 284) | def test_graph_in_subthread(test_case): FILE: python/oneflow/test/graph/test_graph_activation_checkpoint.py class TestGraphActivationCheckpoint (line 33) | class TestGraphActivationCheckpoint(flow.unittest.TestCase): method test_activation_checkpoint (line 34) | def test_activation_checkpoint(test_case): FILE: python/oneflow/test/graph/test_graph_arange.py class TestArangeGraph (line 26) | class TestArangeGraph(oneflow.unittest.TestCase): method test_arange_graph (line 27) | def test_arange_graph(test_case): FILE: python/oneflow/test/graph/test_graph_asymmetric_io.py class TestGlobalAsymmetricGraph (line 26) | class TestGlobalAsymmetricGraph(oneflow.unittest.TestCase): method test_global_asymmetric_graph_gpu (line 27) | def test_global_asymmetric_graph_gpu(test_case): FILE: python/oneflow/test/graph/test_graph_block.py class TestGraphBlock (line 32) | class TestGraphBlock(flow.unittest.TestCase): method test_module_has_custom_func (line 33) | def test_module_has_custom_func(test_case): method test_module_has_special_attr (line 60) | def test_module_has_special_attr(test_case): method test_block_with_parameter (line 87) | def test_block_with_parameter(test_case): method test_block_get_class_in_forward (line 140) | def test_block_get_class_in_forward(test_case): method test_block_with_not_registered_module (line 198) | def test_block_with_not_registered_module(test_case): method test_block_with_seq_container (line 257) | def test_block_with_seq_container(test_case): method test_block_with_list_container (line 303) | def test_block_with_list_container(test_case): method test_module_list_slice (line 358) | def test_module_list_slice(test_case): method test_block_with_dict_container (line 391) | def test_block_with_dict_container(test_case): method test_block_with_dict_container_nto1 (line 446) | def test_block_with_dict_container_nto1(test_case): method test_block_with_para_list_container (line 503) | def test_block_with_para_list_container(test_case): method test_block_with_para_dict_container (line 538) | def test_block_with_para_dict_container(test_case): method test_mixin_module (line 574) | def test_mixin_module(test_case): FILE: python/oneflow/test/graph/test_graph_buffer_limit.py function _test_graph_buffer_limit (line 26) | def _test_graph_buffer_limit(test_case): class TestGraphPipelineBufferLimit (line 86) | class TestGraphPipelineBufferLimit(oneflow.unittest.TestCase): method test_graph_buffer_limit (line 87) | def test_graph_buffer_limit(test_case): FILE: python/oneflow/test/graph/test_graph_clip_grad_norm.py class MyModule1 (line 25) | class MyModule1(flow.nn.Module): method __init__ (line 26) | def __init__(self, param): method forward (line 30) | def forward(self, input): class MyModule2 (line 35) | class MyModule2(flow.nn.Module): method __init__ (line 36) | def __init__(self, param): method forward (line 40) | def forward(self, input, target): function _make_optimizer (line 47) | def _make_optimizer(params, norm_type, max_norm): class MyGraph (line 61) | class MyGraph(flow.nn.Graph): method __init__ (line 62) | def __init__(self, module1, module2, optimizer=None, acc=1): method build (line 82) | def build(self, input, target): class TensorGenerator (line 91) | class TensorGenerator(object): method __init__ (line 92) | def __init__( method local_input (line 168) | def local_input(self): method local_target (line 171) | def local_target(self): method local_param1 (line 174) | def local_param1(self): method local_param2 (line 177) | def local_param2(self): method global_input (line 180) | def global_input(self): method global_target (line 186) | def global_target(self): method global_param1 (line 192) | def global_param1(self): method global_param2 (line 198) | def global_param2(self): function _compare_with_eager (line 205) | def _compare_with_eager( class TestGraphClipGradNorm (line 292) | class TestGraphClipGradNorm(flow.unittest.TestCase): method test_local (line 294) | def test_local(test_case): method test_acc (line 298) | def test_acc(test_case): method test_dp (line 302) | def test_dp(test_case): method test_mp (line 306) | def test_mp(test_case): method test_pp (line 310) | def test_pp(test_case): method test_pp_acc (line 314) | def test_pp_acc(test_case): method test_dp_mp (line 318) | def test_dp_mp(test_case): method test_mp_pp (line 322) | def test_mp_pp(test_case): method test_dp_pp (line 326) | def test_dp_pp(test_case): method test_mp_pp_acc (line 330) | def test_mp_pp_acc(test_case): method test_dp_pp_acc (line 334) | def test_dp_pp_acc(test_case): class TestGraphClipGradNormInf (line 339) | class TestGraphClipGradNormInf(flow.unittest.TestCase): method test_local (line 341) | def test_local(test_case): method test_acc (line 345) | def test_acc(test_case): method test_dp (line 351) | def test_dp(test_case): method test_mp (line 361) | def test_mp(test_case): method test_pp (line 371) | def test_pp(test_case): method test_pp_acc (line 381) | def test_pp_acc(test_case): method test_dp_mp (line 393) | def test_dp_mp(test_case): method test_mp_pp (line 403) | def test_mp_pp(test_case): method test_dp_pp (line 413) | def test_dp_pp(test_case): method test_mp_pp_acc (line 423) | def test_mp_pp_acc(test_case): method test_dp_pp_acc (line 435) | def test_dp_pp_acc(test_case): FILE: python/oneflow/test/graph/test_graph_copy.py class TestCopyGraph (line 26) | class TestCopyGraph(oneflow.unittest.TestCase): method test_copy_graph (line 28) | def test_copy_graph(test_case): FILE: python/oneflow/test/graph/test_graph_debug.py function _graph_debug (line 29) | def _graph_debug(test_case, v_level=0, ranks=None, max_py_stack_depth=2): class TestGraphDebug (line 62) | class TestGraphDebug(oneflow.unittest.TestCase): method test_graph_debug_rank_null (line 63) | def test_graph_debug_rank_null(test_case): method test_graph_debug_rank_0 (line 66) | def test_graph_debug_rank_0(test_case): method test_graph_debug_rank_1 (line 69) | def test_graph_debug_rank_1(test_case): method test_graph_debug_rank_1_and_2 (line 72) | def test_graph_debug_rank_1_and_2(test_case): method test_graph_debug_rank_all (line 75) | def test_graph_debug_rank_all(test_case): method test_graph_debug_mode_closed (line 78) | def test_graph_debug_mode_closed(test_case): method test_graph_debug_mode_opened (line 81) | def test_graph_debug_mode_opened(test_case): method test_graph_debug_max_py_stack_depth_2 (line 84) | def test_graph_debug_max_py_stack_depth_2(test_case): method test_graph_debug_max_py_stack_depth_8 (line 87) | def test_graph_debug_max_py_stack_depth_8(test_case): FILE: python/oneflow/test/graph/test_graph_depend.py function _build_graph_and_test (line 35) | def _build_graph_and_test(TestModel, in_data, test_case): class TestDependGraph (line 56) | class TestDependGraph(oneflow.unittest.TestCase): method test_depend_graph_case0 (line 57) | def test_depend_graph_case0(test_case): method test_depend_graph_case1 (line 74) | def test_depend_graph_case1(test_case): method test_depend_graph_case2 (line 93) | def test_depend_graph_case2(test_case): method test_depend_graph_case3 (line 112) | def test_depend_graph_case3(test_case): method test_depend_graph_case4 (line 133) | def test_depend_graph_case4(test_case): method test_depend_graph_case5 (line 149) | def test_depend_graph_case5(test_case): method test_depend_graph_case6 (line 169) | def test_depend_graph_case6(test_case): method test_depend_graph_case7 (line 188) | def test_depend_graph_case7(test_case): method test_depend_graph_case8 (line 207) | def test_depend_graph_case8(test_case): FILE: python/oneflow/test/graph/test_graph_eye.py class TestEyeGraph (line 25) | class TestEyeGraph(oneflow.unittest.TestCase): method test_eye_graph (line 26) | def test_eye_graph(test_case): FILE: python/oneflow/test/graph/test_graph_free_eager_tensor.py class TestGraphWithEagerTensorCaught (line 26) | class TestGraphWithEagerTensorCaught(oneflow.unittest.TestCase): method test_eager_tensor_forward_graph (line 27) | def test_eager_tensor_forward_graph(test_case): method test_eager_tensor_to (line 60) | def test_eager_tensor_to(test_case): method test_two_graph_caught_same_free_eager_tensor (line 87) | def test_two_graph_caught_same_free_eager_tensor(test_case): method test_graph_return_free_eager_tensor (line 119) | def test_graph_return_free_eager_tensor(test_case): method test_graph_return_inplace_free_eager_tensor (line 145) | def test_graph_return_inplace_free_eager_tensor(test_case): class GlobalFreeEagerTensorGraphTestCase (line 185) | class GlobalFreeEagerTensorGraphTestCase(oneflow.unittest.TestCase): method test_global_eager_tensor_to (line 186) | def test_global_eager_tensor_to(test_case): FILE: python/oneflow/test/graph/test_graph_grad_acc.py function _test_grad_acc_graph (line 24) | def _test_grad_acc_graph(test_case, device): class TestGradAccGraph (line 117) | class TestGradAccGraph(oneflow.unittest.TestCase): method test_grad_acc_graph_gpu (line 118) | def test_grad_acc_graph_gpu(test_case): method test_grad_acc_graph_cpu (line 121) | def test_grad_acc_graph_cpu(test_case): FILE: python/oneflow/test/graph/test_graph_image_gpu_decoder.py class OFRecordDataLoader (line 24) | class OFRecordDataLoader(flow.nn.Module): method __init__ (line 25) | def __init__(self): method forward (line 67) | def forward(self) -> (flow.Tensor, flow.Tensor): class TestImageGpuDecoderGraph (line 81) | class TestImageGpuDecoderGraph(oneflow.unittest.TestCase): method test_image_gpu_decoder_graph (line 83) | def test_image_gpu_decoder_graph(test_case): FILE: python/oneflow/test/graph/test_graph_inplace_add.py function _test_graph_lazy_inplace (line 24) | def _test_graph_lazy_inplace(test_case, x, y): class TestLocalInplace (line 39) | class TestLocalInplace(oneflow.unittest.TestCase): method test_graph_inplace_gpu (line 40) | def test_graph_inplace_gpu(test_case): method test_graph_inplace_cpu (line 45) | def test_graph_inplace_cpu(test_case): class TestGlobalInplace (line 53) | class TestGlobalInplace(oneflow.unittest.TestCase): method test_graph_inplace_gpu (line 54) | def test_graph_inplace_gpu(test_case): method test_graph_inplace_cpu (line 66) | def test_graph_inplace_cpu(test_case): FILE: python/oneflow/test/graph/test_graph_io_check.py class BaseOutput (line 34) | class BaseOutput(OrderedDict): method __post_init__ (line 35) | def __post_init__(self): method __delitem__ (line 56) | def __delitem__(self, *args, **kwargs): method setdefault (line 61) | def setdefault(self, *args, **kwargs): method pop (line 66) | def pop(self, *args, **kwargs): method update (line 71) | def update(self, *args, **kwargs): method __getitem__ (line 76) | def __getitem__(self, k): method __setattr__ (line 94) | def __setattr__(self, name, value): method __setitem__ (line 100) | def __setitem__(self, key, value): method to_tuple (line 106) | def to_tuple(self) -> Tuple[Any]: class CustomDataClass (line 114) | class CustomDataClass(BaseOutput): class TestGraphIOCheck (line 120) | class TestGraphIOCheck(flow.unittest.TestCase): method test_io_node (line 121) | def test_io_node(test_case): method test_io_node_with_simple_tuple_or_list_input (line 162) | def test_io_node_with_simple_tuple_or_list_input(self): method test_custom_class (line 213) | def test_custom_class(test_case): method test_non_tensor_types_of_module (line 239) | def test_non_tensor_types_of_module(test_case): method test_graph_return_size_0_tuple (line 297) | def test_graph_return_size_0_tuple(test_case): method test_graph_return_dict_tuple (line 348) | def test_graph_return_dict_tuple(test_case): method test_graph_outputs_buffer (line 390) | def test_graph_outputs_buffer(test_case): FILE: python/oneflow/test/graph/test_graph_linear.py function _test_linear_graph (line 24) | def _test_linear_graph(test_case, device): function _test_linear_graph_func (line 62) | def _test_linear_graph_func(test_case, device): class TestLinearGraph (line 96) | class TestLinearGraph(oneflow.unittest.TestCase): method test_linear_graph_gpu (line 97) | def test_linear_graph_gpu(test_case): method test_linear_graph_cpu (line 100) | def test_linear_graph_cpu(test_case): method test_linear_graph_func_gpu (line 103) | def test_linear_graph_func_gpu(test_case): method test_linear_graph_func_cpu (line 106) | def test_linear_graph_func_cpu(test_case): FILE: python/oneflow/test/graph/test_graph_linear_train.py function _test_linear_train_graph (line 24) | def _test_linear_train_graph(test_case, device): class TestLinearTrainGraph (line 129) | class TestLinearTrainGraph(oneflow.unittest.TestCase): method test_linear_train_graph_gpu (line 130) | def test_linear_train_graph_gpu(test_case): method test_linear_train_graph_cpu (line 133) | def test_linear_train_graph_cpu(test_case): FILE: python/oneflow/test/graph/test_graph_loss.py function compare_loss (line 28) | def compare_loss(device_type, dim, reduction, cls, data_generator): function generate_necessity_default (line 46) | def generate_necessity_default(dim: int, device: str): function generate_necessity_for_cross_entropy_or_nll_loss (line 55) | def generate_necessity_for_cross_entropy_or_nll_loss(dim: int, device: s... class TestKLDivLossGraph (line 66) | class TestKLDivLossGraph(oneflow.unittest.TestCase): method test_kl_div_loss_graph (line 67) | def test_kl_div_loss_graph(testcase): class TestSmoothL1LossGraph (line 79) | class TestSmoothL1LossGraph(oneflow.unittest.TestCase): method test_smooth_l1_loss_graph (line 80) | def test_smooth_l1_loss_graph(testcase): class TestBCELossOrWithLogitsGraph (line 92) | class TestBCELossOrWithLogitsGraph(flow.unittest.TestCase): method test_bce_loss_graph (line 93) | def test_bce_loss_graph(testcase): class TestCrossEntropyOrNllLossGraph (line 105) | class TestCrossEntropyOrNllLossGraph(flow.unittest.TestCase): method test_cross_entropy_loss_or_nll_loss_graph (line 106) | def test_cross_entropy_loss_or_nll_loss_graph(testcase): FILE: python/oneflow/test/graph/test_graph_lr_scale.py class _Block (line 26) | class _Block(flow.nn.Module): method __init__ (line 27) | def __init__(self, feats, device=None, placement=None): method forward (line 36) | def forward(self, x): class _MyModule (line 40) | class _MyModule(flow.nn.Module): method __init__ (line 41) | def __init__(self, feats, depth, device=None, placement=None): method forward (line 50) | def forward(self, x): class _MyGraph (line 56) | class _MyGraph(flow.nn.Graph): method __init__ (line 57) | def __init__(self, model, optimizer, lr_scheduler): method build (line 62) | def build(self, input): function _lrs_param_groups (line 68) | def _lrs_param_groups(model, base_scale): function _rand_input (line 78) | def _rand_input(shape, device=None, placement=None, requires_grad=False): function _test_lrs (line 89) | def _test_lrs(test_case, **kwargs): class LRScaleTest (line 159) | class LRScaleTest(flow.unittest.TestCase): method test_lr_scale (line 160) | def test_lr_scale(self): class LRScaleParallelTest (line 182) | class LRScaleParallelTest(flow.unittest.TestCase): method test_lr_scale_parallel (line 183) | def test_lr_scale_parallel(self): FILE: python/oneflow/test/graph/test_graph_lr_scheduler.py class MyModule (line 25) | class MyModule(flow.nn.Module): method __init__ (line 26) | def __init__(self): method forward (line 30) | def forward(self, input): class MyGraph (line 34) | class MyGraph(flow.nn.Graph): method __init__ (line 35) | def __init__(self, module, optimizer, lr_scheduler): method build (line 40) | def build(self, input): function _rand_input (line 46) | def _rand_input(): function _get_graph_lrs_from_log (line 50) | def _get_graph_lrs_from_log(log_path): class _DebugMode (line 66) | class _DebugMode(object): method __enter__ (line 67) | def __enter__(self): method __exit__ (line 70) | def __exit__(self, type, value, traceback): function _compare_graph_lr_scheduler_with_eager (line 74) | def _compare_graph_lr_scheduler_with_eager(test_case, **kwargs): class TestGraphLRSchedulerWithEager (line 138) | class TestGraphLRSchedulerWithEager(flow.unittest.TestCase): method test_constant_lr (line 139) | def test_constant_lr(self): method test_linear_lr (line 149) | def test_linear_lr(self): method test_linear_lr_end_factor (line 160) | def test_linear_lr_end_factor(self): method test_step_lr (line 171) | def test_step_lr(self): method test_multi_step_lr (line 181) | def test_multi_step_lr(self): method test_polynomial_lr (line 191) | def test_polynomial_lr(self): method test_exponential_lr (line 213) | def test_exponential_lr(self): method test_cosine_decay_lr (line 223) | def test_cosine_decay_lr(self): method test_cosine_annealing_lr (line 234) | def test_cosine_annealing_lr(self): method test_linear_warmup_cosine_annealing_lr (line 245) | def test_linear_warmup_cosine_annealing_lr(self): method test_linear_warmup_prefix_cosine_annealing_lr (line 260) | def test_linear_warmup_prefix_cosine_annealing_lr(self): method test_linear_warmup_multistep_lr (line 275) | def test_linear_warmup_multistep_lr(self): method test_constant_warmup_cosine_decay_lr (line 288) | def test_constant_warmup_cosine_decay_lr(self): method test_constant_warmup_prefix_cosine_decay_lr (line 302) | def test_constant_warmup_prefix_cosine_decay_lr(self): method test_only_warmup (line 317) | def test_only_warmup(self): method test_warmup_iters_equal_to_zero (line 328) | def test_warmup_iters_equal_to_zero(self): method test_cosine_annealing_warm_restarts (line 340) | def test_cosine_annealing_warm_restarts(self): method test_cosine_annealing_warm_restarts_mult_2 (line 352) | def test_cosine_annealing_warm_restarts_mult_2(self): method test_cosine_annealing_warm_restarts_limit (line 364) | def test_cosine_annealing_warm_restarts_limit(self): FILE: python/oneflow/test/graph/test_graph_lr_with_warmup.py function _test_linear_graph_train_with_lr_sch (line 26) | def _test_linear_graph_train_with_lr_sch( function _sgd_cosine_fn (line 146) | def _sgd_cosine_fn(parameters): function _sgd_cosine_constant_fn (line 156) | def _sgd_cosine_constant_fn(parameters): function _sgd_constant_fn (line 169) | def _sgd_constant_fn(parameters): function _sgd_cosine_linear_fn (line 179) | def _sgd_cosine_linear_fn(parameters): function _sgd_linear_fn (line 192) | def _sgd_linear_fn(parameters): class TestLinearGraphTrainWithCosineLrScheduler (line 204) | class TestLinearGraphTrainWithCosineLrScheduler(flow.unittest.TestCase): method test_graph_cosine (line 205) | def test_graph_cosine(test_case): method test_graph_cosine_constant (line 213) | def test_graph_cosine_constant(test_case): method test_graph_constant (line 221) | def test_graph_constant(test_case): method test_graph_cosine_linear (line 229) | def test_graph_cosine_linear(test_case): method test_graph_linear (line 237) | def test_graph_linear(test_case): FILE: python/oneflow/test/graph/test_graph_lrs.py function _test_linear_graph_train_with_lr_sch (line 27) | def _test_linear_graph_train_with_lr_sch( class TestGraphLRs (line 149) | class TestGraphLRs(flow.unittest.TestCase): method test_step_lr (line 150) | def test_step_lr(test_case): method test_multistep_lr (line 160) | def test_multistep_lr(test_case): method test_cosine_annealing_lr (line 173) | def test_cosine_annealing_lr(test_case): method test_polynomial_lr (line 185) | def test_polynomial_lr(test_case): FILE: python/oneflow/test/graph/test_graph_masked_fill.py class TestMaskedFillGraph (line 28) | class TestMaskedFillGraph(flow.unittest.TestCase): method test_masked_fill_graph (line 29) | def test_masked_fill_graph(test_case): method test_masked_fill_by_generate_graph (line 54) | def test_masked_fill_by_generate_graph(test_case): FILE: python/oneflow/test/graph/test_graph_nccl_logical_fusion.py class TestGraphNcclLogicalFusion (line 29) | class TestGraphNcclLogicalFusion(flow.unittest.TestCase): method test_graph_nccl_fusion_1d (line 30) | def test_graph_nccl_fusion_1d(test_case): method test_graph_nccl_fusion_2d (line 97) | def test_graph_nccl_fusion_2d(test_case): FILE: python/oneflow/test/graph/test_graph_non_contiguous_tensors.py class ModuleTest (line 23) | class ModuleTest(flow.nn.Module): method __init__ (line 24) | def __init__(self, contiguous: bool, device): method forward (line 33) | def forward(self, input): function _test_graph_non_contiguous_tensors (line 38) | def _test_graph_non_contiguous_tensors(test_case, device): class TestGraphNonContiguousTensor (line 96) | class TestGraphNonContiguousTensor(oneflow.unittest.TestCase): method test_graph_non_contiguous_tensors_cpu (line 97) | def test_graph_non_contiguous_tensors_cpu(test_case): method test_graph_non_contiguous_tensors_gpu (line 101) | def test_graph_non_contiguous_tensors_gpu(test_case): FILE: python/oneflow/test/graph/test_graph_normal_inplace.py function _test_data_local (line 42) | def _test_data_local(test_case, device, fn): function _test_data_global (line 63) | def _test_data_global(test_case, data_1, data_2, placement, sbp, fn): class TestNormalOpInplaceData (line 84) | class TestNormalOpInplaceData(flow.unittest.TestCase): method test_normal_op_data_local_with_eager_and_lazy (line 87) | def test_normal_op_data_local_with_eager_and_lazy(test_case): method test_normal_op_data_consistent_with_eager_and_lazy (line 95) | def test_normal_op_data_consistent_with_eager_and_lazy(test_case): FILE: python/oneflow/test/graph/test_graph_ofrecord_reader.py class OFRecordDataLoader (line 24) | class OFRecordDataLoader(flow.nn.Module): method __init__ (line 25) | def __init__(self): method forward (line 67) | def forward(self) -> (flow.Tensor, flow.Tensor): class TestOFRecordReaderGraph (line 78) | class TestOFRecordReaderGraph(oneflow.unittest.TestCase): method test_ofrecord_reader_graph (line 79) | def test_ofrecord_reader_graph(test_case): FILE: python/oneflow/test/graph/test_graph_optim_adadelta.py function compare_with_numpy_adadelta (line 27) | def compare_with_numpy_adadelta( function compare_with_numpy_adadelta_clip_grad (line 122) | def compare_with_numpy_adadelta_clip_grad( class TestAdadelta (line 221) | class TestAdadelta(flow.unittest.TestCase): method test_adadelta (line 223) | def test_adadelta(test_case): method test_adadelta_clip_grad (line 237) | def test_adadelta_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optim_adagrad.py function compare_with_numpy_adagrad (line 28) | def compare_with_numpy_adagrad( function compare_with_numpy_adagrad_clip_grad (line 115) | def compare_with_numpy_adagrad_clip_grad( class TestAdagrad (line 213) | class TestAdagrad(flow.unittest.TestCase): method test_adagrad (line 214) | def test_adagrad(test_case): method test_adagrad_clip_grad (line 230) | def test_adagrad_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optim_adam.py function compare_with_numpy_adam (line 27) | def compare_with_numpy_adam( function compare_with_numpy_adam_clip_grad (line 140) | def compare_with_numpy_adam_clip_grad( class TestAdam (line 257) | class TestAdam(flow.unittest.TestCase): method test_adam (line 258) | def test_adam(test_case): method test_adam_clip_grad (line 273) | def test_adam_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optim_adamw.py function compare_with_numpy_adamw (line 27) | def compare_with_numpy_adamw( function compare_with_numpy_adamw_clip_grad (line 135) | def compare_with_numpy_adamw_clip_grad( class TestAdamW (line 251) | class TestAdamW(flow.unittest.TestCase): method test_adamw (line 252) | def test_adamw(test_case): method test_adamw_clip_grad (line 267) | def test_adamw_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optim_ftrl.py function compare_with_numpy_ftrl (line 29) | def compare_with_numpy_ftrl( function compare_with_numpy_ftrl_clip_grad (line 130) | def compare_with_numpy_ftrl_clip_grad( class Testftrl (line 240) | class Testftrl(flow.unittest.TestCase): method test_ftrl (line 241) | def test_ftrl(test_case): method test_ftrl_clip_grad (line 259) | def test_ftrl_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optim_lamb.py function compare_with_numpy_lamb (line 27) | def compare_with_numpy_lamb( class TestLamb (line 160) | class TestLamb(flow.unittest.TestCase): method test_lamb (line 161) | def test_lamb(test_case): FILE: python/oneflow/test/graph/test_graph_optim_rmsprop.py function compare_with_numpy_rmsprop (line 30) | def compare_with_numpy_rmsprop( function compare_with_numpy_rmsprop_clip_grad (line 131) | def compare_with_numpy_rmsprop_clip_grad( class TestRMSprop (line 237) | class TestRMSprop(flow.unittest.TestCase): method test_rmsprop (line 238) | def test_rmsprop(test_case): method test_rmsprop_clip_grad (line 255) | def test_rmsprop_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optim_sgd.py function compare_with_numpy_sgd (line 27) | def compare_with_numpy_sgd( function compare_with_numpy_sgd_clip_grad (line 127) | def compare_with_numpy_sgd_clip_grad( class TestGraphSGD (line 236) | class TestGraphSGD(flow.unittest.TestCase): method test_sgd (line 237) | def test_sgd(test_case): method test_sgd_with_clip_grad (line 251) | def test_sgd_with_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_optimizer.py class TestGraphOptimizer (line 28) | class TestGraphOptimizer(flow.unittest.TestCase): method test_optimizer (line 29) | def test_optimizer(test_case): method test_multi_optimizer_conf (line 78) | def test_multi_optimizer_conf(test_case): method test_optimizer_with_clip_grad (line 158) | def test_optimizer_with_clip_grad(test_case): FILE: python/oneflow/test/graph/test_graph_pipeline.py class OFRecordDataLoader (line 29) | class OFRecordDataLoader(flow.nn.Module): method __init__ (line 30) | def __init__( method __len__ (line 70) | def __len__(self): method forward (line 73) | def forward(self): function _train_with_graph (line 83) | def _train_with_graph(iter_num=3): function _train_with_module (line 199) | def _train_with_module(iter_num=3, data=None): function _test_graph_pipeline (line 277) | def _test_graph_pipeline(test_case): class TestGraphPipeline (line 292) | class TestGraphPipeline(oneflow.unittest.TestCase): method test_graph_pipeline (line 293) | def test_graph_pipeline(test_case): FILE: python/oneflow/test/graph/test_graph_pipeline_delay.py function _test_graph_pipeline_delay_output (line 26) | def _test_graph_pipeline_delay_output(test_case): class TestGraphPipelineDelayOutput (line 119) | class TestGraphPipelineDelayOutput(oneflow.unittest.TestCase): method test_graph_pipeline_delay_output (line 120) | def test_graph_pipeline_delay_output(test_case): FILE: python/oneflow/test/graph/test_graph_random_seed.py function _inspect_rand_op_and_args (line 26) | def _inspect_rand_op_and_args(rand_op, **kwargs): function _test_rand_op_in_graph (line 50) | def _test_rand_op_in_graph(test_case, rand_op, input=None, **kwargs): function _get_shape_and_device_from_args (line 96) | def _get_shape_and_device_from_args(pop_device=False, **kwargs): function _test_rand_op_in_FRB (line 120) | def _test_rand_op_in_FRB(test_case, rand_op, input=None, **kwargs): function _test_split_rand_op_in_graph (line 173) | def _test_split_rand_op_in_graph(test_case, rand_op, input=None, **kwargs): function _test_broadcast_rand_op_in_graph (line 220) | def _test_broadcast_rand_op_in_graph(test_case, rand_op, input=None, **k... class TestRandOpInGraph (line 270) | class TestRandOpInGraph(oneflow.unittest.TestCase): method test_usual_rand_op (line 272) | def test_usual_rand_op(self): method test_source_rand_op (line 286) | def test_source_rand_op(self): method test_bernoulli (line 298) | def test_bernoulli(self): method test_random_mask_like (line 306) | def test_random_mask_like(self): class TestRandOpInFRB (line 319) | class TestRandOpInFRB(oneflow.unittest.TestCase): method test_usual_rand_op (line 321) | def test_usual_rand_op(self): method test_source_rand_op (line 335) | def test_source_rand_op(self): method test_bernoulli (line 347) | def test_bernoulli(self): method test_random_mask_like (line 355) | def test_random_mask_like(self): class TestGlobalRandInGraph (line 368) | class TestGlobalRandInGraph(oneflow.unittest.TestCase): method test_usual_rand_op_with_split (line 370) | def test_usual_rand_op_with_split(self): method test_usual_rand_op_with_broadcast (line 383) | def test_usual_rand_op_with_broadcast(self): method test_source_rand_op_with_split (line 396) | def test_source_rand_op_with_split(self): method test_source_rand_op_with_broadcast (line 408) | def test_source_rand_op_with_broadcast(self): method test_bernoulli_with_split (line 419) | def test_bernoulli_with_split(self): method test_bernoulli_with_broadcast (line 425) | def test_bernoulli_with_broadcast(self): method test_random_mask_like_with_split (line 432) | def test_random_mask_like_with_split(self): method test_random_mask_like_with_broadcast (line 444) | def test_random_mask_like_with_broadcast(self): FILE: python/oneflow/test/graph/test_graph_relu.py class TestReluGraph (line 24) | class TestReluGraph(oneflow.unittest.TestCase): method test_relu_graph (line 25) | def test_relu_graph(test_case): FILE: python/oneflow/test/graph/test_graph_reshape_acc.py function _test_graph_reshape_acc (line 25) | def _test_graph_reshape_acc(test_case): class TestGraphReshapeAcc (line 98) | class TestGraphReshapeAcc(oneflow.unittest.TestCase): method test_graph_reshape_acc (line 99) | def test_graph_reshape_acc(test_case): FILE: python/oneflow/test/graph/test_graph_reuse_var.py class TestGraphResueVar (line 29) | class TestGraphResueVar(flow.unittest.TestCase): method test_graph_reuse_var (line 30) | def test_graph_reuse_var(test_case): FILE: python/oneflow/test/graph/test_graph_save_load.py function _test_linear_graph_save_load (line 25) | def _test_linear_graph_save_load(test_case, device): class TestLinearGraphSaveLoad (line 128) | class TestLinearGraphSaveLoad(oneflow.unittest.TestCase): method test_linear_graph_save_load_gpu (line 129) | def test_linear_graph_save_load_gpu(test_case): method _test_linear_graph_save_load_cpu (line 132) | def _test_linear_graph_save_load_cpu(test_case): function _test_linear_graph_save_load_global (line 136) | def _test_linear_graph_save_load_global(test_case, device): class TestLinearGraphSaveLoadGlobal (line 268) | class TestLinearGraphSaveLoadGlobal(oneflow.unittest.TestCase): method test_linear_graph_save_load_gpu (line 269) | def test_linear_graph_save_load_gpu(test_case): method _test_linear_graph_save_load_cpu (line 272) | def _test_linear_graph_save_load_cpu(test_case): FILE: python/oneflow/test/graph/test_graph_save_load_global_b_s.py function _test_linear_graph_save_load_global_broadcast (line 26) | def _test_linear_graph_save_load_global_broadcast( function _test_graph_save_load_global_split_2 (line 178) | def _test_graph_save_load_global_split_2( function _test_graph_save_load_global_split_4 (line 435) | def _test_graph_save_load_global_split_4( class TestGraphSaveLoadGlobal2d (line 772) | class TestGraphSaveLoadGlobal2d(oneflow.unittest.TestCase): method test_linear_graph_save_load_gpu_1_broadcast (line 773) | def test_linear_graph_save_load_gpu_1_broadcast(test_case): method test_linear_graph_save_load_cpu_1_broadcast (line 780) | def test_linear_graph_save_load_cpu_1_broadcast(test_case): method test_graph_save_load_gpu_2_split (line 787) | def test_graph_save_load_gpu_2_split(test_case): method test_graph_save_load_cpu_2_split (line 795) | def test_graph_save_load_cpu_2_split(test_case): class TestGraphSaveLoadGlobal4d (line 805) | class TestGraphSaveLoadGlobal4d(oneflow.unittest.TestCase): method test_graph_save_load_gpu_2_split_2_none (line 806) | def test_graph_save_load_gpu_2_split_2_none(test_case): method test_graph_save_load_cpu_2_split_2_none (line 814) | def test_graph_save_load_cpu_2_split_2_none(test_case): FILE: python/oneflow/test/graph/test_graph_scalar.py function _test_scalar_graph (line 24) | def _test_scalar_graph(test_case, device): function _test_scalar_train_graph (line 51) | def _test_scalar_train_graph(test_case, device): function _test_scalar_global_train_graph (line 100) | def _test_scalar_global_train_graph(test_case, placement): class TestScalarGraph (line 155) | class TestScalarGraph(oneflow.unittest.TestCase): method test_scalar_graph_gpu (line 156) | def test_scalar_graph_gpu(test_case): method test_scalar_graph_cpu (line 159) | def test_scalar_graph_cpu(test_case): method test_scalar_train_graph_gpu (line 162) | def test_scalar_train_graph_gpu(test_case): method test_scalar_train_graph_cpu (line 165) | def test_scalar_train_graph_cpu(test_case): method test_scalar_global_train_graph_gpu (line 168) | def test_scalar_global_train_graph_gpu(test_case): method test_scalar_global_train_graph_cpu (line 171) | def test_scalar_global_train_graph_cpu(test_case): FILE: python/oneflow/test/graph/test_graph_separate_compile.py function modified_environ (line 27) | def modified_environ(*remove, **update): function run_testcase_with_sep_compile (line 58) | def run_testcase_with_sep_compile(test_case_cls): function _get_comb1to2d_test (line 71) | def _get_comb1to2d_test(): class TestSeparationCompile (line 185) | class TestSeparationCompile(oneflow.unittest.TestCase): method test_test_alexnet_auto_parallel (line 186) | def test_test_alexnet_auto_parallel(test_case): method _test_comb1to2d (line 191) | def _test_comb1to2d(test_case): method test_graph_zero (line 194) | def test_graph_zero(test_case): method test_graph_clip_grad_norm (line 199) | def test_graph_clip_grad_norm(test_case): method test_graph_pipeline_grad_acc_and_activatioin_checkpointing (line 204) | def test_graph_pipeline_grad_acc_and_activatioin_checkpointing(test_ca... FILE: python/oneflow/test/graph/test_graph_session_env_destruct.py class LinearGraphDestruct (line 34) | class LinearGraphDestruct(flow.nn.Graph): method __init__ (line 35) | def __init__(self): method build (line 39) | def build(self, x): class TestLinearGraphDestruct (line 47) | class TestLinearGraphDestruct(oneflow.unittest.TestCase): method test_linear_graph_destruct (line 48) | def test_linear_graph_destruct(test_case): FILE: python/oneflow/test/graph/test_graph_session_env_destruct1.py class LinearGraphDestruct1 (line 34) | class LinearGraphDestruct1(flow.nn.Graph): method __init__ (line 35) | def __init__(self): method build (line 39) | def build(self, x): FILE: python/oneflow/test/graph/test_graph_sparse_optimizer.py class MyModule (line 23) | class MyModule(flow.nn.Module): method __init__ (line 24) | def __init__(self, placement=None, sbp=None): method forward (line 29) | def forward(self, input): class MyGraph (line 33) | class MyGraph(flow.nn.Graph): method __init__ (line 34) | def __init__(self, module): method build (line 40) | def build(self, input): function _rand_input (line 45) | def _rand_input(placement=None, sbp=None): class GraphSparseOptimizerTest (line 53) | class GraphSparseOptimizerTest(oneflow.unittest.TestCase): method test (line 55) | def test(test_case): FILE: python/oneflow/test/graph/test_graph_sparse_softmax_cross_entropy.py class CrossEntropyModule (line 23) | class CrossEntropyModule(flow.nn.Module): method __init__ (line 24) | def __init__(self, pred): method forward (line 40) | def forward(self, pred, label): class CrossEntropyGraph (line 46) | class CrossEntropyGraph(flow.nn.Graph): method __init__ (line 47) | def __init__(self, module): method build (line 52) | def build(self, pred, label): function _compare_with_nn_cross_entropy_loss (line 58) | def _compare_with_nn_cross_entropy_loss( class TestSparseSoftmaxCrossEntropyGraph (line 104) | class TestSparseSoftmaxCrossEntropyGraph(oneflow.unittest.TestCase): method test_local (line 106) | def test_local(test_case): method test_data_split (line 112) | def test_data_split(test_case): method test_model_split (line 123) | def test_model_split(test_case): method test_2d_split (line 134) | def test_2d_split(test_case): FILE: python/oneflow/test/graph/test_graph_tensor_clone.py class TestTensorCloneGraph (line 26) | class TestTensorCloneGraph(oneflow.unittest.TestCase): method test_tensor_clone_graph (line 27) | def test_tensor_clone_graph(test_case): FILE: python/oneflow/test/graph/test_graph_tensor_detach.py class TestTensorDetachGraph (line 26) | class TestTensorDetachGraph(oneflow.unittest.TestCase): method test_tensor_detach_graph (line 27) | def test_tensor_detach_graph(test_case): FILE: python/oneflow/test/graph/test_graph_with_global.py function _test_linear_train_graph_with_ddp (line 27) | def _test_linear_train_graph_with_ddp(test_case): function _test_global_mode (line 225) | def _test_global_mode(test_case): function _test_global_mode_with_default_placement_and_sbp (line 272) | def _test_global_mode_with_default_placement_and_sbp(test_case): class TestLinearTrainGraphWithDDP (line 292) | class TestLinearTrainGraphWithDDP(oneflow.unittest.TestCase): method test_linear_train_graph_with_ddp (line 293) | def test_linear_train_graph_with_ddp(test_case): method test_global_mode (line 297) | def test_global_mode(test_case): FILE: python/oneflow/test/graph/test_graph_zero.py function _test_linear_train_graph_with_zero (line 25) | def _test_linear_train_graph_with_zero(test_case, zero_stage=1): function _test_linear_train_graph_2d_with_zero (line 118) | def _test_linear_train_graph_2d_with_zero(test_case, zero_stage=1): class TestLinearTrainGraphWithZeRO (line 229) | class TestLinearTrainGraphWithZeRO(oneflow.unittest.TestCase): method test_linear_train_graph_with_zero_1 (line 230) | def test_linear_train_graph_with_zero_1(test_case): method test_linear_train_graph_with_zero_2 (line 233) | def test_linear_train_graph_with_zero_2(test_case): method test_linear_train_graph_with_zero_3 (line 236) | def test_linear_train_graph_with_zero_3(test_case): class TestLinearTrainGraph2DWithZeRO (line 242) | class TestLinearTrainGraph2DWithZeRO(oneflow.unittest.TestCase): method test_linear_train_graph_2d_with_zero_3 (line 243) | def test_linear_train_graph_2d_with_zero_3(test_case): method test_linear_train_graph_2d_with_zero_2 (line 246) | def test_linear_train_graph_2d_with_zero_2(test_case): method test_linear_train_graph_2d_with_zero_1 (line 249) | def test_linear_train_graph_2d_with_zero_1(test_case): FILE: python/oneflow/test/graph/test_input_op_expr.py class TestFeedInputTensor (line 34) | class TestFeedInputTensor(unittest.TestCase): method test_feed_input_tensor (line 35) | def test_feed_input_tensor(test_case): FILE: python/oneflow/test/graph/test_long_add_n_pass.py function _test_long_add_n_graph (line 26) | def _test_long_add_n_graph(test_case, device): function _test_add_n_consume_multi_add_n_graph (line 77) | def _test_add_n_consume_multi_add_n_graph(test_case, device): class TestLongAddNGraph (line 110) | class TestLongAddNGraph(oneflow.unittest.TestCase): method test_add_n (line 111) | def test_add_n(test_case): method test_consume_multi_add_n (line 115) | def test_consume_multi_add_n(test_case): FILE: python/oneflow/test/graph/test_modify_module_forward.py class TestModifyForwardOfModule (line 30) | class TestModifyForwardOfModule(oneflow.unittest.TestCase): method test_modify_forward (line 31) | def test_modify_forward(test_case): FILE: python/oneflow/test/graph/test_multi_client_session.py class TestMultiClientSession (line 28) | class TestMultiClientSession(unittest.TestCase): method test_case1 (line 29) | def test_case1(self): method test_case2 (line 35) | def test_case2(self): FILE: python/oneflow/test/graph/test_multi_graph.py class TestMultiGraph (line 26) | class TestMultiGraph(oneflow.unittest.TestCase): method test_multi_graph (line 27) | def test_multi_graph(test_case): FILE: python/oneflow/test/graph/test_multi_tensor_adam_update_with_cast.py function compare_with_numpy_adam (line 27) | def compare_with_numpy_adam( class TestMultiTensorAdam (line 176) | class TestMultiTensorAdam(flow.unittest.TestCase): method test_multi_tensor_adam (line 177) | def test_multi_tensor_adam(test_case): FILE: python/oneflow/test/graph/test_multi_tensor_sgd_update_with_cast.py function compare_with_numpy_sgd (line 27) | def compare_with_numpy_sgd( class TestMultiTensorSGD (line 142) | class TestMultiTensorSGD(flow.unittest.TestCase): method test_multi_tensor_sgd (line 143) | def test_multi_tensor_sgd(test_case): FILE: python/oneflow/test/graph/test_nccl_logical_send_recv.py function _test_nccl_logical_send_recv_2d (line 29) | def _test_nccl_logical_send_recv_2d(test_case, src_nd_sbp, dst_nd_sbp): function gen_2d_sbp (line 86) | def gen_2d_sbp(): class TestNcclLogicalSendRecv2D (line 103) | class TestNcclLogicalSendRecv2D(flow.unittest.TestCase): method test_nccl_logical_send_recv_2d (line 104) | def test_nccl_logical_send_recv_2d(test_case): function _test_nccl_logical_send_recv_1d (line 113) | def _test_nccl_logical_send_recv_1d(test_case, src_nd_sbp, dst_nd_sbp): function gen_1d_sbp (line 158) | def gen_1d_sbp(): class TestNcclLogicalSendRecv1D (line 176) | class TestNcclLogicalSendRecv1D(flow.unittest.TestCase): method test_nccl_logical_send_recv_1d (line 177) | def test_nccl_logical_send_recv_1d(test_case): FILE: python/oneflow/test/graph/test_neq_device_process_num.py class TestGraphNeqDeviceProcessNum (line 30) | class TestGraphNeqDeviceProcessNum(flow.unittest.TestCase): method test_graph_process_num_greater_than_device (line 31) | def test_graph_process_num_greater_than_device(test_case): FILE: python/oneflow/test/graph/test_oneflow_compiler.py class TorchModule (line 30) | class TorchModule(torch.nn.Module): method __init__ (line 31) | def __init__(self): method forward (line 35) | def forward(self, x): class FlowModule (line 41) | class FlowModule(flow.nn.Module): method __init__ (line 42) | def __init__(self): method forward (line 46) | def forward(self, x): class TestOneflowInferCompiler (line 54) | class TestOneflowInferCompiler(flow.unittest.TestCase): method setUp (line 55) | def setUp(self): method test_compile_from_torch (line 58) | def test_compile_from_torch(test_case): FILE: python/oneflow/test/graph/test_optimization_conf.py class TestGraphWithSysConf (line 28) | class TestGraphWithSysConf(flow.unittest.TestCase): method test_graph_config (line 29) | def test_graph_config(test_case): FILE: python/oneflow/test/graph/test_output_op_expr.py class TestFetchOutputTensor (line 34) | class TestFetchOutputTensor(unittest.TestCase): method test_fetch_output_tensor (line 35) | def test_fetch_output_tensor(test_case): FILE: python/oneflow/test/graph/test_run_global_graph_by_vm.py class TestGlobalInterpreter (line 26) | class TestGlobalInterpreter(flow.unittest.TestCase): method test_data_parallel_run_by_vm (line 28) | def test_data_parallel_run_by_vm(test_case): method test_module_parallel_run_by_vm (line 56) | def test_module_parallel_run_by_vm(test_case): method test_boxing_data_parallel_run_by_vm (line 83) | def test_boxing_data_parallel_run_by_vm(test_case): method test_empty_inputs (line 116) | def test_empty_inputs(test_case): FILE: python/oneflow/test/graph/test_run_graph_by_vm.py class EnvVar (line 21) | class EnvVar(object): method __init__ (line 22) | def __init__(self, env_list: dict): method __enter__ (line 25) | def __enter__(self): method __exit__ (line 28) | def __exit__(self, *args): class RunGraphByVmEnv (line 34) | class RunGraphByVmEnv(EnvVar): method __init__ (line 35) | def __init__(self): class Graph (line 45) | class Graph(flow.nn.Graph): method __init__ (line 46) | def __init__(self, m): method build (line 50) | def build(self, x): class M (line 54) | class M(flow.nn.Module): method __init__ (line 55) | def __init__(self): method forward (line 59) | def forward(self, x): function test_run_graph_by_vm (line 67) | def test_run_graph_by_vm(capsys): FILE: python/oneflow/test/graph/test_to_global.py class MyModule1 (line 95) | class MyModule1(flow.nn.Module): method __init__ (line 96) | def __init__(self, weight): method forward (line 102) | def forward(self, x): class MyModule2 (line 115) | class MyModule2(flow.nn.Module): method __init__ (line 116) | def __init__(self, weight): method forward (line 122) | def forward(self, x): class MyModule3 (line 133) | class MyModule3(flow.nn.Module): method __init__ (line 134) | def __init__(self, transpose_a=False, transpose_b=False): method forward (line 140) | def forward(self, x, y): class GlobalToModule (line 147) | class GlobalToModule(flow.nn.Module): method __init__ (line 148) | def __init__(self, device="cuda"): method forward (line 152) | def forward(self, x): class FreeTensorModule (line 156) | class FreeTensorModule(flow.nn.Module): method __init__ (line 157) | def __init__(self, shape, placement, sbp): method forward (line 163) | def forward(self, x): class ToPlacementModule (line 170) | class ToPlacementModule(flow.nn.Module): method __init__ (line 171) | def __init__(self, placement): method forward (line 175) | def forward(self, x): class MyGraph (line 179) | class MyGraph(flow.nn.Graph): method __init__ (line 180) | def __init__(self, module, optimizer=None): method build (line 186) | def build(self, *arg): class ToGlobalGraphTestCase (line 195) | class ToGlobalGraphTestCase(oneflow.unittest.TestCase): method test_fwd_P2B (line 197) | def test_fwd_P2B(test_case): method test_bwd_P2B (line 233) | def test_bwd_P2B(test_case): method test_multi_graph (line 278) | def test_multi_graph(test_case): method test_global_to (line 331) | def test_global_to(test_case): method test_free_tensor_to_global (line 352) | def test_free_tensor_to_global(test_case): method test_to_placement (line 375) | def test_to_placement(test_case): method test_to_dtype (line 413) | def test_to_dtype(test_case): class MyModule5 (line 447) | class MyModule5(flow.nn.Module): method __init__ (line 448) | def __init__(self, transpose_a=False, transpose_b=False, sbp=[]): method forward (line 454) | def forward(self, x, y): class ToGlobal2DGraphTestCase (line 464) | class ToGlobal2DGraphTestCase(oneflow.unittest.TestCase): method test_matmul (line 465) | def test_matmul(test_case): class TestLazy1dTo2dGlobal (line 483) | class TestLazy1dTo2dGlobal(flow.unittest.TestCase): method test_lazy_1d_to_2d_sbp (line 484) | def test_lazy_1d_to_2d_sbp(test_case): FILE: python/oneflow/test/graph/test_tvm_frontend_dependency_on_graph.py class TvmFrontedGraph (line 27) | class TvmFrontedGraph(flow.nn.Graph): method __init__ (line 28) | def __init__(self, module): method build (line 32) | def build(self, x): function parse_attr (line 37) | def parse_attr(attr): function is_user_op (line 75) | def is_user_op(node): class TestConvertDependency (line 82) | class TestConvertDependency(flow.unittest.TestCase): method test_get_params (line 83) | def test_get_params(test_case): method test_infos_of_nodes (line 98) | def test_infos_of_nodes(test_case): method test_buffer_convert_dependence (line 208) | def test_buffer_convert_dependence(test_case): FILE: python/oneflow/test/graph/test_user_op_expr.py function _get_c_tensor (line 32) | def _get_c_tensor(t): function _test_user_op_graph (line 39) | def _test_user_op_graph(test_case, is_cuda): class TestUserOpGraph (line 145) | class TestUserOpGraph(unittest.TestCase): method test_user_op_graph_cpu (line 146) | def test_user_op_graph_cpu(test_case): method test_user_op_graph_gpu (line 150) | def test_user_op_graph_gpu(test_case): FILE: python/oneflow/test/graph/test_util.py function GenCartesianProduct (line 28) | def GenCartesianProduct(sets): function GenArgList (line 38) | def GenArgList(arg_dict): function GenArgDict (line 45) | def GenArgDict(arg_dict): class Args (line 49) | class Args: method __init__ (line 50) | def __init__(self, flow_args, tf_args=None): method __str__ (line 57) | def __str__(self): method __repr__ (line 60) | def __repr__(self): function FlattenArray (line 84) | def FlattenArray(input_array): function Array2Numpy (line 91) | def Array2Numpy(input_array, target_shape): function Index2Coordinate (line 95) | def Index2Coordinate(idx, tensor_shape): function Coordinate2Index (line 106) | def Coordinate2Index(coordinate, tensor_shape): function generate_graph (line 118) | def generate_graph(func): FILE: python/oneflow/test/graph/test_variable_op_expr.py class TestFeedVariableTensor (line 34) | class TestFeedVariableTensor(unittest.TestCase): method test_feed_var_tensor (line 35) | def test_feed_var_tensor(test_case): FILE: python/oneflow/test/misc/mock_example.py function f (line 21) | def f(): FILE: python/oneflow/test/misc/test_autograd_functional.py function _func_tensor (line 25) | def _func_tensor(x): function _func_scalar (line 29) | def _func_scalar(x): function _func_multi_tensor (line 33) | def _func_multi_tensor(x, y): function _func_multi_scalar (line 37) | def _func_multi_scalar(x, y): function _func_scalar2tensor (line 41) | def _func_scalar2tensor(x): class TestAutogradFunctional (line 46) | class TestAutogradFunctional(flow.unittest.TestCase): method test_vjp (line 48) | def test_vjp(test_case): method test_jvp (line 62) | def test_jvp(test_case): method test_vhp (line 83) | def test_vhp(test_case): method test_hvp (line 99) | def test_hvp(test_case): method test_jacobian (line 116) | def test_jacobian(test_case): method test_hessian (line 141) | def test_hessian(test_case): FILE: python/oneflow/test/misc/test_distributed_env_vars.py class TestDistributedEnvVars (line 22) | class TestDistributedEnvVars(flow.unittest.TestCase): method test_default (line 24) | def test_default(test_case): method test_1n2d (line 35) | def test_1n2d(test_case): FILE: python/oneflow/test/misc/test_empty_cache.py class TestEmptyCache (line 24) | class TestEmptyCache(flow.unittest.TestCase): method test_cuda_to_cpu_empty_cache (line 25) | def test_cuda_to_cpu_empty_cache(test_case): FILE: python/oneflow/test/misc/test_env_cuda.py class TestEnv (line 26) | class TestEnv(flow.unittest.TestCase): method test_get_device_count (line 27) | def test_get_device_count(test_case): method test_current_device_idx (line 30) | def test_current_device_idx(test_case): method test_cuda_is_available (line 33) | def test_cuda_is_available(test_case): method test_cuda_synchronize (line 36) | def test_cuda_synchronize(test_case): method test_cuda_get_device_name (line 52) | def test_cuda_get_device_name(test_case): method test_cuda_get_device_capability (line 55) | def test_cuda_get_device_capability(test_case): method test_cuda_mem_get_info (line 58) | def test_cuda_mem_get_info(test_case): FILE: python/oneflow/test/misc/test_manual_seed_api.py class TestManualSeedApi (line 28) | class TestManualSeedApi(flow.unittest.TestCase): method test_cuda_manual_seed_all (line 29) | def test_cuda_manual_seed_all(test_case): method test_cuda_manual_seed (line 35) | def test_cuda_manual_seed(test_case): method test_manual_seed (line 45) | def test_manual_seed(test_case): method test_set_get_rng_state (line 51) | def test_set_get_rng_state(test_case): FILE: python/oneflow/test/misc/test_mock_diffusers.py class TestMock (line 26) | class TestMock(flow.unittest.TestCase): method test_mock_diffusers (line 27) | def test_mock_diffusers(test_case): FILE: python/oneflow/test/misc/test_mock_scope.py class TestMock (line 39) | class TestMock(flow.unittest.TestCase): method test_with (line 40) | def test_with(test_case): method test_simple (line 50) | def test_simple(test_case): method test_import_from (line 62) | def test_import_from(test_case): method test_error (line 77) | def test_error(test_case): method test_nested_with (line 101) | def test_nested_with(test_case): method test_noop_disable (line 113) | def test_noop_disable(test_case): method test_3rd_party (line 120) | def test_3rd_party(test_case): method test_env_var (line 126) | def test_env_var(test_case): method test_dummy_obj_fallback (line 136) | def test_dummy_obj_fallback(test_case): method test_mock_torchvision (line 144) | def test_mock_torchvision(test_case): method test_mock_lazy_for_loop (line 151) | def test_mock_lazy_for_loop(test_case): method test_mock_lazy_in_if (line 159) | def test_mock_lazy_in_if(test_case): method test_hazard_list (line 166) | def test_hazard_list(test_case): method test_isinstance (line 173) | def test_isinstance(test_case): method test_with_statement (line 179) | def test_with_statement(test_case): method test_setattr (line 191) | def test_setattr(test_case): method test_hasattr_and_getattr_in_lazy_mode (line 200) | def test_hasattr_and_getattr_in_lazy_mode(test_case): method test_mock_extra_dict (line 216) | def test_mock_extra_dict(test_case): function test_verbose (line 224) | def test_verbose(capsys): FILE: python/oneflow/test/misc/test_np_dtype_converter.py class TestNpDtypeConverter (line 25) | class TestNpDtypeConverter(flow.unittest.TestCase): method test_np_dtype_converter (line 26) | def test_np_dtype_converter(test_case): FILE: python/oneflow/test/misc/test_placement.py class TestPlacement (line 22) | class TestPlacement(flow.unittest.TestCase): method test_placement_all_cuda (line 24) | def test_placement_all_cuda(test_case): method test_placement_all_cpu (line 33) | def test_placement_all_cpu(test_case): FILE: python/oneflow/test/misc/test_pybind11_caster.py class TestPybind11Caster (line 22) | class TestPybind11Caster(flow.unittest.TestCase): method test_optional (line 23) | def test_optional(test_case): method test_maybe (line 31) | def test_maybe(test_case): method test_maybe_void (line 34) | def test_maybe_void(test_case): method test_return_maybe_shared_ptr (line 37) | def test_return_maybe_shared_ptr(test_case): method test_pass_optional_shared_ptr (line 48) | def test_pass_optional_shared_ptr(test_case): FILE: python/oneflow/test/modules/image_test_util.py function get_coco (line 32) | def get_coco(anno_file): function random_sample_images_from_coco (line 41) | def random_sample_images_from_coco( function read_images_by_cv (line 64) | def read_images_by_cv(image_files, dtype, channels=3): function read_images_by_pil (line 73) | def read_images_by_pil(image_files, dtype, channels=3): function infer_images_static_shape (line 99) | def infer_images_static_shape(images, channels=3): function compute_keep_aspect_ratio_resized_size (line 123) | def compute_keep_aspect_ratio_resized_size( function infer_keep_aspect_ratio_resized_images_static_shape (line 143) | def infer_keep_aspect_ratio_resized_images_static_shape( FILE: python/oneflow/test/modules/optimizer_test_util.py function clip_grad_norm_np (line 19) | def clip_grad_norm_np(np_grad, max_norm, norm_type): FILE: python/oneflow/test/modules/sync_batchnorm_test_util.py function md5 (line 24) | def md5(fname): function download_file (line 36) | def download_file(out_path: str, url): function ensure_datas (line 52) | def ensure_datas(): FILE: python/oneflow/test/modules/test_0_dim_tensor.py function _test_0_dim_tensor (line 29) | def _test_0_dim_tensor(test_case, device): function _test_scalar_mul (line 45) | def _test_scalar_mul(test_case, device): function _test_slice (line 52) | def _test_slice(test_case, device): function _test_slice_backward (line 61) | def _test_slice_backward(test_case, device): function _test_slice_scalar_graph (line 82) | def _test_slice_scalar_graph(test_case, device): function _test_slice_scalar_train_graph (line 111) | def _test_slice_scalar_train_graph(test_case, device): class TestZeroDimensionTensor (line 161) | class TestZeroDimensionTensor(flow.unittest.TestCase): method test_0_dim_tensor (line 162) | def test_0_dim_tensor(test_case): FILE: python/oneflow/test/modules/test_TripletMarginLoss.py class TestTripletMarginLoss (line 27) | class TestTripletMarginLoss(flow.unittest.TestCase): method test_triplet_marginloss_with_random_data (line 30) | def test_triplet_marginloss_with_random_data(test_case): FILE: python/oneflow/test/modules/test_abs.py class TestAbsModule (line 25) | class TestAbsModule(flow.unittest.TestCase): method test_abs_with_0_size_data (line 27) | def test_abs_with_0_size_data(test_case): method test_abs_with_0dim_data (line 34) | def test_abs_with_0dim_data(test_case): method profile_abs (line 41) | def profile_abs(test_case): FILE: python/oneflow/test/modules/test_activation.py class TestReLUModule (line 32) | class TestReLUModule(flow.unittest.TestCase): method test_relu_module_with_random_data (line 34) | def test_relu_module_with_random_data(test_case): method test_relu_module_with_0dim_data (line 44) | def test_relu_module_with_0dim_data(test_case): method test_relu_module_with_0_size_data (line 54) | def test_relu_module_with_0_size_data(test_case): method profile_relu (line 64) | def profile_relu(test_case): class TestReLU6Module (line 72) | class TestReLU6Module(flow.unittest.TestCase): method test_relu6_module_with_random_data (line 74) | def test_relu6_module_with_random_data(test_case): method test_relu6_module_with_0dim_data (line 84) | def test_relu6_module_with_0dim_data(test_case): method test_relu6_module_with_0_size_data (line 94) | def test_relu6_module_with_0_size_data(test_case): class TestTanh (line 105) | class TestTanh(flow.unittest.TestCase): method test_tanh_module_with_random_data (line 107) | def test_tanh_module_with_random_data(test_case): method test_tanh_module_with_0dim_data (line 117) | def test_tanh_module_with_0dim_data(test_case): method test_tanh_module_with_0_size_data (line 127) | def test_tanh_module_with_0_size_data(test_case): method test_flow_tanh_with_random_data (line 137) | def test_flow_tanh_with_random_data(test_case): method test_flow_tanh_with_0dim_data (line 144) | def test_flow_tanh_with_0dim_data(test_case): method test_flow_tanh_with_0_size_data (line 151) | def test_flow_tanh_with_0_size_data(test_case): method profile_tanh (line 158) | def profile_tanh(test_case): class TestELUModule (line 164) | class TestELUModule(flow.unittest.TestCase): method test_elu_module_with_random_data (line 166) | def test_elu_module_with_random_data(test_case): method test_elu_module_with_0dim_data (line 176) | def test_elu_module_with_0dim_data(test_case): method test_elu_module_with_0_size_data (line 186) | def test_elu_module_with_0_size_data(test_case): method profile_elu (line 196) | def profile_elu(test_case): class TestCELUModule (line 202) | class TestCELUModule(flow.unittest.TestCase): method test_celu_module_with_random_data (line 204) | def test_celu_module_with_random_data(test_case): method test_celu_module_with_0dim_data (line 214) | def test_celu_module_with_0dim_data(test_case): method test_celu_module_with_0_size_data (line 224) | def test_celu_module_with_0_size_data(test_case): method test_inplace_celu_module (line 234) | def test_inplace_celu_module(test_case): method profile_celu (line 244) | def profile_celu(test_case): class TestGelu (line 252) | class TestGelu(flow.unittest.TestCase): method test_gelu_module_with_random_data (line 254) | def test_gelu_module_with_random_data(test_case): method test_gelu_module_with_0dim_data (line 264) | def test_gelu_module_with_0dim_data(test_case): method profile_gelu (line 274) | def profile_gelu(test_case): class TestFastGelu (line 284) | class TestFastGelu(flow.unittest.TestCase): method test_fast_gelu (line 286) | def test_fast_gelu(test_case): method test_fast_gelu_fp16 (line 293) | def test_fast_gelu_fp16(test_case): method test_fast_gelu_scalar (line 299) | def test_fast_gelu_scalar(test_case): method profile_fast_gelu (line 305) | def profile_fast_gelu(test_case): class TestSigmoidModule (line 311) | class TestSigmoidModule(flow.unittest.TestCase): method test_sigmoid_flow_with_half_data (line 314) | def test_sigmoid_flow_with_half_data(test_case): method test_sigmoid_module_with_random_data (line 321) | def test_sigmoid_module_with_random_data(test_case): method test_sigmoid_module_with_0dim_data (line 331) | def test_sigmoid_module_with_0dim_data(test_case): method test_sigmoid_flow_with_random_data (line 341) | def test_sigmoid_flow_with_random_data(test_case): method test_sigmoid_flow_with_0dim_data (line 348) | def test_sigmoid_flow_with_0dim_data(test_case): method test_sigmoid_tensor_with_random_data (line 355) | def test_sigmoid_tensor_with_random_data(test_case): method test_sigmoid_tensor_with_0dim_data (line 362) | def test_sigmoid_tensor_with_0dim_data(test_case): method profile_sigmoid (line 369) | def profile_sigmoid(test_case): class TestHardsigmoidModule (line 375) | class TestHardsigmoidModule(flow.unittest.TestCase): method test_hardsigmoid_inplace (line 376) | def test_hardsigmoid_inplace(test_case): method test_hardsigmoid_module_with_random_data (line 415) | def test_hardsigmoid_module_with_random_data(test_case): method test_hardsigmoid_module_with_0dim_data (line 425) | def test_hardsigmoid_module_with_0dim_data(test_case): method test_functional_hardsigmoid_with_random_data (line 435) | def test_functional_hardsigmoid_with_random_data(test_case): method test_functional_hardsigmoid_with_0dim_data (line 442) | def test_functional_hardsigmoid_with_0dim_data(test_case): method profile_hardsigmoid (line 449) | def profile_hardsigmoid(test_case): function do_test_softmax (line 456) | def do_test_softmax(batch_size: int, log_softmax: bool = False): class TestSoftmax (line 474) | class TestSoftmax(flow.unittest.TestCase): method test_softmax_module_with_random_data (line 476) | def test_softmax_module_with_random_data(test_case): method test_softmax_module_with_batch_size_equal_1024 (line 480) | def test_softmax_module_with_batch_size_equal_1024(test_case): method test_softmax_module_with_batch_size_equal_5120 (line 484) | def test_softmax_module_with_batch_size_equal_5120(test_case): method test_softmax_module_with_batch_size_equal_10240 (line 488) | def test_softmax_module_with_batch_size_equal_10240(test_case): method profile_softmax (line 492) | def profile_softmax(test_case): class TestLogSoftmaxModule (line 498) | class TestLogSoftmaxModule(flow.unittest.TestCase): method test_logsoftmax_module_with_random_data (line 500) | def test_logsoftmax_module_with_random_data(test_case): method test_softmax_module_with_batch_size_equal_1024 (line 504) | def test_softmax_module_with_batch_size_equal_1024(test_case): method test_softmax_module_with_batch_size_equal_5120 (line 508) | def test_softmax_module_with_batch_size_equal_5120(test_case): method test_softmax_module_with_batch_size_equal_10240 (line 512) | def test_softmax_module_with_batch_size_equal_10240(test_case): method profile_logsoftmax (line 516) | def profile_logsoftmax(test_case): class TestLogSigmoidModule (line 522) | class TestLogSigmoidModule(flow.unittest.TestCase): method test_logsigmoid_module_with_random_data (line 524) | def test_logsigmoid_module_with_random_data(test_case): method test_logsigmoid_module_with_0dim_data (line 534) | def test_logsigmoid_module_with_0dim_data(test_case): method profile_logsigmoid (line 544) | def profile_logsigmoid(test_case): function numpy_softplus (line 549) | def numpy_softplus(x, beta, threshold): function _test_softplus (line 555) | def _test_softplus(test_case, device): function _test_softplus_beta (line 564) | def _test_softplus_beta(test_case, device): function _test_softplus_threshold (line 573) | def _test_softplus_threshold(test_case, device): function _test_softplus_backward (line 585) | def _test_softplus_backward(test_case, device): class TestSoftplusModule (line 597) | class TestSoftplusModule(flow.unittest.TestCase): method test_softplus (line 598) | def test_softplus(test_case): method test_softplus_module_with_random_data (line 612) | def test_softplus_module_with_random_data(test_case): method profile_softplus (line 622) | def profile_softplus(test_case): class TestHardswishModule (line 628) | class TestHardswishModule(flow.unittest.TestCase): method test_hardswish_module_with_random_data (line 630) | def test_hardswish_module_with_random_data(test_case): method test_hardswish_module_with_0dim_data (line 640) | def test_hardswish_module_with_0dim_data(test_case): method profile_hardswish (line 650) | def profile_hardswish(test_case): class TestHardtanhModule (line 656) | class TestHardtanhModule(flow.unittest.TestCase): method test_hardtanh_module_with_random_data (line 658) | def test_hardtanh_module_with_random_data(test_case): method test_hardtanh_module_with_0dim_data (line 671) | def test_hardtanh_module_with_0dim_data(test_case): method profile_hardtanh (line 684) | def profile_hardtanh(test_case): class TestLeakyReLUModule (line 694) | class TestLeakyReLUModule(flow.unittest.TestCase): method test_leakyrelu_module_with_random_data (line 696) | def test_leakyrelu_module_with_random_data(test_case): method test_leakyrelu_module_with_inplace_arg (line 706) | def test_leakyrelu_module_with_inplace_arg(test_case): method test_leakyrelu_module_with_0dim_data (line 718) | def test_leakyrelu_module_with_0dim_data(test_case): method profile_leaky_relu (line 728) | def profile_leaky_relu(test_case): class TestMishModule (line 736) | class TestMishModule(flow.unittest.TestCase): method test_mish_module_with_random_data (line 738) | def test_mish_module_with_random_data(test_case): method test_mish_module_with_0dim_data (line 748) | def test_mish_module_with_0dim_data(test_case): method profile_mish (line 758) | def profile_mish(test_case): class TestSiluModule (line 764) | class TestSiluModule(flow.unittest.TestCase): method test_silu_module_with_random_data (line 766) | def test_silu_module_with_random_data(test_case): method test_silu_module_with_0dim_data (line 776) | def test_silu_module_with_0dim_data(test_case): method profile_silu (line 786) | def profile_silu(test_case): class TestSeluModule (line 792) | class TestSeluModule(flow.unittest.TestCase): method test_selu_module_with_random_data (line 794) | def test_selu_module_with_random_data(test_case): method test_selu_module_with_0dim_data (line 804) | def test_selu_module_with_0dim_data(test_case): method profile_selu (line 814) | def profile_selu(test_case): class TestSoftsignModule (line 820) | class TestSoftsignModule(flow.unittest.TestCase): method test_softsign_module_with_random_data (line 822) | def test_softsign_module_with_random_data(test_case): method profile_softsign (line 833) | def profile_softsign(test_case): class TestReluFunction (line 839) | class TestReluFunction(flow.unittest.TestCase): method test_flow_relu_with_random_data (line 841) | def test_flow_relu_with_random_data(test_case): method test_flow_relu_with_0dim_data (line 848) | def test_flow_relu_with_0dim_data(test_case): class TestRelu6Function (line 856) | class TestRelu6Function(flow.unittest.TestCase): method test_flow_nn_functional_relu6_with_random_data (line 858) | def test_flow_nn_functional_relu6_with_random_data(test_case): method test_flow_nn_functional_relu6_with_0dim_data (line 865) | def test_flow_nn_functional_relu6_with_0dim_data(test_case): method profile_relu6 (line 872) | def profile_relu6(test_case): class TestLogSigmoidFunction (line 878) | class TestLogSigmoidFunction(flow.unittest.TestCase): method test_flow_nn_functional_logsigmoid_with_random_data (line 880) | def test_flow_nn_functional_logsigmoid_with_random_data(test_case): method test_flow_nn_functional_logsigmoid_with_0dim_data (line 887) | def test_flow_nn_functional_logsigmoid_with_0dim_data(test_case): class TestHardshrinkModule (line 895) | class TestHardshrinkModule(flow.unittest.TestCase): method test_hardshrink_module_with_random_data (line 897) | def test_hardshrink_module_with_random_data(test_case): method test_hardshrink_module_with_0dim_data (line 907) | def test_hardshrink_module_with_0dim_data(test_case): method test_hardshrink_module_with_0_size_data (line 917) | def test_hardshrink_module_with_0_size_data(test_case): method profile_hardshrink (line 927) | def profile_hardshrink(test_case): class TestSoftshrinkModule (line 933) | class TestSoftshrinkModule(flow.unittest.TestCase): method test_softshrink_module_with_random_data (line 935) | def test_softshrink_module_with_random_data(test_case): method test_softshrink_module_with_0dim_data (line 945) | def test_softshrink_module_with_0dim_data(test_case): method test_softshrink_module_with_0_size_data (line 955) | def test_softshrink_module_with_0_size_data(test_case): method profile_softshrink (line 965) | def profile_softshrink(test_case): class TestThresholdModule (line 971) | class TestThresholdModule(flow.unittest.TestCase): method test_threshold_module_with_random_data (line 973) | def test_threshold_module_with_random_data(test_case): method test_threshold_module_with_0dim_data (line 985) | def test_threshold_module_with_0dim_data(test_case): method test_threshold_module_with_0_size_data (line 997) | def test_threshold_module_with_0_size_data(test_case): method profile_threshold (line 1009) | def profile_threshold(test_case): FILE: python/oneflow/test/modules/test_adaptive_max_pool.py class TestAdaptiveMaxPool (line 33) | class TestAdaptiveMaxPool(flow.unittest.TestCase): method test_adaptive_maxpool1d (line 36) | def test_adaptive_maxpool1d(test_case): method test_adaptive_maxpool2d_manually (line 46) | def test_adaptive_maxpool2d_manually(test_case): method profile_adaptive_max_pool1d (line 96) | def profile_adaptive_max_pool1d(test_case): method profile_adaptive_max_pool2d (line 100) | def profile_adaptive_max_pool2d(test_case): method profile_adaptive_max_pool3d (line 105) | def profile_adaptive_max_pool3d(test_case): FILE: python/oneflow/test/modules/test_adaptive_pool.py class TestAdaptiveAvgPool (line 37) | class TestAdaptiveAvgPool(flow.unittest.TestCase): method test_adaptive_avgpool1d (line 39) | def test_adaptive_avgpool1d(test_case): method profile_adaptive_avg_pool1d (line 49) | def profile_adaptive_avg_pool1d(test_case): method test_adaptive_avgpool2d (line 53) | def test_adaptive_avgpool2d(test_case): method profile_adaptive_avg_pool2d (line 63) | def profile_adaptive_avg_pool2d(test_case): method test_adaptive_avgpool3d (line 72) | def test_adaptive_avgpool3d(test_case): method profile_adaptive_avg_pool3d (line 82) | def profile_adaptive_avg_pool3d(test_case): class TestAdaptiveAvgPoolFunctional (line 88) | class TestAdaptiveAvgPoolFunctional(flow.unittest.TestCase): method test_adaptive_avgpool1d_functional (line 90) | def test_adaptive_avgpool1d_functional(test_case): method test_adaptive_avgpool2d_functional (line 96) | def test_adaptive_avgpool2d_functional(test_case): method test_adaptive_avgpool3d_functional (line 106) | def test_adaptive_avgpool3d_functional(test_case): FILE: python/oneflow/test/modules/test_adaptive_pool_fp16.py class Test_CpuFp16_AdaptiveAvgPool (line 34) | class Test_CpuFp16_AdaptiveAvgPool(flow.unittest.TestCase): method test_adaptive_avgpool1d (line 36) | def test_adaptive_avgpool1d(test_case): method profile_adaptive_avg_pool1d (line 47) | def profile_adaptive_avg_pool1d(test_case): method test_adaptive_avgpool2d (line 51) | def test_adaptive_avgpool2d(test_case): method profile_adaptive_avg_pool2d (line 62) | def profile_adaptive_avg_pool2d(test_case): method test_adaptive_avgpool3d (line 71) | def test_adaptive_avgpool3d(test_case): method profile_adaptive_avg_pool3d (line 82) | def profile_adaptive_avg_pool3d(test_case): class Test_CpuFp16_AdaptiveAvgPoolFunctional (line 90) | class Test_CpuFp16_AdaptiveAvgPoolFunctional(flow.unittest.TestCase): method test_adaptive_avgpool1d_functional (line 92) | def test_adaptive_avgpool1d_functional(test_case): method test_adaptive_avgpool2d_functional (line 99) | def test_adaptive_avgpool2d_functional(test_case): method test_adaptive_avgpool3d_functional (line 106) | def test_adaptive_avgpool3d_functional(test_case): FILE: python/oneflow/test/modules/test_add.py function _test_add_forward (line 29) | def _test_add_forward(test_case, shape, device): function _test_add_backward (line 57) | def _test_add_backward(test_case, shape, device): function _test_inplace_add (line 69) | def _test_inplace_add(test_case, shape, device): function _test_inplace_add_with_type_promotion (line 159) | def _test_inplace_add_with_type_promotion(test_case, shape, device): function _test_inplace_add_0_size_tensor (line 170) | def _test_inplace_add_0_size_tensor(test_case, shape, device): class TestAddModule (line 178) | class TestAddModule(flow.unittest.TestCase): method test_add (line 179) | def test_add(test_case): method test_0_size_add (line 194) | def test_0_size_add(test_case): method test_0dim_inplace_add (line 202) | def test_0dim_inplace_add(test_case): method test_0dim_two_inplace_add (line 210) | def test_0dim_two_inplace_add(test_case): method test_add_with_alpha (line 218) | def test_add_with_alpha(test_case): method test_bool_add (line 232) | def test_bool_add(test_case): method test_0shape_bool_add (line 240) | def test_0shape_bool_add(test_case): method test_0dim_bool_inplace_add (line 248) | def test_0dim_bool_inplace_add(test_case): method test_0dim_two_inplace_add (line 256) | def test_0dim_two_inplace_add(test_case): method test_add_with_alpha_0dim (line 264) | def test_add_with_alpha_0dim(test_case): method profile_add (line 278) | def profile_add(test_case): method test_non_contiguous_inplace_add (line 283) | def test_non_contiguous_inplace_add(test_case): method test_scalar_add_with_random_devices (line 292) | def test_scalar_add_with_random_devices(test_case): FILE: python/oneflow/test/modules/test_addcdiv.py class TestAddcdiv (line 23) | class TestAddcdiv(flow.unittest.TestCase): method test_addcdiv (line 25) | def test_addcdiv(test_case): method test_tensor_addcdiv (line 37) | def test_tensor_addcdiv(test_case): method test_tensor_addcdiv_inplace (line 49) | def test_tensor_addcdiv_inplace(test_case): method profile_addcdiv (line 62) | def profile_addcdiv(test_case): FILE: python/oneflow/test/modules/test_addcmul.py class TestAddcmul (line 23) | class TestAddcmul(flow.unittest.TestCase): method test_addcmul (line 25) | def test_addcmul(test_case): method test_tensor_addcmul (line 38) | def test_tensor_addcmul(test_case): method test_tensor_addcmul_inplace (line 51) | def test_tensor_addcmul_inplace(test_case): method profile_addcmul (line 65) | def profile_addcmul(test_case): FILE: python/oneflow/test/modules/test_addmm.py function _test_addmm (line 29) | def _test_addmm(test_case, shape, alpha, beta, device): function _test_addmm_backward (line 41) | def _test_addmm_backward(test_case, shape, alpha, beta, device): class TestAddmm (line 59) | class TestAddmm(flow.unittest.TestCase): method test_addmm (line 60) | def test_addmm(test_case): method test_addmm_flow_with_random_data (line 71) | def test_addmm_flow_with_random_data(test_case): method test_addmm_broadcast_flow_with_random_data (line 86) | def test_addmm_broadcast_flow_with_random_data(test_case): method profile_addmm (line 101) | def profile_addmm(test_case): FILE: python/oneflow/test/modules/test_affine_grid.py class TestAffineGrid (line 30) | class TestAffineGrid(flow.unittest.TestCase): method test_affine_grid_2d (line 31) | def test_affine_grid_2d(test_case): method test_affine_grid_3d (line 49) | def test_affine_grid_3d(test_case): method test_flow_affine_grid_2d_with_random_data (line 94) | def test_flow_affine_grid_2d_with_random_data(test_case): method test_flow_affine_grid_3d_with_random_data (line 108) | def test_flow_affine_grid_3d_with_random_data(test_case): method profile_affine_grid (line 123) | def profile_affine_grid(test_case): FILE: python/oneflow/test/modules/test_allclose.py function _perturbate (line 29) | def _perturbate(x): class TestAllClose (line 43) | class TestAllClose(flow.unittest.TestCase): method test_allclose_with_random_data (line 45) | def test_allclose_with_random_data(test_case): method test_allclose_with_0dim_data (line 54) | def test_allclose_with_0dim_data(test_case): method test_tensor_allclose_with_random_data (line 63) | def test_tensor_allclose_with_random_data(test_case): method test_allclose_broadcast (line 72) | def test_allclose_broadcast(test_case): FILE: python/oneflow/test/modules/test_allreduce.py class TestAllReduce (line 26) | class TestAllReduce(flow.unittest.TestCase): method test_all_reduce (line 28) | def test_all_reduce(test_case): method test_all_reduce_2nodes (line 42) | def test_all_reduce_2nodes(test_case): FILE: python/oneflow/test/modules/test_amax.py function __check (line 26) | def __check(test_case, input, dim, keepdim, device): function _test_amax_with_negative_dim (line 35) | def _test_amax_with_negative_dim(test_case, device): function _test_amax_with_positive_dim (line 44) | def _test_amax_with_positive_dim(test_case, device): function _test_amax_with_multiple_axes (line 53) | def _test_amax_with_multiple_axes(test_case, device): function _test_amax_with_empty_dim (line 65) | def _test_amax_with_empty_dim(test_case, device): function _test_amax_keepdim (line 73) | def _test_amax_keepdim(test_case, device): function _test_amax_not_keepdim (line 82) | def _test_amax_not_keepdim(test_case, device): class TestAmax (line 92) | class TestAmax(flow.unittest.TestCase): method test_amax (line 93) | def test_amax(test_case): method test_amax_with_random_data_single_dim (line 108) | def test_amax_with_random_data_single_dim(test_case): method test_amax_with_random_data_empty_dim (line 116) | def test_amax_with_random_data_empty_dim(test_case): method test_amax_with_random_data_multi_dims (line 124) | def test_amax_with_random_data_multi_dims(test_case): method profile_amax (line 135) | def profile_amax(test_case): FILE: python/oneflow/test/modules/test_amin.py function __check (line 25) | def __check(test_case, input, dim, keepdim, device): function _test_amin_with_negative_dim (line 34) | def _test_amin_with_negative_dim(test_case, device): function _test_amin_with_positive_dim (line 43) | def _test_amin_with_positive_dim(test_case, device): function _test_amin_with_multiple_axes (line 52) | def _test_amin_with_multiple_axes(test_case, device): function _test_amin_with_empty_dim (line 64) | def _test_amin_with_empty_dim(test_case, device): function _test_amin_keepdim (line 72) | def _test_amin_keepdim(test_case, device): function _test_amin_not_keepdim (line 81) | def _test_amin_not_keepdim(test_case, device): class TestAmin (line 91) | class TestAmin(flow.unittest.TestCase): method test_amin (line 92) | def test_amin(test_case): method test_amin_with_random_data_single_dim (line 107) | def test_amin_with_random_data_single_dim(test_case): method test_amin_with_random_data_empty_dim (line 115) | def test_amin_with_random_data_empty_dim(test_case): method test_amin_with_random_data_multi_dims (line 123) | def test_amin_with_random_data_multi_dims(test_case): method profile_amin (line 134) | def profile_amin(test_case): FILE: python/oneflow/test/modules/test_arange.py function _test_arange (line 29) | def _test_arange(test_case, device): function _test_arange_step_prarm (line 39) | def _test_arange_step_prarm(test_case, device): function _test_arange_more_params (line 45) | def _test_arange_more_params(test_case, device): function _test_arange_backward (line 51) | def _test_arange_backward(test_case, device): function _test_arange_input_tensor_type (line 65) | def _test_arange_input_tensor_type(test_case, device): class TestArange (line 78) | class TestArange(flow.unittest.TestCase): method test_arange (line 79) | def test_arange(test_case): method test_arange_with_random_data (line 93) | def test_arange_with_random_data(test_case): method test_arange_with_float_delta (line 103) | def test_arange_with_float_delta(test_case): method test_arange_input_float_scalar_tensor (line 113) | def test_arange_input_float_scalar_tensor(test_case): method test_arange_input_float16_scalar_tensor (line 122) | def test_arange_input_float16_scalar_tensor(test_case): method test_global_naive (line 131) | def test_global_naive(test_case): method profile_arange (line 139) | def profile_arange(test_case): FILE: python/oneflow/test/modules/test_argmax.py function _test_argmax_axis_negative (line 29) | def _test_argmax_axis_negative(test_case, device): function _test_tensor_argmax (line 39) | def _test_tensor_argmax(test_case, device): function _test_argmax_axis_postive (line 50) | def _test_argmax_axis_postive(test_case, device): function _test_argmax_keepdims (line 60) | def _test_argmax_keepdims(test_case, device): function _test_argmax_dim_equal_none (line 72) | def _test_argmax_dim_equal_none(test_case, device): class TestArgmax (line 82) | class TestArgmax(flow.unittest.TestCase): method test_argmax (line 83) | def test_argmax(test_case): method test_argmax_with_random_data (line 97) | def test_argmax_with_random_data(test_case): method profile_argmax (line 105) | def profile_argmax(test_case): FILE: python/oneflow/test/modules/test_argmin.py function _test_argmin_axis_negative (line 29) | def _test_argmin_axis_negative(test_case, device): function _test_tensor_argmin (line 39) | def _test_tensor_argmin(test_case, device): function _test_argmin_axis_postive (line 50) | def _test_argmin_axis_postive(test_case, device): function _test_argmin_keepdims (line 60) | def _test_argmin_keepdims(test_case, device): function _test_argmin_dim_equal_none (line 72) | def _test_argmin_dim_equal_none(test_case, device): class TestArgmin (line 82) | class TestArgmin(flow.unittest.TestCase): method test_argmin (line 83) | def test_argmin(test_case): method test_argmin_with_random_data (line 97) | def test_argmin_with_random_data(test_case): FILE: python/oneflow/test/modules/test_argsort.py function _test_argsort (line 29) | def _test_argsort(test_case, data_shape, axis, descending, data_type, de... function _test_tensor_argsort (line 45) | def _test_tensor_argsort(test_case, data_shape, axis, descending, data_t... class TestArgsort (line 63) | class TestArgsort(flow.unittest.TestCase): method test_argsort (line 64) | def test_argsort(test_case): method test_argsort_with_random_data (line 76) | def test_argsort_with_random_data(test_case): method test_argsort_bool_with_random_data (line 85) | def test_argsort_bool_with_random_data(test_case): method profile_argsort (line 93) | def profile_argsort(test_case): FILE: python/oneflow/test/modules/test_argwhere.py function _test_argwhere (line 30) | def _test_argwhere(test_case, shape, device): class TestArgwhere (line 40) | class TestArgwhere(flow.unittest.TestCase): method test_argwhere (line 41) | def test_argwhere(test_case): method test_argwhere_with_random_data (line 51) | def test_argwhere_with_random_data(test_case): method profile_argwhere (line 65) | def profile_argwhere(test_case): FILE: python/oneflow/test/modules/test_as_strided.py class TestAsStrided (line 26) | class TestAsStrided(flow.unittest.TestCase): method test_flow_AsStrided (line 28) | def test_flow_AsStrided(test_case): method test_tensor_as_strided (line 48) | def test_tensor_as_strided(test_case): method test_flow_as_strided_tensor_method (line 68) | def test_flow_as_strided_tensor_method(test_case): method test_flow_as_strided_with_stride (line 78) | def test_flow_as_strided_with_stride(test_case): method test_flow_as_strided_bool (line 94) | def test_flow_as_strided_bool(test_case): method test_flow_as_strided_int8 (line 115) | def test_flow_as_strided_int8(test_case): method test_flow_as_strided_uint8 (line 136) | def test_flow_as_strided_uint8(test_case): method profile_as_strided (line 157) | def profile_as_strided(test_case): FILE: python/oneflow/test/modules/test_as_tensor.py class TestAsTensor (line 39) | class TestAsTensor(flow.unittest.TestCase): method test_tensor_type (line 40) | def test_tensor_type(test_case): method test_numpy_type (line 70) | def test_numpy_type(test_case): method test_other_type (line 103) | def test_other_type(test_case): method test_numpy_dtype_bug (line 147) | def test_numpy_dtype_bug(test_case): FILE: python/oneflow/test/modules/test_asyncs_thread.py class TestLocalThread (line 28) | class TestLocalThread(flow.unittest.TestCase): method test_stream (line 29) | def test_stream(test_case): class TestGlobalThread (line 35) | class TestGlobalThread(flow.unittest.TestCase): method test_cpu_stream (line 36) | def test_cpu_stream(test_case): method test_cuda_stream (line 47) | def test_cuda_stream(test_case): FILE: python/oneflow/test/modules/test_atleast.py class TestAtLeast (line 26) | class TestAtLeast(flow.unittest.TestCase): method test_atleast_1d_with_list_random_data (line 28) | def test_atleast_1d_with_list_random_data(test_case): method test_atleast_1d_with_random_data (line 36) | def test_atleast_1d_with_random_data(test_case): method test_atleast_2d_with_list_random_data (line 43) | def test_atleast_2d_with_list_random_data(test_case): method test_atleast_2d_with_random_data (line 52) | def test_atleast_2d_with_random_data(test_case): method test_atleast_3d_with_list_random_data (line 59) | def test_atleast_3d_with_list_random_data(test_case): method test_atleast_3d_with_random_data (line 69) | def test_atleast_3d_with_random_data(test_case): FILE: python/oneflow/test/modules/test_auto_to_global.py function _test_auto_to_global (line 28) | def _test_auto_to_global(test_case, device): class TestAutoToGlobal (line 47) | class TestAutoToGlobal(flow.unittest.TestCase): method test_auto_to_global (line 50) | def test_auto_to_global(test_case): FILE: python/oneflow/test/modules/test_autograd.py function _test_autograd_backward (line 28) | def _test_autograd_backward(test_case, shape, device): function _test_autograd_grad (line 60) | def _test_autograd_grad(test_case, shape, device): class TestAutograd (line 80) | class TestAutograd(flow.unittest.TestCase): method test_autograd_interface (line 81) | def test_autograd_interface(test_case): method test_accumulate_grad (line 90) | def test_accumulate_grad(test_case): method test_0dim_accumulate_grad (line 98) | def test_0dim_accumulate_grad(test_case): method test_scalar_leaf_tensor_backward (line 106) | def test_scalar_leaf_tensor_backward(test_case): method test_out_grad_with_different_dtype (line 113) | def test_out_grad_with_different_dtype(test_case): method test_grad_grad (line 120) | def test_grad_grad(test_case): method test_autograd_multiple_times (line 138) | def test_autograd_multiple_times(test_case): method test_autograd_set_acc_grad_and_backward (line 152) | def test_autograd_set_acc_grad_and_backward(test_case): method test_requires_grad_tensor_inplace_and_backward (line 170) | def test_requires_grad_tensor_inplace_and_backward(test_case): method test_retain_grad_for_leaf_tensor (line 178) | def test_retain_grad_for_leaf_tensor(test_case): method test_run_backward_and_grad_for_same_tensor (line 186) | def test_run_backward_and_grad_for_same_tensor(test_case): method test_no_grad_domain_call_backward (line 205) | def test_no_grad_domain_call_backward(test_case): method test_acc_grad_inplace_update (line 215) | def test_acc_grad_inplace_update(test_case): method test_autograd_grad_allow_unused (line 230) | def test_autograd_grad_allow_unused(test_case): method test_autograd_is_grads_batched (line 286) | def test_autograd_is_grads_batched(test_case): method test_autograd_grad_none_list (line 339) | def test_autograd_grad_none_list(test_case): FILE: python/oneflow/test/modules/test_autograd_function.py class TestAutogradFunction (line 27) | class TestAutogradFunction(flow.unittest.TestCase): method test_simple_input (line 29) | def test_simple_input(test_case): method test_multi_input (line 56) | def test_multi_input(test_case): method test_non_differentiable_interface (line 84) | def test_non_differentiable_interface(test_case): method test_partial_inputs_requires_grad (line 116) | def test_partial_inputs_requires_grad(test_case): method test_dynamic_attr_for_ctx (line 141) | def test_dynamic_attr_for_ctx(test_case): method test_backward_error_message (line 161) | def test_backward_error_message(test_case): method test_graph_test_multi_input (line 185) | def test_graph_test_multi_input(test_case): method test_autograd_function_memory (line 244) | def test_autograd_function_memory(test_case): FILE: python/oneflow/test/modules/test_autograd_mode.py class TestAutogradMode (line 23) | class TestAutogradMode(oneflow.unittest.TestCase): method test_grad_mode (line 24) | def test_grad_mode(test_case): method test_inference_mode (line 27) | def test_inference_mode(test_case): method test_enable_grad (line 50) | def test_enable_grad(test_case): method test_no_grad (line 62) | def test_no_grad(test_case): method test_set_grad_enabled (line 74) | def test_set_grad_enabled(test_case): FILE: python/oneflow/test/modules/test_avgpool.py class TestAvgPoolingModule (line 26) | class TestAvgPoolingModule(flow.unittest.TestCase): method test_avgpool1d_with_random_data (line 28) | def test_avgpool1d_with_random_data(test_case): method test_avgpool2d_with_random_data (line 44) | def test_avgpool2d_with_random_data(test_case): method test_avgpool2d_with_half_data (line 63) | def test_avgpool2d_with_half_data(test_case): method test_avgpool3d_with_random_data (line 86) | def test_avgpool3d_with_random_data(test_case): class TestAvgPoolingFunctional (line 106) | class TestAvgPoolingFunctional(flow.unittest.TestCase): method test_avgpool1d_functional (line 108) | def test_avgpool1d_functional(test_case): method test_avgpool2d_functional (line 122) | def test_avgpool2d_functional(test_case): method test_avgpool3d_functional (line 136) | def test_avgpool3d_functional(test_case): method profile_avgpool2d (line 152) | def profile_avgpool2d(test_case): FILE: python/oneflow/test/modules/test_baddbmm.py class TestBaddBmmModule (line 29) | class TestBaddBmmModule(flow.unittest.TestCase): method test_baddbmm_with_torch (line 31) | def test_baddbmm_with_torch(test_case): method test_baddbmm_in_sd2_with_torch (line 40) | def test_baddbmm_in_sd2_with_torch(test_case): method test_baddbmm_no_attr_with_torch (line 51) | def test_baddbmm_no_attr_with_torch(test_case): method test_baddbmm_broadcast_with_torch (line 60) | def test_baddbmm_broadcast_with_torch(test_case): method profile_baddbmm (line 69) | def profile_baddbmm(test_case): FILE: python/oneflow/test/modules/test_batch_gather.py function _test_batch_gather (line 28) | def _test_batch_gather(test_case, shape, device): class TestBatchGather (line 73) | class TestBatchGather(flow.unittest.TestCase): method test_batch_gather (line 74) | def test_batch_gather(test_case): FILE: python/oneflow/test/modules/test_batchnorm.py class TestBatchNormModule (line 30) | class TestBatchNormModule(flow.unittest.TestCase): method test_batchnorm1d_module_with_random_data (line 35) | def test_batchnorm1d_module_with_random_data(test_case): method test_batchnorm2d_module_with_random_data (line 53) | def test_batchnorm2d_module_with_random_data(test_case): method test_batchnorm3d_module_with_random_data (line 71) | def test_batchnorm3d_module_with_random_data(test_case): method test_functional_batchnorm_with_random_data (line 85) | def test_functional_batchnorm_with_random_data(test_case): method test_batchnorm2d_module_with_half_random_data (line 104) | def test_batchnorm2d_module_with_half_random_data(test_case): method profile_batchnorm (line 122) | def profile_batchnorm(test_case): FILE: python/oneflow/test/modules/test_batchnorm_add_relu.py function _test_bn_add_relu (line 28) | def _test_bn_add_relu(test_case, device, batch, channel, height, width): function _test_bn_relu (line 126) | def _test_bn_relu(test_case, device, batch, channel, height, width): function _test_bn_relu_track_running_states_false (line 210) | def _test_bn_relu_track_running_states_false( function _test_bn_add_relu_track_running_states_false (line 285) | def _test_bn_add_relu_track_running_states_false( function _test_bn_add_relu_eval (line 375) | def _test_bn_add_relu_eval(test_case, device, batch, channel, height, wi... function _test_bn_relu_eval (line 415) | def _test_bn_relu_eval(test_case, device, batch, channel, height, width): class TestBnAddRelu (line 452) | class TestBnAddRelu(flow.unittest.TestCase): method test_bn_add_relu2d (line 453) | def test_bn_add_relu2d(test_case): FILE: python/oneflow/test/modules/test_bernoulli.py function _test_bernoulli (line 28) | def _test_bernoulli(test_case, shape, p, dtype): function _test_bernoulli_with_generator (line 42) | def _test_bernoulli_with_generator(test_case, shape): class TestBernoulli (line 55) | class TestBernoulli(flow.unittest.TestCase): method test_bernoulli (line 56) | def test_bernoulli(test_case): method test_flow_bernoulli_with_random_data (line 67) | def test_flow_bernoulli_with_random_data(test_case): FILE: python/oneflow/test/modules/test_binary_math_ops_dtype.py function get_dtype_str (line 26) | def get_dtype_str(dtype): class TestBinaryMathOpsDtype (line 41) | class TestBinaryMathOpsDtype(flow.unittest.TestCase): method test_binary_math_ops_dtype (line 43) | def test_binary_math_ops_dtype(test_case): FILE: python/oneflow/test/modules/test_bincount.py class TestBinCount (line 25) | class TestBinCount(flow.unittest.TestCase): method test_bincount (line 27) | def test_bincount(test_case): method test_bincount_weight (line 34) | def test_bincount_weight(test_case): method test_bincount_minlength (line 41) | def test_bincount_minlength(test_case): method test_bincount_0element (line 49) | def test_bincount_0element(test_case): method profile_bincount (line 57) | def profile_bincount(test_case): FILE: python/oneflow/test/modules/test_bitwise.py function _test_bitwise_op (line 26) | def _test_bitwise_op(test_case, op): function _test_scalar_bitwise (line 45) | def _test_scalar_bitwise(test_case, op): class TestBitwiseAndModule (line 69) | class TestBitwiseAndModule(flow.unittest.TestCase): method test_bitwise_and (line 71) | def test_bitwise_and(test_case): method test_scalar_bitwise_and (line 75) | def test_scalar_bitwise_and(test_case): class TestBitwiseOrModule (line 80) | class TestBitwiseOrModule(flow.unittest.TestCase): method test_bitwise_or (line 82) | def test_bitwise_or(test_case): method test_scalar_bitwise_or (line 86) | def test_scalar_bitwise_or(test_case): class TestBitwiseXorModule (line 91) | class TestBitwiseXorModule(flow.unittest.TestCase): method test_bitwise_xor (line 93) | def test_bitwise_xor(test_case): method test_scalar_bitwise_xor (line 97) | def test_scalar_bitwise_xor(test_case): class TestBitwiseNotModule (line 102) | class TestBitwiseNotModule(flow.unittest.TestCase): method test_bitwise_not (line 104) | def test_bitwise_not(test_case): FILE: python/oneflow/test/modules/test_bmm.py function _test_bmm (line 28) | def _test_bmm(test_case, device): function _test_bmm_backward (line 40) | def _test_bmm_backward(test_case, device): class TestModule (line 92) | class TestModule(flow.unittest.TestCase): method test_bmm (line 93) | def test_bmm(test_case): method test_bmm_with_torch (line 101) | def test_bmm_with_torch(test_case): method profile_bmm (line 109) | def profile_bmm(test_case): FILE: python/oneflow/test/modules/test_broadcast_like.py function _test_broadcast_like (line 27) | def _test_broadcast_like(test_case, device): function _test_broadcast_like_one (line 43) | def _test_broadcast_like_one(test_case, device): function _test_broadcast_like_different_dim (line 59) | def _test_broadcast_like_different_dim(test_case, device): function _test_broadcast_like_different_dim_with_input_axisvec (line 75) | def _test_broadcast_like_different_dim_with_input_axisvec(test_case, dev... function _test_broadcast_like_3dim (line 91) | def _test_broadcast_like_3dim(test_case, device): function _test_broadcast_like_4dim (line 107) | def _test_broadcast_like_4dim(test_case, device): function _test_broadcast_like_empty_axisvec (line 123) | def _test_broadcast_like_empty_axisvec(test_case, device): function _test_broadcast_like_backward (line 139) | def _test_broadcast_like_backward(test_case, device): class TestBroadCastLike (line 160) | class TestBroadCastLike(flow.unittest.TestCase): method test_broadcast_like (line 161) | def test_broadcast_like(test_case): FILE: python/oneflow/test/modules/test_broadcast_ops.py class TestBroadcastOps (line 50) | class TestBroadcastOps(flow.unittest.TestCase): method test_broadcast_elementwise (line 52) | def test_broadcast_elementwise(test_case): method test_broadcast_matrix_row (line 62) | def test_broadcast_matrix_row(test_case): method test_broadcast_matrix_col (line 72) | def test_broadcast_matrix_col(test_case): method test_cpu_scalar_tensor_auto_cast (line 82) | def test_cpu_scalar_tensor_auto_cast(test_case): method test_broadcast_scalar (line 106) | def test_broadcast_scalar(test_case): method profile_broadcast_matrix_row (line 115) | def profile_broadcast_matrix_row(test_case): method profile_broadcast_matrix_col (line 121) | def profile_broadcast_matrix_col(test_case): method profile_broadcast_elementwise (line 127) | def profile_broadcast_elementwise(test_case): method profile_broadcast_scalar (line 133) | def profile_broadcast_scalar(test_case): method profile_broadcast_general (line 138) | def profile_broadcast_general(test_case): class TestBroadcastOpsOther (line 145) | class TestBroadcastOpsOther(flow.unittest.TestCase): method test_broadcast_shapes (line 146) | def test_broadcast_shapes(test_case): method test_broadcast_tensors (line 153) | def test_broadcast_tensors(test_case): method test_broadcast_to (line 159) | def test_broadcast_to(test_case): FILE: python/oneflow/test/modules/test_cast.py function _test_cast_float2int (line 30) | def _test_cast_float2int(test_case, device, shape): function _test_cast_int2float (line 38) | def _test_cast_int2float(test_case, device, shape): function _test_cast_bool2int16 (line 46) | def _test_cast_bool2int16(test_case, device, shape): function _test_cast_with_non_contiguous_input (line 54) | def _test_cast_with_non_contiguous_input(test_case, device, shape): function _test_cast_backward (line 67) | def _test_cast_backward(test_case, device, shape): function random_expand (line 79) | def random_expand(x, ndim, expand_size): class TestCast (line 87) | class TestCast(flow.unittest.TestCase): method test_cast (line 88) | def test_cast(test_case): method test_cast_with_0_size_data (line 102) | def test_cast_with_0_size_data(test_case): method test_cast_with_strided_input (line 114) | def test_cast_with_strided_input(test_case): method test_cast_with_expanded_input (line 125) | def test_cast_with_expanded_input(test_case): method test_cast_with_expanded_input_2 (line 138) | def test_cast_with_expanded_input_2(test_case): method test_cast_with_squeezed_input (line 147) | def test_cast_with_squeezed_input(test_case): method test_cast_with_sliced_input (line 155) | def test_cast_with_sliced_input(test_case): method test_cast_with_scalar_input (line 171) | def test_cast_with_scalar_input(test_case): method test_cast_with_complex_float2complex (line 179) | def test_cast_with_complex_float2complex(test_case): FILE: python/oneflow/test/modules/test_ceil.py class TestCeilModule (line 27) | class TestCeilModule(flow.unittest.TestCase): method test_ceil_flow_with_random_data (line 29) | def test_ceil_flow_with_random_data(test_case): method test_ceil_flow_with_random_0d_data (line 36) | def test_ceil_flow_with_random_0d_data(test_case): method test_ceil_with_0_size_data (line 43) | def test_ceil_with_0_size_data(test_case): method test_ceil_with_0shape_0d_data (line 50) | def test_ceil_with_0shape_0d_data(test_case): method profile_ceil (line 57) | def profile_ceil(test_case): FILE: python/oneflow/test/modules/test_check_meta_consistency.py class TestGlobalCastModule_1n2d (line 30) | class TestGlobalCastModule_1n2d(flow.unittest.TestCase): method test_check_meta_consistency (line 31) | def test_check_meta_consistency(test_case): FILE: python/oneflow/test/modules/test_checkpointing.py class TestCheckpointing (line 26) | class TestCheckpointing(flow.unittest.TestCase): method test_checkpointing (line 27) | def test_checkpointing(test_case): FILE: python/oneflow/test/modules/test_chunk.py class TestChunk (line 30) | class TestChunk(flow.unittest.TestCase): method test_flow_chunk_list_with_random_data (line 32) | def test_flow_chunk_list_with_random_data(test_case): method test_flow_chunk_list_with_random_data (line 46) | def test_flow_chunk_list_with_random_data(test_case): method test_flow_chunk_list_with_stride (line 62) | def test_flow_chunk_list_with_stride(test_case): method test_flow_chunk_list_bool_with_random_data (line 78) | def test_flow_chunk_list_bool_with_random_data(test_case): method test_flow_chunk_list_with_random_data_negative_dim (line 92) | def test_flow_chunk_list_with_random_data_negative_dim(test_case): method profile_chunk (line 107) | def profile_chunk(test_case): FILE: python/oneflow/test/modules/test_clamp.py function _test_clamp (line 29) | def _test_clamp(test_case, shape, device, dtype): function _test_tensor_clamp (line 38) | def _test_tensor_clamp(test_case, shape, device, dtype): function _test_clamp_scalar_min (line 47) | def _test_clamp_scalar_min(test_case, shape, device, dtype): function _test_clamp_scalar_max (line 56) | def _test_clamp_scalar_max(test_case, shape, device, dtype): function _test_clamp_integral (line 65) | def _test_clamp_integral(test_case, shape, device, dtype): function _numpy_clamp_grad (line 74) | def _numpy_clamp_grad(arr, min, max): function _test_clamp_backward (line 80) | def _test_clamp_backward(test_case, shape, device, dtype): class TestClampModule (line 97) | class TestClampModule(flow.unittest.TestCase): method test_clamp (line 98) | def test_clamp(test_case): method test_clamp_flow_with_random_data (line 121) | def test_clamp_flow_with_random_data(test_case): method test_clamp_min_none_flow_with_random_data (line 128) | def test_clamp_min_none_flow_with_random_data(test_case): method test_clamp_max_none_flow_with_random_data (line 135) | def test_clamp_max_none_flow_with_random_data(test_case): method profile_clamp (line 144) | def profile_clamp(test_case): method test_clip_flow_with_random_data (line 149) | def test_clip_flow_with_random_data(test_case): method test_clip_min_none_flow_with_random_data (line 156) | def test_clip_min_none_flow_with_random_data(test_case): method test_clip_max_none_flow_with_random_data (line 163) | def test_clip_max_none_flow_with_random_data(test_case): method profile_clip (line 172) | def profile_clip(test_case): method test_clamp_with_0_size_data (line 177) | def test_clamp_with_0_size_data(test_case): function _test_clamp_min (line 184) | def _test_clamp_min(test_case, shape, device): function _test_clamp_min_integral (line 193) | def _test_clamp_min_integral(test_case, shape, device): function _test_clamp_min_backward (line 200) | def _test_clamp_min_backward(test_case, shape, device): class TestClampMinModule (line 217) | class TestClampMinModule(flow.unittest.TestCase): method test_clamp_min (line 218) | def test_clamp_min(test_case): method test_clamp_min_flow_with_random_data (line 231) | def test_clamp_min_flow_with_random_data(test_case): method test_clamp_min_with_0_size_data (line 238) | def test_clamp_min_with_0_size_data(test_case): function _test_clamp_max (line 245) | def _test_clamp_max(test_case, shape, device): function _test_clamp_max_integral (line 254) | def _test_clamp_max_integral(test_case, shape, device): function _test_clamp_max_backward (line 261) | def _test_clamp_max_backward(test_case, shape, device): class TestClampMaxModule (line 278) | class TestClampMaxModule(flow.unittest.TestCase): method test_clamp_min (line 279) | def test_clamp_min(test_case): method test_clamp_max_flow_with_random_data (line 292) | def test_clamp_max_flow_with_random_data(test_case): method test_clamp_max_with_0_size_data (line 299) | def test_clamp_max_with_0_size_data(test_case): FILE: python/oneflow/test/modules/test_clip_grad.py function _clip_grad_norm_np (line 26) | def _clip_grad_norm_np(input, max_norm, norm_type): function _test_clip_grad_norm_impl (line 50) | def _test_clip_grad_norm_impl(test_case, shape, device, max_norm, norm_t... function _clip_grad_value_np (line 69) | def _clip_grad_value_np(input, clip_value): function _test_clip_grad_value_impl (line 79) | def _test_clip_grad_value_impl(test_case, shape, device, clip_value): class ReluGraph (line 94) | class ReluGraph(flow.nn.Graph): method __init__ (line 95) | def __init__(self, clip_value) -> None: method build (line 99) | def build(self, x): function _test_graph_clip_grad_value_impl (line 104) | def _test_graph_clip_grad_value_impl(test_case, shape, device, clip_value): function _test_clip_grad_norm_global_impl (line 120) | def _test_clip_grad_norm_global_impl( class TestClipGrad (line 151) | class TestClipGrad(flow.unittest.TestCase): method test_clip_grad (line 152) | def test_clip_grad(test_case): method test_clip_value (line 162) | def test_clip_value(test_case): class TestClipGradGlobal (line 173) | class TestClipGradGlobal(flow.unittest.TestCase): method test_clip_grad_global (line 175) | def test_clip_grad_global(test_case): FILE: python/oneflow/test/modules/test_clone.py class TestClone (line 25) | class TestClone(flow.unittest.TestCase): method test_clone_with_random_data (line 27) | def test_clone_with_random_data(test_case): FILE: python/oneflow/test/modules/test_coco_reader.py class COCODataLoader (line 24) | class COCODataLoader(flow.nn.Module): method __init__ (line 25) | def __init__( method forward (line 51) | def forward(self): class DataLoaderGraph (line 61) | class DataLoaderGraph(flow.nn.Graph): method __init__ (line 62) | def __init__(self, loader): method build (line 66) | def build(self): class COCODataLoaderDistributedTestCase (line 71) | class COCODataLoaderDistributedTestCase(oneflow.unittest.TestCase): method test_case1 (line 72) | def test_case1(test_case): FILE: python/oneflow/test/modules/test_coin_flip.py function _test_coin_flip_impl (line 27) | def _test_coin_flip_impl(test_case, batch_size, random_seed, probability... class TestCoinFlipModule (line 35) | class TestCoinFlipModule(flow.unittest.TestCase): method test_coin_flip (line 36) | def test_coin_flip(test_case): FILE: python/oneflow/test/modules/test_comb2to2d.py class _TestModuleDiffHierarchy (line 30) | class _TestModuleDiffHierarchy(nn.Module): method forward (line 31) | def forward(self, x): class _TestModuleDiffPlacement (line 62) | class _TestModuleDiffPlacement(nn.Module): method forward (line 63) | def forward(self, x): class _TestGraph (line 94) | class _TestGraph(nn.Graph): method __init__ (line 95) | def __init__(self, model): method build (line 99) | def build(self, x): class TestLazyAllSbpCombinationTesting (line 106) | class TestLazyAllSbpCombinationTesting(flow.unittest.TestCase): method test_lazy_boxing_2d_all_combination (line 107) | def test_lazy_boxing_2d_all_combination(test_case): FILE: python/oneflow/test/modules/test_combined_margin_loss.py function _scatter_add_numpy (line 26) | def _scatter_add_numpy(src, dim, index, outshape): function _np_one_hot (line 39) | def _np_one_hot(indices, depth): function _np_gather_with_batch_dims (line 43) | def _np_gather_with_batch_dims(params, indices, axis): function _np_gather_with_batch_dims_grad (line 52) | def _np_gather_with_batch_dims_grad(params, indices, axis, output): function _np_combined_margin_loss (line 61) | def _np_combined_margin_loss(np_input, np_label, m1, m2, m3): function _np_combined_margin_loss_grad (line 87) | def _np_combined_margin_loss_grad(np_input, np_label, m1, m2, m3): function _test_combined_margin_loss (line 121) | def _test_combined_margin_loss( class TestCombinedMarginLoss (line 144) | class TestCombinedMarginLoss(flow.unittest.TestCase): method test_combined_margin_loss (line 145) | def test_combined_margin_loss(test_case): FILE: python/oneflow/test/modules/test_comm.py class TestComm (line 28) | class TestComm(flow.unittest.TestCase): method _test_send_recv (line 29) | def _test_send_recv(test_case, x0, src, dst): method test_send_recv_2_devices (line 51) | def test_send_recv_2_devices(test_case): method test_send_recv_4_devices (line 58) | def test_send_recv_4_devices(test_case): method _test_send_recv_without_sending_meta (line 64) | def _test_send_recv_without_sending_meta(test_case, x0, src, dst): method test_send_recv_without_sending_meta_2_devices (line 86) | def test_send_recv_without_sending_meta_2_devices(test_case): method test_send_recv_without_sending_meta_4_devices (line 93) | def test_send_recv_without_sending_meta_4_devices(test_case): method test_comm_in_thread (line 100) | def test_comm_in_thread(test_case): FILE: python/oneflow/test/modules/test_comm_ops.py class TestAllReduce (line 29) | class TestAllReduce(flow.unittest.TestCase): method test_all_reduce_1n2d (line 31) | def test_all_reduce_1n2d(test_case): method test_all_reduce_2n2d (line 38) | def test_all_reduce_2n2d(test_case): class TestAllGather (line 46) | class TestAllGather(flow.unittest.TestCase): method test_all_gather_into_tensor_1n2d (line 48) | def test_all_gather_into_tensor_1n2d(test_case): method test_all_gather_1n2d (line 73) | def test_all_gather_1n2d(test_case): class TestBroadCast (line 90) | class TestBroadCast(flow.unittest.TestCase): method test_broadcast_1n2d (line 92) | def test_broadcast_1n2d(test_case): class TestScatter (line 107) | class TestScatter(flow.unittest.TestCase): method test_scatter_1n4d (line 109) | def test_scatter_1n4d(test_case): class TestGather (line 129) | class TestGather(flow.unittest.TestCase): method test_gather_1n4d (line 131) | def test_gather_1n4d(test_case): class TestReduce (line 151) | class TestReduce(flow.unittest.TestCase): method test_reduce_1n2d (line 153) | def test_reduce_1n2d(test_case): class TestAllToAll (line 171) | class TestAllToAll(flow.unittest.TestCase): method test_all_to_all_1n4d (line 173) | def test_all_to_all_1n4d(test_case): class TestReduceScatter (line 190) | class TestReduceScatter(flow.unittest.TestCase): method test_reduce_scatter_1n4d (line 192) | def test_reduce_scatter_1n4d(test_case): method test_reduce_scatter_tensor_1n2d (line 204) | def test_reduce_scatter_tensor_1n2d(test_case): class TestDocs (line 237) | class TestDocs(flow.unittest.TestCase): method test_docs (line 238) | def test_docs(test_case): FILE: python/oneflow/test/modules/test_concat.py function _test_concat_origin (line 29) | def _test_concat_origin(test_case, device): function _test_concat_with_empty_input (line 41) | def _test_concat_with_empty_input(test_case, device): function _test_concat_with_axis_one (line 75) | def _test_concat_with_axis_one(test_case, device): function _test_concat_with_three_tensor (line 87) | def _test_concat_with_three_tensor(test_case, device): function _test_concat_with_three_tensor_backward (line 102) | def _test_concat_with_three_tensor_backward(test_case, device): function _test_concat_grad_and_no_grad (line 135) | def _test_concat_grad_and_no_grad(test_case, device): function _test_concat_single_input_type (line 156) | def _test_concat_single_input_type(test_case, device): function _test_concat_grad_fn_name (line 166) | def _test_concat_grad_fn_name(test_case, device): class TestModule (line 184) | class TestModule(flow.unittest.TestCase): method test_concat (line 185) | def test_concat(test_case): method test_cat_with_random_data (line 202) | def test_cat_with_random_data(test_case): method test_cat_with_diff_dtypes (line 208) | def test_cat_with_diff_dtypes(test_case): method test_cat_with_diff_dtype_corner_case (line 216) | def test_cat_with_diff_dtype_corner_case(test_case): method test_concat_with_input_0_size_data (line 228) | def test_concat_with_input_0_size_data(test_case): method test_concat_with_output_0_size_data (line 236) | def test_concat_with_output_0_size_data(test_case): method test_cat_bool_with_random_data (line 245) | def test_cat_bool_with_random_data(test_case): method test_cat_only_one_tensor (line 251) | def test_cat_only_one_tensor(test_case): method profile_cat (line 257) | def profile_cat(test_case): FILE: python/oneflow/test/modules/test_constant.py function _test_different_dtype (line 29) | def _test_different_dtype(test_case, device, shape): class TestConstantModule (line 47) | class TestConstantModule(flow.unittest.TestCase): method test_flow_zeros_list_with_random_data (line 49) | def test_flow_zeros_list_with_random_data(test_case): method profile_zeros (line 62) | def profile_zeros(test_case): method test_flow_ones_list_with_random_data (line 68) | def test_flow_ones_list_with_random_data(test_case): method profile_ones (line 79) | def profile_ones(test_case): method test_flow_zeros_like_list_with_random_data (line 85) | def test_flow_zeros_like_list_with_random_data(test_case): method profile_zeros_like (line 92) | def profile_zeros_like(test_case): method test_flow_zeros_like_list_with_random_data_and_requires_grad (line 101) | def test_flow_zeros_like_list_with_random_data_and_requires_grad(test_... method test_flow_zeros_like_list_with_0dim_data (line 108) | def test_flow_zeros_like_list_with_0dim_data(test_case): method test_flow_ones_like_list_with_random_data (line 115) | def test_flow_ones_like_list_with_random_data(test_case): method profile_ones_like (line 122) | def profile_ones_like(test_case): method test_flow_ones_like_list_with_random_data_and_requires_grad (line 131) | def test_flow_ones_like_list_with_random_data_and_requires_grad(test_c... method test_flow_ones_like_list_with_0dim_data (line 138) | def test_flow_ones_like_list_with_0dim_data(test_case): method test_flow_new_ones_list_with_random_data (line 145) | def test_flow_new_ones_list_with_random_data(test_case): method profile_new_ones (line 156) | def profile_new_ones(test_case): method test_flow_new_ones_list_with_0dim_data (line 164) | def test_flow_new_ones_list_with_0dim_data(test_case): method test_new_zeros (line 175) | def test_new_zeros(test_case): method profile_new_zeros (line 186) | def profile_new_zeros(test_case): method test_new_full (line 193) | def test_new_full(test_case): method test_new_full_with_scalar (line 205) | def test_new_full_with_scalar(test_case): method test_full_with_scalar (line 212) | def test_full_with_scalar(test_case): method test_full_with_random_data_int (line 218) | def test_full_with_random_data_int(test_case): method test_full_with_random_data_numpy_scalar (line 225) | def test_full_with_random_data_numpy_scalar(test_case): method test_full_with_scalar_tensor (line 232) | def test_full_with_scalar_tensor(test_case): method profile_full_with_scalar_tensor (line 244) | def profile_full_with_scalar_tensor(test_case): method profile_full (line 250) | def profile_full(test_case): method test_full_with_random_data_float (line 256) | def test_full_with_random_data_float(test_case): method test_full_like_with_random_data_float (line 263) | def test_full_like_with_random_data_float(test_case): method profile_full_like (line 270) | def profile_full_like(test_case): method test_cast (line 275) | def test_cast(test_case): FILE: python/oneflow/test/modules/test_constant_pad.py class TestConstantPad1d (line 31) | class TestConstantPad1d(flow.unittest.TestCase): method test_constantpad1d_with_random_data (line 33) | def test_constantpad1d_with_random_data(test_case): method test_constantpad1d_with_random_int_data (line 45) | def test_constantpad1d_with_random_int_data(test_case): class TestConstantPad2d (line 62) | class TestConstantPad2d(flow.unittest.TestCase): method test_constantpad2d_with_random_data (line 64) | def test_constantpad2d_with_random_data(test_case): method test_constantpad2d_with_random_int_data (line 78) | def test_constantpad2d_with_random_int_data(test_case): class TestConstantPad3d (line 95) | class TestConstantPad3d(flow.unittest.TestCase): method test_constantpad3d_with_random_data (line 97) | def test_constantpad3d_with_random_data(test_case): method test_constantpad3d_with_random_int_data (line 115) | def test_constantpad3d_with_random_int_data(test_case): class TestFunctionalConstantPad2d (line 136) | class TestFunctionalConstantPad2d(flow.unittest.TestCase): method test_functional_constantpad2d (line 138) | def test_functional_constantpad2d(test_case): method test_functional_constantpad2d_int_data (line 153) | def test_functional_constantpad2d_int_data(test_case): method profile_pad (line 171) | def profile_pad(test_case): FILE: python/oneflow/test/modules/test_contiguous.py class TestContiguous (line 30) | class TestContiguous(flow.unittest.TestCase): method test_transpose_with_random_data (line 32) | def test_transpose_with_random_data(test_case): method test_transpose_with_bool_data (line 40) | def test_transpose_with_bool_data(test_case): method test_transpose_with_int_data (line 48) | def test_transpose_with_int_data(test_case): method test_contiguous_with_half_data (line 56) | def test_contiguous_with_half_data(test_case): method test_permute2d_tensor_with_random_data (line 64) | def test_permute2d_tensor_with_random_data(test_case): method test_permute3d_tensor_with_random_data (line 77) | def test_permute3d_tensor_with_random_data(test_case): method test_permute4d_tensor_with_random_data (line 93) | def test_permute4d_tensor_with_random_data(test_case): method profile_contiguous (line 110) | def profile_contiguous(test_case): function _test_inplace_contiguous (line 115) | def _test_inplace_contiguous(test_case, device): class TestInplaceContiguous (line 133) | class TestInplaceContiguous(flow.unittest.TestCase): method test_inplace_contiguous (line 134) | def test_inplace_contiguous(test_case): FILE: python/oneflow/test/modules/test_conv1d.py function _test_conv1d_bias_false (line 31) | def _test_conv1d_bias_false(test_case, device): function _test_conv1d_bias_true (line 65) | def _test_conv1d_bias_true(test_case, device): function _test_conv1d_dilation (line 127) | def _test_conv1d_dilation(test_case, device): function _test_conv1d_stride (line 163) | def _test_conv1d_stride(test_case, device): function _test_conv1d_group_bias_true (line 199) | def _test_conv1d_group_bias_true(test_case, device): function _test_conv1d_group_large_out_bias_true (line 249) | def _test_conv1d_group_large_out_bias_true(test_case, device): function _test_conv1d_group_large_in_bias_true (line 305) | def _test_conv1d_group_large_in_bias_true(test_case, device): function _test_conv1d_compilcate (line 356) | def _test_conv1d_compilcate(test_case, device): class TestConv1d (line 423) | class TestConv1d(flow.unittest.TestCase): method test_conv1d (line 424) | def test_conv1d(test_case): method test_nn_functional_conv1d (line 442) | def test_nn_functional_conv1d(test_case): method test_nn_functional_conv1d_2dinput (line 454) | def test_nn_functional_conv1d_2dinput(test_case): method profile_conv1d (line 462) | def profile_conv1d(test_case): method test_conv1d_with_random_data (line 495) | def test_conv1d_with_random_data(test_case): method test_conv1d_group_with_random_data (line 516) | def test_conv1d_group_with_random_data(test_case): FILE: python/oneflow/test/modules/test_conv2d.py function _test_conv2d (line 1154) | def _test_conv2d( function _test_conv2d_backward (line 1167) | def _test_conv2d_backward( function _test_conv2d_large_in_channel (line 1199) | def _test_conv2d_large_in_channel(test_case, device): function _test_conv2d_large_out_channel (line 1471) | def _test_conv2d_large_out_channel(test_case, device): class TestConv2d (line 1584) | class TestConv2d(flow.unittest.TestCase): method test_conv2d_default_init (line 1585) | def test_conv2d_default_init(test_case): method test_nn_functional_conv2d (line 1630) | def test_nn_functional_conv2d(test_case): method test_nn_functional_conv2d_3dinput (line 1642) | def test_nn_functional_conv2d_3dinput(test_case): method test_conv2d (line 1649) | def test_conv2d(test_case): method test_conv2d_backward (line 1664) | def test_conv2d_backward(test_case): method test_conv2d_with_bias (line 1682) | def test_conv2d_with_bias(test_case): method test_conv2d_group (line 1698) | def test_conv2d_group(test_case): method test_conv2d_group_backward (line 1715) | def test_conv2d_group_backward(test_case): method test_conv2d_padding (line 1733) | def test_conv2d_padding(test_case): method test_conv2d_padding_backward (line 1750) | def test_conv2d_padding_backward(test_case): method test_conv2d_stride (line 1769) | def test_conv2d_stride(test_case): method test_conv2d_stride_backward (line 1786) | def test_conv2d_stride_backward(test_case): method test_conv2d_kernel (line 1805) | def test_conv2d_kernel(test_case): method test_conv2d_kernel_backward (line 1822) | def test_conv2d_kernel_backward(test_case): method test_conv2d_dilation (line 1839) | def test_conv2d_dilation(test_case): method test_conv2d_dilation_backward (line 1856) | def test_conv2d_dilation_backward(test_case): method test_large_in_channel_group_conv (line 1874) | def test_large_in_channel_group_conv(test_case): method test_large_out_channel_group_conv (line 1883) | def test_large_out_channel_group_conv(test_case): method test_conv2d_with_random_data (line 1893) | def test_conv2d_with_random_data(test_case): method test_conv2d_auto_squeeze_with_random_data (line 1917) | def test_conv2d_auto_squeeze_with_random_data(test_case): method test_conv2d_0size_with_random_data (line 1938) | def test_conv2d_0size_with_random_data(test_case): method test_conv2d_group_with_random_data (line 1959) | def test_conv2d_group_with_random_data(test_case): method test_conv2d_NHWC_with_random_data (line 1983) | def test_conv2d_NHWC_with_random_data(test_case): method profile_conv2d (line 2084) | def profile_conv2d(test_case): FILE: python/oneflow/test/modules/test_copy.py class Test_Copy_module (line 29) | class Test_Copy_module(flow.unittest.TestCase): method test_copy_broadcast_tensor (line 30) | def test_copy_broadcast_tensor(test_case): method test_non_contiguous_sliced_tensor_copy (line 41) | def test_non_contiguous_sliced_tensor_copy(test_case): method test_non_contiguous_permuted_tensor_copy (line 50) | def test_non_contiguous_permuted_tensor_copy(test_case): method test_copy_fp16 (line 59) | def test_copy_fp16(test_case): method test_tensor_inplace_copy_with_diff_dtype (line 65) | def test_tensor_inplace_copy_with_diff_dtype(test_case): method test_tensor_inplace_copy_with_diff_dtype_and_device (line 72) | def test_tensor_inplace_copy_with_diff_dtype_and_device(test_case): method test_global_tensor_inplace_copy_with_diff_dtype_and_device (line 80) | def test_global_tensor_inplace_copy_with_diff_dtype_and_device(test_ca... FILE: python/oneflow/test/modules/test_cosine_similarity.py class TestCosineSimilarity (line 25) | class TestCosineSimilarity(flow.unittest.TestCase): method test_cosine_similartiy_module_with_random_data (line 27) | def test_cosine_similartiy_module_with_random_data(test_case): method test_cosine_similartiy_functional_with_random_data (line 37) | def test_cosine_similartiy_functional_with_random_data(test_case): method test_cosine_similartiy_broadcast_with_random_data (line 46) | def test_cosine_similartiy_broadcast_with_random_data(test_case): method test_cosine_similartiy_module_with_nonequal_dim_data (line 54) | def test_cosine_similartiy_module_with_nonequal_dim_data(test_case): method profile_cosine_similarity (line 67) | def profile_cosine_similarity(test_case): FILE: python/oneflow/test/modules/test_ctc_greedy_decoder.py function log_softmax (line 29) | def log_softmax(logits, axis=0): function np_ctc_greedy_decoder (line 37) | def np_ctc_greedy_decoder(log_probs, input_lengths, merge_repeated=True): function compare_with_np (line 59) | def compare_with_np( function gen_arg_list (line 98) | def gen_arg_list(): class TestCTCGreedyDecoder1n1d (line 110) | class TestCTCGreedyDecoder1n1d(flow.unittest.TestCase): method test_ctc_greedy_decoder (line 111) | def test_ctc_greedy_decoder(test_case): FILE: python/oneflow/test/modules/test_ctc_loss.py class TestCTCLoss1n1d (line 30) | class TestCTCLoss1n1d(flow.unittest.TestCase): method test_ctc_loss_with_diff_device_input (line 36) | def test_ctc_loss_with_diff_device_input(test_case): method test_ctc_loss_functional (line 57) | def test_ctc_loss_functional(test_case): FILE: python/oneflow/test/modules/test_cublas_fused_mlp.py function _matmul_bias_relu (line 24) | def _matmul_bias_relu(x, weight, bias, skip_activation): function _test_fused_matmul_bias_add_relu (line 31) | def _test_fused_matmul_bias_add_relu( class TestFusedMatmulBiasAddRelu (line 172) | class TestFusedMatmulBiasAddRelu(flow.unittest.TestCase): method test_fused_matmul_op (line 173) | def test_fused_matmul_op(test_case): FILE: python/oneflow/test/modules/test_cum_ops.py class TestCumOp (line 28) | class TestCumOp(flow.unittest.TestCase): method test_cumsum (line 30) | def test_cumsum(test_case): method test_cumprod (line 38) | def test_cumprod(test_case): method test_cumop_with_dtype (line 45) | def test_cumop_with_dtype(test_case): method test_cumsum (line 53) | def test_cumsum(test_case): method test_cumprod_with_user_dy (line 61) | def test_cumprod_with_user_dy(test_case): method test_cumprod_with_zero (line 69) | def test_cumprod_with_zero(test_case): method test_cumsum_graph_backward (line 92) | def test_cumsum_graph_backward(test_case): method profile_cumsum (line 123) | def profile_cumsum(test_case): method profile_cumprod (line 129) | def profile_cumprod(test_case): FILE: python/oneflow/test/modules/test_dataset.py class TestOFRecordModule (line 29) | class TestOFRecordModule(flow.unittest.TestCase): method test_record (line 30) | def test_record(test_case): class TestGlobalOFRecordModule (line 96) | class TestGlobalOFRecordModule(flow.unittest.TestCase): method test_global_record (line 97) | def test_global_record(test_case): function _coco (line 173) | def _coco(anno_file): function _get_coco_image_samples (line 182) | def _get_coco_image_samples(anno_file, image_dir, image_ids): function _get_category_id_to_contiguous_id_map (line 198) | def _get_category_id_to_contiguous_id_map(coco): function _read_images_with_cv (line 202) | def _read_images_with_cv(coco, image_dir, image_ids): function _bbox_convert_from_xywh_to_xyxy (line 216) | def _bbox_convert_from_xywh_to_xyxy(bbox, image_h, image_w): function _read_bbox (line 230) | def _read_bbox(coco, image_ids): function _read_label (line 252) | def _read_label(coco, image_ids, category_id_to_contiguous_id_map): function _read_segm_poly (line 270) | def _read_segm_poly(coco, image_ids): function _segm_poly_list_to_tensor (line 291) | def _segm_poly_list_to_tensor(img_segm_poly_list): class TestCocoReader (line 313) | class TestCocoReader(flow.unittest.TestCase): method test_coco_reader (line 314) | def test_coco_reader(test_case): class TestOFRecordBytesDecoder (line 364) | class TestOFRecordBytesDecoder(flow.unittest.TestCase): method test_OFRecordBytesDecoder (line 365) | def test_OFRecordBytesDecoder(test_case): FILE: python/oneflow/test/modules/test_ddp.py function np_allclose_with_shape (line 29) | def np_allclose_with_shape(a, b, *args, **kwargs): class TestDDP (line 39) | class TestDDP(flow.unittest.TestCase): method _test_ddp_basic (line 40) | def _test_ddp_basic(test_case, dev_type): method test_ddp_basic (line 67) | def test_ddp_basic(test_case): method _test_ddp_multiple_buckets (line 71) | def _test_ddp_multiple_buckets(test_case, dev_type, use_bucket): method test_ddp_multiple_buckets (line 108) | def test_ddp_multiple_buckets(test_case): method _test_ddp_with_unused_param (line 112) | def _test_ddp_with_unused_param(test_case, dev_type): method test_ddp_with_unused_param (line 148) | def test_ddp_with_unused_param(test_case): method _test_out_of_order_execution (line 152) | def _test_out_of_order_execution(test_case, dev_type): method test_out_of_order_execution (line 189) | def test_out_of_order_execution(test_case): method _test_ddp_with_partial_requires_grad_parameter (line 193) | def _test_ddp_with_partial_requires_grad_parameter(test_case, dev_type): method test_ddp_with_partial_requires_grad_parameter (line 229) | def test_ddp_with_partial_requires_grad_parameter(test_case): method _test_ddp_two_iters (line 233) | def _test_ddp_two_iters(test_case, dev_type): method test_ddp_two_iters (line 260) | def test_ddp_two_iters(test_case): method _test_broadcast_buffer (line 264) | def _test_broadcast_buffer(test_case, dev_type): method test_broadcast_buffer (line 307) | def test_broadcast_buffer(test_case): FILE: python/oneflow/test/modules/test_ddp_multi_outputs.py class Model (line 40) | class Model(flow.nn.Module): method __init__ (line 41) | def __init__(self): method forward (line 48) | def forward(self, x, label): function train (line 56) | def train(test_case, train_x, device, output, requires_grad): class TestDdpMultmpleOutputs (line 81) | class TestDdpMultmpleOutputs(flow.unittest.TestCase): method test_outputs_float32 (line 82) | def test_outputs_float32(test_case): method test_outputs_int32 (line 90) | def test_outputs_int32(test_case): FILE: python/oneflow/test/modules/test_deconv2d.py function _test_deconv_bias_false (line 31) | def _test_deconv_bias_false(test_case, device): function _test_deconv_bias_true (line 137) | def _test_deconv_bias_true(test_case, device): function _test_deconv_group_bias_false (line 243) | def _test_deconv_group_bias_false(test_case, device): function _test_deconv_group_bias_true (line 360) | def _test_deconv_group_bias_true(test_case, device): function _test_deconv_group_large_out_channel (line 476) | def _test_deconv_group_large_out_channel(test_case, device): function _test_deconv_group_large_in_channel (line 716) | def _test_deconv_group_large_in_channel(test_case, device): class TestDeconv2d (line 861) | class TestDeconv2d(flow.unittest.TestCase): method test_deconv2d (line 862) | def test_deconv2d(test_case): method test_deconv2d_with_random_data (line 877) | def test_deconv2d_with_random_data(test_case): method test_deconv2d_auto_squeeze_with_random_data (line 902) | def test_deconv2d_auto_squeeze_with_random_data(test_case): method test_deconv2d_0size_with_random_data (line 923) | def test_deconv2d_0size_with_random_data(test_case): method test_deconv2d_group_with_random_data (line 946) | def test_deconv2d_group_with_random_data(test_case): method profile_conv_transpose2d (line 969) | def profile_conv_transpose2d(test_case): FILE: python/oneflow/test/modules/test_default_dtype.py class TestDefaultDTypeInferface (line 35) | class TestDefaultDTypeInferface(oneflow.unittest.TestCase): method test_set_default_dtype (line 36) | def test_set_default_dtype(test_case): method test_set_default_tensor_type (line 56) | def test_set_default_tensor_type(test_case): method test_behavior_for_oneflow_tensor (line 76) | def test_behavior_for_oneflow_tensor(test_case): FILE: python/oneflow/test/modules/test_deform_conv2d.py function GetRandomData (line 29) | def GetRandomData(max_batch_sz): function GetFunArgs (line 71) | def GetFunArgs(device, max_batch_size): function _test_deform_conv2d_forward (line 111) | def _test_deform_conv2d_forward( function _test_deform_conv2d_backward (line 154) | def _test_deform_conv2d_backward( function _test_forward_and_backward (line 228) | def _test_forward_and_backward(test_case, device): class TestDeformConv2d (line 243) | class TestDeformConv2d(flow.unittest.TestCase): method test_deform_conv2d (line 244) | def test_deform_conv2d(test_case): FILE: python/oneflow/test/modules/test_det.py function det_random_device (line 24) | def det_random_device(): class TestLinalgDet (line 33) | class TestLinalgDet(flow.unittest.TestCase): method test_det_3by3_with_random_data (line 35) | def test_det_3by3_with_random_data(test_case): method test_det_batch_3by3_with_random_data (line 41) | def test_det_batch_3by3_with_random_data(test_case): method test_det_random_square_with_random_data (line 47) | def test_det_random_square_with_random_data(test_case): method profile_linalg_det (line 54) | def profile_linalg_det(test_case): FILE: python/oneflow/test/modules/test_diag.py class Test_Diag_module (line 29) | class Test_Diag_module(flow.unittest.TestCase): method test_diag_one_dim (line 31) | def test_diag_one_dim(test_case): method test_diag_other_dim (line 37) | def test_diag_other_dim(test_case): method test_diag_one_dim (line 43) | def test_diag_one_dim(test_case): method test_diag_0size_tensor (line 48) | def test_diag_0size_tensor(test_case): method profile_diag (line 66) | def profile_diag(test_case): FILE: python/oneflow/test/modules/test_diagonal.py class TestDiagonal (line 25) | class TestDiagonal(flow.unittest.TestCase): method test_flow_diagonal_with_random_data (line 27) | def test_flow_diagonal_with_random_data(test_case): method test_flow_diagonal_with_random_data (line 44) | def test_flow_diagonal_with_random_data(test_case): method profile_diagonal (line 61) | def profile_diagonal(test_case): FILE: python/oneflow/test/modules/test_div.py function _test_div_impl (line 32) | def _test_div_impl(test_case, shape, device): class TestDiv (line 94) | class TestDiv(flow.unittest.TestCase): method test_div (line 95) | def test_div(test_case): method test_random_dim_div (line 103) | def test_random_dim_div(test_case): method test_random_dim_scalar_div (line 113) | def test_random_dim_scalar_div(test_case): method test_0_size_div (line 123) | def test_0_size_div(test_case): method test_0dim_div (line 131) | def test_0dim_div(test_case): method test_non_contiguous_inplace_div (line 139) | def test_non_contiguous_inplace_div(test_case): method test_int_dtype_inplace_div (line 148) | def test_int_dtype_inplace_div(test_case): method test_scalar_div_with_random_devices (line 155) | def test_scalar_div_with_random_devices(test_case): method profile_div (line 164) | def profile_div(test_case): class TestDivRoundmode (line 171) | class TestDivRoundmode(flow.unittest.TestCase): method test_random_dim_div_floor (line 173) | def test_random_dim_div_floor(test_case): method test_random_dim_div_trunc (line 183) | def test_random_dim_div_trunc(test_case): method test_scalar_div_mode_floor (line 193) | def test_scalar_div_mode_floor(test_case): method test_scalar_div_mode_trunc (line 201) | def test_scalar_div_mode_trunc(test_case): method test_scalar_div_mode_floor2 (line 209) | def test_scalar_div_mode_floor2(test_case): method test_scalar_div_mode_trunc2 (line 217) | def test_scalar_div_mode_trunc2(test_case): FILE: python/oneflow/test/modules/test_dlpack.py function are_tensors_equal (line 39) | def are_tensors_equal(a, b): class TestPack (line 56) | class TestPack(flow.unittest.TestCase): method test_same_data (line 57) | def test_same_data(test_case): method test_use_ops (line 70) | def test_use_ops(test_case): method test_more_dtype (line 78) | def test_more_dtype(test_case): method test_non_contiguous_input (line 92) | def test_non_contiguous_input(test_case): method test_scalar_tensor (line 100) | def test_scalar_tensor(test_case): method test_0_size_tensor (line 106) | def test_0_size_tensor(test_case): method test_lifecycle (line 112) | def test_lifecycle(test_case): method test_subview (line 133) | def test_subview(test_case): FILE: python/oneflow/test/modules/test_dot.py class TestDot (line 24) | class TestDot(flow.unittest.TestCase): method test_dot (line 26) | def test_dot(test_case): method profile_dot (line 35) | def profile_dot(test_case): FILE: python/oneflow/test/modules/test_dropout.py function do_test_dropout_numpy_p0 (line 29) | def do_test_dropout_numpy_p0(test_case, shape, device, dtype): function do_test_dropout_numpy_p1 (line 42) | def do_test_dropout_numpy_p1(test_case, shape, device, dtype): function do_test_dropout_numpy_fp16_p0 (line 55) | def do_test_dropout_numpy_fp16_p0(test_case, shape): function do_test_dropout_numpy_fp16_p1 (line 71) | def do_test_dropout_numpy_fp16_p1(test_case, shape): function do_test_dropout_addend_numpy_p0 (line 88) | def do_test_dropout_addend_numpy_p0(test_case, shape, device, dtype): function do_test_dropout_addend_numpy_p1 (line 109) | def do_test_dropout_addend_numpy_p1(test_case, shape, device, dtype): function do_test_dropout_addend_numpy_fp16_p0 (line 129) | def do_test_dropout_addend_numpy_fp16_p0(test_case, shape): function do_test_dropout_addend_numpy_fp16_p1 (line 155) | def do_test_dropout_addend_numpy_fp16_p1(test_case, shape): function fixed_cpu_seed_dropout_test (line 181) | def fixed_cpu_seed_dropout_test(test_case): function fixed_gpu_seed_dropout_test (line 209) | def fixed_gpu_seed_dropout_test(test_case): class TestModule (line 238) | class TestModule(flow.unittest.TestCase): method test_dropout_numpy_case (line 239) | def test_dropout_numpy_case(test_case): method test_dropout_fp16_numpy_case (line 251) | def test_dropout_fp16_numpy_case(test_case): method test_dropout_addend_numpy_case (line 261) | def test_dropout_addend_numpy_case(test_case): method test_dropout_addend_fp16_numpy_case (line 274) | def test_dropout_addend_fp16_numpy_case(test_case): method test_cpu_fixed_dropout (line 284) | def test_cpu_fixed_dropout(test_case): method test_gpu_fixed_dropout (line 293) | def test_gpu_fixed_dropout(test_case): method test_dropout_p0 (line 302) | def test_dropout_p0(test_case): method test_dropout1d_p0 (line 310) | def test_dropout1d_p0(test_case): method test_dropout2d_p0 (line 317) | def test_dropout2d_p0(test_case): method test_dropout3d_p0 (line 328) | def test_dropout3d_p0(test_case): method test_dropout_p1 (line 335) | def test_dropout_p1(test_case): method test_dropout1d_p1 (line 343) | def test_dropout1d_p1(test_case): method test_dropout2d_p1 (line 351) | def test_dropout2d_p1(test_case): method test_dropout3d_p1 (line 362) | def test_dropout3d_p1(test_case): method test_functional_dropout1d_p1 (line 370) | def test_functional_dropout1d_p1(test_case): method test_functional_dropout2d_p1 (line 376) | def test_functional_dropout2d_p1(test_case): method test_functional_dropout3d_p1 (line 386) | def test_functional_dropout3d_p1(test_case): method test_dropout_eval (line 392) | def test_dropout_eval(test_case): method test_dropout1d_eval (line 401) | def test_dropout1d_eval(test_case): method test_dropout2d_eval (line 409) | def test_dropout2d_eval(test_case): method test_dropout3d_eval (line 421) | def test_dropout3d_eval(test_case): method test_0dim_dropout_eval (line 429) | def test_0dim_dropout_eval(test_case): method profile_dropout (line 437) | def profile_dropout(test_case): class TestDropoutOnNonDefaultDevice (line 444) | class TestDropoutOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 445) | def test_non_default_device(test_case): FILE: python/oneflow/test/modules/test_dynamic_allocation_gradient_shuffle_shuffle_global.py function get_tensors (line 31) | def get_tensors(batch_size, num_tables): function round_half_away_from_zero (line 46) | def round_half_away_from_zero(x): function _test_embedding_gradient_shuffle (line 55) | def _test_embedding_gradient_shuffle(test_case, enable_quantize, fp16, e... class DataShuffleTestCase (line 182) | class DataShuffleTestCase(flow.unittest.TestCase): method test_embedding_gradient_shuffle (line 183) | def test_embedding_gradient_shuffle(test_case): FILE: python/oneflow/test/modules/test_eager_boxing.py function _test_eager_boxing_with_non_overlapping_placement_p_to_s1 (line 28) | def _test_eager_boxing_with_non_overlapping_placement_p_to_s1( function _test_eager_boxing_with_non_overlapping_placement_b_to_s1 (line 71) | def _test_eager_boxing_with_non_overlapping_placement_b_to_s1( function _test_eager_boxing_with_non_overlapping_placement_s0_to_s1 (line 114) | def _test_eager_boxing_with_non_overlapping_placement_s0_to_s1( function _test_eager_boxing_with_non_overlapping_placement_s1_to_s1 (line 172) | def _test_eager_boxing_with_non_overlapping_placement_s1_to_s1( function _test_eager_boxing_with_non_overlapping_placement_s1_to_s0 (line 231) | def _test_eager_boxing_with_non_overlapping_placement_s1_to_s0( function _test_eager_boxing_with_non_overlapping_placement_s1_to_b (line 281) | def _test_eager_boxing_with_non_overlapping_placement_s1_to_b( function _test_eager_boxing_with_non_overlapping_placement_s1_to_p (line 349) | def _test_eager_boxing_with_non_overlapping_placement_s1_to_p( function _test_eager_boxing_with_overlapping_placement_p_to_s1 (line 417) | def _test_eager_boxing_with_overlapping_placement_p_to_s1( function _test_eager_boxing_with_overlapping_placement_b_to_s1 (line 460) | def _test_eager_boxing_with_overlapping_placement_b_to_s1( function _test_eager_boxing_with_overlapping_placement_s0_to_s1 (line 503) | def _test_eager_boxing_with_overlapping_placement_s0_to_s1( function _test_eager_boxing_with_overlapping_placement_s1_to_s1 (line 578) | def _test_eager_boxing_with_overlapping_placement_s1_to_s1( function _test_eager_boxing_with_overlapping_placement_s1_to_s0 (line 677) | def _test_eager_boxing_with_overlapping_placement_s1_to_s0( function _test_eager_boxing_with_overlapping_placement_s1_to_b (line 764) | def _test_eager_boxing_with_overlapping_placement_s1_to_b( function _test_eager_boxing_with_overlapping_placement_s1_to_p (line 863) | def _test_eager_boxing_with_overlapping_placement_s1_to_p( function _test_eager_boxing_with_in_placement_contain_out_placement_p_to_s1 (line 962) | def _test_eager_boxing_with_in_placement_contain_out_placement_p_to_s1( function _test_eager_boxing_with_in_placement_contain_out_placement_b_to_s1 (line 1005) | def _test_eager_boxing_with_in_placement_contain_out_placement_b_to_s1( function _test_eager_boxing_with_in_placement_contain_out_placement_s0_to_s1 (line 1048) | def _test_eager_boxing_with_in_placement_contain_out_placement_s0_to_s1( function _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s1 (line 1123) | def _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s1( function _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s0 (line 1167) | def _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s0( function _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_p (line 1211) | def _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_p( function _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_b (line 1261) | def _test_eager_boxing_with_in_placement_contain_out_placement_s1_to_b( function _test_eager_boxing_with_out_placement_contain_in_placement_p_to_s1 (line 1311) | def _test_eager_boxing_with_out_placement_contain_in_placement_p_to_s1( function _test_eager_boxing_with_out_placement_contain_in_placement_b_to_s1 (line 1368) | def _test_eager_boxing_with_out_placement_contain_in_placement_b_to_s1( function _test_eager_boxing_with_out_placement_contain_in_placement_s0_to_s1 (line 1425) | def _test_eager_boxing_with_out_placement_contain_in_placement_s0_to_s1( function _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_b (line 1494) | def _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_b( function _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_p (line 1569) | def _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_p( function _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s0 (line 1714) | def _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s0( function _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s1 (line 1807) | def _test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s1( function _test_eager_boxing_with_same_placement_p_to_s1 (line 1918) | def _test_eager_boxing_with_same_placement_p_to_s1(test_case, in_device,... function _test_eager_boxing_with_same_placement_b_to_s1 (line 2005) | def _test_eager_boxing_with_same_placement_b_to_s1(test_case, in_device,... function _test_eager_boxing_with_same_placement_s0_to_s1 (line 2090) | def _test_eager_boxing_with_same_placement_s0_to_s1(test_case, in_device... function _test_eager_boxing_with_same_placement_s1_to_s1 (line 2208) | def _test_eager_boxing_with_same_placement_s1_to_s1(test_case, in_device... function _test_eager_boxing_with_same_placement_s1_to_s0 (line 2327) | def _test_eager_boxing_with_same_placement_s1_to_s0(test_case, in_device... function _test_eager_boxing_with_same_placement_s1_to_p (line 2422) | def _test_eager_boxing_with_same_placement_s1_to_p(test_case, in_device,... function _test_eager_boxing_with_same_placement_s1_to_b (line 2541) | def _test_eager_boxing_with_same_placement_s1_to_b(test_case, in_device,... function _test_eager_boxing_b_to_s (line 2661) | def _test_eager_boxing_b_to_s( function _test_eager_boxing_s_to_b (line 2702) | def _test_eager_boxing_s_to_b( function _test_eager_boxing_p_to_s (line 2728) | def _test_eager_boxing_p_to_s( function _test_eager_boxing_p_to_b (line 2770) | def _test_eager_boxing_p_to_b( function _test_eager_naive_boxing_s_to_s (line 2794) | def _test_eager_naive_boxing_s_to_s( class TestEagerBoxingWithNonOverlappingPlacement (line 2844) | class TestEagerBoxingWithNonOverlappingPlacement(flow.unittest.TestCase): method test_eager_boxing_with_non_overlapping_placement_p_to_s1 (line 2845) | def test_eager_boxing_with_non_overlapping_placement_p_to_s1(test_case): method test_eager_boxing_with_non_overlapping_placement_b_to_s1 (line 2852) | def test_eager_boxing_with_non_overlapping_placement_b_to_s1(test_case): method test_eager_boxing_with_non_overlapping_placement_s0_to_s1 (line 2859) | def test_eager_boxing_with_non_overlapping_placement_s0_to_s1(test_case): method test_eager_boxing_with_non_overlapping_placement_s1_to_s1 (line 2866) | def test_eager_boxing_with_non_overlapping_placement_s1_to_s1(test_case): method test_eager_boxing_with_non_overlapping_placement_s1_to_s0 (line 2873) | def test_eager_boxing_with_non_overlapping_placement_s1_to_s0(test_case): method test_eager_boxing_with_non_overlapping_placement_s1_to_b (line 2880) | def test_eager_boxing_with_non_overlapping_placement_s1_to_b(test_case): method test_eager_boxing_with_non_overlapping_placement_s1_to_p (line 2888) | def test_eager_boxing_with_non_overlapping_placement_s1_to_p(test_case): class TestEagerBoxingWithOverlappingPlacement (line 2897) | class TestEagerBoxingWithOverlappingPlacement(flow.unittest.TestCase): method test_eager_boxing_with_overlapping_placement_p_to_s1 (line 2898) | def test_eager_boxing_with_overlapping_placement_p_to_s1(test_case): method test_eager_boxing_with_overlapping_placement_b_to_s1 (line 2905) | def test_eager_boxing_with_overlapping_placement_b_to_s1(test_case): method test_eager_boxing_with_overlapping_placement_s0_to_s1 (line 2912) | def test_eager_boxing_with_overlapping_placement_s0_to_s1(test_case): method test_eager_boxing_with_overlapping_placement_s1_to_s1 (line 2919) | def test_eager_boxing_with_overlapping_placement_s1_to_s1(test_case): method test_eager_boxing_with_overlapping_placement_s1_to_s0 (line 2926) | def test_eager_boxing_with_overlapping_placement_s1_to_s0(test_case): method test_eager_boxing_with_overlapping_placement_s1_to_b (line 2933) | def test_eager_boxing_with_overlapping_placement_s1_to_b(test_case): method test_eager_boxing_with_overlapping_placement_s1_to_p (line 2941) | def test_eager_boxing_with_overlapping_placement_s1_to_p(test_case): class TestEagerBoxingWithInPlacementContainOutPlacement (line 2950) | class TestEagerBoxingWithInPlacementContainOutPlacement(flow.unittest.Te... method test_eager_boxing_with_in_placement_contain_out_placement_p_to_s1 (line 2951) | def test_eager_boxing_with_in_placement_contain_out_placement_p_to_s1(... method test_eager_boxing_with_in_placement_contain_out_placement_b_to_s1 (line 2960) | def test_eager_boxing_with_in_placement_contain_out_placement_b_to_s1(... method test_eager_boxing_with_in_placement_contain_out_placement_s0_to_s1 (line 2969) | def test_eager_boxing_with_in_placement_contain_out_placement_s0_to_s1... method test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s1 (line 2978) | def test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s1... method test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s0 (line 2987) | def test_eager_boxing_with_in_placement_contain_out_placement_s1_to_s0... method test_eager_boxing_with_in_placement_contain_out_placement_s1_to_p (line 2997) | def test_eager_boxing_with_in_placement_contain_out_placement_s1_to_p(... method test_eager_boxing_with_in_placement_contain_out_placement_s1_to_b (line 3006) | def test_eager_boxing_with_in_placement_contain_out_placement_s1_to_b(... class TestEagerBoxingWithOutPlacementContainInPlacement (line 3017) | class TestEagerBoxingWithOutPlacementContainInPlacement(flow.unittest.Te... method test_eager_boxing_with_out_placement_contain_in_placement_p_to_s1 (line 3018) | def test_eager_boxing_with_out_placement_contain_in_placement_p_to_s1(... method test_eager_boxing_with_out_placement_contain_in_placement_b_to_s1 (line 3027) | def test_eager_boxing_with_out_placement_contain_in_placement_b_to_s1(... method test_eager_boxing_with_out_placement_contain_in_placement_s0_to_s1 (line 3036) | def test_eager_boxing_with_out_placement_contain_in_placement_s0_to_s1... method test_eager_boxing_with_out_placement_contain_in_placement_s1_to_b (line 3045) | def test_eager_boxing_with_out_placement_contain_in_placement_s1_to_b(... method test_eager_boxing_with_out_placement_contain_in_placement_s1_to_p (line 3055) | def test_eager_boxing_with_out_placement_contain_in_placement_s1_to_p(... method test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s0 (line 3064) | def test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s0... method test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s1 (line 3073) | def test_eager_boxing_with_out_placement_contain_in_placement_s1_to_s1... class TestEagerBoxingWithSameInOutPlacement (line 3084) | class TestEagerBoxingWithSameInOutPlacement(flow.unittest.TestCase): method test_eager_boxing_with_same_placement_s0_to_s1 (line 3085) | def test_eager_boxing_with_same_placement_s0_to_s1(test_case): method test_eager_boxing_with_same_placement_p_to_s1 (line 3093) | def test_eager_boxing_with_same_placement_p_to_s1(test_case): method test_eager_boxing_with_same_placement_b_to_s1 (line 3100) | def test_eager_boxing_with_same_placement_b_to_s1(test_case): method test_eager_boxing_with_same_placement_s1_to_s1 (line 3107) | def test_eager_boxing_with_same_placement_s1_to_s1(test_case): method test_eager_boxing_with_same_placement_s1_to_s0 (line 3114) | def test_eager_boxing_with_same_placement_s1_to_s0(test_case): method test_eager_boxing_with_same_placement_s1_to_p (line 3121) | def test_eager_boxing_with_same_placement_s1_to_p(test_case): method test_eager_boxing_with_same_placement_s1_to_b (line 3129) | def test_eager_boxing_with_same_placement_s1_to_b(test_case): class TestEagerBoxingBToS (line 3138) | class TestEagerBoxingBToS(flow.unittest.TestCase): method test_eager_boxing_b_to_s (line 3139) | def test_eager_boxing_b_to_s(test_case): class TestEagerBoxingPToS (line 3151) | class TestEagerBoxingPToS(flow.unittest.TestCase): method test_eager_boxing_p_to_s (line 3152) | def test_eager_boxing_p_to_s(test_case): class TestEagerBoxingSToB (line 3164) | class TestEagerBoxingSToB(flow.unittest.TestCase): method test_eager_boxing_s_to_b (line 3165) | def test_eager_boxing_s_to_b(test_case): class TestEagerBoxingPToB (line 3177) | class TestEagerBoxingPToB(flow.unittest.TestCase): method test_eager_boxing_p_to_b (line 3178) | def test_eager_boxing_p_to_b(test_case): class TestEagerNaiveBoxingSToS (line 3189) | class TestEagerNaiveBoxingSToS(flow.unittest.TestCase): method test_eager_naive_boxing_s_to_s (line 3190) | def test_eager_naive_boxing_s_to_s(test_case): method test_eager_global_with_0_size_data (line 3377) | def test_eager_global_with_0_size_data(test_case): class TestEagerGlobalCastWithSamePlacementAndSBP (line 3204) | class TestEagerGlobalCastWithSamePlacementAndSBP(flow.unittest.TestCase): method test_eager_global_cast_with_same_placement_and_sbp (line 3205) | def test_eager_global_cast_with_same_placement_and_sbp(test_case): class TestEagerGlobalCast1DTo2DSBP (line 3221) | class TestEagerGlobalCast1DTo2DSBP(flow.unittest.TestCase): method test_eager_global_cast_1d_to_2d_sbp (line 3222) | def test_eager_global_cast_1d_to_2d_sbp(test_case): class TestEagerGlobalCast2DTo1DSBP (line 3244) | class TestEagerGlobalCast2DTo1DSBP(flow.unittest.TestCase): method test_eager_global_cast_2d_to_1d_sbp (line 3245) | def test_eager_global_cast_2d_to_1d_sbp(test_case): function _test_eager_global_cast_1d_uneven_split (line 3263) | def _test_eager_global_cast_1d_uneven_split(test_case, device_type, shape): class TestEagerGlobalCastOneDUnevenSplit (line 3312) | class TestEagerGlobalCastOneDUnevenSplit(flow.unittest.TestCase): method test_eager_global_cast_1d_uneven_split (line 3313) | def test_eager_global_cast_1d_uneven_split(test_case): function _test_eager_global_n_dim_reduce (line 3321) | def _test_eager_global_n_dim_reduce(test_case, device_type, src_sbp, dst... class TestEagerGlobalCastNDimReduceBoxing (line 3345) | class TestEagerGlobalCastNDimReduceBoxing(flow.unittest.TestCase): method test_eager_global_n_dim_reduce (line 3346) | def test_eager_global_n_dim_reduce(test_case): function _test_eager_global_with_0_size_data (line 3355) | def _test_eager_global_with_0_size_data( class TestEagerNaiveBoxingSToS (line 3376) | class TestEagerNaiveBoxingSToS(flow.unittest.TestCase): method test_eager_naive_boxing_s_to_s (line 3190) | def test_eager_naive_boxing_s_to_s(test_case): method test_eager_global_with_0_size_data (line 3377) | def test_eager_global_with_0_size_data(test_case): function _test_eager_boxing_one_to_n_with_diff_dim (line 3400) | def _test_eager_boxing_one_to_n_with_diff_dim( function _test_eager_boxing_n_to_one_with_diff_dim (line 3420) | def _test_eager_boxing_n_to_one_with_diff_dim( class TestEagerBoxingOneToNWithDiffDim (line 3440) | class TestEagerBoxingOneToNWithDiffDim(flow.unittest.TestCase): method test_eager_boxing_one_to_n_with_diff_dim (line 3441) | def test_eager_boxing_one_to_n_with_diff_dim(test_case): class TestEagerBoxingNToOneWithDiffDim (line 3450) | class TestEagerBoxingNToOneWithDiffDim(flow.unittest.TestCase): method test_eager_boxing_n_to_one_with_diff_dim (line 3451) | def test_eager_boxing_n_to_one_with_diff_dim(test_case): function _test_asymmetric_mix_1d_2d_eager_boxing_with_random_placement (line 3459) | def _test_asymmetric_mix_1d_2d_eager_boxing_with_random_placement( class TestEagerBoxingAsymmetricMix1d2dWithRandomPlacement (line 3488) | class TestEagerBoxingAsymmetricMix1d2dWithRandomPlacement(flow.unittest.... method test_eager_boxing_asymmetric_mix_1d_2d_with_random_placement (line 3489) | def test_eager_boxing_asymmetric_mix_1d_2d_with_random_placement(test_... class TestEagerBoxing2DLocalToGlobalWithBalancedSplitSize (line 3566) | class TestEagerBoxing2DLocalToGlobalWithBalancedSplitSize(flow.unittest.... method test_eager_boxing_2d_local_to_globa_with_balanced_size (line 3567) | def test_eager_boxing_2d_local_to_globa_with_balanced_size(test_case): FILE: python/oneflow/test/modules/test_eager_boxing_exhaustive.py function _test_eager_boxing_normal_1d_exhaustive_testing (line 30) | def _test_eager_boxing_normal_1d_exhaustive_testing( function _test_eager_boxing_symmetric_2d_exhaustive_testing (line 48) | def _test_eager_boxing_symmetric_2d_exhaustive_testing( function _test_eager_boxing_1d_special_split_axis (line 70) | def _test_eager_boxing_1d_special_split_axis( function _test_eager_boxing_2d_special_split_axis (line 89) | def _test_eager_boxing_2d_special_split_axis(test_case, in_device, out_d... class TestEagerBoxingSymmetricExhaustiveTesting (line 111) | class TestEagerBoxingSymmetricExhaustiveTesting(flow.unittest.TestCase): method test_eager_boxing_normal_1d_exhaustive_testing (line 113) | def test_eager_boxing_normal_1d_exhaustive_testing(test_case): method test_eager_boxing_symmetric_2d_exhaustive_testing (line 124) | def test_eager_boxing_symmetric_2d_exhaustive_testing(test_case): class TestEagerBoxingSpecialSplitAxisExhaustiveTesting (line 134) | class TestEagerBoxingSpecialSplitAxisExhaustiveTesting(flow.unittest.Tes... method test_eager_boxing_1d_special_split_axis (line 136) | def test_eager_boxing_1d_special_split_axis(test_case): method test_eager_boxing_2d_special_split_axis (line 146) | def test_eager_boxing_2d_special_split_axis(test_case): FILE: python/oneflow/test/modules/test_empty.py function _test_local_empty (line 25) | def _test_local_empty(test_case, shape, dtype, device, requires_grad): function _test_new_empty (line 52) | def _test_new_empty(test_case, shape, dtype, device, requires_grad): function _test_local_empty_strided (line 75) | def _test_local_empty_strided(test_case, shape, stride, dtype, device, r... class TestEmptyOp (line 92) | class TestEmptyOp(flow.unittest.TestCase): method test_local_empty (line 93) | def test_local_empty(test_case): method test_local_empty_strided (line 103) | def test_local_empty_strided(test_case): FILE: python/oneflow/test/modules/test_eq.py class TestEq (line 30) | class TestEq(flow.unittest.TestCase): method test_eq_with_0_size_data (line 32) | def test_eq_with_0_size_data(test_case): method test_eq_with_0shape_0d_data (line 40) | def test_eq_with_0shape_0d_data(test_case): method test_flow_eq_with_random_data (line 48) | def test_flow_eq_with_random_data(test_case): method test_flow_tensor_eq_with_random_data (line 56) | def test_flow_tensor_eq_with_random_data(test_case): method test_flow_eq_with_random_0d_data (line 64) | def test_flow_eq_with_random_0d_data(test_case): method test_flow_eq_with_same_random_data (line 72) | def test_flow_eq_with_same_random_data(test_case): method test_flow_eq_bool_with_random_data (line 79) | def test_flow_eq_bool_with_random_data(test_case): method test_flow_eq_with_same_random_0d_data (line 91) | def test_flow_eq_with_same_random_0d_data(test_case): method profile_eq (line 98) | def profile_eq(test_case): FILE: python/oneflow/test/modules/test_equal.py class TestEqual (line 31) | class TestEqual(flow.unittest.TestCase): method test_eq_with_0_size_data (line 33) | def test_eq_with_0_size_data(test_case): method test_equal_with_0shape_0d_data (line 41) | def test_equal_with_0shape_0d_data(test_case): method test_flow_equal_with_random_data (line 49) | def test_flow_equal_with_random_data(test_case): method test_flow_tensor_equal_with_random_data (line 57) | def test_flow_tensor_equal_with_random_data(test_case): method test_flow_equal_with_random_0d_data (line 65) | def test_flow_equal_with_random_0d_data(test_case): method test_flow_equal_with_same_random_data (line 73) | def test_flow_equal_with_same_random_data(test_case): method test_flow_equal_complex_with_same_random_data (line 80) | def test_flow_equal_complex_with_same_random_data(test_case): method test_flow_equal_bool_with_random_data (line 89) | def test_flow_equal_bool_with_random_data(test_case): method test_flow_equal_complex_with_random_data (line 101) | def test_flow_equal_complex_with_random_data(test_case): method test_flow_not_equal_complex_with_random_data (line 113) | def test_flow_not_equal_complex_with_random_data(test_case): method test_flow_equal_with_same_random_0d_data (line 125) | def test_flow_equal_with_same_random_0d_data(test_case): method profile_equal (line 132) | def profile_equal(test_case): FILE: python/oneflow/test/modules/test_erf.py class TestErfModule (line 31) | class TestErfModule(flow.unittest.TestCase): method test_flow_erf_with_random_data (line 33) | def test_flow_erf_with_random_data(test_case): method test_flow_erf_with_0dim_data (line 40) | def test_flow_erf_with_0dim_data(test_case): method profile_erf (line 47) | def profile_erf(test_case): FILE: python/oneflow/test/modules/test_erfc.py class TestErfcModule (line 31) | class TestErfcModule(flow.unittest.TestCase): method test_flow_erfc_with_random_data (line 33) | def test_flow_erfc_with_random_data(test_case): method test_flow_erfc_with_0dim_data (line 40) | def test_flow_erfc_with_0dim_data(test_case): method profile_erfc (line 47) | def profile_erfc(test_case): FILE: python/oneflow/test/modules/test_erfinv.py function _test_flow_erfinv_with_inf_data (line 30) | def _test_flow_erfinv_with_inf_data(test_case, device): function _test_flow_erfinv_with_nan_data (line 37) | def _test_flow_erfinv_with_nan_data(test_case, device): class TestErfinvModule (line 47) | class TestErfinvModule(flow.unittest.TestCase): method test_flow_erfinv (line 48) | def test_flow_erfinv(test_case): method test_flow_erfinv_with_random_data (line 59) | def test_flow_erfinv_with_random_data(test_case): method profile_erfinv (line 66) | def profile_erfinv(test_case): FILE: python/oneflow/test/modules/test_expand.py function _np_get_expand (line 27) | def _np_get_expand(input_shape, expand_size): function _test_expand_new_dims (line 85) | def _test_expand_new_dims(test_case, device): function _test_expand_same_dim (line 96) | def _test_expand_same_dim(test_case, device): function _test_expand_same_dim_negative (line 106) | def _test_expand_same_dim_negative(test_case, device): function _test_expand_same_int (line 116) | def _test_expand_same_int(test_case, device): function _test_expand_flow_size (line 126) | def _test_expand_flow_size(test_case, device): function _test_expand_backward_same_dim (line 136) | def _test_expand_backward_same_dim(test_case, device): function _test_expand_backward (line 165) | def _test_expand_backward(test_case, device): function random_expand (line 185) | def random_expand(x, ndim, expand_size): class TestExpand (line 193) | class TestExpand(flow.unittest.TestCase): method test_flow_tensor_expand_with_random_data (line 195) | def test_flow_tensor_expand_with_random_data(test_case): method test_flow_tensor_expand_bool_with_random_data (line 201) | def test_flow_tensor_expand_bool_with_random_data(test_case): method test_expand_compare_with_numpy (line 206) | def test_expand_compare_with_numpy(test_case): method test_flow_expand_with_0_size (line 222) | def test_flow_expand_with_0_size(test_case): FILE: python/oneflow/test/modules/test_expand_stride.py function _cmp_expand_stride (line 28) | def _cmp_expand_stride( function _cmp_expand_non_contiguous_stride (line 56) | def _cmp_expand_non_contiguous_stride( function _cmp_lazy_expand_stride (line 88) | def _cmp_lazy_expand_stride( class ExpandStrideTestCase (line 139) | class ExpandStrideTestCase(flow.unittest.TestCase): method test_on_cpu (line 184) | def test_on_cpu(self): method test_stride (line 194) | def test_stride(self): method test_non_contiguous_stride (line 204) | def test_non_contiguous_stride(self): method test_lazy (line 226) | def test_lazy(self): FILE: python/oneflow/test/modules/test_expm1.py function _test_expm1_impl (line 29) | def _test_expm1_impl(test_case, device, shape): class TestExpm1Module (line 45) | class TestExpm1Module(flow.unittest.TestCase): method test_expm1 (line 46) | def test_expm1(test_case): method test_expm1_flow_with_random_data (line 55) | def test_expm1_flow_with_random_data(test_case): method test_expm1_with_0_size_data (line 62) | def test_expm1_with_0_size_data(test_case): method test_expm1_flow_with_0dim_data (line 69) | def test_expm1_flow_with_0dim_data(test_case): method profile_expm1 (line 76) | def profile_expm1(test_case): FILE: python/oneflow/test/modules/test_eye.py function _test_eye_forward (line 27) | def _test_eye_forward(test_case, device, n, m): function _test_eye_backward (line 33) | def _test_eye_backward(test_case, device, n, m): function _test_eye_with_1n2d (line 41) | def _test_eye_with_1n2d(test_case, n, m, device): class TestEye (line 49) | class TestEye(flow.unittest.TestCase): method test_eye (line 50) | def test_eye(test_case): method test_eye_with_random_data (line 63) | def test_eye_with_random_data(test_case): method test_eye_with_random_data (line 72) | def test_eye_with_random_data(test_case): method test_eye_bool_with_random_data (line 79) | def test_eye_bool_with_random_data(test_case): method test_eye_with_0dim_data (line 89) | def test_eye_with_0dim_data(test_case): method profile_eye (line 99) | def profile_eye(test_case): class TestGlobalEye (line 105) | class TestGlobalEye(flow.unittest.TestCase): method test_eye_with_1n2d (line 106) | def test_eye_with_1n2d(test_case): FILE: python/oneflow/test/modules/test_fake_quantization.py function gen_quant_scale_for_min_max_symmetric (line 35) | def gen_quant_scale_for_min_max_symmetric(weight, quantization_bit): function gen_quant_scale_for_min_max_affine (line 41) | def gen_quant_scale_for_min_max_affine(weight, quantization_bit): function gen_quant_scale_for_min_max_cambricon (line 50) | def gen_quant_scale_for_min_max_cambricon(weight, quantization_bit): function product (line 56) | def product(tu): function fake_quant_per_layer_symmetric (line 60) | def fake_quant_per_layer_symmetric(input, quantization_bit, scale): function fake_quant_per_layer_affine (line 66) | def fake_quant_per_layer_affine(input, quantization_bit, scale, zero_poi... function fake_quant_per_layer_cambricon (line 75) | def fake_quant_per_layer_cambricon(input, quantization_bit, shift): function _check_fake_quantize (line 82) | def _check_fake_quantize( function _run_test_fake_quantize (line 139) | def _run_test_fake_quantize( class TestFakeQuantize (line 183) | class TestFakeQuantize(flow.unittest.TestCase): method test_fake_quantize (line 184) | def test_fake_quantize(test_case): FILE: python/oneflow/test/modules/test_fft.py function is_cufft_available (line 31) | def is_cufft_available(): function is_complex_dtype (line 39) | def is_complex_dtype(dtype): function gen_params_1d_fft (line 54) | def gen_params_1d_fft(lower_n_dims=1, upper_n_dims=5): function gen_params_2d_fft (line 80) | def gen_params_2d_fft(lower_n_dims=2, upper_n_dims=5): function gen_params_nd_fft (line 117) | def gen_params_nd_fft(lower_n_dims=2, upper_n_dims=5): function _test_fft (line 159) | def _test_fft(test_case): function _test_ifft (line 188) | def _test_ifft(test_case): function _test_rfft (line 219) | def _test_rfft(test_case): function _test_irfft (line 243) | def _test_irfft(test_case): function _test_hfft (line 266) | def _test_hfft(test_case): function _test_ihfft (line 289) | def _test_ihfft(test_case): function _test_fft2 (line 312) | def _test_fft2(test_case): function _test_ifft2 (line 342) | def _test_ifft2(test_case): function _test_rfft2 (line 373) | def _test_rfft2(test_case): function _test_irfft2 (line 397) | def _test_irfft2(test_case): function _test_hfft2 (line 420) | def _test_hfft2(test_case): function _test_ihfft2 (line 443) | def _test_ihfft2(test_case): function _test_fftn (line 466) | def _test_fftn(test_case): function _test_ifftn (line 496) | def _test_ifftn(test_case): function _test_rfftn (line 527) | def _test_rfftn(test_case): function _test_irfftn (line 551) | def _test_irfftn(test_case): function _test_hfftn (line 574) | def _test_hfftn(test_case): function _test_ihfftn (line 597) | def _test_ihfftn(test_case): class TestComplex64Fft (line 622) | class TestComplex64Fft(flow.unittest.TestCase): method setUp (line 623) | def setUp(test_case): method initTestFft (line 640) | def initTestFft(test_case): method test_1d_fft (line 825) | def test_1d_fft(test_case): method test_2d_fft_except_hfft2 (line 838) | def test_2d_fft_except_hfft2(test_case): method test_2d_fft_hfft2 (line 853) | def test_2d_fft_hfft2(test_case): method test_nd_fft_except_hfftn (line 859) | def test_nd_fft_except_hfftn(test_case): method test_nd_fft_hfftn (line 874) | def test_nd_fft_hfftn(test_case): class TestComplex128Fft (line 883) | class TestComplex128Fft(TestComplex64Fft): method setUp (line 884) | def setUp(test_case): FILE: python/oneflow/test/modules/test_flatten.py function _test_flatten (line 29) | def _test_flatten(test_case, device): function _test_flatten_backward (line 50) | def _test_flatten_backward(test_case, device): class TestFlattenModule (line 62) | class TestFlattenModule(flow.unittest.TestCase): method test_cast (line 63) | def test_cast(test_case): method test_flatten_module_with_random_data (line 71) | def test_flatten_module_with_random_data(test_case): method test_flatten_with_random_data (line 83) | def test_flatten_with_random_data(test_case): method test_flatten_bool_with_random_data (line 94) | def test_flatten_bool_with_random_data(test_case): method test_flatten_with_0dim_data (line 105) | def test_flatten_with_0dim_data(test_case): method profile_flatten (line 116) | def profile_flatten(test_case): FILE: python/oneflow/test/modules/test_flip.py class TestFlip (line 30) | class TestFlip(flow.unittest.TestCase): method test_flow_flip_list_with_random_data (line 32) | def test_flow_flip_list_with_random_data(test_case): method test_flow_flip_tuple_with_random_data (line 41) | def test_flow_flip_tuple_with_random_data(test_case): method test_flow_flip_bool_tuple_with_random_data (line 50) | def test_flow_flip_bool_tuple_with_random_data(test_case): method test_flow_flip_list_lastdim_with_random_data (line 58) | def test_flow_flip_list_lastdim_with_random_data(test_case): method profile_flip (line 67) | def profile_flip(test_case): FILE: python/oneflow/test/modules/test_floor.py function _test_floor (line 29) | def _test_floor(test_case, shape, device): class TestFloor (line 48) | class TestFloor(flow.unittest.TestCase): method test_floor (line 49) | def test_floor(test_case): method test_flow_floor_with_random_data (line 57) | def test_flow_floor_with_random_data(test_case): method test_flow_floor_inplace_with_random_data (line 64) | def test_flow_floor_inplace_with_random_data(test_case): method test_flow_floor_with_0dim_data (line 72) | def test_flow_floor_with_0dim_data(test_case): method profile_floor (line 79) | def profile_floor(test_case): FILE: python/oneflow/test/modules/test_fmod.py class TestFmodModule (line 34) | class TestFmodModule(flow.unittest.TestCase): method test_flow_fmod_element_with_random_data (line 41) | def test_flow_fmod_element_with_random_data(test_case): method test_flow_fmod_element_with_0dim_data (line 50) | def test_flow_fmod_element_with_0dim_data(test_case): method test_flow_fmod_broadcast_with_random_data (line 57) | def test_flow_fmod_broadcast_with_random_data(test_case): method test_flow_fmod_scalar_with_random_data (line 66) | def test_flow_fmod_scalar_with_random_data(test_case): method test_fmod_with_0_size_data (line 75) | def test_fmod_with_0_size_data(test_case): method profile_fmod (line 82) | def profile_fmod(test_case): FILE: python/oneflow/test/modules/test_fold.py class TestFold (line 26) | class TestFold(flow.unittest.TestCase): method test_fold_with_random_data_1 (line 28) | def test_fold_with_random_data_1(test_case): method test_fold_with_random_data_2 (line 54) | def test_fold_with_random_data_2(test_case): method test_fold_with_random_data_3 (line 80) | def test_fold_with_random_data_3(test_case): method test_fold_with_random_data_4 (line 106) | def test_fold_with_random_data_4(test_case): method profile_fold (line 132) | def profile_fold(test_case): FILE: python/oneflow/test/modules/test_fork_sub_process.py function _test_fork_sub_process (line 27) | def _test_fork_sub_process(id): class TestForkSubProcess (line 39) | class TestForkSubProcess(flow.unittest.TestCase): method test_fork_sub_process (line 40) | def test_fork_sub_process(test_case): FILE: python/oneflow/test/modules/test_frac.py class TestFrac (line 23) | class TestFrac(flow.unittest.TestCase): method test_frac (line 25) | def test_frac(test_case): method test_tensor_frac (line 34) | def test_tensor_frac(test_case): method test_tensor_frac_inplace (line 43) | def test_tensor_frac_inplace(test_case): FILE: python/oneflow/test/modules/test_from_numpy.py class TestFromNumpy (line 28) | class TestFromNumpy(flow.unittest.TestCase): method test_same_data (line 29) | def test_same_data(test_case): method test_use_ops (line 40) | def test_use_ops(test_case): method test_more_dtype (line 46) | def test_more_dtype(test_case): method test_non_contiguous_input (line 62) | def test_non_contiguous_input(test_case): FILE: python/oneflow/test/modules/test_from_torch.py function torch_device_to_flow (line 25) | def torch_device_to_flow(device): class TestFromTroch (line 34) | class TestFromTroch(flow.unittest.TestCase): method test_from_torch_cpu (line 36) | def test_from_torch_cpu(test_case): method test_from_torch_cpu_with_0_size_data (line 62) | def test_from_torch_cpu_with_0_size_data(test_case): method test_from_torch_cpu_with_0dim_data (line 73) | def test_from_torch_cpu_with_0dim_data(test_case): method test_from_torch_gpu (line 90) | def test_from_torch_gpu(test_case): FILE: python/oneflow/test/modules/test_functional_docstr.py function _run_functional_doctest (line 27) | def _run_functional_doctest( class TestFunctionalDocstrModule (line 66) | class TestFunctionalDocstrModule(flow.unittest.TestCase): method test_functional_docstr (line 67) | def test_functional_docstr(test_case): FILE: python/oneflow/test/modules/test_functional_scalar_tensor_param.py class TestFunctionalWithScalarTensorParam (line 27) | class TestFunctionalWithScalarTensorParam(flow.unittest.TestCase): method test_scalar_tensor_transfer_to_scalar (line 30) | def test_scalar_tensor_transfer_to_scalar(test_case): method test_scalar_tensor_transfer_to_double (line 38) | def test_scalar_tensor_transfer_to_double(test_case): method test_scalar_tensor_transfer_to_int (line 45) | def test_scalar_tensor_transfer_to_int(test_case): FILE: python/oneflow/test/modules/test_fused_attention_ops.py function _ref (line 27) | def _ref( function _to_layout (line 82) | def _to_layout(ts, layout, tensor_index, seq_len=None): function _fused_mha (line 157) | def _fused_mha( function _test_fused_attention_concat_past_key_value (line 241) | def _test_fused_attention_concat_past_key_value( function _test_fused_multi_head_attention_inference (line 301) | def _test_fused_multi_head_attention_inference( function _test_fused_multi_head_attention_inference_with_attn_bias (line 357) | def _test_fused_multi_head_attention_inference_with_attn_bias( function _test_fused_multi_head_attention_inference_variable_length (line 441) | def _test_fused_multi_head_attention_inference_variable_length( class TestFusedMultiHeadAttentionInference (line 520) | class TestFusedMultiHeadAttentionInference(flow.unittest.TestCase): method test_multi_head_attention_inference (line 521) | def test_multi_head_attention_inference(test_case): method test_multi_head_attention_inference_with_attn_bias (line 573) | def test_multi_head_attention_inference_with_attn_bias(test_case): method test_multi_head_attention_inference_with_layout (line 606) | def test_multi_head_attention_inference_with_layout(test_case): method test_multi_head_attention_inference_with_output_layout (line 637) | def test_multi_head_attention_inference_with_output_layout(test_case): method test_multi_head_attention_inference_variable_length (line 666) | def test_multi_head_attention_inference_variable_length(test_case): class TestFusedAttentionConcatPastKeyValue (line 715) | class TestFusedAttentionConcatPastKeyValue(flow.unittest.TestCase): method test_fused_attention_concat_past_key_value (line 716) | def test_fused_attention_concat_past_key_value(test_case): FILE: python/oneflow/test/modules/test_fused_bias_add_dropout.py function _test_fused_bias_add_dropout (line 28) | def _test_fused_bias_add_dropout(test_case, shape, axis, drop_prob): class TestFusedBiasAddDropout (line 76) | class TestFusedBiasAddDropout(flow.unittest.TestCase): method test_fuse_bias_add_dropout (line 77) | def test_fuse_bias_add_dropout(test_case): FILE: python/oneflow/test/modules/test_fused_bias_add_gelu.py function _test_fused_bias_add_gelu (line 28) | def _test_fused_bias_add_gelu(test_case, channel, axis): class TestFusedBiasAddGelu (line 74) | class TestFusedBiasAddGelu(flow.unittest.TestCase): method test_gather (line 75) | def test_gather(test_case): FILE: python/oneflow/test/modules/test_fused_bias_add_scale_mask_softmax_dropout.py function _torch_bias_add_scale_mask_softmax_dropout (line 28) | def _torch_bias_add_scale_mask_softmax_dropout(x, bias, mask, fill, scal... function _test_bias_add_fused_scale_mask_softmax_dropout (line 37) | def _test_bias_add_fused_scale_mask_softmax_dropout( class TestFusedBiasAddScaleMaskSoftmaxDropout (line 140) | class TestFusedBiasAddScaleMaskSoftmaxDropout(flow.unittest.TestCase): method test_real_case (line 141) | def test_real_case(test_case): method test_different_broadcast_dim (line 155) | def test_different_broadcast_dim(test_case): method test_same_broadcast_dim (line 160) | def test_same_broadcast_dim(test_case): method test_broadcast_bias (line 165) | def test_broadcast_bias(test_case): method test_broadcast_mask (line 170) | def test_broadcast_mask(test_case): FILE: python/oneflow/test/modules/test_fused_center.py function torch_center (line 28) | def torch_center(b1_x1, b1_x2, b2_x1, b2_x2, b1_y1, b1_y2, b2_y1, b2_y2): function _test_fused_get_center_dist_impl (line 34) | def _test_fused_get_center_dist_impl(test_case, device, shape): class TestGetCenterDistModule (line 118) | class TestGetCenterDistModule(flow.unittest.TestCase): method test_fused_get_center_dist (line 119) | def test_fused_get_center_dist(test_case): FILE: python/oneflow/test/modules/test_fused_codegeex_qkv_reshape.py function _test_codegeex_qkv_reshape_impl (line 26) | def _test_codegeex_qkv_reshape_impl(test_case, device, shape, num_attent... class TestFusedCodegeexQkvReshapeModule (line 62) | class TestFusedCodegeexQkvReshapeModule(flow.unittest.TestCase): method test_codegeex_qkv_reshape (line 63) | def test_codegeex_qkv_reshape(test_case): FILE: python/oneflow/test/modules/test_fused_cross_interaction.py function _test_fused_cross_feature_interaction_v1 (line 25) | def _test_fused_cross_feature_interaction_v1( function _test_fused_cross_feature_interaction_v2 (line 74) | def _test_fused_cross_feature_interaction_v2( class TestFusedCrossFeatureInteraction (line 129) | class TestFusedCrossFeatureInteraction(flow.unittest.TestCase): method test_fused_cross_feature_interaction_v1 (line 130) | def test_fused_cross_feature_interaction_v1(test_case): method test_fused_cross_feature_interaction_v2 (line 141) | def test_fused_cross_feature_interaction_v2(test_case): FILE: python/oneflow/test/modules/test_fused_dot_feature_interaction.py function _test_fused_dot_feature_interaction (line 26) | def _test_fused_dot_feature_interaction( function _test_fused_dot_feature_interaction_pooling_sum (line 119) | def _test_fused_dot_feature_interaction_pooling_sum( class FusedDotFeatureInteractionTestCase (line 176) | class FusedDotFeatureInteractionTestCase(flow.unittest.TestCase): method test_fused_dot_feature_interaction (line 177) | def test_fused_dot_feature_interaction(test_case): method test_fused_dot_feature_interaction_pooling_sum (line 187) | def test_fused_dot_feature_interaction_pooling_sum(test_case): FILE: python/oneflow/test/modules/test_fused_gelu_mul.py function _test_fused_fast_gelu_mul (line 26) | def _test_fused_fast_gelu_mul(test_case, shape, dtype=flow.float32): class TestFusedFastGeluMul (line 62) | class TestFusedFastGeluMul(flow.unittest.TestCase): method test_fused_fast_gelu_mul (line 63) | def test_fused_fast_gelu_mul(test_case): FILE: python/oneflow/test/modules/test_fused_get_boundding_boxes_coord.py function _test_get_boundding_boxes_coord_impl (line 28) | def _test_get_boundding_boxes_coord_impl(test_case, device, shape): class TestGetBounddingBoxesCoordModule (line 132) | class TestGetBounddingBoxesCoordModule(flow.unittest.TestCase): method test_get_boundding_boxes_coord (line 133) | def test_get_boundding_boxes_coord(test_case): FILE: python/oneflow/test/modules/test_fused_get_ciou_diagonal_angle.py function torch_get_ciou_diagonal_angle (line 29) | def torch_get_ciou_diagonal_angle(w1, h1, w2, h2, eps=1e-8): function _test_fused_get_ciou_diagonal_angle_impl (line 35) | def _test_fused_get_ciou_diagonal_angle_impl(test_case, device, shape): class TestGetCiouDiagonalAngle (line 87) | class TestGetCiouDiagonalAngle(flow.unittest.TestCase): method test_fused_get_ciou_diagonal_angle (line 88) | def test_fused_get_ciou_diagonal_angle(test_case): FILE: python/oneflow/test/modules/test_fused_get_ciou_result.py function _test_get_ciou_result_impl (line 28) | def _test_get_ciou_result_impl(test_case, device, shape): class TestGetCiouResultModule (line 81) | class TestGetCiouResultModule(flow.unittest.TestCase): method test_get_ciou_result (line 82) | def test_get_ciou_result(test_case): FILE: python/oneflow/test/modules/test_fused_get_convex_diagonal_squared.py function torch_fused_get_convex_diagonal_squared (line 28) | def torch_fused_get_convex_diagonal_squared( function _test_fused_get_convex_diagonal_squared_impl (line 37) | def _test_fused_get_convex_diagonal_squared_impl(test_case, device, shape): class TestGetCenterDistModule (line 123) | class TestGetCenterDistModule(flow.unittest.TestCase): method test_fused_get_convex_diagonal_squared (line 124) | def test_fused_get_convex_diagonal_squared(test_case): FILE: python/oneflow/test/modules/test_fused_get_intersection_area.py function torch_get_intersection_area (line 27) | def torch_get_intersection_area(b1_x1, b1_x2, b2_x1, b2_x2, b1_y1, b1_y2... function _test_fused_get_intersection_area_impl (line 33) | def _test_fused_get_intersection_area_impl(test_case, device, shape): class TestGetIntersectionAreaModule (line 117) | class TestGetIntersectionAreaModule(flow.unittest.TestCase): method test_fused_get_inter_intersection_area (line 118) | def test_fused_get_inter_intersection_area(test_case): FILE: python/oneflow/test/modules/test_fused_get_iou.py function _test_get_iou_impl (line 28) | def _test_get_iou_impl(test_case, device, shape): class TestGetIouModule (line 81) | class TestGetIouModule(flow.unittest.TestCase): method test_get_iou (line 82) | def test_get_iou(test_case): FILE: python/oneflow/test/modules/test_fused_glu.py class Glu (line 31) | class Glu(nn.Module): method __init__ (line 32) | def __init__(self): method forward (line 35) | def forward( function tensor_builder (line 86) | def tensor_builder(params: dict, dtype=flow.float32, is_split_mode=True): function compare_result (line 123) | def compare_result(test_case, a, b, rtol=1e-5, atol=1e-8): function _test_fused_glu (line 130) | def _test_fused_glu(test_case, params: dict, dtype=flow.float32): function _test_fused_glu_without_bias (line 180) | def _test_fused_glu_without_bias(test_case, params: dict, dtype=flow.flo... function _test_fused_glu_split (line 226) | def _test_fused_glu_split(test_case, params: dict, dtype=flow.float32): function _test_fused_glu_split_without_bias (line 289) | def _test_fused_glu_split_without_bias(test_case, params: dict, dtype=fl... class TestFusedGlu (line 344) | class TestFusedGlu(flow.unittest.TestCase): method test_gather (line 346) | def test_gather(test_case): FILE: python/oneflow/test/modules/test_fused_matmul_bias.py function _matmul_bias (line 25) | def _matmul_bias(x, weight, bias, add_to_output): function _test_fused_matmul_add_bias (line 34) | def _test_fused_matmul_add_bias( class TestFusedMatmulBiasAddRelu (line 97) | class TestFusedMatmulBiasAddRelu(flow.unittest.TestCase): method test_fused_matmul_op (line 98) | def test_fused_matmul_op(test_case): FILE: python/oneflow/test/modules/test_fused_matmul_bias_add_relu_dropout.py function _matmul_bias_relu (line 24) | def _matmul_bias_relu(x, weight, bias, skip_activate): function _test_fused_matmul_bias_add_relu_dropout (line 32) | def _test_fused_matmul_bias_add_relu_dropout( class TestFusedMatmulBiasAddReluDropout (line 175) | class TestFusedMatmulBiasAddReluDropout(flow.unittest.TestCase): method test_fused_matmul_bias_add_relu_dropout (line 176) | def test_fused_matmul_bias_add_relu_dropout(test_case): FILE: python/oneflow/test/modules/test_fused_rotary_embedding.py function plane_shuffle (line 26) | def plane_shuffle(x): function shuffle_adjacent_two_elem (line 31) | def shuffle_adjacent_two_elem(x): function parseDims (line 39) | def parseDims(dims, x_layout): function naive_embedding (line 92) | def naive_embedding( function _test_without_position (line 176) | def _test_without_position( function _test_without_position_sinuous (line 362) | def _test_without_position_sinuous( function _test_with_position_sinuous (line 505) | def _test_with_position_sinuous( function _test_with_position (line 700) | def _test_with_position( function _test_plane (line 849) | def _test_plane( class TestFusedRotaryEmbedding (line 1015) | class TestFusedRotaryEmbedding(flow.unittest.TestCase): method test_fused_rotary_embedding_op_plane (line 1017) | def test_fused_rotary_embedding_op_plane(test_case): method test_fused_rotary_embedding_op_interval_2d (line 1034) | def test_fused_rotary_embedding_op_interval_2d(test_case): method test_fused_rotary_embedding_op_interval_1d (line 1051) | def test_fused_rotary_embedding_op_interval_1d(test_case): FILE: python/oneflow/test/modules/test_fused_scale_mask_bias_softmax.py function timing (line 28) | def timing(fn): function permute_final_dims (line 45) | def permute_final_dims(tensor: flow.Tensor, inds: List[int]): function _fused_op (line 52) | def _fused_op(x, v, scale, mask, bias, inplace=False): function _ref_op (line 59) | def _ref_op(x, v, scale, mask, bias=None, inplace=False): function _test_fused_scale_mask_bias_softmax (line 66) | def _test_fused_scale_mask_bias_softmax( class TestFusedMsaSoftmax (line 144) | class TestFusedMsaSoftmax(flow.unittest.TestCase): method test_fused_msa_softmax (line 145) | def test_fused_msa_softmax(test_case): FILE: python/oneflow/test/modules/test_fused_scale_mask_softmax.py function _test_fused_scale_mask_softmax (line 28) | def _test_fused_scale_mask_softmax( class TestFusedScaleMaskSoftmax (line 73) | class TestFusedScaleMaskSoftmax(flow.unittest.TestCase): method test_fused_op (line 74) | def test_fused_op(test_case): FILE: python/oneflow/test/modules/test_fused_scale_mask_softmax_dropout.py function _test_fused_scale_mask_softmax_dropout (line 29) | def _test_fused_scale_mask_softmax_dropout( class TestFusedScaleMaskSoftmaxDropout (line 85) | class TestFusedScaleMaskSoftmaxDropout(flow.unittest.TestCase): method test_fused_op (line 86) | def test_fused_op(test_case): FILE: python/oneflow/test/modules/test_fused_scale_tril.py function _np_tril (line 26) | def _np_tril(x, diagonal, fill_value, scale): function _test_fused_scale_tril (line 37) | def _test_fused_scale_tril( class FusedScaleTrilTestCase (line 77) | class FusedScaleTrilTestCase(flow.unittest.TestCase): method test_fused_scale_tril (line 78) | def test_fused_scale_tril(test_case): FILE: python/oneflow/test/modules/test_fused_self_attention.py function _test_fused_self_attention (line 27) | def _test_fused_self_attention(test_case, batch_size, seq_len, num_heads... class TestFusedSelfAttention (line 91) | class TestFusedSelfAttention(flow.unittest.TestCase): method _test_fused_self_attention (line 92) | def _test_fused_self_attention(test_case): FILE: python/oneflow/test/modules/test_fused_tril_softmax_mask_scale.py function _test_fused_tril_softmax_mask_scale (line 28) | def _test_fused_tril_softmax_mask_scale( class TestFusedTrilSoftmaxMaskScale (line 66) | class TestFusedTrilSoftmaxMaskScale(flow.unittest.TestCase): method test_fused_tril_softmax_dropout (line 67) | def test_fused_tril_softmax_dropout(test_case): FILE: python/oneflow/test/modules/test_fused_weighted_sum.py function _ref (line 27) | def _ref(inputs, weights, alpha, init_grad, device, dtype): function _fused_weighted_sum (line 40) | def _fused_weighted_sum(inputs, weights, alpha, init_grad, device, dtype): function _test_fused_weighted_sum (line 50) | def _test_fused_weighted_sum(test_case, shape, n, device, dtype): class TestFusedWeightedSum (line 64) | class TestFusedWeightedSum(flow.unittest.TestCase): method test_fused_weighted_sum (line 65) | def test_fused_weighted_sum(test_case): FILE: python/oneflow/test/modules/test_gather.py function _scatter_add_numpy (line 28) | def _scatter_add_numpy(src, dim, index, outshape): function _test_gather (line 41) | def _test_gather(test_case, device): function _test_gather_tensor_function (line 53) | def _test_gather_tensor_function(test_case, device): function _test_gather_random_array (line 63) | def _test_gather_random_array(test_case, device): function _test_gather_backward (line 89) | def _test_gather_backward(test_case, device): function _test_gather_index_0dim_tensor (line 106) | def _test_gather_index_0dim_tensor(test_case, device): function _test_gather_input_index_0dim_tensor (line 116) | def _test_gather_input_index_0dim_tensor(test_case, device): function _test_gather_input_0dim_tensor (line 126) | def _test_gather_input_0dim_tensor(test_case, device): class TestGather (line 137) | class TestGather(flow.unittest.TestCase): method test_gather (line 138) | def test_gather(test_case): method test_flow_gather_with_random_data (line 154) | def test_flow_gather_with_random_data(test_case): method test_flow_gather_bool_with_random_data (line 170) | def test_flow_gather_bool_with_random_data(test_case): method profile_gather (line 188) | def profile_gather(test_case): FILE: python/oneflow/test/modules/test_gather_nd.py function _test_gather_nd (line 27) | def _test_gather_nd(test_case, device): function _test_gather_nd_t (line 38) | def _test_gather_nd_t(test_case, device): function _test_gather_nd_backward (line 49) | def _test_gather_nd_backward(test_case, device): function _test_gather_nd_backward_t (line 66) | def _test_gather_nd_backward_t(test_case, device): class TestGather_nd (line 84) | class TestGather_nd(flow.unittest.TestCase): method test_gather_nd (line 85) | def test_gather_nd(test_case): FILE: python/oneflow/test/modules/test_gelu_approximate.py class NewGELUActivation (line 29) | class NewGELUActivation(torch.nn.Module): method forward (line 35) | def forward(self, input: torch.Tensor) -> torch.Tensor: function _test_gelu_approximate (line 49) | def _test_gelu_approximate(test_case, device): class TestModule (line 67) | class TestModule(flow.unittest.TestCase): method test_gelu_approximate (line 68) | def test_gelu_approximate(test_case): FILE: python/oneflow/test/modules/test_generator.py class TestGenerator (line 25) | class TestGenerator(flow.unittest.TestCase): method test_different_devices (line 26) | def test_different_devices(test_case): method test_generator_manual_seed (line 36) | def test_generator_manual_seed(test_case): method test_generator_in_dropout (line 43) | def test_generator_in_dropout(test_case): class TestDefaultGenerator (line 56) | class TestDefaultGenerator(flow.unittest.TestCase): method test_different_devices (line 57) | def test_different_devices(test_case): method test_generator_manual_seed (line 72) | def test_generator_manual_seed(test_case): method test_generator_seed (line 85) | def test_generator_seed(test_case): method test_generator_getstate (line 97) | def test_generator_getstate(test_case): method test_generator_setstate (line 107) | def test_generator_setstate(test_case): method test_get_rng_state (line 138) | def test_get_rng_state(test_case): method test_set_rng_state (line 149) | def test_set_rng_state(test_case): method test_tensor_init (line 181) | def test_tensor_init(test_case): FILE: python/oneflow/test/modules/test_global_0_dim_tensor.py function _test_0_dim_tensor (line 24) | def _test_0_dim_tensor(test_case, placement, sbp): function _test_1dim_slice (line 33) | def _test_1dim_slice(test_case, placement, sbp): class TestZeroDimensionTensor (line 38) | class TestZeroDimensionTensor(flow.unittest.TestCase): method test_0_dim_tensor (line 41) | def test_0_dim_tensor(test_case): FILE: python/oneflow/test/modules/test_global_TripletMarginLoss.py function _test_global_triplet_marginloss_with_random_data (line 27) | def _test_global_triplet_marginloss_with_random_data(test_case, placemen... class TestGlobalTripletMarginLoss (line 41) | class TestGlobalTripletMarginLoss(flow.unittest.TestCase): method test_global_triplet_marginloss_with_random_data (line 44) | def test_global_triplet_marginloss_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_abs.py function _test_abs_with_ndim_data (line 25) | def _test_abs_with_ndim_data(test_case, ndim, placement, sbp): class TestAbsModule (line 32) | class TestAbsModule(flow.unittest.TestCase): method test_abs_with_ndim_data (line 35) | def test_abs_with_ndim_data(test_case): FILE: python/oneflow/test/modules/test_global_activation.py function build_module (line 24) | def build_module(act_type): function _test_activation_module_with_random_data (line 70) | def _test_activation_module_with_random_data(test_case, act_type, ndim, ... function _test_activation_module_with_0dim_data (line 80) | def _test_activation_module_with_0dim_data(test_case, act_type, placemen... function _test_activation_module_with_0_size_data (line 89) | def _test_activation_module_with_0_size_data( function _test_activation_module (line 102) | def _test_activation_module(test_case, act_type): class TestReLUModule (line 123) | class TestReLUModule(flow.unittest.TestCase): method test_relu_module (line 124) | def test_relu_module(test_case): class TestReLU6Module (line 128) | class TestReLU6Module(flow.unittest.TestCase): method test_relu6_module (line 129) | def test_relu6_module(test_case): class TestTanh (line 133) | class TestTanh(flow.unittest.TestCase): method test_tanh_module (line 134) | def test_tanh_module(test_case): class TestELUModule (line 138) | class TestELUModule(flow.unittest.TestCase): method test_elu_module (line 139) | def test_elu_module(test_case): class TestCELUModule (line 143) | class TestCELUModule(flow.unittest.TestCase): method test_celu_module (line 144) | def test_celu_module(test_case): class TestGelu (line 148) | class TestGelu(flow.unittest.TestCase): method test_gelu_module (line 149) | def test_gelu_module(test_case): class TestSigmoidModule (line 153) | class TestSigmoidModule(flow.unittest.TestCase): method test_sigmoid_module (line 154) | def test_sigmoid_module(test_case): class TestHardsigmoidModule (line 158) | class TestHardsigmoidModule(flow.unittest.TestCase): method test_hardsigmoid_module (line 159) | def test_hardsigmoid_module(test_case): class TestHardshrinkModule (line 163) | class TestHardshrinkModule(flow.unittest.TestCase): method test_hardshrink_module (line 164) | def test_hardshrink_module(test_case): class TestLogSigmoidModule (line 168) | class TestLogSigmoidModule(flow.unittest.TestCase): method test_logsigmoid_module (line 169) | def test_logsigmoid_module(test_case): class TestHardswishModule (line 173) | class TestHardswishModule(flow.unittest.TestCase): method test_hardswish_module (line 174) | def test_hardswish_module(test_case): class TestHardtanhModule (line 178) | class TestHardtanhModule(flow.unittest.TestCase): method test_hardtanh_module (line 179) | def test_hardtanh_module(test_case): class TestLeakyReLUModule (line 183) | class TestLeakyReLUModule(flow.unittest.TestCase): method test_leakyrelu_module (line 184) | def test_leakyrelu_module(test_case): class TestMishModule (line 188) | class TestMishModule(flow.unittest.TestCase): method test_mish_module (line 189) | def test_mish_module(test_case): class TestSiluModule (line 193) | class TestSiluModule(flow.unittest.TestCase): method test_silu_module (line 194) | def test_silu_module(test_case): class TestSeluModule (line 198) | class TestSeluModule(flow.unittest.TestCase): method test_selu_module (line 199) | def test_selu_module(test_case): class TestThresholdModule (line 203) | class TestThresholdModule(flow.unittest.TestCase): method test_threshold_module (line 204) | def test_threshold_module(test_case): class TestSoftplusModule (line 208) | class TestSoftplusModule(flow.unittest.TestCase): method test_softplus_module (line 209) | def test_softplus_module(test_case): class TestSoftshrinkModule (line 213) | class TestSoftshrinkModule(flow.unittest.TestCase): method test_softshrink_module (line 214) | def test_softshrink_module(test_case): FILE: python/oneflow/test/modules/test_global_adaptive_pool.py function _test_adaptive_avgpoolnd (line 35) | def _test_adaptive_avgpoolnd(test_case, ndim, pool_size, placement, sbp): function _test_adaptive_avgpoolnd_functional (line 52) | def _test_adaptive_avgpoolnd_functional(test_case, ndim, pool_size, plac... class TestAdaptiveAvgPool (line 63) | class TestAdaptiveAvgPool(flow.unittest.TestCase): method test_adaptive_avgpool (line 65) | def test_adaptive_avgpool(test_case): FILE: python/oneflow/test/modules/test_global_add.py function _test_add_with_alpha (line 25) | def _test_add_with_alpha(test_case, ndim, placement, sbp): function _test_add_with_0size (line 40) | def _test_add_with_0size(test_case, ndim, zerodim, placement, sbp): class TestAddModule (line 49) | class TestAddModule(flow.unittest.TestCase): method test_add_with_alpha (line 52) | def test_add_with_alpha(test_case): FILE: python/oneflow/test/modules/test_global_addcdiv.py function _test_addcdiv (line 24) | def _test_addcdiv(test_case, ndim, placement, sbp): class TestModule (line 34) | class TestModule(flow.unittest.TestCase): method test_addcdiv (line 36) | def test_addcdiv(test_case): FILE: python/oneflow/test/modules/test_global_addcmul.py function _test_addcmul (line 24) | def _test_addcmul(test_case, ndim, placement, sbp): class TestModule (line 35) | class TestModule(flow.unittest.TestCase): method test_addcmul (line 37) | def test_addcmul(test_case): FILE: python/oneflow/test/modules/test_global_addmm.py function _test_addmm_with_random_data (line 27) | def _test_addmm_with_random_data(test_case, placement, sbp): function _test_addmm_broadcast_with_random_data (line 43) | def _test_addmm_broadcast_with_random_data(test_case, placement, sbp): class TestAddmm (line 58) | class TestAddmm(flow.unittest.TestCase): method test_addmm (line 60) | def test_addmm(test_case): FILE: python/oneflow/test/modules/test_global_affine_grid.py function _test_affine_grid_2d_with_random_data (line 26) | def _test_affine_grid_2d_with_random_data(test_case, placement, sbp): function _test_affine_grid_3d_with_random_data (line 42) | def _test_affine_grid_3d_with_random_data(test_case, placement, sbp): class TestAffineGrid (line 58) | class TestAffineGrid(flow.unittest.TestCase): method test_affine_grid (line 60) | def test_affine_grid(test_case): FILE: python/oneflow/test/modules/test_global_argmax.py function _test_argmax_with_random_data (line 25) | def _test_argmax_with_random_data(test_case, ndim, placement, sbp): class TestArgmax (line 33) | class TestArgmax(flow.unittest.TestCase): method test_argmax (line 35) | def test_argmax(test_case): FILE: python/oneflow/test/modules/test_global_argmin.py function _test_argmin_with_random_data (line 25) | def _test_argmin_with_random_data(test_case, ndim, placement, sbp): class TestArgmin (line 33) | class TestArgmin(flow.unittest.TestCase): method test_argmin (line 35) | def test_argmin(test_case): FILE: python/oneflow/test/modules/test_global_argsort.py function _test_argsort_with_random_data (line 25) | def _test_argsort_with_random_data(test_case, ndim, placement, sbp): class TestArgsort (line 35) | class TestArgsort(flow.unittest.TestCase): method test_argsort (line 37) | def test_argsort(test_case): FILE: python/oneflow/test/modules/test_global_argwhere.py function _test_argwhere_with_random_data (line 26) | def _test_argwhere_with_random_data(test_case, ndim, placement, sbp): class TestArgwhere (line 37) | class TestArgwhere(flow.unittest.TestCase): method test_argwhere (line 39) | def test_argwhere(test_case): FILE: python/oneflow/test/modules/test_global_atleast.py function _test_atleast1d_with_random_data (line 26) | def _test_atleast1d_with_random_data(test_case, placement, sbp): function _test_atleast2d_with_random_data (line 34) | def _test_atleast2d_with_random_data(test_case, placement, sbp): function _test_atleast3d_with_random_data (line 43) | def _test_atleast3d_with_random_data(test_case, placement, sbp): class TestAtLeastModule (line 52) | class TestAtLeastModule(flow.unittest.TestCase): method test_atleast1d_with_random_data (line 54) | def test_atleast1d_with_random_data(test_case): method test_atleast2d_with_random_data (line 60) | def test_atleast2d_with_random_data(test_case): method test_atleast3d_with_random_data (line 66) | def test_atleast3d_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_avgpool.py function _test_avgpool1d_with_random_data (line 25) | def _test_avgpool1d_with_random_data(test_case, placement, sbp): function _test_avgpool2d_with_random_data (line 43) | def _test_avgpool2d_with_random_data(test_case, placement, sbp): function _test_avgpool3d_with_random_data (line 62) | def _test_avgpool3d_with_random_data(test_case, placement, sbp): function _test_functional_avgpool1d_with_random_data (line 81) | def _test_functional_avgpool1d_with_random_data(test_case, placement, sbp): function _test_functional_avgpool2d_with_random_data (line 97) | def _test_functional_avgpool2d_with_random_data(test_case, placement, sbp): function _test_functional_avgpool3d_with_random_data (line 113) | def _test_functional_avgpool3d_with_random_data(test_case, placement, sbp): class TestAvgPoolingModule (line 128) | class TestAvgPoolingModule(flow.unittest.TestCase): method test_avg_pooling (line 130) | def test_avg_pooling(test_case): FILE: python/oneflow/test/modules/test_global_batch_gather.py function _test_batch_gather (line 27) | def _test_batch_gather(test_case, ndim, placement, sbp): class TestBatchGather (line 58) | class TestBatchGather(flow.unittest.TestCase): method test_batch_gather (line 60) | def test_batch_gather(test_case): FILE: python/oneflow/test/modules/test_global_bincount.py function _test_bincount (line 23) | def _test_bincount(test_case, placement, sbp): class TestBinCountModule (line 30) | class TestBinCountModule(flow.unittest.TestCase): method test_bincount (line 32) | def test_bincount(test_case): FILE: python/oneflow/test/modules/test_global_bitwise.py function _test_bitwise_ops_with_random_data (line 26) | def _test_bitwise_ops_with_random_data(test_case, op, placement, sbp): function _test_bitwise_not_with_random_data (line 34) | def _test_bitwise_not_with_random_data(test_case, placement, sbp): class TestBitwiseModule (line 39) | class TestBitwiseModule(flow.unittest.TestCase): method test_bitwise_and_with_random_data (line 41) | def test_bitwise_and_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_broadcase_like.py function _test_broadcast_like (line 25) | def _test_broadcast_like(test_case, placement, sbp): function _test_broadcast_like_expand_dims (line 39) | def _test_broadcast_like_expand_dims(test_case, placement, sbp): class TestGlobalBroadcaseLike (line 53) | class TestGlobalBroadcaseLike(flow.unittest.TestCase): method test_broadcase_like (line 55) | def test_broadcase_like(test_case): FILE: python/oneflow/test/modules/test_global_broadcast_matmul.py function _test_flow_tensor_global_broadcast_matmul_with_random_data (line 23) | def _test_flow_tensor_global_broadcast_matmul_with_random_data( function _test_flow_tensor_global_x_broadcast_y_matmul (line 36) | def _test_flow_tensor_global_x_broadcast_y_matmul(test_case, placement, ... function _test_flow_tensor_global_broadcast_matmul_with_same_dims (line 48) | def _test_flow_tensor_global_broadcast_matmul_with_same_dims( class TestGlobalBroadcastMatmulModule (line 62) | class TestGlobalBroadcastMatmulModule(flow.unittest.TestCase): method test_global_broadcast_matmul_with_random_data (line 64) | def test_global_broadcast_matmul_with_random_data(test_case): method test_global_x_broadcast_y_matmul (line 73) | def test_global_x_broadcast_y_matmul(test_case): method test_global_broadcast_matmul_with_same_dims (line 82) | def test_global_broadcast_matmul_with_same_dims(test_case): FILE: python/oneflow/test/modules/test_global_broadcast_ops.py function _test_global_broadcast_tensors (line 23) | def _test_global_broadcast_tensors( class TestGlobalBroadcastOps (line 35) | class TestGlobalBroadcastOps(flow.unittest.TestCase): method test_global_tensors (line 40) | def test_global_tensors(test_case): FILE: python/oneflow/test/modules/test_global_cast.py class TestGlobalCastModule_1n4d (line 31) | class TestGlobalCastModule_1n4d(flow.unittest.TestCase): method test_to_global_flatten_hierarchy (line 32) | def test_to_global_flatten_hierarchy(test_case): method test_to_global_flatten_hierarchy_cpu_to_gpu (line 46) | def test_to_global_flatten_hierarchy_cpu_to_gpu(test_case): method test_to_global_flatten_hierarchy_gpu_to_cpu (line 60) | def test_to_global_flatten_hierarchy_gpu_to_cpu(test_case): method test_to_global_broadcast_shape_dtype (line 73) | def test_to_global_broadcast_shape_dtype(test_case): method test_local_to_global_2d_sbp (line 87) | def test_local_to_global_2d_sbp(test_case): method test_local_to_global_sp_2_bb (line 98) | def test_local_to_global_sp_2_bb(test_case): method test_local_to_global_ps0_2_s0s0 (line 118) | def test_local_to_global_ps0_2_s0s0(test_case): method test_local_to_global_s0p_2_s0s0 (line 140) | def test_local_to_global_s0p_2_s0s0(test_case): method test_to_global_loop_broadcast_shape_dtype (line 162) | def test_to_global_loop_broadcast_shape_dtype(test_case): class TestGlobalCastModule_1n2d (line 183) | class TestGlobalCastModule_1n2d(flow.unittest.TestCase): method test_to_global_broadcast_shape_dtype (line 184) | def test_to_global_broadcast_shape_dtype(test_case): method test_local_to_global_broadcast_data (line 197) | def test_local_to_global_broadcast_data(test_case): method test_cuda_global_to_global_cpu_s2b (line 212) | def test_cuda_global_to_global_cpu_s2b(test_case): method test_cuda_global_to_global_s2b (line 226) | def test_cuda_global_to_global_s2b(test_case): method test_cuda_global_to_global_cpu_s2p (line 239) | def test_cuda_global_to_global_cpu_s2p(test_case): method test_cuda_global_to_global_s2p (line 278) | def test_cuda_global_to_global_s2p(test_case): method test_cuda_global_to_global_b2p (line 317) | def test_cuda_global_to_global_b2p(test_case): method test_cuda_global_to_global_b2s (line 338) | def test_cuda_global_to_global_b2s(test_case): method test_cuda_global_to_global_cpu_p2s (line 351) | def test_cuda_global_to_global_cpu_p2s(test_case): method test_cuda_global_to_global_p2s (line 367) | def test_cuda_global_to_global_p2s(test_case): method test_cuda_global_to_global_cuda_h2d (line 383) | def test_cuda_global_to_global_cuda_h2d(test_case): method test_cuda_global_to_global_cpu_p2b (line 397) | def test_cuda_global_to_global_cpu_p2b(test_case): method test_cuda_global_to_global_p2b (line 418) | def test_cuda_global_to_global_p2b(test_case): class TestGlobalCastModule_1n1d (line 435) | class TestGlobalCastModule_1n1d(flow.unittest.TestCase): method test_to_global (line 436) | def test_to_global(test_case): function _test_cpu_p2b_with_random_parameter (line 446) | def _test_cpu_p2b_with_random_parameter(test_case, device_list): function _test_cpu_s2b_with_random_parameter (line 492) | def _test_cpu_s2b_with_random_parameter(test_case, device_list): function _test_cpu_p2s_with_random_parameter (line 538) | def _test_cpu_p2s_with_random_parameter(test_case, device_list): class TestGlobalCast (line 588) | class TestGlobalCast(flow.unittest.TestCase): method test_cpu_local_tensor_to_gpu_placement (line 589) | def test_cpu_local_tensor_to_gpu_placement(test_case): method test_cpu_p2b_with_random_parameter (line 607) | def test_cpu_p2b_with_random_parameter(test_case): method test_cpu_s2b_with_random_parameter (line 613) | def test_cpu_s2b_with_random_parameter(test_case): method test_cpu_p2s_with_random_parameter (line 619) | def test_cpu_p2s_with_random_parameter(test_case): method test_local_to_global_with_wrong_device (line 625) | def test_local_to_global_with_wrong_device(test_case): class TestGlobalCast_S2S (line 642) | class TestGlobalCast_S2S(flow.unittest.TestCase): method test_global_to_global_s0_to_s1 (line 643) | def test_global_to_global_s0_to_s1(test_case): method test_global_to_global_s1_to_s0 (line 698) | def test_global_to_global_s1_to_s0(test_case): method test_global_to_global_s0_to_s1_cpu (line 746) | def test_global_to_global_s0_to_s1_cpu(test_case): method test_global_to_global_s1_to_s0_cpu (line 773) | def test_global_to_global_s1_to_s0_cpu(test_case): class TestGlobalCast_XToB (line 807) | class TestGlobalCast_XToB(flow.unittest.TestCase): method test_global_to_global_btb_gpu_to_gpu (line 808) | def test_global_to_global_btb_gpu_to_gpu(test_case): method test_global_to_global_stb_gpu_to_gpu (line 847) | def test_global_to_global_stb_gpu_to_gpu(test_case): method test_global_to_global_ptb_gpu_to_gpu (line 898) | def test_global_to_global_ptb_gpu_to_gpu(test_case): class TestGlobalCast_1ToN (line 944) | class TestGlobalCast_1ToN(flow.unittest.TestCase): method test_global_to_global_1tob (line 945) | def test_global_to_global_1tob(test_case): method test_global_to_global_1top (line 974) | def test_global_to_global_1top(test_case): method test_global_to_global_1tos (line 1015) | def test_global_to_global_1tos(test_case): class TestGlobalCast_NTo1 (line 1065) | class TestGlobalCast_NTo1(flow.unittest.TestCase): method test_global_to_global_bt1 (line 1066) | def test_global_to_global_bt1(test_case): method test_global_to_global_st1 (line 1095) | def test_global_to_global_st1(test_case): method test_global_to_global_pt1 (line 1133) | def test_global_to_global_pt1(test_case): class TestGlobalCast_1To1 (line 1170) | class TestGlobalCast_1To1(flow.unittest.TestCase): method test_global_to_global_1to1_gpu_to_gpu (line 1171) | def test_global_to_global_1to1_gpu_to_gpu(test_case): method test_global_to_global_1to1_cpu_to_cpu (line 1200) | def test_global_to_global_1to1_cpu_to_cpu(test_case): method test_global_to_global_1to1_gpu_to_cpu (line 1229) | def test_global_to_global_1to1_gpu_to_cpu(test_case): method test_global_to_global_1to1_cpu_to_gpu (line 1258) | def test_global_to_global_1to1_cpu_to_gpu(test_case): class GraphTestModel (line 1288) | class GraphTestModel(nn.Graph): method __init__ (line 1289) | def __init__(self, model): method build (line 1293) | def build(self, x): class TestToGlobalAndLocal (line 1298) | class TestToGlobalAndLocal(flow.unittest.TestCase): method __all_global (line 1305) | def __all_global(test_case, input, placement, sbp): method __all_local (line 1328) | def __all_local(test_case, input): method _test_any_input (line 1337) | def _test_any_input(test_case): method _test_any_input_get_sbp_func (line 1380) | def _test_any_input_get_sbp_func(test_case): method _test_tensor_to_global (line 1424) | def _test_tensor_to_global(test_case): method _test_tensor_to_local (line 1457) | def _test_tensor_to_local(test_case): method __test_state_dict_to_global (line 1467) | def __test_state_dict_to_global(test_case, local_state_dict): method __test_state_dict_to_local (line 1492) | def __test_state_dict_to_local(test_case, global_state_dict): method _test_eagar_state_dict (line 1500) | def _test_eagar_state_dict(test_case): method _test_graph_state_dict (line 1507) | def _test_graph_state_dict(test_case): method test_to_global_local (line 1515) | def test_to_global_local(test_case): FILE: python/oneflow/test/modules/test_global_chunk.py function _test_chunk (line 25) | def _test_chunk(test_case, ndim, placement, sbp): class TestModule (line 35) | class TestModule(flow.unittest.TestCase): method test_chunk (line 37) | def test_chunk(test_case): FILE: python/oneflow/test/modules/test_global_clone.py function do_test_clone_impl (line 24) | def do_test_clone_impl(test_case, ndim, placement, sbp): class TestCloneConsistent (line 32) | class TestCloneConsistent(flow.unittest.TestCase): method test_clone (line 34) | def test_clone(test_case): FILE: python/oneflow/test/modules/test_global_coin_flip.py function _test_global_coin_flip (line 27) | def _test_global_coin_flip( function _test_graph_coin_flip (line 40) | def _test_graph_coin_flip( class TestCoinFlipGlobal (line 61) | class TestCoinFlipGlobal(flow.unittest.TestCase): method test_coin_flip_global (line 63) | def test_coin_flip_global(test_case): method test_coin_flip_graph (line 81) | def test_coin_flip_graph(test_case): FILE: python/oneflow/test/modules/test_global_concat.py function _test_cat_with_random_data (line 30) | def _test_cat_with_random_data(test_case, placement, sbp): function _test_concat_with_input_0_size_data (line 36) | def _test_concat_with_input_0_size_data(test_case, placement, sbp): function _test_concat_with_output_0_size_data (line 46) | def _test_concat_with_output_0_size_data(test_case, placement, sbp): function _test_cat_only_one_tensor (line 54) | def _test_cat_only_one_tensor(test_case, placement, sbp): class TestModule (line 61) | class TestModule(flow.unittest.TestCase): method test_cat_with_random_data (line 63) | def test_cat_with_random_data(test_case): method test_cat_only_one_tensor (line 69) | def test_cat_only_one_tensor(test_case): method test_concat_with_input_0_size_data (line 75) | def test_concat_with_input_0_size_data(test_case): method test_concat_with_output_0_size_data (line 81) | def test_concat_with_output_0_size_data(test_case): FILE: python/oneflow/test/modules/test_global_constant.py function _test_global_new_full (line 28) | def _test_global_new_full(test_case, shape, full_value, placement, sbp): function _test_global_graph_new_full (line 45) | def _test_global_graph_new_full(test_case, shape, full_value, placement,... function _test_global_constant (line 72) | def _test_global_constant(test_case, func, shape, placement, sbp): function _test_graph_constant (line 102) | def _test_graph_constant(test_case, func, shape, placement, sbp): class TestConstantGlobal (line 141) | class TestConstantGlobal(flow.unittest.TestCase): method test_constant_global (line 143) | def test_constant_global(test_case): method test_constant_graph (line 169) | def test_constant_graph(test_case): FILE: python/oneflow/test/modules/test_global_ctc_loss.py function log_softmax (line 28) | def log_softmax(logits, axis=0): function _compare_torch_and_oneflow (line 36) | def _compare_torch_and_oneflow( function _test_ctc_loss_impl (line 131) | def _test_ctc_loss_impl( class TestCTCLossGlobal (line 166) | class TestCTCLossGlobal(oneflow.unittest.TestCase): method test_ctc_loss_global (line 168) | def test_ctc_loss_global(test_case): FILE: python/oneflow/test/modules/test_global_cumprod.py function _test_cumprod_impl (line 25) | def _test_cumprod_impl(test_case, ndim, placement, sbp): class TestCumprodGlobal (line 34) | class TestCumprodGlobal(flow.unittest.TestCase): method test_cumprod (line 36) | def test_cumprod(test_case): FILE: python/oneflow/test/modules/test_global_cumsum.py function _test_cumsum_impl (line 25) | def _test_cumsum_impl(test_case, ndim, placement, sbp): class TestCumsumGlobal (line 35) | class TestCumsumGlobal(flow.unittest.TestCase): method test_cumsum (line 37) | def test_cumsum(test_case): FILE: python/oneflow/test/modules/test_global_deconv2d.py function _test_deconv2d_impl (line 25) | def _test_deconv2d_impl(test_case, placement, input_sbp): class TestDeconv2dGlobal (line 66) | class TestDeconv2dGlobal(flow.unittest.TestCase): method test_deconv2d (line 69) | def test_deconv2d(test_case): FILE: python/oneflow/test/modules/test_global_deform_conv2d.py function _test_deform_conv2d (line 27) | def _test_deform_conv2d(test_case, placement): class TestGlobalDeformConv2d (line 119) | class TestGlobalDeformConv2d(flow.unittest.TestCase): method test_deform_conv2d (line 121) | def test_deform_conv2d(test_case): FILE: python/oneflow/test/modules/test_global_det.py function det_all_placement (line 24) | def det_all_placement(): function _test_det (line 35) | def _test_det(test_case, placement, sbp, ndim): class TestDet (line 47) | class TestDet(flow.unittest.TestCase): method test_det (line 49) | def test_det(test_case): FILE: python/oneflow/test/modules/test_global_diag.py function do_test_diag_impl (line 26) | def do_test_diag_impl(test_case, ndim, placement, sbp): class TestDiagGlobal (line 33) | class TestDiagGlobal(flow.unittest.TestCase): method test_diag (line 35) | def test_diag(test_case): FILE: python/oneflow/test/modules/test_global_diagonal.py function _test_diagonal_impl (line 24) | def _test_diagonal_impl(test_case, placement, sbp): class TestDiagonalGlobal (line 42) | class TestDiagonalGlobal(flow.unittest.TestCase): method test_diagonal (line 44) | def test_diagonal(test_case): FILE: python/oneflow/test/modules/test_global_div.py function do_test_div_impl (line 25) | def do_test_div_impl(test_case, ndim, placement, sbp): class TestDivGlobal (line 36) | class TestDivGlobal(flow.unittest.TestCase): method test_div (line 38) | def test_div(test_case): FILE: python/oneflow/test/modules/test_global_dot.py function do_test_dot_impl (line 24) | def do_test_dot_impl(test_case, placement, sbp): class TestDotGlobal (line 32) | class TestDotGlobal(flow.unittest.TestCase): method test_dot (line 35) | def test_dot(test_case): FILE: python/oneflow/test/modules/test_global_dropout.py function _test_dropout_p01 (line 24) | def _test_dropout_p01(test_case, placement, sbp, ndim, p): function _test_dropout_eval_p01 (line 33) | def _test_dropout_eval_p01(test_case, placement, sbp, ndim, p): class TestDropoutGlobal (line 42) | class TestDropoutGlobal(flow.unittest.TestCase): method test_dropout_p01 (line 44) | def test_dropout_p01(test_case): method test_dropout_eval (line 53) | def test_dropout_eval(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase1.py function _test_einsum_alphaflod_usecase1 (line 26) | def _test_einsum_alphaflod_usecase1(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase1 (line 39) | def test_einsum_alphaflod_usecase1(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase10.py function _test_einsum_alphaflod_usecase10 (line 26) | def _test_einsum_alphaflod_usecase10(test_case, placement, sbp): class TestEinsumGlobal (line 39) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase10 (line 41) | def test_einsum_alphaflod_usecase10(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase11.py function _test_einsum_alphaflod_usecase11 (line 26) | def _test_einsum_alphaflod_usecase11(test_case, placement, sbp): class TestEinsumGlobal (line 36) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase11 (line 38) | def test_einsum_alphaflod_usecase11(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase2.py function _test_einsum_alphaflod_usecase2 (line 26) | def _test_einsum_alphaflod_usecase2(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase2 (line 39) | def test_einsum_alphaflod_usecase2(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase3.py function _test_einsum_alphaflod_usecase3 (line 26) | def _test_einsum_alphaflod_usecase3(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase3 (line 40) | def test_einsum_alphaflod_usecase3(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase4.py function _test_einsum_alphaflod_usecase4 (line 26) | def _test_einsum_alphaflod_usecase4(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase4 (line 39) | def test_einsum_alphaflod_usecase4(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase5.py function _test_einsum_alphaflod_usecase5 (line 26) | def _test_einsum_alphaflod_usecase5(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase5 (line 39) | def test_einsum_alphaflod_usecase5(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase6.py function _test_einsum_alphaflod_usecase6 (line 26) | def _test_einsum_alphaflod_usecase6(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase6 (line 39) | def test_einsum_alphaflod_usecase6(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase7.py function _test_einsum_alphaflod_usecase7 (line 26) | def _test_einsum_alphaflod_usecase7(test_case, placement, sbp): class TestEinsumGlobal (line 39) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase7 (line 41) | def test_einsum_alphaflod_usecase7(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase8.py function _test_einsum_alphaflod_usecase8 (line 26) | def _test_einsum_alphaflod_usecase8(test_case, placement, sbp): class TestEinsumGlobal (line 36) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase8 (line 38) | def test_einsum_alphaflod_usecase8(test_case): FILE: python/oneflow/test/modules/test_global_einsum_alphaflod_usecase9.py function _test_einsum_alphaflod_usecase9 (line 26) | def _test_einsum_alphaflod_usecase9(test_case, placement, sbp): class TestEinsumGlobal (line 36) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_alphaflod_usecase9 (line 38) | def test_einsum_alphaflod_usecase9(test_case): FILE: python/oneflow/test/modules/test_global_einsum_attention.py function _test_einsum_attention (line 26) | def _test_einsum_attention(test_case, placement, sbp): class TestEinsumGlobal (line 39) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_attention (line 41) | def test_einsum_attention(test_case): FILE: python/oneflow/test/modules/test_global_einsum_batch_matmul.py function _test_einsum_batch_matmul (line 26) | def _test_einsum_batch_matmul(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_batch_matmul (line 39) | def test_einsum_batch_matmul(test_case): FILE: python/oneflow/test/modules/test_global_einsum_batch_matmul2.py function _test_einsum_batch_matmul2 (line 26) | def _test_einsum_batch_matmul2(test_case, placement, sbp): class TestEinsumGlobal (line 38) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_batch_matmul2 (line 40) | def test_einsum_batch_matmul2(test_case): FILE: python/oneflow/test/modules/test_global_einsum_batch_matmul3.py function _test_einsum_batch_matmul3 (line 26) | def _test_einsum_batch_matmul3(test_case, placement, sbp): class TestEinsumGlobal (line 39) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_batch_matmul3 (line 41) | def test_einsum_batch_matmul3(test_case): FILE: python/oneflow/test/modules/test_global_einsum_batch_matmul4.py function _test_einsum_batch_matmul4 (line 26) | def _test_einsum_batch_matmul4(test_case, placement, sbp): class TestEinsumGlobal (line 39) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_batch_matmul4 (line 41) | def test_einsum_batch_matmul4(test_case): FILE: python/oneflow/test/modules/test_global_einsum_batch_matrix_vector_multiply.py function _test_einsum_batch_matrix_vector_multiply (line 26) | def _test_einsum_batch_matrix_vector_multiply(test_case, placement, sbp): class TestEinsumGlobal (line 38) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_batch_matrix_vector_multiply (line 41) | def test_einsum_batch_matrix_vector_multiply(test_case): FILE: python/oneflow/test/modules/test_global_einsum_batch_permute.py function _test_einsum_batch_permute (line 26) | def _test_einsum_batch_permute(test_case, placement, sbp): class TestEinsumGlobal (line 40) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_batch_permute (line 42) | def test_einsum_batch_permute(test_case): FILE: python/oneflow/test/modules/test_global_einsum_bilinear_transformation.py function _test_einsum_bilinear_transformation (line 26) | def _test_einsum_bilinear_transformation(test_case, placement, sbp): class TestEinsumGlobal (line 40) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_bilinear_transformation (line 42) | def test_einsum_bilinear_transformation(test_case): FILE: python/oneflow/test/modules/test_global_einsum_eltwise_mul_sum_row.py function _test_einsum_eltwise_mul_sum_row (line 26) | def _test_einsum_eltwise_mul_sum_row(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_eltwise_mul_sum_row (line 39) | def test_einsum_eltwise_mul_sum_row(test_case): FILE: python/oneflow/test/modules/test_global_einsum_eltwise_mul_then_reduce_sum.py function _test_einsum_eltwise_mul_then_reduce_sum (line 26) | def _test_einsum_eltwise_mul_then_reduce_sum(test_case, placement, sbp): class TestEinsumGlobal (line 38) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_eltwise_mul_then_reduce_sum (line 40) | def test_einsum_eltwise_mul_then_reduce_sum(test_case): FILE: python/oneflow/test/modules/test_global_einsum_eltwise_multiply.py function _test_einsum_eltwise_multiply (line 26) | def _test_einsum_eltwise_multiply(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_eltwise_multiply (line 39) | def test_einsum_eltwise_multiply(test_case): FILE: python/oneflow/test/modules/test_global_einsum_get_diagonal.py function _test_einsum_get_diagonal (line 26) | def _test_einsum_get_diagonal(test_case, placement, sbp): class TestEinsumGlobal (line 34) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_get_diagonal (line 36) | def test_einsum_get_diagonal(test_case): FILE: python/oneflow/test/modules/test_global_einsum_matmul.py function _test_einsum_matmul (line 26) | def _test_einsum_matmul(test_case, placement, sbp): class TestEinsumGlobal (line 39) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_matmul (line 42) | def test_einsum_matmul(test_case): FILE: python/oneflow/test/modules/test_global_einsum_matmul2.py function _test_einsum_matmul2 (line 26) | def _test_einsum_matmul2(test_case, placement, sbp): class TestEinsumGlobal (line 36) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_matmul2 (line 39) | def test_einsum_matmul2(test_case): FILE: python/oneflow/test/modules/test_global_einsum_matrix_column_sum.py function _test_einsum_matrix_column_sum (line 26) | def _test_einsum_matrix_column_sum(test_case, placement, sbp): class TestEinsumGlobal (line 33) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_matrix_column_sum (line 36) | def test_einsum_matrix_column_sum(test_case): FILE: python/oneflow/test/modules/test_global_einsum_matrix_transpose.py function _test_einsum_matrix_transpose (line 26) | def _test_einsum_matrix_transpose(test_case, placement, sbp): class TestEinsumGlobal (line 33) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_matrix_transpose (line 35) | def test_einsum_matrix_transpose(test_case): FILE: python/oneflow/test/modules/test_global_einsum_matrix_vector_multiply.py function _test_einsum_matrix_vector_multiply (line 26) | def _test_einsum_matrix_vector_multiply(test_case, placement, sbp): class TestEinsumGlobal (line 38) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_matrix_vector_multiply (line 40) | def test_einsum_matrix_vector_multiply(test_case): FILE: python/oneflow/test/modules/test_global_einsum_reduce_sum.py function _test_einsum_reduce_sum (line 26) | def _test_einsum_reduce_sum(test_case, placement, sbp): class TestEinsumGlobal (line 33) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_reduce_sum (line 35) | def test_einsum_reduce_sum(test_case): FILE: python/oneflow/test/modules/test_global_einsum_tensor_contraction.py function _test_einsum_tensor_contraction (line 27) | def _test_einsum_tensor_contraction(test_case, placement, sbp): class TestEinsumGlobal (line 47) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_tensor_contraction (line 49) | def test_einsum_tensor_contraction(test_case): FILE: python/oneflow/test/modules/test_global_einsum_tensor_contraction2.py function _test_einsum_tensor_contraction2 (line 26) | def _test_einsum_tensor_contraction2(test_case, placement, sbp): class TestEinsumGlobal (line 42) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_tensor_contraction2 (line 45) | def test_einsum_tensor_contraction2(test_case): FILE: python/oneflow/test/modules/test_global_einsum_vector_inner_product.py function _test_einsum_vector_inner_product (line 26) | def _test_einsum_vector_inner_product(test_case, placement, sbp): class TestEinsumGlobal (line 37) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_vector_inner_product (line 39) | def test_einsum_vector_inner_product(test_case): FILE: python/oneflow/test/modules/test_global_einsum_vector_outer_product.py function _test_einsum_vector_outer_product (line 26) | def _test_einsum_vector_outer_product(test_case, placement, sbp): class TestEinsumGlobal (line 36) | class TestEinsumGlobal(flow.unittest.TestCase): method test_einsum_vector_outer_product (line 38) | def test_einsum_vector_outer_product(test_case): FILE: python/oneflow/test/modules/test_global_empty.py function _test_global_empty (line 27) | def _test_global_empty(test_case, func, shape, placement, sbp): function _test_graph_empty (line 54) | def _test_graph_empty(test_case, func, shape, placement, sbp): class TestEmptyGlobal (line 90) | class TestEmptyGlobal(flow.unittest.TestCase): method test_empty_global (line 92) | def test_empty_global(test_case): method test_empty_graph (line 109) | def test_empty_graph(test_case): FILE: python/oneflow/test/modules/test_global_eq.py function do_test_eq_impl (line 25) | def do_test_eq_impl(test_case, ndim, placement, sbp): class TestEqGlobal (line 36) | class TestEqGlobal(flow.unittest.TestCase): method test_eq (line 38) | def test_eq(test_case): FILE: python/oneflow/test/modules/test_global_erf.py function do_test_erf_impl (line 25) | def do_test_erf_impl(test_case, ndim, placement, sbp): class TestErfGlobal (line 33) | class TestErfGlobal(flow.unittest.TestCase): method test_erf (line 35) | def test_erf(test_case): FILE: python/oneflow/test/modules/test_global_erfc.py function do_test_erfc_impl (line 25) | def do_test_erfc_impl(test_case, ndim, placement, sbp): class TestErfcGlobal (line 33) | class TestErfcGlobal(flow.unittest.TestCase): method test_erfc (line 35) | def test_erfc(test_case): FILE: python/oneflow/test/modules/test_global_expand_op.py function _test_global_expand (line 27) | def _test_global_expand( class ExpandGlobalTestCase (line 81) | class ExpandGlobalTestCase(oneflow.unittest.TestCase): method test_global_expand (line 82) | def test_global_expand(test_case): method test_split_expand (line 99) | def test_split_expand(test_case): method test_broadcast_scalar_expand (line 113) | def test_broadcast_scalar_expand(test_case): FILE: python/oneflow/test/modules/test_global_expm1.py function do_test_expm1_impl (line 25) | def do_test_expm1_impl(test_case, ndim, placement, sbp): class TestExpm1Global (line 33) | class TestExpm1Global(flow.unittest.TestCase): method test_expm1 (line 35) | def test_expm1(test_case): FILE: python/oneflow/test/modules/test_global_eye.py function do_test_eye_impl (line 24) | def do_test_eye_impl(test_case, placement, sbp): class TestEyeGlobal (line 37) | class TestEyeGlobal(flow.unittest.TestCase): method test_eye (line 39) | def test_eye(test_case): FILE: python/oneflow/test/modules/test_global_fill.py function _test_fill_ (line 23) | def _test_fill_(test_case, ndim, placement, sbp): function _test_fill_tensor_ (line 33) | def _test_fill_tensor_(test_case, ndim, placement, sbp): class TestFillModule (line 51) | class TestFillModule(flow.unittest.TestCase): method test_fill_ (line 53) | def test_fill_(test_case): FILE: python/oneflow/test/modules/test_global_flatten.py function do_test_flatten_impl (line 25) | def do_test_flatten_impl(test_case, ndim, placement, sbp): class TestFlattenGlobal (line 36) | class TestFlattenGlobal(flow.unittest.TestCase): method test_flatten (line 38) | def test_flatten(test_case): FILE: python/oneflow/test/modules/test_global_flip.py function _test_flip_impl (line 29) | def _test_flip_impl(test_case, ndim, placement, sbp): class TestFlipGlobal (line 38) | class TestFlipGlobal(flow.unittest.TestCase): method test_flip (line 40) | def test_flip(test_case): FILE: python/oneflow/test/modules/test_global_floor.py function do_test_floor_impl (line 25) | def do_test_floor_impl(test_case, ndim, placement, sbp): class TestFloorGlobal (line 33) | class TestFloorGlobal(flow.unittest.TestCase): method test_floor (line 35) | def test_floor(test_case): FILE: python/oneflow/test/modules/test_global_fmod.py function do_test_fmod_impl (line 31) | def do_test_fmod_impl(test_case, ndim, placement, sbp): class TestFmodGlobal (line 42) | class TestFmodGlobal(flow.unittest.TestCase): method test_fmod (line 44) | def test_fmod(test_case): FILE: python/oneflow/test/modules/test_global_fold.py function _test_fold_impl (line 25) | def _test_fold_impl(test_case, placement, sbp): class TestFold (line 50) | class TestFold(flow.unittest.TestCase): method test_fold (line 52) | def test_fold(test_case): FILE: python/oneflow/test/modules/test_global_frac.py function _test_frac (line 24) | def _test_frac(test_case, ndim, placement, sbp): class TestModule (line 31) | class TestModule(flow.unittest.TestCase): method test_frac (line 33) | def test_frac(test_case): FILE: python/oneflow/test/modules/test_global_full.py function _test_global_full (line 27) | def _test_global_full(test_case, shape, placement, sbp): function _test_global_full_tensor_scalar (line 35) | def _test_global_full_tensor_scalar(test_case, shape, placement, sbp): function _test_graph_full (line 44) | def _test_graph_full(test_case, shape, placement, sbp): function _test_graph_full_tensor_scalar (line 61) | def _test_graph_full_tensor_scalar(test_case, shape, placement, sbp): class TestFullGlobal (line 87) | class TestFullGlobal(flow.unittest.TestCase): method test_full_global (line 89) | def test_full_global(test_case): method test_full_graph (line 101) | def test_full_graph(test_case): FILE: python/oneflow/test/modules/test_global_full_like.py function _test_global_full_like (line 27) | def _test_global_full_like(test_case, shape, placement, sbp): function _test_graph_full_like (line 36) | def _test_graph_full_like(test_case, shape, placement, sbp): class TestFillLikeGlobal (line 54) | class TestFillLikeGlobal(flow.unittest.TestCase): method test_full_like_global (line 56) | def test_full_like_global(test_case): method test_full_like_graph (line 67) | def test_full_like_graph(test_case): FILE: python/oneflow/test/modules/test_global_greater.py function _test_greater_impl (line 30) | def _test_greater_impl(test_case, ndim, placement, sbp): class TestGreaterGlobal (line 42) | class TestGreaterGlobal(flow.unittest.TestCase): method test_greater (line 44) | def test_greater(test_case): FILE: python/oneflow/test/modules/test_global_greater_equal.py function do_test_greater_equal_impl (line 25) | def do_test_greater_equal_impl(test_case, ndim, placement, sbp): class TestGreaterEqualGlobal (line 36) | class TestGreaterEqualGlobal(flow.unittest.TestCase): method test_greater_equal (line 38) | def test_greater_equal(test_case): FILE: python/oneflow/test/modules/test_global_grid_sample.py function _test_flow_grid_sample_cudnn (line 27) | def _test_flow_grid_sample_cudnn(test_case, placement, sbp): function _test_flow_grid_sample_4d (line 63) | def _test_flow_grid_sample_4d(test_case, placement, sbp): function _test_flow_grid_sample_5d (line 89) | def _test_flow_grid_sample_5d(test_case, placement, sbp): class TestGridSample (line 116) | class TestGridSample(flow.unittest.TestCase): method test_grid_sample (line 119) | def test_grid_sample(test_case): FILE: python/oneflow/test/modules/test_global_groupnorm.py function _test_global_group_norm (line 30) | def _test_global_group_norm(test_case, placement, input_sbp, affine): class TestGroupNormModule (line 55) | class TestGroupNormModule(flow.unittest.TestCase): method test_global_group_norm_with_random_data (line 57) | def test_global_group_norm_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_gru_cell.py function _test_gru_cell (line 27) | def _test_gru_cell(test_case, placement, sbp): class TestRNNCellGlobal (line 66) | class TestRNNCellGlobal(flow.unittest.TestCase): method test_gru_cell (line 68) | def test_gru_cell(test_case): FILE: python/oneflow/test/modules/test_global_hann_window.py function _test_global_hann_window (line 26) | def _test_global_hann_window(test_case, placement, sbp): function _test_graph_hann_window (line 33) | def _test_graph_hann_window(test_case, placement, sbp): class TestHannWindowGlobal (line 49) | class TestHannWindowGlobal(flow.unittest.TestCase): method test_hann_window_global (line 54) | def test_hann_window_global(test_case): method test_hann_window_graph (line 63) | def test_hann_window_graph(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_activation.py function _assert_true (line 28) | def _assert_true(test_case, value1, value2): function _test_activation_grad_grad_impl (line 39) | def _test_activation_grad_grad_impl(test_case, op_name, placement, *args... function _test_prelu_activation_grad_grad_impl (line 62) | def _test_prelu_activation_grad_grad_impl( function _test_hardswish_activation_grad_grad_impl (line 99) | def _test_hardswish_activation_grad_grad_impl( function _test_hardsigmoid_activation_grad_grad_impl (line 142) | def _test_hardsigmoid_activation_grad_grad_impl( class TestActivationHigherDerivative (line 181) | class TestActivationHigherDerivative(flow.unittest.TestCase): method test_activation_grad_grad (line 183) | def test_activation_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_conv.py function _test_convnd_grad_grad_impl (line 27) | def _test_convnd_grad_grad_impl(test_case, ndim, placement): class TestGlobalConvHigherDerivative (line 152) | class TestGlobalConvHigherDerivative(flow.unittest.TestCase): method test_conv1d_grad_grad (line 154) | def test_conv1d_grad_grad(test_case): method test_conv2d_grad_grad (line 160) | def test_conv2d_grad_grad(test_case): method test_conv3d_grad_grad (line 166) | def test_conv3d_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_div.py function _test_global_div_grad_grad_impl (line 24) | def _test_global_div_grad_grad_impl(test_case, placement): class TestGlobalDivHigherDerivative (line 93) | class TestGlobalDivHigherDerivative(flow.unittest.TestCase): method test_global_div_grad_grad (line 96) | def test_global_div_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_loss.py function _assert_true (line 24) | def _assert_true(test_case, value1, value2, name=""): function generate_grads_for_variables (line 31) | def generate_grads_for_variables(variables): function calculate_and_compare_loss (line 49) | def calculate_and_compare_loss(test_case, input, target, model, order=2): function generate_necessity_for_default_loss (line 75) | def generate_necessity_for_default_loss(placement): function generate_necessity_for_nll_loss (line 90) | def generate_necessity_for_nll_loss(placement): function generate_necessity_for_bce_loss (line 115) | def generate_necessity_for_bce_loss(placement): function _test_smooth_l1_loss_grad_grad_impl (line 156) | def _test_smooth_l1_loss_grad_grad_impl(test_case, placement): function _test_kl_div_loss_grad_grad_impl (line 166) | def _test_kl_div_loss_grad_grad_impl(test_case, placement): function _test_bce_loss_grad_grad_impl (line 177) | def _test_bce_loss_grad_grad_impl(test_case, placement, with_logits=False): function _test_nll_loss_grad_grad_impl (line 200) | def _test_nll_loss_grad_grad_impl(test_case, placement): class TestGlobalLossHigherDerivative (line 212) | class TestGlobalLossHigherDerivative(flow.unittest.TestCase): method test_smooth_l1_loss_grad_grad (line 214) | def test_smooth_l1_loss_grad_grad(test_case): method test_kl_div_loss_grad_grad (line 219) | def test_kl_div_loss_grad_grad(test_case): method test_nll_loss_grad_grad (line 224) | def test_nll_loss_grad_grad(test_case): method test_bce_loss_grad_grad (line 229) | def test_bce_loss_grad_grad(test_case): method test_bce_with_logits_loss_grad_grad (line 234) | def test_bce_with_logits_loss_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_matmul.py function _test_broadcast_matmul_grad_b_grad_impl (line 27) | def _test_broadcast_matmul_grad_b_grad_impl(test_case, placement): class TestGlobalMatmulHigherDerivative (line 155) | class TestGlobalMatmulHigherDerivative(flow.unittest.TestCase): method test_broadcast_matmul_grad_b_grad (line 158) | def test_broadcast_matmul_grad_b_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_neg.py function _global_neg_grad_grad_impl (line 24) | def _global_neg_grad_grad_impl(test_case, placement, sbp): class TestGlobalNegHigherDerivative (line 41) | class TestGlobalNegHigherDerivative(flow.unittest.TestCase): method test_global_neg_grad_grad (line 43) | def test_global_neg_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_pool.py function _check_equal (line 24) | def _check_equal(test_case, lhs, rhs, name="", rtol=1e-5, atol=1e-5): function _test_avg_pool_grad_grad_impl (line 35) | def _test_avg_pool_grad_grad_impl(test_case, placement, ndim): function _test_max_pool_grad_grad_impl (line 62) | def _test_max_pool_grad_grad_impl(test_case, placement, ndim): function _test_adaptive_pool_grad_grad_impl (line 90) | def _test_adaptive_pool_grad_grad_impl(test_case, placement, ndim, mode): class TestGlobalPoolHigherDerivative (line 121) | class TestGlobalPoolHigherDerivative(flow.unittest.TestCase): method test_max_pool_1d_grad_grad (line 123) | def test_max_pool_1d_grad_grad(test_case): method test_max_pool_2d_grad_grad (line 128) | def test_max_pool_2d_grad_grad(test_case): method test_max_pool_3d_grad_grad (line 133) | def test_max_pool_3d_grad_grad(test_case): method test_avg_pool_1d_grad_grad (line 138) | def test_avg_pool_1d_grad_grad(test_case): method test_avg_pool_2d_grad_grad (line 143) | def test_avg_pool_2d_grad_grad(test_case): method test_avg_pool_3d_grad_grad (line 148) | def test_avg_pool_3d_grad_grad(test_case): method test_adaptive_avg_pool_1d_grad_grad (line 153) | def test_adaptive_avg_pool_1d_grad_grad(test_case): method test_adaptive_avg_pool_2d_grad_grad (line 158) | def test_adaptive_avg_pool_2d_grad_grad(test_case): method test_adaptive_avg_pool_3d_grad_grad (line 163) | def test_adaptive_avg_pool_3d_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_pow.py function _check_equal (line 24) | def _check_equal(test_case, lhs, rhs, rtol=1e-3, atol=1e-3): function _test_global_pow_grad_grad_impl (line 35) | def _test_global_pow_grad_grad_impl(test_case, placement): class TestGlobalPowHigherDerivative (line 72) | class TestGlobalPowHigherDerivative(flow.unittest.TestCase): method test_global_pow_grad_grad (line 74) | def test_global_pow_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_scalar_pow.py function _check_equal (line 24) | def _check_equal(test_case, lhs, rhs, rtol=1e-4, atol=1e-4, name=""): function _test_scalar_pow_grad_grad_impl (line 35) | def _test_scalar_pow_grad_grad_impl(test_case, placement, reverse=False): class TestGlobalScalarPowHigherDerivative (line 59) | class TestGlobalScalarPowHigherDerivative(flow.unittest.TestCase): method test_global_scalar_pow_grad_grad (line 61) | def test_global_scalar_pow_grad_grad(test_case): method test_global_scalar_reverse_pow_grad_grad (line 67) | def test_global_scalar_reverse_pow_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_slice.py function _global_slice_grad_grad_impl (line 24) | def _global_slice_grad_grad_impl(test_case, placement, sbp): class TestGlobalSliceHigherDerivative (line 60) | class TestGlobalSliceHigherDerivative(flow.unittest.TestCase): method test_global_slice_grad_grad (line 62) | def test_global_slice_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_higher_derivative_softmax.py function _assert_true (line 24) | def _assert_true(test_case, value1, value2): function _test_global_softmax_grad_grad_impl (line 35) | def _test_global_softmax_grad_grad_impl(test_case, op_name, placement, s... class TestGlobalSoftmaxHigherDerivative (line 67) | class TestGlobalSoftmaxHigherDerivative(flow.unittest.TestCase): method test_global_softmax_grad_grad (line 69) | def test_global_softmax_grad_grad(test_case): method test_global_logsoftmax_grad_grad (line 77) | def test_global_logsoftmax_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_inv.py function _test_inv (line 24) | def _test_inv(test_case, placement, sbp, ndim): class TestInv (line 36) | class TestInv(flow.unittest.TestCase): method test_inv (line 38) | def test_inv(test_case): FILE: python/oneflow/test/modules/test_global_lerp.py function _test_lerp (line 24) | def _test_lerp(test_case, placement, sbp, ndim): class TestLerp (line 34) | class TestLerp(flow.unittest.TestCase): method test_lerp (line 36) | def test_lerp(test_case): FILE: python/oneflow/test/modules/test_global_linalg_cross.py function _test_linalg_cross (line 24) | def _test_linalg_cross(test_case, index_size_equal_3, ndim, placement, s... class TestLinalgCrossGlobal (line 36) | class TestLinalgCrossGlobal(flow.unittest.TestCase): method test_linalg_cross (line 38) | def test_linalg_cross(test_case): FILE: python/oneflow/test/modules/test_global_linear.py function _test_linear_with_random_data (line 30) | def _test_linear_with_random_data(test_case, placement, input_sbp): class TestLinearModule (line 49) | class TestLinearModule(flow.unittest.TestCase): method test_linear_with_random_data (line 51) | def test_linear_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_linspace.py function _test_global_linspace (line 26) | def _test_global_linspace(test_case, placement, sbp): function _test_graph_linspace (line 33) | def _test_graph_linspace(test_case, start, end, steps, placement, sbp): class TestLinspaceGlobal (line 49) | class TestLinspaceGlobal(flow.unittest.TestCase): method test_linspace_global (line 51) | def test_linspace_global(test_case): method test_linspace_graph (line 58) | def test_linspace_graph(test_case): FILE: python/oneflow/test/modules/test_global_logspace.py function _test_global_logspace (line 27) | def _test_global_logspace(test_case, placement, sbp): function _test_graph_logspace (line 34) | def _test_graph_logspace(test_case, start, end, steps, placement, sbp): class TestLogspaceGlobal (line 50) | class TestLogspaceGlobal(flow.unittest.TestCase): method test_logspace_global (line 55) | def test_logspace_global(test_case): method test_logspace_graph (line 64) | def test_logspace_graph(test_case): FILE: python/oneflow/test/modules/test_global_lstm_cell.py function _test_lstm_cell (line 27) | def _test_lstm_cell(test_case, placement, sbp): class TestRNNCellGlobal (line 73) | class TestRNNCellGlobal(flow.unittest.TestCase): method test_lstm_cell (line 75) | def test_lstm_cell(test_case): FILE: python/oneflow/test/modules/test_global_masked_fill.py function _test_masked_fill (line 25) | def _test_masked_fill(test_case, placement, sbp): function _test_masked_fill_with_0dim_data (line 35) | def _test_masked_fill_with_0dim_data(test_case, placement, sbp): function _test_masked_fill_with_broadcast_way (line 43) | def _test_masked_fill_with_broadcast_way(test_case, placement, sbp): class TestMaskedFill (line 56) | class TestMaskedFill(flow.unittest.TestCase): method test_masked_fill (line 58) | def test_masked_fill(test_case): FILE: python/oneflow/test/modules/test_global_masked_select.py function _test_masked_select (line 27) | def _test_masked_select(test_case, placement, sbp): function _test_masked_select_broadcast (line 39) | def _test_masked_select_broadcast(test_case, placement, input_sbp, mask_... class TestMaskedSelect (line 51) | class TestMaskedSelect(flow.unittest.TestCase): method test_masked_select (line 53) | def test_masked_select(test_case): method test_masked_select_broadcast (line 59) | def test_masked_select_broadcast(test_case): FILE: python/oneflow/test/modules/test_global_math_op_higher_derivative.py function _global_math_op_grad_grad_impl (line 24) | def _global_math_op_grad_grad_impl(test_case, op_name, placement, sbp): class TestGlobalMathOpHigherDerivative (line 68) | class TestGlobalMathOpHigherDerivative(flow.unittest.TestCase): method test_global_sin_grad_grad (line 70) | def test_global_sin_grad_grad(test_case): method test_global_cos_grad_grad (line 76) | def test_global_cos_grad_grad(test_case): method test_global_tan_grad_grad (line 82) | def test_global_tan_grad_grad(test_case): method test_global_sinh_grad_grad (line 88) | def test_global_sinh_grad_grad(test_case): method test_global_cosh_grad_grad (line 94) | def test_global_cosh_grad_grad(test_case): method test_global_tanh_grad_grad (line 100) | def test_global_tanh_grad_grad(test_case): method test_global_asin_grad_grad (line 106) | def test_global_asin_grad_grad(test_case): method test_global_acos_grad_grad (line 112) | def test_global_acos_grad_grad(test_case): method test_global_atan_grad_grad (line 118) | def test_global_atan_grad_grad(test_case): method test_global_asinh_grad_grad (line 124) | def test_global_asinh_grad_grad(test_case): method test_global_acosh_grad_grad (line 130) | def test_global_acosh_grad_grad(test_case): method test_global_atanh_grad_grad (line 136) | def test_global_atanh_grad_grad(test_case): method test_global_erf_grad_grad (line 142) | def test_global_erf_grad_grad(test_case): method test_global_erfc_grad_grad (line 148) | def test_global_erfc_grad_grad(test_case): method test_global_exp_grad_grad (line 154) | def test_global_exp_grad_grad(test_case): method test_global_exp2_grad_grad (line 160) | def test_global_exp2_grad_grad(test_case): method test_global_expm1_grad_grad (line 166) | def test_global_expm1_grad_grad(test_case): method test_global_log_grad_grad (line 172) | def test_global_log_grad_grad(test_case): method test_global_logsigmoid_grad_grad (line 178) | def test_global_logsigmoid_grad_grad(test_case): method test_global_log2_grad_grad (line 186) | def test_global_log2_grad_grad(test_case): method test_global_log1p_grad_grad (line 192) | def test_global_log1p_grad_grad(test_case): method test_global_reciprocal_grad_grad (line 198) | def test_global_reciprocal_grad_grad(test_case): method test_global_rsqrt_grad_grad (line 204) | def test_global_rsqrt_grad_grad(test_case): method test_global_sqrt_grad_grad (line 210) | def test_global_sqrt_grad_grad(test_case): method test_global_square_grad_grad (line 216) | def test_global_square_grad_grad(test_case): method test_global_sigmoid_grad_grad (line 222) | def test_global_sigmoid_grad_grad(test_case): method test_global_abs_grad_grad (line 228) | def test_global_abs_grad_grad(test_case): FILE: python/oneflow/test/modules/test_global_math_ops.py function _test_sinh (line 23) | def _test_sinh(test_case, placement, sbp, ndim): function _test_sin (line 31) | def _test_sin(test_case, placement, sbp, ndim): function _test_inplace_sin (line 39) | def _test_inplace_sin(test_case, placement, sbp, ndim): function _test_cos (line 48) | def _test_cos(test_case, placement, sbp, ndim): function _test_log (line 56) | def _test_log(test_case, placement, sbp, ndim): function _test_sqrt (line 64) | def _test_sqrt(test_case, placement, sbp, ndim): function _test_exp (line 72) | def _test_exp(test_case, placement, sbp, ndim): function _test_exp2 (line 80) | def _test_exp2(test_case, placement, sbp, ndim): function _test_rsqrt (line 88) | def _test_rsqrt(test_case, placement, sbp, ndim): function _test_square (line 96) | def _test_square(test_case, placement, sbp, ndim): function _test_pow_with_scalar (line 104) | def _test_pow_with_scalar(test_case, placement, sbp, ndim): function _test_floordiv_with_scalar (line 113) | def _test_floordiv_with_scalar(test_case, placement, sbp, ndim): function _test_arccos (line 124) | def _test_arccos(test_case, placement, sbp, ndim): function _test_acos (line 132) | def _test_acos(test_case, placement, sbp, ndim): function _test_arccosh (line 140) | def _test_arccosh(test_case, placement, sbp, ndim): function _test_acosh (line 148) | def _test_acosh(test_case, placement, sbp, ndim): function _test_floordiv (line 156) | def _test_floordiv(test_case, placement, sbp, ndim): function _test_atan2 (line 167) | def _test_atan2(test_case, placement, sbp, ndim): function _test_digamma (line 176) | def _test_digamma(test_case, placement, sbp, ndim): class TestMathOps (line 183) | class TestMathOps(flow.unittest.TestCase): method test_math_ops (line 185) | def test_math_ops(test_case): FILE: python/oneflow/test/modules/test_global_matmul.py function _test_matmul (line 25) | def _test_matmul(test_case, placement, x_sbp, y_sbp): function _test_tensor_broadcast_matmul (line 32) | def _test_tensor_broadcast_matmul(test_case, placement, x_sbp, y_sbp): class TestMatMulModule (line 40) | class TestMatMulModule(flow.unittest.TestCase): method test_matmul (line 42) | def test_matmul(test_case): method test_broadcast_matmul (line 49) | def test_broadcast_matmul(test_case): FILE: python/oneflow/test/modules/test_global_max.py function _np_max (line 26) | def _np_max(shape, dim, keepdims): function _test_max (line 41) | def _test_max( class TestMaxModule (line 65) | class TestMaxModule(flow.unittest.TestCase): method test_eager_global_max (line 68) | def test_eager_global_max(test_case): FILE: python/oneflow/test/modules/test_global_maximum_minimum.py function _test_broadcast_maximum (line 32) | def _test_broadcast_maximum(test_case, placement, x_sbp, y_sbp): function _test_broadcast_minimum (line 50) | def _test_broadcast_minimum(test_case, placement, x_sbp, y_sbp): function _test_maximum_with_same_input (line 68) | def _test_maximum_with_same_input(test_case, placement, sbp): function _test_minimum_with_same_input (line 83) | def _test_minimum_with_same_input(test_case, placement, sbp): class TestMaximumMinimumOps (line 91) | class TestMaximumMinimumOps(flow.unittest.TestCase): method test_maximum_minimum_with_same_input (line 93) | def test_maximum_minimum_with_same_input(test_case): method test_broadcast_maximum_minimum (line 100) | def test_broadcast_maximum_minimum(test_case): FILE: python/oneflow/test/modules/test_global_maxpool.py function _test_maxpool1d_functional (line 29) | def _test_maxpool1d_functional(test_case, placement, sbp): function _test_maxpool2d_functional (line 52) | def _test_maxpool2d_functional(test_case, placement, sbp): function _test_maxpool3d_functional (line 76) | def _test_maxpool3d_functional(test_case, placement, sbp): function _test_maxpool1d (line 105) | def _test_maxpool1d(test_case, placement, sbp): function _test_maxpool2d (line 129) | def _test_maxpool2d(test_case, placement, sbp): function _test_maxpool3d (line 153) | def _test_maxpool3d(test_case, placement, sbp): function _test_maxpool2d_channel_last (line 182) | def _test_maxpool2d_channel_last( class TestMaxPool (line 219) | class TestMaxPool(flow.unittest.TestCase): method test_maxpool (line 221) | def test_maxpool(test_case): method test_maxpool2d_channel_last (line 239) | def test_maxpool2d_channel_last(test_case): FILE: python/oneflow/test/modules/test_global_maxunpool.py function _get_valid_output_size (line 28) | def _get_valid_output_size( function _test_module_unpoolnd (line 57) | def _test_module_unpoolnd(test_case, placement, sbp, n): function _test_functional_unpoolnd (line 116) | def _test_functional_unpoolnd(test_case, placement, sbp, n): class TestMaxPool (line 179) | class TestMaxPool(flow.unittest.TestCase): method test_maxpool (line 181) | def test_maxpool(test_case): FILE: python/oneflow/test/modules/test_global_mean.py function _test_mean (line 24) | def _test_mean(test_case, placement, sbp, ndim): class TestMean (line 31) | class TestMean(flow.unittest.TestCase): method test_mean (line 33) | def test_mean(test_case): FILE: python/oneflow/test/modules/test_global_median.py function _test_median (line 26) | def _test_median(test_case, placement, sbp, ndim): function _test_median_with_indices (line 33) | def _test_median_with_indices(test_case, placement, sbp, ndim): class TestMedian (line 46) | class TestMedian(flow.unittest.TestCase): method test_median (line 48) | def test_median(test_case): FILE: python/oneflow/test/modules/test_global_meshgrid.py function _test_meshgrid (line 25) | def _test_meshgrid(test_case, placement): class TestMeshGrid (line 34) | class TestMeshGrid(flow.unittest.TestCase): method test_meshgrid (line 36) | def test_meshgrid(test_case): FILE: python/oneflow/test/modules/test_global_min.py function _np_min (line 26) | def _np_min(shape, dim, keepdims): function _test_min (line 41) | def _test_min( class TestMinModule (line 65) | class TestMinModule(flow.unittest.TestCase): method test_eager_global_min (line 69) | def test_eager_global_min(test_case): FILE: python/oneflow/test/modules/test_global_min_max_observer.py function _run_test_min_max_observer (line 26) | def _run_test_min_max_observer( class TestMinMaxObserver (line 61) | class TestMinMaxObserver(flow.unittest.TestCase): method test_min_max_observer (line 63) | def test_min_max_observer(test_case): FILE: python/oneflow/test/modules/test_global_movedim.py function _test_movedim (line 23) | def _test_movedim(test_case, placement, sbp): class TestMovedim (line 35) | class TestMovedim(flow.unittest.TestCase): method test_movedim (line 37) | def test_movedim(test_case): FILE: python/oneflow/test/modules/test_global_moving_average_max_min_observer.py function _run_test_moving_average_min_max_observer (line 25) | def _run_test_moving_average_min_max_observer( class TestMovingAverageMinMaxObserver (line 81) | class TestMovingAverageMinMaxObserver(flow.unittest.TestCase): method test_moving_average_min_max_observer (line 83) | def test_moving_average_min_max_observer(test_case): FILE: python/oneflow/test/modules/test_global_mul.py function _test_broadcast_mul (line 23) | def _test_broadcast_mul(test_case, placement, sbp): function _test_mul_with_scalar (line 32) | def _test_mul_with_scalar(test_case, ndim, placement, sbp): class TestMulModule (line 39) | class TestMulModule(flow.unittest.TestCase): method test_broadcast_mul (line 41) | def test_broadcast_mul(test_case): method test_mul_with_scalar (line 47) | def test_mul_with_scalar(test_case): FILE: python/oneflow/test/modules/test_global_mv.py function _test_mv (line 23) | def _test_mv(test_case, placement, sbp): class TestMvModule (line 30) | class TestMvModule(flow.unittest.TestCase): method test_mv (line 32) | def test_mv(test_case): FILE: python/oneflow/test/modules/test_global_nansum.py function _test_global_nansum_against_pytorch (line 29) | def _test_global_nansum_against_pytorch(test_case, placement, sbp): function _test_global_nansum_with_0_size_tensor (line 38) | def _test_global_nansum_with_0_size_tensor(test_case, placement, sbp): class TestGlobalNanSumModule (line 46) | class TestGlobalNanSumModule(flow.unittest.TestCase): method test_global_nansum_against_pytorch (line 48) | def test_global_nansum_against_pytorch(test_case): method test_global_nansum_with_0_size_tensor (line 54) | def test_global_nansum_with_0_size_tensor(test_case): FILE: python/oneflow/test/modules/test_global_narrow.py function _test_narrow (line 23) | def _test_narrow(test_case, ndim, placement, sbp): class TestNarrow (line 33) | class TestNarrow(flow.unittest.TestCase): method test_narrow (line 35) | def test_narrow(test_case): FILE: python/oneflow/test/modules/test_global_ne.py function _test_ne (line 23) | def _test_ne(test_case, placement, sbp): class TestNe (line 29) | class TestNe(flow.unittest.TestCase): method test_ne (line 31) | def test_ne(test_case): FILE: python/oneflow/test/modules/test_global_negative.py function _test_negative (line 23) | def _test_negative(test_case, placement, sbp, ndim): class TestNegative (line 29) | class TestNegative(flow.unittest.TestCase): method test_negative (line 31) | def test_negative(test_case): FILE: python/oneflow/test/modules/test_global_nms.py function _test_nms (line 26) | def _test_nms(test_case, placement, sbp): class TestNMS (line 48) | class TestNMS(flow.unittest.TestCase): method test_nms (line 50) | def test_nms(test_case): FILE: python/oneflow/test/modules/test_global_normal.py function _test_global_normal (line 24) | def _test_global_normal( class TestNormalGlobal (line 45) | class TestNormalGlobal(flow.unittest.TestCase): method test_normal_global (line 47) | def test_normal_global(test_case): FILE: python/oneflow/test/modules/test_global_normalize.py function _test_functional_normalize (line 25) | def _test_functional_normalize(test_case, placement, sbp): class TestModule (line 34) | class TestModule(flow.unittest.TestCase): method test_normalize_with_random_data (line 36) | def test_normalize_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_nozero.py function _test_nonzero (line 25) | def _test_nonzero(test_case, placement, sbp, ndim): class TestNonZero (line 31) | class TestNonZero(flow.unittest.TestCase): method test_nonzero (line 33) | def test_nonzero(test_case): FILE: python/oneflow/test/modules/test_global_ones_like.py function _test_ones_like_float (line 27) | def _test_ones_like_float(test_case, placement, sbp, shape, device): function _test_ones_like_int (line 43) | def _test_ones_like_int(test_case, placement, sbp, shape, device): class TestModule (line 54) | class TestModule(flow.unittest.TestCase): method test_ones_like (line 57) | def test_ones_like(test_case): FILE: python/oneflow/test/modules/test_global_pad.py function _test_pad_1d_impl (line 25) | def _test_pad_1d_impl(test_case, placement, sbp): function _test_pad_2d_impl (line 35) | def _test_pad_2d_impl(test_case, placement, sbp): class TestPad (line 48) | class TestPad(flow.unittest.TestCase): method test_pad_1d (line 50) | def test_pad_1d(test_case): FILE: python/oneflow/test/modules/test_global_partical_fc.py class TestParitalFC (line 26) | class TestParitalFC(flow.unittest.TestCase): method test_parital_fc (line 28) | def test_parital_fc(test_case): FILE: python/oneflow/test/modules/test_global_permute.py function _test_permute4d_tensor_with_random_data (line 25) | def _test_permute4d_tensor_with_random_data(test_case, placement, sbp): class TestModule (line 35) | class TestModule(flow.unittest.TestCase): method test_permute4d_tensor_with_random_data (line 37) | def test_permute4d_tensor_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_rand.py function _test_global_rand (line 27) | def _test_global_rand(test_case, shape, placement, sbp): function _test_graph_rand (line 35) | def _test_graph_rand(test_case, shape, placement, sbp): class TestRandGlobal (line 52) | class TestRandGlobal(flow.unittest.TestCase): method test_rand_global (line 54) | def test_rand_global(test_case): method test_rand_graph (line 65) | def test_rand_graph(test_case): FILE: python/oneflow/test/modules/test_global_randint.py function _test_global_randint (line 27) | def _test_global_randint(test_case, shape, placement, sbp, dtype): function _test_graph_randint (line 36) | def _test_graph_randint(test_case, shape, placement, sbp, dtype): class TestRandintGlobal (line 54) | class TestRandintGlobal(flow.unittest.TestCase): method test_randint_global (line 56) | def test_randint_global(test_case): method test_randint_graph (line 76) | def test_randint_graph(test_case): FILE: python/oneflow/test/modules/test_global_randint_like.py function _test_consistent_randint_like (line 27) | def _test_consistent_randint_like(test_case, shape, placement, sbp, dtype): function _test_graph_randint_like (line 37) | def _test_graph_randint_like(test_case, shape, placement, sbp, dtype): class TestRandIntLikeConsistent (line 56) | class TestRandIntLikeConsistent(flow.unittest.TestCase): method test_randint_like_consistent (line 58) | def test_randint_like_consistent(test_case): method test_randint_like_graph (line 80) | def test_randint_like_graph(test_case): FILE: python/oneflow/test/modules/test_global_randn.py function _test_global_randn (line 28) | def _test_global_randn(test_case, shape, placement, sbp): function _test_different_dtype (line 37) | def _test_different_dtype(test_case, shape, placement, sbp): function _test_backward (line 44) | def _test_backward(test_case, shape, placement, sbp): function _test_with_generator (line 53) | def _test_with_generator(test_case, shape, placement, sbp): function _test_randn_tuple_shape (line 62) | def _test_randn_tuple_shape(test_case, shape, placement, sbp): function _test_graph_randn (line 70) | def _test_graph_randn(test_case, shape, placement, sbp): class TestRandnGlobal (line 87) | class TestRandnGlobal(flow.unittest.TestCase): method test_randn_global (line 89) | def test_randn_global(test_case): method test_randn_graph (line 105) | def test_randn_graph(test_case): FILE: python/oneflow/test/modules/test_global_random_op_data.py function _test_data_consistent (line 44) | def _test_data_consistent(test_case, shape, placement, sbp, fn): class TestGlobalRandomOpData (line 74) | class TestGlobalRandomOpData(flow.unittest.TestCase): method test_random_op_data_consistent_with_eager_and_lazy (line 77) | def test_random_op_data_consistent_with_eager_and_lazy(test_case): method test_random_op_data_correctness (line 88) | def test_random_op_data_correctness(test_case): FILE: python/oneflow/test/modules/test_global_randperm.py function _test_global_randperm (line 26) | def _test_global_randperm(test_case, N, placement, sbp, dtype): function _test_graph_randperm (line 34) | def _test_graph_randperm(test_case, N, placement, sbp, dtype): class TestRandpermGlobal (line 55) | class TestRandpermGlobal(flow.unittest.TestCase): method test_randperm_global (line 57) | def test_randperm_global(test_case): method test_randperm_graph (line 72) | def test_randperm_graph(test_case): FILE: python/oneflow/test/modules/test_global_reciprocal.py function _test_reciprocal_impl (line 25) | def _test_reciprocal_impl(test_case, ndim, placement, sbp): class TestReciprocalGlobal (line 33) | class TestReciprocalGlobal(flow.unittest.TestCase): method test_reciprocal (line 35) | def test_reciprocal(test_case): FILE: python/oneflow/test/modules/test_global_reflection_pad2d.py function _test_reflection_pad2d_impl (line 26) | def _test_reflection_pad2d_impl(test_case, padding, placement, sbp): class TestReflectionPad2dGlobal (line 35) | class TestReflectionPad2dGlobal(flow.unittest.TestCase): method test_reflection_pad2d (line 37) | def test_reflection_pad2d(test_case): FILE: python/oneflow/test/modules/test_global_repeat.py function _test_repeat_impl (line 25) | def _test_repeat_impl(test_case, ndim, placement, sbp): class TestRepeatGlobal (line 34) | class TestRepeatGlobal(flow.unittest.TestCase): method test_repeat (line 37) | def test_repeat(test_case): FILE: python/oneflow/test/modules/test_global_replication_pad2d.py function _test_replication_pad2d_impl (line 26) | def _test_replication_pad2d_impl(test_case, padding, placement, sbp): class TestReplicationPad2dGlobal (line 35) | class TestReplicationPad2dGlobal(flow.unittest.TestCase): method test_replication_pad2d (line 37) | def test_replication_pad2d(test_case): FILE: python/oneflow/test/modules/test_global_reshape.py function _test_reshape_impl (line 27) | def _test_reshape_impl(test_case, pair, placement, sbp): function _test_reshape_like_impl (line 35) | def _test_reshape_like_impl(test_case, pair, placement, in_sbp, like_sbp): class TestReshapeGlobal (line 57) | class TestReshapeGlobal(flow.unittest.TestCase): method test_reshape (line 59) | def test_reshape(test_case): method test_reshape_like (line 73) | def test_reshape_like(test_case): FILE: python/oneflow/test/modules/test_global_rnn.py function _compare_torch_and_oneflow (line 28) | def _compare_torch_and_oneflow( function _test_rnn_impl (line 85) | def _test_rnn_impl( function _test_lstm_impl (line 124) | def _test_lstm_impl( function _test_gru_impl (line 163) | def _test_gru_impl( class TestRNNGlobal (line 199) | class TestRNNGlobal(oneflow.unittest.TestCase): method test_rnn (line 201) | def test_rnn(test_case): method test_lstm (line 227) | def test_lstm(test_case): method test_gru (line 253) | def test_gru(test_case): FILE: python/oneflow/test/modules/test_global_rnn_cell.py function _test_rnn_relu_cell (line 26) | def _test_rnn_relu_cell(test_case, placement, sbp): function _test_rnn_tanh_cell (line 71) | def _test_rnn_tanh_cell(test_case, placement, sbp): class TestRNNCellGlobal (line 116) | class TestRNNCellGlobal(flow.unittest.TestCase): method test_rnn_relu_cell (line 118) | def test_rnn_relu_cell(test_case): method test_rnn_tanh_cell (line 124) | def test_rnn_tanh_cell(test_case): FILE: python/oneflow/test/modules/test_global_roi_align.py function _get_np_rois (line 27) | def _get_np_rois(): function _test_roi_align (line 51) | def _test_roi_align(test_case, placement, rois_sbp): function _test_roi_align_in_fixed_data_impl (line 111) | def _test_roi_align_in_fixed_data_impl(test_case, placement, sbp): class TestGlobalRoiAlign (line 134) | class TestGlobalRoiAlign(flow.unittest.TestCase): method test_global_roi_align_in_fixed_data (line 147) | def test_global_roi_align_in_fixed_data(test_case): FILE: python/oneflow/test/modules/test_global_roll.py function _test_roll_impl (line 25) | def _test_roll_impl(test_case, placement, sbp): class TestRollGlobal (line 40) | class TestRollGlobal(flow.unittest.TestCase): method test_roll (line 42) | def test_roll(test_case): FILE: python/oneflow/test/modules/test_global_round.py function _test_round_impl (line 26) | def _test_round_impl(test_case, ndim, placement, sbp): class TestRoundGlobal (line 34) | class TestRoundGlobal(flow.unittest.TestCase): method test_round (line 36) | def test_round(test_case): FILE: python/oneflow/test/modules/test_global_scatter_nd.py function _test_scatter_nd (line 25) | def _test_scatter_nd(test_case, placement, sbp): class TestScatterNd (line 54) | class TestScatterNd(flow.unittest.TestCase): method test_scatter_nd (line 56) | def test_scatter_nd(test_case): FILE: python/oneflow/test/modules/test_global_scatter_ops.py function _test_scatter_random_data (line 26) | def _test_scatter_random_data(test_case, placement): function _test_scatter_scalar_random_data (line 43) | def _test_scatter_scalar_random_data(test_case, placement): function _test_scatter_add_random_data (line 57) | def _test_scatter_add_random_data(test_case, placement): class TestScatterOps (line 74) | class TestScatterOps(flow.unittest.TestCase): method test_scatter_ops (line 76) | def test_scatter_ops(test_case): FILE: python/oneflow/test/modules/test_global_searchsorted.py function _test_search_sorted (line 26) | def _test_search_sorted(test_case, placement, sbp, ndim): class TestSearchSorted_Global (line 36) | class TestSearchSorted_Global(flow.unittest.TestCase): method test_search_sorted (line 38) | def test_search_sorted(test_case): function _test_search_sorted_scalar (line 46) | def _test_search_sorted_scalar(test_case, placement, sbp): class TestSearchSortedScalar_Global (line 56) | class TestSearchSortedScalar_Global(flow.unittest.TestCase): method test_search_sorted_scalar (line 59) | def test_search_sorted_scalar(test_case): FILE: python/oneflow/test/modules/test_global_sign.py function _test_sign_impl (line 25) | def _test_sign_impl(test_case, ndim, placement, sbp): class TestSign (line 32) | class TestSign(flow.unittest.TestCase): method test_sign (line 34) | def test_sign(test_case): FILE: python/oneflow/test/modules/test_global_slice.py function _check_forward_and_backward (line 25) | def _check_forward_and_backward(test_case, input, of_out, torch_out): function _test_slice_random_data (line 40) | def _test_slice_random_data(test_case, placement, sbp): function _test_slice_empty (line 51) | def _test_slice_empty(test_case, placement, sbp): function _test_slice_1dim (line 62) | def _test_slice_1dim(test_case, placement, sbp): function _test_negative_index (line 72) | def _test_negative_index(test_case, placement, sbp): function _test_slice_ellipsis_type (line 82) | def _test_slice_ellipsis_type(test_case, placement, sbp): function _test_slice_with_bool (line 92) | def _test_slice_with_bool(test_case, placement, sbp): function _test_slice_with_grad (line 105) | def _test_slice_with_grad(test_case, placement): class TestSlice (line 169) | class TestSlice(flow.unittest.TestCase): method test_slice (line 171) | def test_slice(test_case): method test_graph_slice (line 183) | def test_graph_slice(test_case): FILE: python/oneflow/test/modules/test_global_slice_update.py function _test_slice_update (line 25) | def _test_slice_update(test_case, placement, sbp): function _test_graph_slice_update (line 53) | def _test_graph_slice_update(test_case, placement, sbp): class TestGlobalSliceUpdate (line 118) | class TestGlobalSliceUpdate(flow.unittest.TestCase): method test_slice_update (line 120) | def test_slice_update(test_case): FILE: python/oneflow/test/modules/test_global_sort.py function _test_sort_impl (line 25) | def _test_sort_impl(test_case, placement): class TestSortGlobal (line 38) | class TestSortGlobal(flow.unittest.TestCase): method test_sort (line 40) | def test_sort(test_case): FILE: python/oneflow/test/modules/test_global_sparse.py function _test_embedding (line 24) | def _test_embedding(test_case, ndim, placement, sbp): class TestEmbedding (line 43) | class TestEmbedding(flow.unittest.TestCase): method test_embedding (line 45) | def test_embedding(test_case): FILE: python/oneflow/test/modules/test_global_sparse_softmax_cross_entropy.py function _compare_eager_global_with_torch (line 31) | def _compare_eager_global_with_torch( function _compare_lazy_global_with_torch (line 78) | def _compare_lazy_global_with_torch( class TestGlobalSparseSoftmaxCrossEntropyWithLogits (line 127) | class TestGlobalSparseSoftmaxCrossEntropyWithLogits(flow.unittest.TestCa... method test_eager_global_sparse_softmax_cross_entropy (line 129) | def test_eager_global_sparse_softmax_cross_entropy(test_case): FILE: python/oneflow/test/modules/test_global_split.py function _test_flow_split_with_random_data (line 29) | def _test_flow_split_with_random_data(test_case, placement, sbp): function _test_flow_split_sizes_with_random_data (line 42) | def _test_flow_split_sizes_with_random_data(test_case, placement, sbp): function _test_flow_split_sizes_neg_dim_with_random_data (line 54) | def _test_flow_split_sizes_neg_dim_with_random_data(test_case, placement... class TestGlobalSplitModule (line 65) | class TestGlobalSplitModule(flow.unittest.TestCase): method test_flow_split_with_random_data (line 67) | def test_flow_split_with_random_data(test_case): method test_flow_split_sizes_with_random_data (line 73) | def test_flow_split_sizes_with_random_data(test_case): method test_flow_split_sizes_neg_dim_with_random_data (line 79) | def test_flow_split_sizes_neg_dim_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_sqrt_square_sum.py function _test_sqrt_sum_with_cpu_random_data (line 29) | def _test_sqrt_sum_with_cpu_random_data(test_case, placement, sbp): function _test_scalar_random_data (line 38) | def _test_scalar_random_data(test_case, placement, sbp): class TestGlobalLinalgVectorNorm2D (line 46) | class TestGlobalLinalgVectorNorm2D(flow.unittest.TestCase): method test_sqrt_sum_with_cpu_random_data (line 48) | def test_sqrt_sum_with_cpu_random_data(test_case): method test_scalar_random_data (line 54) | def test_scalar_random_data(test_case): FILE: python/oneflow/test/modules/test_global_squeeze.py function _test_squeeze_1d_input (line 29) | def _test_squeeze_1d_input(test_case, placement, sbp): function _test_flow_squeeze_with_random_data (line 36) | def _test_flow_squeeze_with_random_data(test_case, placement, sbp): function _test_squeeze_with_0_size_data (line 43) | def _test_squeeze_with_0_size_data(test_case, placement, sbp): class TestGlobalSqueeze (line 49) | class TestGlobalSqueeze(flow.unittest.TestCase): method test_squeeze_1d_input (line 51) | def test_squeeze_1d_input(test_case): method test_flow_squeeze_with_random_data (line 57) | def test_flow_squeeze_with_random_data(test_case): method test_squeeze_with_0_size_data (line 63) | def test_squeeze_with_0_size_data(test_case): FILE: python/oneflow/test/modules/test_global_stack.py function _test_stack_with_random_data (line 26) | def _test_stack_with_random_data(test_case, placement, sbp): class TestStackModule (line 38) | class TestStackModule(flow.unittest.TestCase): method test_stack_with_random_data (line 40) | def test_stack_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_stateful_kernel_with_cache.py function _test_global_stateful_kernel_with_inpersistent_state (line 25) | def _test_global_stateful_kernel_with_inpersistent_state(test_case, plac... class TestStatefulKernelWithInpersistentState (line 40) | class TestStatefulKernelWithInpersistentState(flow.unittest.TestCase): method test_global_stateful_kernel_with_inpersistent_state (line 42) | def test_global_stateful_kernel_with_inpersistent_state(test_case): FILE: python/oneflow/test/modules/test_global_std.py function _test_global_std_flow_with_random_data (line 24) | def _test_global_std_flow_with_random_data(test_case, placement, sbp): function _test_global_std_tensor_with_random_data (line 38) | def _test_global_std_tensor_with_random_data(test_case, placement, sbp): class TestGlobalStd (line 51) | class TestGlobalStd(flow.unittest.TestCase): method test_global_std_flow_with_random_data (line 53) | def test_global_std_flow_with_random_data(test_case): method test_global_std_tensor_with_random_data (line 59) | def test_global_std_tensor_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_sub.py function _test_global_sub (line 29) | def _test_global_sub(test_case, placement, sbp): function _test_global_sub_with_0_size_data (line 40) | def _test_global_sub_with_0_size_data(test_case, placement, sbp): class TestGlobalSubModule (line 48) | class TestGlobalSubModule(flow.unittest.TestCase): method test_global_sub (line 50) | def test_global_sub(test_case): method test_global_sub_with_0_size_data (line 56) | def test_global_sub_with_0_size_data(test_case): FILE: python/oneflow/test/modules/test_global_sum.py function _test_global_sum_against_pytorch (line 29) | def _test_global_sum_against_pytorch(test_case, placement, sbp): function _test_global_sum_with_0_size_tensor (line 36) | def _test_global_sum_with_0_size_tensor(test_case, placement, sbp): class TestGlobalSumModule (line 42) | class TestGlobalSumModule(flow.unittest.TestCase): method test_global_sum_against_pytorch (line 45) | def test_global_sum_against_pytorch(test_case): method test_global_sum_with_0_size_tensor (line 51) | def test_global_sum_with_0_size_tensor(test_case): FILE: python/oneflow/test/modules/test_global_tensor_new.py function _test_tensor_new (line 23) | def _test_tensor_new(test_case, placement, sbp): class TestTensorNew (line 46) | class TestTensorNew(flow.unittest.TestCase): method test_tensor_new (line 48) | def test_tensor_new(test_case): FILE: python/oneflow/test/modules/test_global_tensor_ops.py function _test_type_as (line 29) | def _test_type_as(test_case, shape, src_dtype, tgt_dtype, placement, sbp): function _test_local_to_global_type_as (line 37) | def _test_local_to_global_type_as( function _test_global_to_local_type_as (line 49) | def _test_global_to_local_type_as( function _test_is_floating_point (line 60) | def _test_is_floating_point(test_case, shape, dtype, placement, sbp): function _test_global_cuda (line 72) | def _test_global_cuda(test_case, placement, sbp): class TestGlobalCuda (line 79) | class TestGlobalCuda(flow.unittest.TestCase): method test_global_cuda (line 81) | def test_global_cuda(test_case): function _test_global_cpu (line 88) | def _test_global_cpu(test_case, placement, sbp): function _test_global_long (line 98) | def _test_global_long(test_case, placement, sbp): function _test_global_int (line 106) | def _test_global_int(test_case, placement, sbp): function _test_global_float (line 114) | def _test_global_float(test_case, placement, sbp): function _test_global_double (line 121) | def _test_global_double(test_case, placement, sbp): function _test_global_item (line 128) | def _test_global_item(test_case, placement, sbp): function _test_global_tolist (line 135) | def _test_global_tolist(test_case, placement, sbp): class TestGlobalTensorOps (line 143) | class TestGlobalTensorOps(flow.unittest.TestCase): method test_global_cpu (line 145) | def test_global_cpu(test_case): method test_global_long (line 151) | def test_global_long(test_case): method test_global_int (line 157) | def test_global_int(test_case): method test_global_float (line 163) | def test_global_float(test_case): method test_global_double (line 169) | def test_global_double(test_case): method test_global_item (line 176) | def test_global_item(test_case): method test_global_tolist (line 182) | def test_global_tolist(test_case): method test_type_as (line 188) | def test_type_as(test_case): method test_is_floating_point (line 201) | def test_is_floating_point(test_case): FILE: python/oneflow/test/modules/test_global_tensor_scatter_nd_update.py class TensorScatterNdUpdate (line 27) | class TensorScatterNdUpdate(flow.nn.Graph): method __init__ (line 28) | def __init__(self): method build (line 31) | def build(self, origin, indices, update): function _test_global_tensor_scatter_nd_update (line 35) | def _test_global_tensor_scatter_nd_update(test_case, placement, sbp, che... function _test_global_tensor_scatter_nd_update_t (line 63) | def _test_global_tensor_scatter_nd_update_t( function _test_eager_global_tensor_scatter_nd_update_backward (line 94) | def _test_eager_global_tensor_scatter_nd_update_backward(test_case, plac... class TestTensorScatterNdUpdate (line 126) | class TestTensorScatterNdUpdate(flow.unittest.TestCase): method test_global_tensor_scatter_nd_update (line 128) | def test_global_tensor_scatter_nd_update(test_case): method test_global_tensor_scatter_nd_update_t (line 140) | def test_global_tensor_scatter_nd_update_t(test_case): method test_global_tensor_scatter_nd_update_backward (line 152) | def test_global_tensor_scatter_nd_update_backward(test_case): FILE: python/oneflow/test/modules/test_global_tensordot.py function _test_global_tensordot_against_pytorch (line 24) | def _test_global_tensordot_against_pytorch(test_case, ndim, placement, s... class TestTensorDotGlobal (line 38) | class TestTensorDotGlobal(flow.unittest.TestCase): method test_tensordot (line 40) | def test_tensordot(test_case): FILE: python/oneflow/test/modules/test_global_tile.py function _test_global_flow_tile_with_random_data (line 25) | def _test_global_flow_tile_with_random_data(test_case, placement, sbp): function _test_global_flow_tensor_tile_with_random_data (line 37) | def _test_global_flow_tensor_tile_with_random_data(test_case, placement,... class TestGlobalTile (line 48) | class TestGlobalTile(flow.unittest.TestCase): method test_global_flow_tile_with_random_data (line 51) | def test_global_flow_tile_with_random_data(test_case): method test_global_flow_tensor_tile_with_random_data (line 58) | def test_global_flow_tensor_tile_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_transpose.py function _test_global_transpose (line 29) | def _test_global_transpose(test_case, placement, sbp): function _test_global_tensor_transpose (line 39) | def _test_global_tensor_transpose(test_case, placement, sbp): function _test_global_tranpose_negative_dim (line 49) | def _test_global_tranpose_negative_dim(test_case, placement, sbp): function _test_global_transpose_backward (line 59) | def _test_global_transpose_backward(test_case, placement, sbp): function _test_global_transpose_backward_v2 (line 72) | def _test_global_transpose_backward_v2(test_case, placement, sbp): function _test_global_transpose_flow_with_random_data (line 86) | def _test_global_transpose_flow_with_random_data(test_case, placement, s... function _test_global_transpose_with_0_size_data (line 93) | def _test_global_transpose_with_0_size_data(test_case, placement, sbp): class TestGlobalTranspose (line 100) | class TestGlobalTranspose(flow.unittest.TestCase): method test_global_transpose (line 102) | def test_global_transpose(test_case): method test_global_transpose_flow_with_random_data (line 117) | def test_global_transpose_flow_with_random_data(test_case): method test_global_transpose_with_0_size_data (line 123) | def test_global_transpose_with_0_size_data(test_case): FILE: python/oneflow/test/modules/test_global_tril.py function _test_global_tril_without_diag (line 24) | def _test_global_tril_without_diag(test_case, placement, sbp): function _test_global_tril_with_diag (line 39) | def _test_global_tril_with_diag(test_case, placement, sbp): class TestGlobalTril (line 54) | class TestGlobalTril(flow.unittest.TestCase): method test_global_tril_without_diag (line 57) | def test_global_tril_without_diag(test_case): method test_global_tril_with_diag (line 63) | def test_global_tril_with_diag(test_case): FILE: python/oneflow/test/modules/test_global_triu.py function _test_global_triu_without_diag (line 24) | def _test_global_triu_without_diag(test_case, placement, sbp): function _test_global_triu_with_diag (line 39) | def _test_global_triu_with_diag(test_case, placement, sbp): class TestGlobalTriu (line 54) | class TestGlobalTriu(flow.unittest.TestCase): method test_global_triu_without_diag (line 56) | def test_global_triu_without_diag(test_case): method test_global_triu_with_diag (line 62) | def test_global_triu_with_diag(test_case): FILE: python/oneflow/test/modules/test_global_unbind.py function _test_unbind (line 25) | def _test_unbind(test_case, placement, sbp): class TestUnbind (line 34) | class TestUnbind(flow.unittest.TestCase): method test_unbind (line 36) | def test_unbind(test_case): FILE: python/oneflow/test/modules/test_global_unfold.py function _test_unfold_with_random_data (line 26) | def _test_unfold_with_random_data(test_case, placement, sbp): class TestUnfold (line 49) | class TestUnfold(flow.unittest.TestCase): method test_unfold_with_random_data (line 51) | def test_unfold_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_unfold_tensor.py function _test_global_unfold_tensor_with_random_data (line 26) | def _test_global_unfold_tensor_with_random_data(test_case, placement, sbp): class TestGlobalUnfoldTensor (line 43) | class TestGlobalUnfoldTensor(flow.unittest.TestCase): method test_global_unfold_tensor_with_random_data (line 45) | def test_global_unfold_tensor_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_unique.py function _test_unique_unsorted (line 24) | def _test_unique_unsorted(test_case, placement, sbp): function _test_unique_sorted (line 60) | def _test_unique_sorted(test_case, placement, sbp): class TestUniqueModule (line 87) | class TestUniqueModule(flow.unittest.TestCase): method test_unique (line 89) | def test_unique(test_case): FILE: python/oneflow/test/modules/test_global_unsqueeze.py function _test_flow_unsqueeze_with_random_data (line 28) | def _test_flow_unsqueeze_with_random_data(test_case, placement, sbp): function _test_tensor_unsqueeze_with_random_data (line 35) | def _test_tensor_unsqueeze_with_random_data(test_case, placement, sbp): function _test_unsqueeze_with_0_size_data (line 42) | def _test_unsqueeze_with_0_size_data(test_case, placement, sbp): class TestGlobalUnsqueeze (line 48) | class TestGlobalUnsqueeze(flow.unittest.TestCase): method test_flow_unsqueeze_with_random_data (line 50) | def test_flow_unsqueeze_with_random_data(test_case): method test_tensor_unsqueeze_with_random_data (line 56) | def test_tensor_unsqueeze_with_random_data(test_case): method test_unsqueeze_with_0_size_data (line 62) | def test_unsqueeze_with_0_size_data(test_case): FILE: python/oneflow/test/modules/test_global_upsample.py function _test_global_upsample2d_nearest (line 29) | def _test_global_upsample2d_nearest(test_case, placement, sbp): function _test_global_upsample2d_linear (line 38) | def _test_global_upsample2d_linear(test_case, placement, sbp): function _test_global_upsample2d_bilinear (line 48) | def _test_global_upsample2d_bilinear(test_case, placement, sbp): function _test_global_upsample2d_bicubic (line 58) | def _test_global_upsample2d_bicubic(test_case, placement, sbp): function _test_global_upsample2d_trilinear (line 68) | def _test_global_upsample2d_trilinear(test_case, placement, sbp): class TestGlobalUpsample2d (line 77) | class TestGlobalUpsample2d(flow.unittest.TestCase): method test_global_upsample2d_nearest (line 82) | def test_global_upsample2d_nearest(test_case): method test_global_upsample2d_linear (line 88) | def test_global_upsample2d_linear(test_case): method test_global_upsample2d_bilinear (line 94) | def test_global_upsample2d_bilinear(test_case): method test_global_upsample2d_bicubic (line 100) | def test_global_upsample2d_bicubic(test_case): method test_global_upsample2d_trilinear (line 106) | def test_global_upsample2d_trilinear(test_case): FILE: python/oneflow/test/modules/test_global_var.py function _test_flow_global_var_all_dim_with_random_data (line 26) | def _test_flow_global_var_all_dim_with_random_data(test_case, placement,... function _test_flow_global_var_one_dim_with_random_data (line 35) | def _test_flow_global_var_one_dim_with_random_data(test_case, placement,... function _test_flow_var_0_size_data_with_random_data (line 49) | def _test_flow_var_0_size_data_with_random_data(test_case, placement, sbp): class TestVar (line 60) | class TestVar(flow.unittest.TestCase): method test_flow_global_var_all_dim_with_random_data (line 62) | def test_flow_global_var_all_dim_with_random_data(test_case): method test_flow_global_var_one_dim_with_random_data (line 70) | def test_flow_global_var_one_dim_with_random_data(test_case): method test_flow_var_0_size_data_with_random_data (line 78) | def test_flow_var_0_size_data_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_vector_matrix_product.py function _test_vector_matrix_product (line 23) | def _test_vector_matrix_product(test_case, placement, sbp): class TestGlobalVectorMatrixProduct (line 32) | class TestGlobalVectorMatrixProduct(flow.unittest.TestCase): method test_vector_matrix_product (line 34) | def test_vector_matrix_product(test_case): FILE: python/oneflow/test/modules/test_global_view.py function _test_global_view (line 26) | def _test_global_view(test_case, placement, sbp): function _test_global_view_size (line 33) | def _test_global_view_size(test_case, placement, sbp): class TestGlobalView (line 40) | class TestGlobalView(flow.unittest.TestCase): method test_global_view (line 42) | def test_global_view(test_case): method test_global_view_size (line 48) | def test_global_view_size(test_case): FILE: python/oneflow/test/modules/test_global_weight_norm.py function _test_global_weight_norm_with_random_data (line 26) | def _test_global_weight_norm_with_random_data(test_case, placement, sbp): class TestGlobalWeightNorm (line 33) | class TestGlobalWeightNorm(flow.unittest.TestCase): method test_global_weight_norm_with_random_data (line 36) | def test_global_weight_norm_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_where.py function _test_global_where (line 26) | def _test_global_where(test_case, placement, sbp): function _test_global_where_broadcast (line 40) | def _test_global_where_broadcast(test_case, placement, sbp): function _test_global_where_scalar (line 54) | def _test_global_where_scalar(test_case, placement, sbp): function _test_where_x_y_none (line 71) | def _test_where_x_y_none(test_case, placement, sbp): function _test_global_where_tensor_with_0dim_data (line 80) | def _test_global_where_tensor_with_0dim_data(test_case, placement, sbp): function _test_flow_where_tensor_broadcast_with_random_data (line 88) | def _test_flow_where_tensor_broadcast_with_random_data(test_case, placem... function _test_flow_where_scalar_x_with_random_data (line 96) | def _test_flow_where_scalar_x_with_random_data(test_case, placement, sbp): function _test_flow_where_scalar_x_broadcast_with_random_data (line 108) | def _test_flow_where_scalar_x_broadcast_with_random_data(test_case, plac... function _test_flow_where_scalar_x_int_with_random_data (line 120) | def _test_flow_where_scalar_x_int_with_random_data(test_case, placement,... function _test_flow_where_scalar_y_with_random_data (line 128) | def _test_flow_where_scalar_y_with_random_data(test_case, placement, sbp): function _test_flow_where_scalar_y_broadcast_with_random_data (line 140) | def _test_flow_where_scalar_y_broadcast_with_random_data(test_case, plac... function _test_flow_where_scalar_y_int_with_random_data (line 152) | def _test_flow_where_scalar_y_int_with_random_data(test_case, placement,... function _test_flow_where_tensor_bool_with_random_data (line 160) | def _test_flow_where_tensor_bool_with_random_data(test_case, placement, ... function _test_flow_where_tensor_broadcast_bool_with_random_data (line 168) | def _test_flow_where_tensor_broadcast_bool_with_random_data(test_case, p... function _test_flow_where_scalar_x_bool_with_random_data (line 176) | def _test_flow_where_scalar_x_bool_with_random_data(test_case, placement... function _test_flow_where_scalar_x_broadcast_bool_with_random_data (line 188) | def _test_flow_where_scalar_x_broadcast_bool_with_random_data( function _test_flow_where_scalar_y_bool_with_random_data (line 202) | def _test_flow_where_scalar_y_bool_with_random_data(test_case, placement... function _test_flow_where_scalar_y_broadcast_bool_with_random_data (line 214) | def _test_flow_where_scalar_y_broadcast_bool_with_random_data( function _test_flow_where_scalar_xy_bool_with_random_data (line 228) | def _test_flow_where_scalar_xy_bool_with_random_data(test_case, placemen... class TestGlobalWhere (line 235) | class TestGlobalWhere(flow.unittest.TestCase): method test_global_where (line 237) | def test_global_where(test_case): method test_global_where_broadcast (line 243) | def test_global_where_broadcast(test_case): method test_global_where_scalar (line 249) | def test_global_where_scalar(test_case): method test_where_x_y_none (line 255) | def test_where_x_y_none(test_case): method test_global_where_tensor_with_0dim_data (line 261) | def test_global_where_tensor_with_0dim_data(test_case): method test_flow_where_tensor_broadcast_with_random_data (line 267) | def test_flow_where_tensor_broadcast_with_random_data(test_case): method test_flow_where_scalar_x_with_random_data (line 275) | def test_flow_where_scalar_x_with_random_data(test_case): method test_flow_where_scalar_x_broadcast_with_random_data (line 281) | def test_flow_where_scalar_x_broadcast_with_random_data(test_case): method test_flow_where_scalar_x_int_with_random_data (line 289) | def test_flow_where_scalar_x_int_with_random_data(test_case): method test_flow_where_scalar_y_with_random_data (line 297) | def test_flow_where_scalar_y_with_random_data(test_case): method test_flow_where_scalar_y_broadcast_with_random_data (line 303) | def test_flow_where_scalar_y_broadcast_with_random_data(test_case): method test_flow_where_scalar_y_int_with_random_data (line 311) | def test_flow_where_scalar_y_int_with_random_data(test_case): method test_flow_where_tensor_bool_with_random_data (line 319) | def test_flow_where_tensor_bool_with_random_data(test_case): method test_flow_where_tensor_broadcast_bool_with_random_data (line 325) | def test_flow_where_tensor_broadcast_bool_with_random_data(test_case): method test_flow_where_scalar_x_bool_with_random_data (line 333) | def test_flow_where_scalar_x_bool_with_random_data(test_case): method test_flow_where_scalar_x_broadcast_bool_with_random_data (line 341) | def test_flow_where_scalar_x_broadcast_bool_with_random_data(test_case): method test_flow_where_scalar_y_bool_with_random_data (line 349) | def test_flow_where_scalar_y_bool_with_random_data(test_case): method test_flow_where_scalar_y_broadcast_bool_with_random_data (line 357) | def test_flow_where_scalar_y_broadcast_bool_with_random_data(test_case): method test_flow_where_scalar_xy_bool_with_random_data (line 365) | def test_flow_where_scalar_xy_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_global_zeropad2d.py function _test_global_ZeroPad2d (line 26) | def _test_global_ZeroPad2d(test_case, placement, sbp, padding): class TestGlobalZeroPad2dModule (line 35) | class TestGlobalZeroPad2dModule(flow.unittest.TestCase): method test_global_ZeroPad2d (line 37) | def test_global_ZeroPad2d(test_case): FILE: python/oneflow/test/modules/test_global_zeros_like.py function _test_zeros_like_float (line 27) | def _test_zeros_like_float(test_case, placement, sbp, shape, device): function _test_zeros_like_int (line 40) | def _test_zeros_like_int(test_case, placement, sbp, shape, device): class TestModule (line 51) | class TestModule(flow.unittest.TestCase): method test_zeros_like (line 53) | def test_zeros_like(test_case): FILE: python/oneflow/test/modules/test_glu.py class TestGluModule (line 26) | class TestGluModule(flow.unittest.TestCase): method test_glu_module_with_random_data (line 28) | def test_glu_module_with_random_data(test_case): method test_glu_module_with_random_data (line 37) | def test_glu_module_with_random_data(test_case): method profile_glu (line 47) | def profile_glu(test_case): FILE: python/oneflow/test/modules/test_gpt_data_loader.py class GPTDataLoader (line 24) | class GPTDataLoader(flow.nn.Module): method __init__ (line 25) | def __init__( method forward (line 52) | def forward(self): class DataLoaderGraph (line 56) | class DataLoaderGraph(flow.nn.Graph): method __init__ (line 57) | def __init__(self, loader): method build (line 61) | def build(self): class GPTDataLoaderDistributedTestCase (line 71) | class GPTDataLoaderDistributedTestCase(oneflow.unittest.TestCase): method test_case1 (line 72) | def test_case1(test_case): FILE: python/oneflow/test/modules/test_greater.py function _test_greater_normal (line 29) | def _test_greater_normal(test_case, device): function _test_greater_symbol (line 45) | def _test_greater_symbol(test_case, device): function _test_greater_int_scalar (line 61) | def _test_greater_int_scalar(test_case, device): function _test_greater_int_tensor_int_scalar (line 70) | def _test_greater_int_tensor_int_scalar(test_case, device): function _test_greater_float_scalar (line 79) | def _test_greater_float_scalar(test_case, device): class TestGreater (line 89) | class TestGreater(flow.unittest.TestCase): method test_greater (line 90) | def test_greater(test_case): method test_greater_with_random_data (line 104) | def test_greater_with_random_data(test_case): method test_tensor_inplace_greater_with_random_data (line 113) | def test_tensor_inplace_greater_with_random_data(test_case): method test_tensor_greater_with_random_data (line 122) | def test_tensor_greater_with_random_data(test_case): method test_greater_with_0_size_data (line 132) | def test_greater_with_0_size_data(test_case): method test_greater_bool_with_random_data (line 141) | def test_greater_bool_with_random_data(test_case): method test_greater_with_0dim_data (line 154) | def test_greater_with_0dim_data(test_case): method profile_gt (line 163) | def profile_gt(test_case): FILE: python/oneflow/test/modules/test_greater_equal.py function _test_greater_equal_normal (line 27) | def _test_greater_equal_normal(test_case, device): function _test_greater_equal_symbol (line 43) | def _test_greater_equal_symbol(test_case, device): function _test_greater_equal_int_scalar (line 59) | def _test_greater_equal_int_scalar(test_case, device): function _test_greater_equal_int_tensor_int_scalr (line 68) | def _test_greater_equal_int_tensor_int_scalr(test_case, device): function _test_greater_equal_float_scalar (line 77) | def _test_greater_equal_float_scalar(test_case, device): class TestGreaterEqual (line 87) | class TestGreaterEqual(flow.unittest.TestCase): method test_greter_equal (line 88) | def test_greter_equal(test_case): FILE: python/oneflow/test/modules/test_grid_sample.py class TestGridSample (line 30) | class TestGridSample(flow.unittest.TestCase): method test_grid_sample_4d (line 31) | def test_grid_sample_4d(test_case): method test_flow_grid_sample_cudnn_with_random_data (line 55) | def test_flow_grid_sample_cudnn_with_random_data(test_case): method test_flow_grid_sample_4d_with_random_data (line 90) | def test_flow_grid_sample_4d_with_random_data(test_case): method test_flow_grid_sample_5d_with_random_data (line 116) | def test_flow_grid_sample_5d_with_random_data(test_case): method profile_grid_sample (line 146) | def profile_grid_sample(test_case): FILE: python/oneflow/test/modules/test_grouped_matmul_bias.py function _ref (line 26) | def _ref(xs, weights, biases): function _grouped (line 39) | def _grouped(xs, weights, biases): function _test_grouped_matmul_bias (line 46) | def _test_grouped_matmul_bias(test_case, dtype, problems, bias): class TestGroupedMatmulBias (line 66) | class TestGroupedMatmulBias(flow.unittest.TestCase): method test_grouped_matmul_bias (line 67) | def test_grouped_matmul_bias(test_case): FILE: python/oneflow/test/modules/test_groupnorm.py function _test_groupnorm (line 28) | def _test_groupnorm(test_case, device): function _test_groupnorm_3d (line 93) | def _test_groupnorm_3d(test_case, device): function _test_groupnorm_backward (line 216) | def _test_groupnorm_backward(test_case, device): function _test_groupnorm_backward_fp16 (line 258) | def _test_groupnorm_backward_fp16(test_case, device): function _test_groupnorm_backward_3d (line 304) | def _test_groupnorm_backward_3d(test_case, device): function _test_groupnorm_backward_3d_fp16 (line 376) | def _test_groupnorm_backward_3d_fp16(test_case, device): function _test_groupnorm_nhwc (line 450) | def _test_groupnorm_nhwc(test_case, shape, num_groups): class TestGroupNorm (line 472) | class TestGroupNorm(flow.unittest.TestCase): method test_groupnorm (line 473) | def test_groupnorm(test_case): method test_groupnorm_grad_fp16 (line 485) | def test_groupnorm_grad_fp16(test_case): method test_group_norm_with_random_data (line 499) | def test_group_norm_with_random_data(test_case): method test_groupnorm_nhwc (line 515) | def test_groupnorm_nhwc(test_case): FILE: python/oneflow/test/modules/test_groupwise_quantization.py function _pack_int8_to_int4 (line 26) | def _pack_int8_to_int4(x): function _unpack_int4_to_int8 (line 37) | def _unpack_int4_to_int8(x): function _quantize (line 46) | def _quantize(num_bits, symmetric, x, group_dim, group_size, quant_type): function _dequantize_ref (line 84) | def _dequantize_ref(num_bits, symmetric, quantized, scale, zero, group_d... function _dequantize (line 106) | def _dequantize(num_bits, symmetric, x, scale, zero, group_dim, group_si... function _test_dequantize (line 118) | def _test_dequantize(test_case, num_bits, shape, group_dim, group_size): function _test_fused_linear (line 138) | def _test_fused_linear(test_case, num_bits, m, k, n, group_dim, group_si... class TestGroupWiseQuantization (line 182) | class TestGroupWiseQuantization(flow.unittest.TestCase): method test_dequantize (line 183) | def test_dequantize(test_case): method test_fused_linear (line 208) | def test_fused_linear(test_case): FILE: python/oneflow/test/modules/test_gumbel_softmax.py function _test_gumbel_softmax (line 29) | def _test_gumbel_softmax(test_case, tau, dim, device, dtype): function _test_gumbel_softmax_hard (line 40) | def _test_gumbel_softmax_hard(test_case, tau, dim, device, dtype): function _test_gumbel_softmax_backward (line 51) | def _test_gumbel_softmax_backward(test_case, tau, dim, device, dtype): function _test_gumbel_softmax_half (line 71) | def _test_gumbel_softmax_half(test_case, tau, dim, device): class TestGumbelSoftmaxModule (line 84) | class TestGumbelSoftmaxModule(flow.unittest.TestCase): method test_gumbel_softmax (line 86) | def test_gumbel_softmax(test_case): method test_leakyrelu_module_with_half_random_data (line 102) | def test_leakyrelu_module_with_half_random_data(test_case): FILE: python/oneflow/test/modules/test_hann_window.py class TestHannWindow (line 24) | class TestHannWindow(flow.unittest.TestCase): method test_hann_window (line 26) | def test_hann_window(test_case): method test_hann_window_global (line 33) | def test_hann_window_global(test_case): method test_hann_window_dtype (line 42) | def test_hann_window_dtype(test_case): method profile_hann_window (line 51) | def profile_hann_window(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_activation.py function _assert_true (line 28) | def _assert_true(test_case, value1, value2): function _test_activation_grad_grad_impl (line 39) | def _test_activation_grad_grad_impl(test_case, op_name, *args, **kwargs): function _test_prelu_activation_grad_grad_impl (line 56) | def _test_prelu_activation_grad_grad_impl(test_case, op_name, *args, **k... function _test_hardswish_activation_grad_grad_impl (line 81) | def _test_hardswish_activation_grad_grad_impl(test_case, op_name, *args,... function _test_hardsigmoid_activation_grad_grad_impl (line 114) | def _test_hardsigmoid_activation_grad_grad_impl(test_case, op_name, *arg... class TestActivationHigherDerivative (line 145) | class TestActivationHigherDerivative(flow.unittest.TestCase): method test_activation_grad_grad (line 147) | def test_activation_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_conv.py function _test_convnd_grad_grad_impl (line 27) | def _test_convnd_grad_grad_impl(test_case, ndim, rtol=1e-4, atol=1e-5): class TestConvHigherDerivative (line 150) | class TestConvHigherDerivative(flow.unittest.TestCase): method test_conv1d_grad_grad (line 151) | def test_conv1d_grad_grad(test_case): method test_conv2d_grad_grad (line 154) | def test_conv2d_grad_grad(test_case): method test_conv3d_grad_grad (line 157) | def test_conv3d_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_div.py function _test_div_grad_grad_impl (line 26) | def _test_div_grad_grad_impl(test_case): class TestDivHigherDerivative (line 87) | class TestDivHigherDerivative(flow.unittest.TestCase): method test_div_grad_grad (line 88) | def test_div_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_loss.py function _assert_true (line 24) | def _assert_true(test_case, value1, value2, name=""): function generate_grads_for_variables (line 31) | def generate_grads_for_variables(variables): function calculate_and_compare_loss (line 48) | def calculate_and_compare_loss(test_case, input, target, model, order=2): function generate_necessity_for_default_loss (line 74) | def generate_necessity_for_default_loss(): function generate_necessity_for_nll_loss (line 90) | def generate_necessity_for_nll_loss(): function generate_necessity_for_bce_loss (line 117) | def generate_necessity_for_bce_loss(): function _test_smooth_l1_loss_grad_grad_impl (line 163) | def _test_smooth_l1_loss_grad_grad_impl(test_case): function _test_kl_div_loss_grad_grad_impl (line 174) | def _test_kl_div_loss_grad_grad_impl(test_case): function _test_bce_loss_grad_grad_impl (line 186) | def _test_bce_loss_grad_grad_impl(test_case, with_logits=False): function _test_nll_loss_grad_grad_impl (line 209) | def _test_nll_loss_grad_grad_impl(test_case): class TestLossHigherDerivative (line 222) | class TestLossHigherDerivative(flow.unittest.TestCase): method test_smooth_l1_loss_grad_grad (line 223) | def test_smooth_l1_loss_grad_grad(test_case): method test_kl_div_loss_grad_grad (line 227) | def test_kl_div_loss_grad_grad(test_case): method test_nll_loss_grad_grad (line 232) | def test_nll_loss_grad_grad(test_case): method test_bce_loss_grad_grad (line 236) | def test_bce_loss_grad_grad(test_case): method test_bce_with_logits_loss_grad_grad (line 240) | def test_bce_with_logits_loss_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_matmul.py class TestMatmulHigherDerivative (line 27) | class TestMatmulHigherDerivative(flow.unittest.TestCase): method test_broadcast_matmul_grad_b_grad (line 28) | def test_broadcast_matmul_grad_b_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_neg.py class TestNegHigherDerivative (line 24) | class TestNegHigherDerivative(flow.unittest.TestCase): method test_neg_grad_grad (line 25) | def test_neg_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_pool.py function _check_equal (line 24) | def _check_equal(test_case, lhs, rhs, name="", rtol=1e-5, atol=1e-5): function _test_avg_pool_grad_grad_impl (line 35) | def _test_avg_pool_grad_grad_impl(test_case, ndim): function _test_max_pool_grad_grad_impl (line 78) | def _test_max_pool_grad_grad_impl(test_case, ndim): function _test_adaptive_pool_grad_grad_impl (line 122) | def _test_adaptive_pool_grad_grad_impl(test_case, ndim, mode): class TestPoolHigherDerivative (line 148) | class TestPoolHigherDerivative(flow.unittest.TestCase): method test_max_pool_1d_grad_grad (line 149) | def test_max_pool_1d_grad_grad(test_case): method test_max_pool_2d_grad_grad (line 152) | def test_max_pool_2d_grad_grad(test_case): method test_max_pool_3d_grad_grad (line 155) | def test_max_pool_3d_grad_grad(test_case): method test_avg_pool_1d_grad_grad (line 158) | def test_avg_pool_1d_grad_grad(test_case): method test_avg_pool_2d_grad_grad (line 161) | def test_avg_pool_2d_grad_grad(test_case): method test_avg_pool_3d_grad_grad (line 164) | def test_avg_pool_3d_grad_grad(test_case): method test_adaptive_avg_pool_1d_grad_grad (line 167) | def test_adaptive_avg_pool_1d_grad_grad(test_case): method test_adaptive_avg_pool_2d_grad_grad (line 170) | def test_adaptive_avg_pool_2d_grad_grad(test_case): method test_adaptive_avg_pool_3d_grad_grad (line 173) | def test_adaptive_avg_pool_3d_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_pow.py function _check_equal (line 24) | def _check_equal(test_case, lhs, rhs, rtol=1e-3, atol=1e-3): function _test_pow_grad_grad_impl (line 35) | def _test_pow_grad_grad_impl(test_case): class TestPowHigherDerivative (line 65) | class TestPowHigherDerivative(flow.unittest.TestCase): method test_pow_grad_grad (line 66) | def test_pow_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_scalar_pow.py function _check_equal (line 24) | def _check_equal(test_case, lhs, rhs, rtol=1e-4, atol=1e-4, name=""): function _test_scalar_pow_grad_grad_impl (line 35) | def _test_scalar_pow_grad_grad_impl(test_case, reverse=False): class TestScalarPowHigherDerivative (line 53) | class TestScalarPowHigherDerivative(flow.unittest.TestCase): method test_scalar_pow_grad_grad (line 54) | def test_scalar_pow_grad_grad(test_case): method test_scalar_reverse_pow_grad_grad (line 58) | def test_scalar_reverse_pow_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_slice.py function random_index (line 24) | def random_index(dim): function random_slice (line 33) | def random_slice(dim_vec): function _test_slice_grad_grad_impl (line 38) | def _test_slice_grad_grad_impl(test_case): class TestSliceHigherDerivative (line 67) | class TestSliceHigherDerivative(flow.unittest.TestCase): method test_slice_grad_grad (line 68) | def test_slice_grad_grad(test_case): FILE: python/oneflow/test/modules/test_higher_derivative_softmax.py function _assert_true (line 24) | def _assert_true(test_case, value1, value2): function _test_softmax_grad_grad_impl (line 35) | def _test_softmax_grad_grad_impl(test_case, op_name): class TestSoftmaxHigherDerivative (line 59) | class TestSoftmaxHigherDerivative(flow.unittest.TestCase): method test_softmax_grad_grad (line 60) | def test_softmax_grad_grad(test_case): method test_logsoftmax_grad_grad (line 63) | def test_logsoftmax_grad_grad(test_case): FILE: python/oneflow/test/modules/test_host_memory_input.py class TestHostMemory (line 26) | class TestHostMemory(oneflow.unittest.TestCase): method test_host_memory (line 28) | def test_host_memory(test_case): method test_host_memory_1n2d (line 55) | def test_host_memory_1n2d(test_case): FILE: python/oneflow/test/modules/test_hsplit.py class TestHsplitVec (line 25) | class TestHsplitVec(flow.unittest.TestCase): method test_flow_hsplit_vec (line 27) | def test_flow_hsplit_vec(test_case): method test_flow_hsplit_vec_with_stride (line 40) | def test_flow_hsplit_vec_with_stride(test_case): class TestHsplitInt (line 57) | class TestHsplitInt(flow.unittest.TestCase): method test_flow_hsplit_int (line 59) | def test_flow_hsplit_int(test_case): FILE: python/oneflow/test/modules/test_hub.py class TestHub (line 24) | class TestHub(flow.unittest.TestCase): method test_hub_list_api (line 25) | def test_hub_list_api(test_case): method test_hub_help_api (line 30) | def test_hub_help_api(test_case): method test_hub_load_api (line 34) | def test_hub_load_api(test_case): method test_hub_download_url_to_file__api (line 41) | def test_hub_download_url_to_file__api(test_case): method test_hub_load_state_dict_from_url_api (line 47) | def test_hub_load_state_dict_from_url_api(test_case): FILE: python/oneflow/test/modules/test_image_batch_align.py function _read_images_by_cv (line 28) | def _read_images_by_cv(image_files): function _get_images_static_shape (line 33) | def _get_images_static_shape(images): function _roundup (line 46) | def _roundup(x, n): class TestImageBatchAlign (line 51) | class TestImageBatchAlign(flow.unittest.TestCase): method test_image_batch_align (line 52) | def test_image_batch_align(test_case): FILE: python/oneflow/test/modules/test_image_decode.py class TestImageDecode (line 27) | class TestImageDecode(flow.unittest.TestCase): method test_image_decode (line 28) | def test_image_decode(test_case): FILE: python/oneflow/test/modules/test_image_flip.py function _of_image_flip (line 26) | def _of_image_flip(images, image_static_shape, flip_code): function _read_images_by_cv (line 33) | def _read_images_by_cv(image_files): function _get_images_static_shape (line 38) | def _get_images_static_shape(images): function _compare_image_flip_with_cv (line 52) | def _compare_image_flip_with_cv(test_case, image_files): class TestImageFlip (line 69) | class TestImageFlip(flow.unittest.TestCase): method test_image_flip (line 70) | def test_image_flip(test_case): FILE: python/oneflow/test/modules/test_image_normalize.py function _of_image_normalize (line 26) | def _of_image_normalize(images, image_static_shape, std, mean): function _read_images_by_cv (line 39) | def _read_images_by_cv(image_files): function _get_images_static_shape (line 44) | def _get_images_static_shape(images): function _compare_image_normalize (line 58) | def _compare_image_normalize(test_case, image_files, std, mean): class TestImageNormalize (line 74) | class TestImageNormalize(flow.unittest.TestCase): method test_image_normalize (line 75) | def test_image_normalize(test_case): FILE: python/oneflow/test/modules/test_image_resize.py function _of_image_resize (line 28) | def _of_image_resize( function _get_resize_size_and_scale (line 76) | def _get_resize_size_and_scale( function _cv_image_resize (line 110) | def _cv_image_resize( function _test_image_resize_with_cv (line 138) | def _test_image_resize_with_cv( class TestImageResize (line 191) | class TestImageResize(flow.unittest.TestCase): method test_image_resize_to_fixed_size (line 192) | def test_image_resize_to_fixed_size(test_case): method test_image_resize_shorter_to_target_size (line 198) | def test_image_resize_shorter_to_target_size(test_case): method test_image_resize_longer_to_target_size (line 208) | def test_image_resize_longer_to_target_size(test_case): method test_image_resize_shorter_to_target_size_with_max_size (line 218) | def test_image_resize_shorter_to_target_size_with_max_size(test_case): method test_image_resize_longer_to_target_size_with_min_size (line 229) | def test_image_resize_longer_to_target_size_with_min_size(test_case): method test_image_resize_to_fixed_size_with_dtype_uint8 (line 240) | def test_image_resize_to_fixed_size_with_dtype_uint8(test_case): method test_image_reisze_shorter_to_target_size_with_max_size_with_dtype_uint8 (line 250) | def test_image_reisze_shorter_to_target_size_with_max_size_with_dtype_... method test_image_resize_uint8_to_float (line 264) | def test_image_resize_uint8_to_float(test_case): FILE: python/oneflow/test/modules/test_in_top_k.py function _topk_np (line 27) | def _topk_np(input, k, dim: int = -1, largest: bool = True, _sorted: boo... function _in_top_k_np (line 68) | def _in_top_k_np(targets, predictions, k): function _test_in_top_k_impl (line 82) | def _test_in_top_k_impl(test_case, shape, k, device): class TestInTopK (line 102) | class TestInTopK(flow.unittest.TestCase): method test_in_top_k (line 103) | def test_in_top_k(test_case): FILE: python/oneflow/test/modules/test_index_add.py function _test_index_add (line 27) | def _test_index_add(test_case, device): class TestIndexAdd (line 63) | class TestIndexAdd(flow.unittest.TestCase): method test_index_add (line 64) | def test_index_add(test_case): method profile_index_add (line 72) | def profile_index_add(test_case): FILE: python/oneflow/test/modules/test_index_select.py class TestIndexSelect (line 28) | class TestIndexSelect(flow.unittest.TestCase): method test_index_select_by_random (line 30) | def test_index_select_by_random(test_case): method test_index_select_bool_by_random (line 61) | def test_index_select_bool_by_random(test_case): FILE: python/oneflow/test/modules/test_info.py function _test_finfo (line 24) | def _test_finfo(test_case, dtype): class TestTypeInfo (line 41) | class TestTypeInfo(flow.unittest.TestCase): method test_iinfo (line 42) | def test_iinfo(test_case): method test_finfo (line 49) | def test_finfo(test_case): FILE: python/oneflow/test/modules/test_initializer.py class DataChecker (line 25) | class DataChecker: method __init__ (line 35) | def __init__(self, **kwargs): method __call__ (line 41) | def __call__(self, test_case, tensor): class TestInitializer (line 169) | class TestInitializer(flow.unittest.TestCase): method test_initializer (line 170) | def test_initializer(test_case): FILE: python/oneflow/test/modules/test_instancenorm.py function _test_instancenorm1d (line 29) | def _test_instancenorm1d(test_case, device): function _test_instancenorm2d (line 71) | def _test_instancenorm2d(test_case, device): function _test_instancenorm3d (line 141) | def _test_instancenorm3d(test_case, device): function _test_instancenorm1d_backward (line 267) | def _test_instancenorm1d_backward(test_case, device): function _test_instancenorm2d_backward (line 295) | def _test_instancenorm2d_backward(test_case, device): function _test_instancenorm3d_backward (line 337) | def _test_instancenorm3d_backward(test_case, device): class TestInstanceNorm (line 408) | class TestInstanceNorm(flow.unittest.TestCase): method test_instancenorm (line 409) | def test_instancenorm(test_case): method test_instancenorm_with_random_data (line 426) | def test_instancenorm_with_random_data(test_case): method test_instancenorm_with_random_data2 (line 444) | def test_instancenorm_with_random_data2(test_case): method test_instancenorm_with_random_data3 (line 463) | def test_instancenorm_with_random_data3(test_case): FILE: python/oneflow/test/modules/test_interpolate.py function _test_interpolate_linear_1d (line 27) | def _test_interpolate_linear_1d(test_case, device): function _test_interpolate_nearest_1d (line 75) | def _test_interpolate_nearest_1d(test_case, device): function _test_interpolate_nearest_2d (line 91) | def _test_interpolate_nearest_2d(test_case, device): function _test_interpolate_nearest_3d (line 118) | def _test_interpolate_nearest_3d(test_case, device): function _test_interpolate_bilinear_2d (line 165) | def _test_interpolate_bilinear_2d(test_case, device): function _test_interpolate_bicubic_2d (line 192) | def _test_interpolate_bicubic_2d(test_case, device): function _test_interpolate_bicubic_same_dim_2d (line 219) | def _test_interpolate_bicubic_same_dim_2d(test_case, device): function _test_interpolate_trilinear_3d (line 235) | def _test_interpolate_trilinear_3d(test_case, device): function _test_interpolate_trilinear_3d_align_corners (line 282) | def _test_interpolate_trilinear_3d_align_corners(test_case, device): function _test_interpolate_area_1d (line 368) | def _test_interpolate_area_1d(test_case, device): function _test_interpolate_area_2d (line 412) | def _test_interpolate_area_2d(test_case, device): function _test_interpolate_area_3d (line 481) | def _test_interpolate_area_3d(test_case, device): function _test_interpolate_output_size_arg_with_scalar (line 656) | def _test_interpolate_output_size_arg_with_scalar(test_case, device): class TestInterpolate (line 678) | class TestInterpolate(flow.unittest.TestCase): method test_interpolate (line 679) | def test_interpolate(test_case): FILE: python/oneflow/test/modules/test_inv.py class TestLinalgInv (line 25) | class TestLinalgInv(flow.unittest.TestCase): method test_inv_3by3_with_random_data (line 28) | def test_inv_3by3_with_random_data(test_case): method test_inv_batch_3by3_with_random_data (line 34) | def test_inv_batch_3by3_with_random_data(test_case): method test_inv_random_square_with_random_data (line 40) | def test_inv_random_square_with_random_data(test_case): method profile_linalg_inv (line 47) | def profile_linalg_inv(test_case): FILE: python/oneflow/test/modules/test_isclose.py function _perturbate (line 29) | def _perturbate(x): class TestIsClose (line 43) | class TestIsClose(flow.unittest.TestCase): method test_isclose_with_random_data (line 45) | def test_isclose_with_random_data(test_case): method test_isclose_with_0dim_data (line 53) | def test_isclose_with_0dim_data(test_case): method test_tensor_isclose_with_random_data (line 61) | def test_tensor_isclose_with_random_data(test_case): method test_isclose_broadcast (line 69) | def test_isclose_broadcast(test_case): FILE: python/oneflow/test/modules/test_jit_script_api.py function _test_jit_script_api (line 29) | def _test_jit_script_api(test_case): function _test_jit_ignore_api (line 39) | def _test_jit_ignore_api(test_case): class TestJitScriptApi (line 50) | class TestJitScriptApi(flow.unittest.TestCase): method test_jit_script (line 51) | def test_jit_script(test_case): FILE: python/oneflow/test/modules/test_layer_norm.py function _layer_norm (line 26) | def _layer_norm(x, normalized_shape, weight=None, bias=None, eps=1e-6): function _test_layer_norm (line 48) | def _test_layer_norm( class TestLayerNorm (line 177) | class TestLayerNorm(flow.unittest.TestCase): method test_no_affine (line 178) | def test_no_affine(test_case): method test_warp_impl (line 183) | def test_warp_impl(test_case): method test_block_smem_impl (line 192) | def test_block_smem_impl(test_case): method test_block_uncached_impl (line 202) | def test_block_uncached_impl(test_case): FILE: python/oneflow/test/modules/test_lerp.py class TestLerp (line 24) | class TestLerp(flow.unittest.TestCase): method test_lerp_with_broadcast_data (line 26) | def test_lerp_with_broadcast_data(test_case): method test_lerp_with_random_data (line 34) | def test_lerp_with_random_data(test_case): method test_tesnor_lerp_with_random_data (line 44) | def test_tesnor_lerp_with_random_data(test_case): method test_tesnor_inplace_lerp_with_random_data (line 52) | def test_tesnor_inplace_lerp_with_random_data(test_case): method profile_lerp (line 60) | def profile_lerp(test_case): FILE: python/oneflow/test/modules/test_less.py function _test_less_normal (line 29) | def _test_less_normal(test_case, device): function _test_less_symbol (line 41) | def _test_less_symbol(test_case, device): function _test_less_int_scalar (line 57) | def _test_less_int_scalar(test_case, device): function _test_less_int_tensor_int_scalr (line 66) | def _test_less_int_tensor_int_scalr(test_case, device): function _test_less_float_scalar (line 75) | def _test_less_float_scalar(test_case, device): class TestLess (line 85) | class TestLess(flow.unittest.TestCase): method test_less (line 86) | def test_less(test_case): method test_less_with_random_data (line 100) | def test_less_with_random_data(test_case): method test_less_with_0dim_data (line 109) | def test_less_with_0dim_data(test_case): method test_tensor_less_with_random_data (line 118) | def test_tensor_less_with_random_data(test_case): method test_less_bool_with_random_data (line 128) | def test_less_bool_with_random_data(test_case): method test_tensor_less_with_0dim_data (line 141) | def test_tensor_less_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_less_equal.py function _test_less_equal_normal (line 27) | def _test_less_equal_normal(test_case, device): function _test_less_equal_symbol (line 39) | def _test_less_equal_symbol(test_case, device): function _test_less_equal_int_scalar (line 55) | def _test_less_equal_int_scalar(test_case, device): function _test_less_equal_int_tensor_int_scalr (line 64) | def _test_less_equal_int_tensor_int_scalr(test_case, device): function _test_less_equal_float_scalar (line 73) | def _test_less_equal_float_scalar(test_case, device): class TestLessEqual (line 83) | class TestLessEqual(flow.unittest.TestCase): method test_less_equal (line 84) | def test_less_equal(test_case): FILE: python/oneflow/test/modules/test_linalg_cross.py class TestLinalgCross (line 25) | class TestLinalgCross(flow.unittest.TestCase): method test_cross_with_random_data_default_dim (line 75) | def test_cross_with_random_data_default_dim(test_case): FILE: python/oneflow/test/modules/test_linear.py function _test_linear_no_bias (line 29) | def _test_linear_no_bias(test_case, device): function _test_linear_with_bias (line 54) | def _test_linear_with_bias(test_case, device): function _test_linear_3_dimension_input (line 83) | def _test_linear_3_dimension_input(test_case, device): function _test_linear_4_dimension_input (line 100) | def _test_linear_4_dimension_input(test_case, device): function _test_identity (line 113) | def _test_identity(test_case, device): function _test_linear_backward_with_bias (line 123) | def _test_linear_backward_with_bias(test_case, device): class TestLinear (line 162) | class TestLinear(flow.unittest.TestCase): method test_linear_forward (line 163) | def test_linear_forward(test_case): method test_linear_backward (line 176) | def test_linear_backward(test_case): method test_linear_with_random_data (line 184) | def test_linear_with_random_data(test_case): method test_linear_with_device_and_dtype (line 197) | def test_linear_with_device_and_dtype(test_case): method test_nn_functional_linear_with_random_data (line 214) | def test_nn_functional_linear_with_random_data(test_case): method test_nn_functional_bias_linear_with_random_data (line 223) | def test_nn_functional_bias_linear_with_random_data(test_case): method test_identity_with_random_data (line 234) | def test_identity_with_random_data(test_case): FILE: python/oneflow/test/modules/test_linspace.py class TestLinspace (line 30) | class TestLinspace(flow.unittest.TestCase): method test_linspace_int_with_random_data (line 32) | def test_linspace_int_with_random_data(test_case): method test_linspace_float_with_random_data (line 42) | def test_linspace_float_with_random_data(test_case): method test_linspace_with_scalar_tensor_as_params (line 52) | def test_linspace_with_scalar_tensor_as_params(test_case): method test_global_naive (line 59) | def test_global_naive(test_case): method test_linspace_in_transformer_bug (line 66) | def test_linspace_in_transformer_bug(test_case): method test_linspace_start_equal_end_bug (line 204) | def test_linspace_start_equal_end_bug(test_case): FILE: python/oneflow/test/modules/test_log1p.py class TestLog1pModule (line 29) | class TestLog1pModule(flow.unittest.TestCase): method test_log1p_with_random_data (line 31) | def test_log1p_with_random_data(test_case): method test_log1p_with_0dim_data (line 37) | def test_log1p_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_logaddexp.py class TestLogAddExpModule (line 25) | class TestLogAddExpModule(flow.unittest.TestCase): method test_log_add_exp_against_pytorch (line 27) | def test_log_add_exp_against_pytorch(test_case): method test_log_add_exp_with_0dim_tensor (line 37) | def test_log_add_exp_with_0dim_tensor(test_case): method test_tensor_log_add_exp_against_pytorch (line 45) | def test_tensor_log_add_exp_against_pytorch(test_case): method test_tensor_log_add_exp_with_0dim_tensor (line 55) | def test_tensor_log_add_exp_with_0dim_tensor(test_case): FILE: python/oneflow/test/modules/test_logical_and.py function _test_logical_and (line 28) | def _test_logical_and(test_case, shape, dtype, device): function _test_tensor_logical_and (line 43) | def _test_tensor_logical_and(test_case, shape, dtype, device): function _test_tensor_scalar_logical_and (line 53) | def _test_tensor_scalar_logical_and(test_case, shape, scalar, dtype, dev... class TestLogicalAndModule (line 62) | class TestLogicalAndModule(flow.unittest.TestCase): method test_logical_and (line 63) | def test_logical_and(test_case): method test_scalar_logical_and (line 75) | def test_scalar_logical_and(test_case): method test_logical_and_with_random_data (line 86) | def test_logical_and_with_random_data(test_case): method test_logical_and_bool_with_random_data (line 95) | def test_logical_and_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_logical_not.py function _test_logical_not (line 25) | def _test_logical_not(test_case, shape, device): function _test_tensor_logical_not (line 33) | def _test_tensor_logical_not(test_case, shape, device): class TestLogicalNotModule (line 42) | class TestLogicalNotModule(flow.unittest.TestCase): method test_logical_not (line 43) | def test_logical_not(test_case): method test_logical_not_with_random_data (line 55) | def test_logical_not_with_random_data(test_case): method test_logical_not_bool_with_random_data (line 63) | def test_logical_not_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_logical_or.py function _test_logical_or (line 28) | def _test_logical_or(test_case, shape, device): function _test_tensor_logical_or (line 38) | def _test_tensor_logical_or(test_case, shape, device): function _test_tensor_scalar_logical_or (line 48) | def _test_tensor_scalar_logical_or(test_case, shape, scalar, dtype, devi... class TestLogicalOrModule (line 57) | class TestLogicalOrModule(flow.unittest.TestCase): method test_logical_or (line 58) | def test_logical_or(test_case): method test_scalar_logical_or (line 69) | def test_scalar_logical_or(test_case): method test_logical_or_with_random_data (line 80) | def test_logical_or_with_random_data(test_case): method test_logical_or_bool_with_random_data (line 89) | def test_logical_or_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_logical_reduce.py class TestLogicalReduce (line 29) | class TestLogicalReduce(flow.unittest.TestCase): method test_sum_with_random_data (line 31) | def test_sum_with_random_data(test_case): method test_mean_with_random_data (line 38) | def test_mean_with_random_data(test_case): method test_all_with_random_data (line 45) | def test_all_with_random_data(test_case): method test_any_with_random_data (line 52) | def test_any_with_random_data(test_case): method test_prod_with_random_data (line 59) | def test_prod_with_random_data(test_case): method test_sum_keepdim_with_random_data (line 66) | def test_sum_keepdim_with_random_data(test_case): method test_mean_keepdim_with_random_data (line 73) | def test_mean_keepdim_with_random_data(test_case): method test_all_keepdim_with_random_data (line 80) | def test_all_keepdim_with_random_data(test_case): method test_any_keepdim_with_random_data (line 87) | def test_any_keepdim_with_random_data(test_case): method test_prod_keepdim_with_random_data (line 94) | def test_prod_keepdim_with_random_data(test_case): method test_scalar_reduce_sum_with_random_data (line 101) | def test_scalar_reduce_sum_with_random_data(test_case): method test_scalar_reduce_mean_with_random_data (line 107) | def test_scalar_reduce_mean_with_random_data(test_case): method test_scalar_reduce_all_with_random_data (line 113) | def test_scalar_reduce_all_with_random_data(test_case): method test_scalar_reduce_any_with_random_data (line 119) | def test_scalar_reduce_any_with_random_data(test_case): method test_scalar_reduce_prod_with_random_data (line 125) | def test_scalar_reduce_prod_with_random_data(test_case): method test_all_bool_input_with_random_data (line 131) | def test_all_bool_input_with_random_data(test_case): method test_max_bool_input_with_random_data (line 140) | def test_max_bool_input_with_random_data(test_case): method test_min_bool_input_with_random_data (line 149) | def test_min_bool_input_with_random_data(test_case): method test_any_bool_input_with_random_data (line 158) | def test_any_bool_input_with_random_data(test_case): method test_reduce_all_0dim_tensor (line 167) | def test_reduce_all_0dim_tensor(test_case): method test_reduce_all_0size_tensor (line 173) | def test_reduce_all_0size_tensor(test_case): FILE: python/oneflow/test/modules/test_logical_xor.py function _test_logical_xor_int (line 27) | def _test_logical_xor_int(test_case, shape, device): function _test_logical_xor_float (line 37) | def _test_logical_xor_float(test_case, shape, device): function _test_tensor_logical_xor_int (line 47) | def _test_tensor_logical_xor_int(test_case, shape, device): function _test_tensor_logical_xor_float (line 57) | def _test_tensor_logical_xor_float(test_case, shape, device): function _test_tensor_scalar_logical_xor (line 67) | def _test_tensor_scalar_logical_xor(test_case, shape, scalar, dtype, dev... class TestLogicalXorModule (line 76) | class TestLogicalXorModule(flow.unittest.TestCase): method test_logical_xor (line 77) | def test_logical_xor(test_case): method test_scalar_logical_xor (line 90) | def test_scalar_logical_xor(test_case): method test_logical_xor_with_random_data (line 101) | def test_logical_xor_with_random_data(test_case): method test_logical_xor_bool_with_random_data (line 110) | def test_logical_xor_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_logspace.py class TestLogspace (line 24) | class TestLogspace(flow.unittest.TestCase): method test_logspace_int_with_random_data (line 26) | def test_logspace_int_with_random_data(test_case): method test_logspace_float_with_random_data (line 36) | def test_logspace_float_with_random_data(test_case): method test_logspace_with_random_base (line 46) | def test_logspace_with_random_base(test_case): method test_global_naive (line 56) | def test_global_naive(test_case): FILE: python/oneflow/test/modules/test_logsumexp.py class TestLogSumExpModule (line 30) | class TestLogSumExpModule(flow.unittest.TestCase): method test_log_sum_exp_against_pytorch (line 32) | def test_log_sum_exp_against_pytorch(test_case): method test_log_sum_exp_with_big_value (line 40) | def test_log_sum_exp_with_big_value(test_case): method test_log_sum_exp_with_0_size_tensor (line 47) | def test_log_sum_exp_with_0_size_tensor(test_case): method test_log_sum_exp_with_0dim_tensor (line 54) | def test_log_sum_exp_with_0dim_tensor(test_case): method test_tensor_log_sum_exp_against_pytorch (line 61) | def test_tensor_log_sum_exp_against_pytorch(test_case): method test_tensor_log_sum_exp_with_big_value (line 69) | def test_tensor_log_sum_exp_with_big_value(test_case): method test_tensor_log_sum_exp_with_0_size_tensor (line 76) | def test_tensor_log_sum_exp_with_0_size_tensor(test_case): method test_tensor_log_sum_exp_with_0dim_tensor (line 83) | def test_tensor_log_sum_exp_with_0dim_tensor(test_case): FILE: python/oneflow/test/modules/test_loss.py function generate_necessity_for_cross_entropy_or_nll_loss (line 29) | def generate_necessity_for_cross_entropy_or_nll_loss(dim: int, prob: boo... function generate_necessity_for_bce_loss (line 65) | def generate_necessity_for_bce_loss(dim: int): function _test_cross_entropy_loss (line 99) | def _test_cross_entropy_loss(dim: int, prob: bool = False): function _test_nn_functional_cross_entropy_loss (line 123) | def _test_nn_functional_cross_entropy_loss(dim: int, prob: bool): class TestCrossEntropyLossModule (line 137) | class TestCrossEntropyLossModule(flow.unittest.TestCase): method test_cross_entropy_loss_with_random_data_dim_2 (line 139) | def test_cross_entropy_loss_with_random_data_dim_2(test_case): method test_cross_entropy_loss_with_random_data_dim_3 (line 143) | def test_cross_entropy_loss_with_random_data_dim_3(test_case): method test_cross_entropy_loss_with_random_data_dim_4 (line 147) | def test_cross_entropy_loss_with_random_data_dim_4(test_case): method test_cross_entropy_loss_with_random_data_dim_5 (line 151) | def test_cross_entropy_loss_with_random_data_dim_5(test_case): method test_nn_functional_cross_entropy_with_random_data_dim (line 155) | def test_nn_functional_cross_entropy_with_random_data_dim(test_case): method test_cross_entropy_prob_loss_with_random_data_dim_2 (line 161) | def test_cross_entropy_prob_loss_with_random_data_dim_2(test_case): method test_cross_entropy_prob_loss_with_random_data_dim_3 (line 165) | def test_cross_entropy_prob_loss_with_random_data_dim_3(test_case): method test_cross_entropy_prob_loss_with_random_data_dim_4 (line 170) | def test_cross_entropy_prob_loss_with_random_data_dim_4(test_case): method test_cross_entropy_prob_loss_with_random_data_dim_5 (line 175) | def test_cross_entropy_prob_loss_with_random_data_dim_5(test_case): method test_nn_functional_prob_cross_entropy_with_random_data_dim (line 179) | def test_nn_functional_prob_cross_entropy_with_random_data_dim(test_ca... function _test_nll_loss (line 184) | def _test_nll_loss(dim=int): class TestNLLLossModule (line 205) | class TestNLLLossModule(flow.unittest.TestCase): method test_nll_loss_with_random_data_dim_2 (line 207) | def test_nll_loss_with_random_data_dim_2(test_case): method test_nll_loss_with_random_data_dim_3 (line 211) | def test_nll_loss_with_random_data_dim_3(test_case): method test_nll_loss_with_random_data_dim_4 (line 215) | def test_nll_loss_with_random_data_dim_4(test_case): method test_nll_loss_with_random_data_dim_5 (line 219) | def test_nll_loss_with_random_data_dim_5(test_case): function _test_bce_loss (line 223) | def _test_bce_loss(dim=int, with_logits: bool = False): function _test_nn_functional_binary_cross_entropy (line 246) | def _test_nn_functional_binary_cross_entropy(dim=int): function _test_nn_functional_binary_cross_entropy_with_logits (line 258) | def _test_nn_functional_binary_cross_entropy_with_logits(dim=int): function _test_nn_functional_binary_cross_entropy_with_logits_different_dtype_float_first (line 269) | def _test_nn_functional_binary_cross_entropy_with_logits_different_dtype... function _test_nn_functional_binary_cross_entropy_with_logits_different_dtype_double_first (line 302) | def _test_nn_functional_binary_cross_entropy_with_logits_different_dtype... class TestBCELossModule (line 336) | class TestBCELossModule(flow.unittest.TestCase): method test_bce_loss_with_random_data_dim_2 (line 338) | def test_bce_loss_with_random_data_dim_2(test_case): method test_bce_loss_with_random_data_dim_3 (line 342) | def test_bce_loss_with_random_data_dim_3(test_case): method test_bce_loss_with_random_data_dim_4 (line 346) | def test_bce_loss_with_random_data_dim_4(test_case): method test_bce_loss_with_random_data_dim_5 (line 350) | def test_bce_loss_with_random_data_dim_5(test_case): method test_nn_functional_binary_cross_entropy (line 354) | def test_nn_functional_binary_cross_entropy(test_case): class TestBCEWithLogitsLossModule (line 360) | class TestBCEWithLogitsLossModule(flow.unittest.TestCase): method test_bce_with_logits_loss_with_random_data_dim_2 (line 362) | def test_bce_with_logits_loss_with_random_data_dim_2(test_case): method test_bce_with_logits_loss_with_random_data_dim_3 (line 366) | def test_bce_with_logits_loss_with_random_data_dim_3(test_case): method test_bce_with_logits_loss_with_random_data_dim_4 (line 370) | def test_bce_with_logits_loss_with_random_data_dim_4(test_case): method test_bce_with_logits_loss_with_random_data_dim_5 (line 374) | def test_bce_with_logits_loss_with_random_data_dim_5(test_case): method test_nn_functional_binary_cross_entropy_with_logits (line 378) | def test_nn_functional_binary_cross_entropy_with_logits(test_case): method test_nn_functional_binary_cross_entropy_with_logits_different_dtype (line 383) | def test_nn_functional_binary_cross_entropy_with_logits_different_dtyp... class TestL1LossModule (line 398) | class TestL1LossModule(flow.unittest.TestCase): method test_l1_loss_with_random_data (line 400) | def test_l1_loss_with_random_data(test_case): method _test_nn_functional_l1_loss (line 415) | def _test_nn_functional_l1_loss(test_case): class TestSmoothL1LossModule (line 429) | class TestSmoothL1LossModule(flow.unittest.TestCase): method test_smooth_l1_loss_with_random_data (line 431) | def test_smooth_l1_loss_with_random_data(test_case): class TestMSELossModule (line 449) | class TestMSELossModule(flow.unittest.TestCase): method test_mse_loss_with_random_data (line 451) | def test_mse_loss_with_random_data(test_case): method _test_nn_functional_mse_loss (line 466) | def _test_nn_functional_mse_loss(test_case): class TestKLDivLossModule (line 480) | class TestKLDivLossModule(flow.unittest.TestCase): method test_kldiv_loss_with_random_data (line 482) | def test_kldiv_loss_with_random_data(test_case): method test_nn_functional_kl_div (line 502) | def test_nn_functional_kl_div(test_case): class TestMarginRankingLossModule (line 519) | class TestMarginRankingLossModule(flow.unittest.TestCase): method test_margin_ranking_loss_with_random_data (line 521) | def test_margin_ranking_loss_with_random_data(test_case): FILE: python/oneflow/test/modules/test_loss_global.py function get_sbp (line 25) | def get_sbp(device: str): function compare_loss (line 32) | def compare_loss(device_type, dim, reduction, cls, data_generator): function generate_necessity_default (line 59) | def generate_necessity_default(dim: int, device: str, placement, sbp): function generate_necessity_for_cross_entropy_or_nll_loss (line 75) | def generate_necessity_for_cross_entropy_or_nll_loss( class TestBCELossOrWithLogitsConsistent (line 94) | class TestBCELossOrWithLogitsConsistent(flow.unittest.TestCase): method test_bce_loss (line 96) | def test_bce_loss(testcase): class TestCrossEntropyOrNllLossConsistent (line 107) | class TestCrossEntropyOrNllLossConsistent(flow.unittest.TestCase): method test_cross_entropy_loss_or_nll_loss (line 109) | def test_cross_entropy_loss_or_nll_loss(testcase): class TestKLDivLossConsistent (line 120) | class TestKLDivLossConsistent(flow.unittest.TestCase): method test_kl_div_loss (line 122) | def test_kl_div_loss(testcase): class TestSmoothL1LossConsistent (line 133) | class TestSmoothL1LossConsistent(flow.unittest.TestCase): method test_smooth_l1_loss (line 135) | def test_smooth_l1_loss(testcase): FILE: python/oneflow/test/modules/test_lr_scheduler.py function compare_with_torch_reduce_lr (line 32) | def compare_with_torch_reduce_lr( class TestLrScheduler (line 79) | class TestLrScheduler(flow.unittest.TestCase): method test_cosine_decay_lr (line 82) | def test_cosine_decay_lr(test_case): method test_cosine_annealing_lr (line 109) | def test_cosine_annealing_lr(test_case): method test_step_lr (line 142) | def test_step_lr(test_case): method test_multistep_lr (line 160) | def test_multistep_lr(test_case): method test_exponential_lr (line 182) | def test_exponential_lr(test_case): method test_lambda_lr (line 199) | def test_lambda_lr(test_case): method test_polynomial_lr (line 222) | def test_polynomial_lr(test_case): method test_reduce_lr_on_plateau (line 269) | def test_reduce_lr_on_plateau(test_case): method test_warmup_scheduler_save_and_load (line 282) | def test_warmup_scheduler_save_and_load(test_case): class WarmupLRTestCase (line 322) | class WarmupLRTestCase(flow.unittest.TestCase): method test_only_warmup (line 323) | def test_only_warmup(test_case): method test_warmup_iters_0_exp_lr (line 354) | def test_warmup_iters_0_exp_lr(test_case): method test_linear_warmup_exp_lr (line 376) | def test_linear_warmup_exp_lr(test_case): method test_linear_warmup_prefix_exp_lr (line 409) | def test_linear_warmup_prefix_exp_lr(test_case): method test_constant_warmup_cosine_annealing (line 446) | def test_constant_warmup_cosine_annealing(test_case): method test_linear_warmup_cosine_annealing (line 490) | def test_linear_warmup_cosine_annealing(test_case): method test_linear_warmup_prefix_cosine_annealing (line 533) | def test_linear_warmup_prefix_cosine_annealing(test_case): method test_linear_warmup_multistep_lr (line 580) | def test_linear_warmup_multistep_lr(test_case): method test_linear_warmup_prefix_multistep_lr (line 622) | def test_linear_warmup_prefix_multistep_lr(test_case): class ConstantLRTestCase (line 673) | class ConstantLRTestCase(flow.unittest.TestCase): method test (line 674) | def test(test_case): class LinearLRTestCase (line 713) | class LinearLRTestCase(flow.unittest.TestCase): method test (line 714) | def test(test_case): method test_end_factor (line 752) | def test_end_factor(test_case): class ChainedSchedulerTestCase (line 792) | class ChainedSchedulerTestCase(flow.unittest.TestCase): method test (line 793) | def test(test_case): class CosineAnnealingWarmRestartsTestCase (line 814) | class CosineAnnealingWarmRestartsTestCase(flow.unittest.TestCase): method test_mult_1 (line 815) | def test_mult_1(test_case): method test_mult_2 (line 835) | def test_mult_2(test_case): method test_mult_2_decay_half_limit_2 (line 855) | def test_mult_2_decay_half_limit_2(test_case): FILE: python/oneflow/test/modules/test_masked_fill.py class TestMaskedFill (line 26) | class TestMaskedFill(flow.unittest.TestCase): method test_flow_masked_fill_with_random_data (line 28) | def test_flow_masked_fill_with_random_data(test_case): method test_flow_masked_fill_with_0dim_data (line 38) | def test_flow_masked_fill_with_0dim_data(test_case): method test_flow_masked_fill_broadcast_with_random_data (line 46) | def test_flow_masked_fill_broadcast_with_random_data(test_case): method test_flow_masked_fill_int_with_random_data (line 56) | def test_flow_masked_fill_int_with_random_data(test_case): method test_flow_masked_fill_bool_with_random_data (line 66) | def test_flow_masked_fill_bool_with_random_data(test_case): method test_flow_masked_fill_inplace_with_random_data (line 78) | def test_flow_masked_fill_inplace_with_random_data(test_case): FILE: python/oneflow/test/modules/test_masked_select.py function _test_masked_select (line 27) | def _test_masked_select(test_case, device): function _test_masked_select_broadcast (line 44) | def _test_masked_select_broadcast(test_case, device): function _test_masked_select_input_zero (line 85) | def _test_masked_select_input_zero(test_case, device): class TestMaskedSelect (line 126) | class TestMaskedSelect(flow.unittest.TestCase): method test_masked_select (line 127) | def test_masked_select(test_case): method test_masked_select_broadcast (line 138) | def test_masked_select_broadcast(test_case): FILE: python/oneflow/test/modules/test_math_op_higher_derivative.py function _test_math_op_grad_grad_impl (line 24) | def _test_math_op_grad_grad_impl(test_case, op_name): class TestMathOpHigherDerivative (line 67) | class TestMathOpHigherDerivative(flow.unittest.TestCase): method test_sin_grad_grad (line 68) | def test_sin_grad_grad(test_case): method test_cos_grad_grad (line 71) | def test_cos_grad_grad(test_case): method test_tan_grad_grad (line 74) | def test_tan_grad_grad(test_case): method test_sinh_grad_grad (line 77) | def test_sinh_grad_grad(test_case): method test_cosh_grad_grad (line 80) | def test_cosh_grad_grad(test_case): method test_tanh_grad_grad (line 83) | def test_tanh_grad_grad(test_case): method test_asin_grad_grad (line 86) | def test_asin_grad_grad(test_case): method test_acos_grad_grad (line 89) | def test_acos_grad_grad(test_case): method test_atan_grad_grad (line 92) | def test_atan_grad_grad(test_case): method test_asinh_grad_grad (line 95) | def test_asinh_grad_grad(test_case): method test_acosh_grad_grad (line 98) | def test_acosh_grad_grad(test_case): method test_atanh_grad_grad (line 101) | def test_atanh_grad_grad(test_case): method test_erf_grad_grad (line 104) | def test_erf_grad_grad(test_case): method test_erfc_grad_grad (line 107) | def test_erfc_grad_grad(test_case): method test_exp_grad_grad (line 110) | def test_exp_grad_grad(test_case): method test_exp2_grad_grad (line 113) | def test_exp2_grad_grad(test_case): method test_expm1_grad_grad (line 116) | def test_expm1_grad_grad(test_case): method test_log_grad_grad (line 119) | def test_log_grad_grad(test_case): method test_logsigmoid_grad_grad (line 122) | def test_logsigmoid_grad_grad(test_case): method test_log2_grad_grad (line 125) | def test_log2_grad_grad(test_case): method test_log1p_grad_grad (line 128) | def test_log1p_grad_grad(test_case): method test_reciprocal_grad_grad (line 131) | def test_reciprocal_grad_grad(test_case): method test_rsqrt_grad_grad (line 134) | def test_rsqrt_grad_grad(test_case): method test_sqrt_grad_grad (line 137) | def test_sqrt_grad_grad(test_case): method test_square_grad_grad (line 140) | def test_square_grad_grad(test_case): method test_sigmoid_grad_grad (line 143) | def test_sigmoid_grad_grad(test_case): method test_abs_grad_grad (line 146) | def test_abs_grad_grad(test_case): FILE: python/oneflow/test/modules/test_math_ops.py class TestSinh (line 36) | class TestSinh(flow.unittest.TestCase): method test_flow_sinh_with_random_data (line 38) | def test_flow_sinh_with_random_data(test_case): class TestSin (line 46) | class TestSin(flow.unittest.TestCase): method test_flow_sin_with_random_data (line 48) | def test_flow_sin_with_random_data(test_case): class TestInplaceSin (line 56) | class TestInplaceSin(flow.unittest.TestCase): method test_flow_inplace_sin_with_random_data (line 58) | def test_flow_inplace_sin_with_random_data(test_case): function _test_cos (line 66) | def _test_cos(test_case, shape, device): function _test_cos_backward (line 75) | def _test_cos_backward(test_case, shape, device): class TestCos (line 90) | class TestCos(flow.unittest.TestCase): method test_cos (line 91) | def test_cos(test_case): class TestLogModule (line 101) | class TestLogModule(flow.unittest.TestCase): method test_log_with_random_data (line 103) | def test_log_with_random_data(test_case): class TestSqrt (line 110) | class TestSqrt(flow.unittest.TestCase): method test_sqrt_flow_with_random_data (line 112) | def test_sqrt_flow_with_random_data(test_case): method test_sqrt_tensor_with_random_data (line 119) | def test_sqrt_tensor_with_random_data(test_case): class TestExp (line 127) | class TestExp(flow.unittest.TestCase): method test_flow_exp_with_random_data (line 129) | def test_flow_exp_with_random_data(test_case): class TestExp2 (line 137) | class TestExp2(flow.unittest.TestCase): method test_flow_exp2_with_random_data (line 139) | def test_flow_exp2_with_random_data(test_case): class TestRsqrt (line 148) | class TestRsqrt(flow.unittest.TestCase): method test_rsqrt_flow_with_random_data (line 150) | def test_rsqrt_flow_with_random_data(test_case): class TestSquare (line 158) | class TestSquare(flow.unittest.TestCase): method test_square_flow_with_random_data (line 160) | def test_square_flow_with_random_data(test_case): method test_square_tensor_with_random_data (line 167) | def test_square_tensor_with_random_data(test_case): class TestPow (line 175) | class TestPow(flow.unittest.TestCase): method test_pow_float_scalar_with_random_data (line 177) | def test_pow_float_scalar_with_random_data(test_case): method test_pow_int_scalar_with_random_data (line 183) | def test_pow_int_scalar_with_random_data(test_case): method test_reverse_pow_int_scalar_with_random_data (line 190) | def test_reverse_pow_int_scalar_with_random_data(test_case): method test_symbolic_reverse_pow_with_random_data (line 197) | def test_symbolic_reverse_pow_with_random_data(test_case): method test_pow_elementwise_with_random_data (line 204) | def test_pow_elementwise_with_random_data(test_case): method test_pow_broadcast_with_random_data (line 211) | def test_pow_broadcast_with_random_data(test_case): method test_pow_broadcast_with_random_data_reverse (line 218) | def test_pow_broadcast_with_random_data_reverse(test_case): method test_scalar_pow_with_random_devices (line 225) | def test_scalar_pow_with_random_devices(test_case): method test_pow_scalar_with_random_data (line 360) | def test_pow_scalar_with_random_data(test_case): method test_pow_elementwise_with_random_data (line 367) | def test_pow_elementwise_with_random_data(test_case): method test_pow_broadcast_with_random_data (line 375) | def test_pow_broadcast_with_random_data(test_case): method test_elementwise_pow_random_data (line 541) | def test_elementwise_pow_random_data(test_case): method test_flow_broadcast_pow_with_random_data (line 549) | def test_flow_broadcast_pow_with_random_data(test_case): method test_tensor_pow_scalar_random_data (line 559) | def test_tensor_pow_scalar_random_data(test_case): class TestAsin (line 235) | class TestAsin(flow.unittest.TestCase): method test_flow_asin_with_random_data (line 237) | def test_flow_asin_with_random_data(test_case): method test_flow_arcsin_with_random_data (line 244) | def test_flow_arcsin_with_random_data(test_case): class TestAsinh (line 252) | class TestAsinh(flow.unittest.TestCase): method test_flow_asinh_with_random_data (line 254) | def test_flow_asinh_with_random_data(test_case): method test_flow_arcsinh_with_random_data (line 261) | def test_flow_arcsinh_with_random_data(test_case): class TestTan (line 269) | class TestTan(flow.unittest.TestCase): method test_flow_tan_with_random_data (line 271) | def test_flow_tan_with_random_data(test_case): class TestAtan (line 279) | class TestAtan(flow.unittest.TestCase): method test_flow_atan_with_random_data (line 281) | def test_flow_atan_with_random_data(test_case): method test_flow_arctan_with_random_data (line 288) | def test_flow_arctan_with_random_data(test_case): method test_flow_atan2_with_random_data (line 295) | def test_flow_atan2_with_random_data(test_case): method test_flow_atan2_with_1elem_data (line 303) | def test_flow_atan2_with_1elem_data(test_case): method test_flow_atanh_with_random_data (line 311) | def test_flow_atanh_with_random_data(test_case): method test_flow_arctanh_with_random_data (line 318) | def test_flow_arctanh_with_random_data(test_case): class TestTopk (line 326) | class TestTopk(flow.unittest.TestCase): method test_flow_topk_with_random_data (line 328) | def test_flow_topk_with_random_data(test_case): class TestTopkReturnValues (line 342) | class TestTopkReturnValues(flow.unittest.TestCase): method test_flow_topk_with_random_data (line 344) | def test_flow_topk_with_random_data(test_case): class TestPow (line 358) | class TestPow(flow.unittest.TestCase): method test_pow_float_scalar_with_random_data (line 177) | def test_pow_float_scalar_with_random_data(test_case): method test_pow_int_scalar_with_random_data (line 183) | def test_pow_int_scalar_with_random_data(test_case): method test_reverse_pow_int_scalar_with_random_data (line 190) | def test_reverse_pow_int_scalar_with_random_data(test_case): method test_symbolic_reverse_pow_with_random_data (line 197) | def test_symbolic_reverse_pow_with_random_data(test_case): method test_pow_elementwise_with_random_data (line 204) | def test_pow_elementwise_with_random_data(test_case): method test_pow_broadcast_with_random_data (line 211) | def test_pow_broadcast_with_random_data(test_case): method test_pow_broadcast_with_random_data_reverse (line 218) | def test_pow_broadcast_with_random_data_reverse(test_case): method test_scalar_pow_with_random_devices (line 225) | def test_scalar_pow_with_random_devices(test_case): method test_pow_scalar_with_random_data (line 360) | def test_pow_scalar_with_random_data(test_case): method test_pow_elementwise_with_random_data (line 367) | def test_pow_elementwise_with_random_data(test_case): method test_pow_broadcast_with_random_data (line 375) | def test_pow_broadcast_with_random_data(test_case): method test_elementwise_pow_random_data (line 541) | def test_elementwise_pow_random_data(test_case): method test_flow_broadcast_pow_with_random_data (line 549) | def test_flow_broadcast_pow_with_random_data(test_case): method test_tensor_pow_scalar_random_data (line 559) | def test_tensor_pow_scalar_random_data(test_case): class TestArccos (line 383) | class TestArccos(flow.unittest.TestCase): method test_arccos_flow_with_random_data (line 385) | def test_arccos_flow_with_random_data(test_case): class TestAcos (line 393) | class TestAcos(flow.unittest.TestCase): method test_acos_flow_with_random_data (line 395) | def test_acos_flow_with_random_data(test_case): class TestArccosh (line 403) | class TestArccosh(flow.unittest.TestCase): method test_arccosh_flow_with_random_data (line 405) | def test_arccosh_flow_with_random_data(test_case): class TestAcosh (line 413) | class TestAcosh(flow.unittest.TestCase): method test_acosh_flow_with_random_data (line 415) | def test_acosh_flow_with_random_data(test_case): class TestAtan2 (line 423) | class TestAtan2(flow.unittest.TestCase): method test_flow_atan2_with_random_data (line 425) | def test_flow_atan2_with_random_data(test_case): class TestMinimum (line 434) | class TestMinimum(flow.unittest.TestCase): method test_flow_elementwise_minimum_with_random_data (line 436) | def test_flow_elementwise_minimum_with_random_data(test_case): method test_flow_broadcast_minimum_with_random_data (line 445) | def test_flow_broadcast_minimum_with_random_data(test_case): class TestMaximum (line 456) | class TestMaximum(flow.unittest.TestCase): method test_flow_elementwise_mximum_with_random_data (line 458) | def test_flow_elementwise_mximum_with_random_data(test_case): method test_flow_broadcast_maximum_with_random_data (line 467) | def test_flow_broadcast_maximum_with_random_data(test_case): class TestFloorDiv (line 478) | class TestFloorDiv(flow.unittest.TestCase): method test_elementwise_floordiv_random_data (line 480) | def test_elementwise_floordiv_random_data(test_case): method test_tensor_floordiv_scalar_random_data (line 494) | def test_tensor_floordiv_scalar_random_data(test_case): class TestFmod (line 506) | class TestFmod(flow.unittest.TestCase): method test_elementwise_fmod_random_data (line 513) | def test_elementwise_fmod_random_data(test_case): method test_flow_broadcast_fmod_with_random_data (line 521) | def test_flow_broadcast_fmod_with_random_data(test_case): method test_tensor_fmod_scalar_random_data (line 531) | def test_tensor_fmod_scalar_random_data(test_case): class TestPow (line 539) | class TestPow(flow.unittest.TestCase): method test_pow_float_scalar_with_random_data (line 177) | def test_pow_float_scalar_with_random_data(test_case): method test_pow_int_scalar_with_random_data (line 183) | def test_pow_int_scalar_with_random_data(test_case): method test_reverse_pow_int_scalar_with_random_data (line 190) | def test_reverse_pow_int_scalar_with_random_data(test_case): method test_symbolic_reverse_pow_with_random_data (line 197) | def test_symbolic_reverse_pow_with_random_data(test_case): method test_pow_elementwise_with_random_data (line 204) | def test_pow_elementwise_with_random_data(test_case): method test_pow_broadcast_with_random_data (line 211) | def test_pow_broadcast_with_random_data(test_case): method test_pow_broadcast_with_random_data_reverse (line 218) | def test_pow_broadcast_with_random_data_reverse(test_case): method test_scalar_pow_with_random_devices (line 225) | def test_scalar_pow_with_random_devices(test_case): method test_pow_scalar_with_random_data (line 360) | def test_pow_scalar_with_random_data(test_case): method test_pow_elementwise_with_random_data (line 367) | def test_pow_elementwise_with_random_data(test_case): method test_pow_broadcast_with_random_data (line 375) | def test_pow_broadcast_with_random_data(test_case): method test_elementwise_pow_random_data (line 541) | def test_elementwise_pow_random_data(test_case): method test_flow_broadcast_pow_with_random_data (line 549) | def test_flow_broadcast_pow_with_random_data(test_case): method test_tensor_pow_scalar_random_data (line 559) | def test_tensor_pow_scalar_random_data(test_case): class TestAbsModule (line 567) | class TestAbsModule(flow.unittest.TestCase): method test_abs_with_random_data (line 569) | def test_abs_with_random_data(test_case): class TestCoshModule (line 576) | class TestCoshModule(flow.unittest.TestCase): method test_cosh_with_random_data (line 578) | def test_cosh_with_random_data(test_case): class TestLgammaModule (line 585) | class TestLgammaModule(flow.unittest.TestCase): method test_lgamma_with_random_data (line 587) | def test_lgamma_with_random_data(test_case): class TestLog2Module (line 594) | class TestLog2Module(flow.unittest.TestCase): method test_log2_with_random_data (line 596) | def test_log2_with_random_data(test_case): class TestLog10Module (line 603) | class TestLog10Module(flow.unittest.TestCase): method test_log10_with_random_data (line 605) | def test_log10_with_random_data(test_case): class TestDigammaModule (line 612) | class TestDigammaModule(flow.unittest.TestCase): method test_digamma_with_random_data (line 614) | def test_digamma_with_random_data(test_case): FILE: python/oneflow/test/modules/test_matmul.py class TestModule (line 28) | class TestModule(flow.unittest.TestCase): method test_flow_matmul_with_random_data (line 30) | def test_flow_matmul_with_random_data(test_case): method test_flow_tensor_matmul_with_random_data_allow_tf32 (line 39) | def test_flow_tensor_matmul_with_random_data_allow_tf32(test_case): method test_flow_tensor_matmul_with_random_data (line 52) | def test_flow_tensor_matmul_with_random_data(test_case): method test_flow_tensor_matmul_with_random_int_data (line 60) | def test_flow_tensor_matmul_with_random_int_data(test_case): method test_flow_tensor_matmul_with_random_fp16_data (line 75) | def test_flow_tensor_matmul_with_random_fp16_data(test_case): method test_flow_tensor_broadcast_matmul_with_random_data (line 89) | def test_flow_tensor_broadcast_matmul_with_random_data(test_case): method test_flow_tensor_x_broadcast_y_matmul (line 97) | def test_flow_tensor_x_broadcast_y_matmul(test_case): method test_flow_tensor_broadcast_matmul_with_same_dims (line 105) | def test_flow_tensor_broadcast_matmul_with_same_dims(test_case): method test_flow_mm_with_random_data (line 113) | def test_flow_mm_with_random_data(test_case): method test_flow_mv_with_random_data (line 122) | def test_flow_mv_with_random_data(test_case): method profile_mv (line 131) | def profile_mv(test_case): method test_flow_vector_matrix_product_with_random_data (line 135) | def test_flow_vector_matrix_product_with_random_data(test_case): FILE: python/oneflow/test/modules/test_max.py function _test_scalar_max (line 25) | def _test_scalar_max(test_case, device): class TestMaxModule (line 35) | class TestMaxModule(flow.unittest.TestCase): method test_scalar_max (line 36) | def test_scalar_max(test_case): method test_max_reduce_random_dim (line 40) | def test_max_reduce_random_dim(test_case): method test_max_reduce_all_dim (line 79) | def test_max_reduce_all_dim(test_case): method test_max_elementwise (line 86) | def test_max_elementwise(test_case): method test_max_elementwise_dtype_promotion (line 95) | def test_max_elementwise_dtype_promotion(test_case): method test_max_broadcast_dtype_promotion (line 104) | def test_max_broadcast_dtype_promotion(test_case): method test_max_with_diff_size (line 114) | def test_max_with_diff_size(test_case): method test_max_return_type (line 122) | def test_max_return_type(test_case): FILE: python/oneflow/test/modules/test_maxpool.py function _test_maxpool2d_channel_last (line 30) | def _test_maxpool2d_channel_last( class TestMaxPooling (line 62) | class TestMaxPooling(flow.unittest.TestCase): method test_maxpool1d_with_random_data (line 64) | def test_maxpool1d_with_random_data(test_case): method test_maxpool1d_with_2d_input_tensor (line 88) | def test_maxpool1d_with_2d_input_tensor(test_case): method test_maxpool2d_with_random_data (line 109) | def test_maxpool2d_with_random_data(test_case): method test_maxpool2d_with_3d_input_tensor (line 133) | def test_maxpool2d_with_3d_input_tensor(test_case): method test_maxpool2d_with_half_data (line 155) | def test_maxpool2d_with_half_data(test_case): method test_maxpool3d_with_random_data (line 180) | def test_maxpool3d_with_random_data(test_case): method test_maxpool3d_with_4d_input_tensor (line 206) | def test_maxpool3d_with_4d_input_tensor(test_case): method test_maxpool2d_channel_last (line 235) | def test_maxpool2d_channel_last(test_case): class TestMaxPoolingFunctional (line 253) | class TestMaxPoolingFunctional(flow.unittest.TestCase): method test_maxpool1d_with_random_data (line 255) | def test_maxpool1d_with_random_data(test_case): method test_maxpool2d_with_random_data (line 277) | def test_maxpool2d_with_random_data(test_case): method test_maxpool3d_with_random_data (line 299) | def test_maxpool3d_with_random_data(test_case): method profile_maxpool2d (line 323) | def profile_maxpool2d(test_case): FILE: python/oneflow/test/modules/test_maxunpool.py function _get_valid_output_size (line 29) | def _get_valid_output_size( function _test_module_unpoolnd (line 58) | def _test_module_unpoolnd(test_case, n): function _test_functional_unpoolnd (line 110) | def _test_functional_unpoolnd(test_case, n): class TestMaxUnpooling (line 167) | class TestMaxUnpooling(flow.unittest.TestCase): method test_max_unpool1d_with_random_data (line 169) | def test_max_unpool1d_with_random_data(test_case): method test_functional_max_unpool1d_with_random_data (line 173) | def test_functional_max_unpool1d_with_random_data(test_case): method test_max_unpool2d_with_random_data (line 177) | def test_max_unpool2d_with_random_data(test_case): method test_functional_max_unpool2d_with_random_data (line 181) | def test_functional_max_unpool2d_with_random_data(test_case): method test_max_unpool3d_with_random_data (line 185) | def test_max_unpool3d_with_random_data(test_case): method test_functional_max_unpool3d_with_random_data (line 189) | def test_functional_max_unpool3d_with_random_data(test_case): method profile_max_unpool1d (line 193) | def profile_max_unpool1d(test_case): method profile_max_unpool2d (line 203) | def profile_max_unpool2d(test_case): method profile_max_unpool3d (line 213) | def profile_max_unpool3d(test_case): FILE: python/oneflow/test/modules/test_mean.py function _test_mean (line 29) | def _test_mean(test_case, shape, device): function _test_mean_negative_dim (line 44) | def _test_mean_negative_dim(test_case, shape, device): function _test_mean_backward (line 55) | def _test_mean_backward(test_case, shape, device): class TestMean (line 69) | class TestMean(flow.unittest.TestCase): method test_mean (line 70) | def test_mean(test_case): method test_mean_with_random_data (line 83) | def test_mean_with_random_data(test_case): method test_mean_with_scalar_data (line 90) | def test_mean_with_scalar_data(test_case): method test_mean_with_float16_data (line 98) | def test_mean_with_float16_data(test_case): FILE: python/oneflow/test/modules/test_median.py class TestMedianModule (line 23) | class TestMedianModule(flow.unittest.TestCase): method test_median_reduce_all_dim (line 25) | def test_median_reduce_all_dim(test_case): method test_median_reduce_one_dim (line 32) | def test_median_reduce_one_dim(test_case): method test_median_reduce_one_dim_keepdim (line 40) | def test_median_reduce_one_dim_keepdim(test_case): method test_median_0size (line 48) | def test_median_0size(test_case): method test_median_reduce_one_dim_0size (line 54) | def test_median_reduce_one_dim_0size(test_case): method test_median_return_type (line 60) | def test_median_return_type(test_case): FILE: python/oneflow/test/modules/test_meshgrid.py function _test_meshgrid_forawd (line 29) | def _test_meshgrid_forawd(test_case, device, indexing): function _test_meshgrid_forawd_scalar (line 41) | def _test_meshgrid_forawd_scalar(test_case, device, indexing): function _test_meshgrid_forawd_3tensor (line 49) | def _test_meshgrid_forawd_3tensor(test_case, device, indexing): class TestMeshGridModule (line 67) | class TestMeshGridModule(flow.unittest.TestCase): method test_meshgrid (line 68) | def test_meshgrid(test_case): method test_meshgrid_with_random_data (line 82) | def test_meshgrid_with_random_data(test_case): method test_meshgrid_with_0dim_data (line 90) | def test_meshgrid_with_0dim_data(test_case): method test_meshgrid_with_random_data_xy (line 98) | def test_meshgrid_with_random_data_xy(test_case): method test_meshgrid_with_random_data_size (line 107) | def test_meshgrid_with_random_data_size(test_case): method test_meshgrid_tuple_list_with_random_data (line 114) | def test_meshgrid_tuple_list_with_random_data(test_case): FILE: python/oneflow/test/modules/test_min.py class TestMinModule (line 26) | class TestMinModule(flow.unittest.TestCase): method test_min_reduce_random_dim (line 28) | def test_min_reduce_random_dim(test_case): method test_min_reduce_all_dim (line 67) | def test_min_reduce_all_dim(test_case): method test_min_elementwise (line 74) | def test_min_elementwise(test_case): method test_min_elementwise_dtype_promotion (line 83) | def test_min_elementwise_dtype_promotion(test_case): method test_min_broadcast_dtype_promotion (line 92) | def test_min_broadcast_dtype_promotion(test_case): method test_min_return_type (line 102) | def test_min_return_type(test_case): FILE: python/oneflow/test/modules/test_min_max_observer.py function gen_quant_scale_for_min_max_symmetric (line 36) | def gen_quant_scale_for_min_max_symmetric(weight, quantization_bit): function gen_quant_scale_for_min_max_affine (line 42) | def gen_quant_scale_for_min_max_affine(weight, quantization_bit): function gen_quant_scale_for_min_max_cambricon (line 51) | def gen_quant_scale_for_min_max_cambricon(weight, quantization_bit): function product (line 57) | def product(tu): function _check_min_max_observer (line 61) | def _check_min_max_observer( function _run_test_min_max_observer (line 103) | def _run_test_min_max_observer( class TestMinMaxObserver (line 135) | class TestMinMaxObserver(flow.unittest.TestCase): method test_min_max_observer (line 136) | def test_min_max_observer(test_case): FILE: python/oneflow/test/modules/test_mock.py class TestMockModule (line 27) | class TestMockModule(flow.unittest.TestCase): method test_mock_device (line 28) | def test_mock_device(test_case): method test_mock_placement (line 32) | def test_mock_placement(test_case): FILE: python/oneflow/test/modules/test_mode.py class TestModeModule (line 23) | class TestModeModule(flow.unittest.TestCase): method test_mode_reduce_one_dim (line 25) | def test_mode_reduce_one_dim(test_case): method test_mode_reduce_one_dim_keepdim (line 33) | def test_mode_reduce_one_dim_keepdim(test_case): method test_mode_0size (line 41) | def test_mode_0size(test_case): method test_mode_reduce_one_dim_0size (line 47) | def test_mode_reduce_one_dim_0size(test_case): FILE: python/oneflow/test/modules/test_module.py function np_relu (line 36) | def np_relu(np_arr): function _test_hooks (line 40) | def _test_hooks(test_case, backward_register_fn): function _test_module_forward_preforward_hook_removable (line 111) | def _test_module_forward_preforward_hook_removable(test_case): function _test_module_forward_forward_hook_removable (line 142) | def _test_module_forward_forward_hook_removable(test_case): class TestModule (line 174) | class TestModule(flow.unittest.TestCase): method test_nested_module (line 176) | def test_nested_module(test_case): method test_relu (line 192) | def test_relu(test_case): method test_load_state_dict (line 200) | def test_load_state_dict(test_case): method test_state_dict (line 217) | def test_state_dict(test_case): method test_parameter (line 235) | def test_parameter(test_case): method test_module_forward (line 243) | def test_module_forward(test_case): method test_train_eval (line 258) | def test_train_eval(test_case): method test_module_setattr (line 267) | def test_module_setattr(test_case): method test_module_apply (line 298) | def test_module_apply(test_case): method test_module_cpu_cuda (line 317) | def test_module_cpu_cuda(test_case): method test_module_float_double (line 339) | def test_module_float_double(test_case): method test_moduledict (line 360) | def test_moduledict(test_case): method test_module_submodule (line 382) | def test_module_submodule(test_case): method test_module_get_parameter (line 399) | def test_module_get_parameter(test_case): method test_module_delattr (line 412) | def test_module_delattr(test_case): method test_hooks_register (line 426) | def test_hooks_register(test_case): method test_register_state_dict_hook_hook (line 433) | def test_register_state_dict_hook_hook(test_case): method test_full_backward_hook (line 462) | def test_full_backward_hook(test_case): method test_full_backward_hook_with_return_value (line 484) | def test_full_backward_hook_with_return_value(test_case): FILE: python/oneflow/test/modules/test_module_to.py class DummyModule (line 35) | class DummyModule(flow.nn.Module): method __init__ (line 36) | def __init__(self): method forward (line 42) | def forward(self, x): function _test_dummy_module (line 46) | def _test_dummy_module(test_case): function _test_dummy_module_to (line 59) | def _test_dummy_module_to(test_case): class TestModuleTo (line 87) | class TestModuleTo(flow.unittest.TestCase): method test_module_to_device (line 89) | def test_module_to_device(test_case): method test_module_to_dtype (line 95) | def test_module_to_dtype(test_case): method test_module_to_tensor (line 102) | def test_module_to_tensor(test_case): method test_module_to_with_var_reuse (line 112) | def test_module_to_with_var_reuse(test_case): FILE: python/oneflow/test/modules/test_module_to_global_or_local.py class TestModuleToGlobalOrLocal (line 29) | class TestModuleToGlobalOrLocal(flow.unittest.TestCase): method test_module_to_global (line 30) | def test_module_to_global(test_case): method test_module_to_local (line 62) | def test_module_to_local(test_case): FILE: python/oneflow/test/modules/test_module_to_half.py class TestModuleToHalf (line 23) | class TestModuleToHalf(flow.unittest.TestCase): method test_module_to_half (line 25) | def test_module_to_half(test_case): FILE: python/oneflow/test/modules/test_movedim.py class TestMovedim (line 25) | class TestMovedim(flow.unittest.TestCase): method test_flow_movedim_with_vector (line 27) | def test_flow_movedim_with_vector(test_case): method test_flow_movedim_with_stride (line 40) | def test_flow_movedim_with_stride(test_case): method test_flow_movedim_with_int (line 56) | def test_flow_movedim_with_int(test_case): FILE: python/oneflow/test/modules/test_moving_average_min_max_observer.py function gen_quant_scale_for_moving_average_min_max_symmetric (line 35) | def gen_quant_scale_for_moving_average_min_max_symmetric( function gen_quant_scale_for_moving_average_min_max_affine (line 48) | def gen_quant_scale_for_moving_average_min_max_affine( function gen_quant_scale_for_moving_average_min_max_cambricon (line 67) | def gen_quant_scale_for_moving_average_min_max_cambricon( function _check_moving_average_min_max_observer (line 79) | def _check_moving_average_min_max_observer( function _run_test_moving_average_min_max_observer (line 133) | def _run_test_moving_average_min_max_observer( class TestMovingAverageMinMaxObserver (line 184) | class TestMovingAverageMinMaxObserver(flow.unittest.TestCase): method test_moving_average_min_max_observer (line 185) | def test_moving_average_min_max_observer(test_case): FILE: python/oneflow/test/modules/test_mul.py function _test_mul_impl (line 30) | def _test_mul_impl(test_case, device): function inplace_mul_tensors_helper (line 125) | def inplace_mul_tensors_helper(test_case, device, arr_0, arr_y): function _test_inplace_mul_tensors (line 145) | def _test_inplace_mul_tensors(test_case, device): function _test_inplace_mul_scalar (line 151) | def _test_inplace_mul_scalar(test_case, device): function _test_mul_inplace_0size_tensor (line 180) | def _test_mul_inplace_0size_tensor(test_case, device): class TestMulModule (line 190) | class TestMulModule(flow.unittest.TestCase): method test_mul (line 191) | def test_mul(test_case): method test_broadcast_mul (line 204) | def test_broadcast_mul(test_case): method test_non_contiguous_inplace_mul (line 213) | def test_non_contiguous_inplace_mul(test_case): method test_scalar_mul_with_random_devices (line 222) | def test_scalar_mul_with_random_devices(test_case): FILE: python/oneflow/test/modules/test_multi_tensor_yolov5_weight_update.py function _test_multi_tensor_weight_update_impl (line 28) | def _test_multi_tensor_weight_update_impl(test_case, device, shape, n, d): class TestMultiTensorWeightUpdateModule (line 83) | class TestMultiTensorWeightUpdateModule(flow.unittest.TestCase): method test_multi_tensor_weight_update (line 84) | def test_multi_tensor_weight_update(test_case): FILE: python/oneflow/test/modules/test_multinomial.py function _test_multinomial (line 28) | def _test_multinomial(test_case, device, seed, replacement, dtype): class TestMultinomial (line 77) | class TestMultinomial(flow.unittest.TestCase): method test_multinomial (line 78) | def test_multinomial(test_case): FILE: python/oneflow/test/modules/test_nansum.py class TestNanSumModule (line 30) | class TestNanSumModule(flow.unittest.TestCase): method test_nansum_without_nan (line 32) | def test_nansum_without_nan(test_case): method test_nansum_with_partial_nan (line 39) | def test_nansum_with_partial_nan(test_case): method test_nansum_with_total_nan (line 48) | def test_nansum_with_total_nan(test_case): method test_nansum_with_partial_nan_dims (line 57) | def test_nansum_with_partial_nan_dims(test_case): method test_nansum_with_total_nan_dims (line 66) | def test_nansum_with_total_nan_dims(test_case): method test_sum_with_0_size_tensor (line 75) | def test_sum_with_0_size_tensor(test_case): method test_sum_with_0dim_tensor (line 82) | def test_sum_with_0dim_tensor(test_case): FILE: python/oneflow/test/modules/test_narrow.py class TestNarrow (line 29) | class TestNarrow(flow.unittest.TestCase): method test_flow_narrow_start_with_random_data (line 31) | def test_flow_narrow_start_with_random_data(test_case): method test_flow_narrow_length_with_random_data (line 41) | def test_flow_narrow_length_with_random_data(test_case): method test_flow_narrow_with_stride (line 51) | def test_flow_narrow_with_stride(test_case): method test_flow_narrow_start_bool_with_random_data (line 65) | def test_flow_narrow_start_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_ne.py function _test_ne (line 29) | def _test_ne(test_case, shape, device): function _test_tensor_ne_operator (line 43) | def _test_tensor_ne_operator(test_case, shape, device): function _test_ne_int (line 53) | def _test_ne_int(test_case, shape, device): function _test_tensor_ne_operator_int (line 62) | def _test_tensor_ne_operator_int(test_case, shape, device): function _test_ne_float (line 71) | def _test_ne_float(test_case, shape, device): function _test_tensor_ne_operator_float (line 80) | def _test_tensor_ne_operator_float(test_case, shape, device): class TestNe (line 90) | class TestNe(flow.unittest.TestCase): method test_ne (line 91) | def test_ne(test_case): method test_ne_with_0_size_data (line 107) | def test_ne_with_0_size_data(test_case): method test_ne_with_0dim_data (line 117) | def test_ne_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_negative.py class TestNegativeModule (line 26) | class TestNegativeModule(flow.unittest.TestCase): method test_ne_with_0_size_data (line 28) | def test_ne_with_0_size_data(test_case): method test_tensor_negative_with_random_data (line 37) | def test_tensor_negative_with_random_data(test_case): method test_negative_with_random_data (line 42) | def test_negative_with_random_data(test_case): method test_neg_with_random_data (line 48) | def test_neg_with_random_data(test_case): method test_tensor_negative_with_0dim_data (line 54) | def test_tensor_negative_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_nll_loss.py function _test_nll_loss (line 27) | def _test_nll_loss( class NLLLossTestCase (line 96) | class NLLLossTestCase(flow.unittest.TestCase): method test_local (line 97) | def test_local(test_case): method test_weighted (line 100) | def test_weighted(test_case): class ParallelNLLLossTestCase (line 105) | class ParallelNLLLossTestCase(flow.unittest.TestCase): method test_data_parallel (line 107) | def test_data_parallel(test_case): method test_data_parallel_weighted (line 111) | def test_data_parallel_weighted(test_case): method test_model_parallel (line 115) | def test_model_parallel(test_case): method test_model_parallel_weighted (line 119) | def test_model_parallel_weighted(test_case): class TowDParallelNLLLossTestCase (line 124) | class TowDParallelNLLLossTestCase(flow.unittest.TestCase): method test_2d_parallel (line 126) | def test_2d_parallel(test_case): method test_2d_parallel_weighted (line 130) | def test_2d_parallel_weighted(test_case): FILE: python/oneflow/test/modules/test_nms.py function box_area (line 24) | def box_area(boxes): function _box_inter_union_np (line 28) | def _box_inter_union_np(boxes1, boxes2): function box_iou_np (line 43) | def box_iou_np(boxes1, boxes2): function nms_np (line 49) | def nms_np(boxes, scores, iou_threshold): function create_tensors_with_iou (line 66) | def create_tensors_with_iou(N, iou_thresh): function _test_nms (line 79) | def _test_nms(test_case, device): class TestNMS (line 90) | class TestNMS(flow.unittest.TestCase): method test_nms (line 91) | def test_nms(test_case): FILE: python/oneflow/test/modules/test_noncontiguous_binary_op.py function _test_op (line 24) | def _test_op(test_case, x, y, inplace): function _test_noncontiguous_binary_op (line 44) | def _test_noncontiguous_binary_op(test_case, dtype, pack_size, ndims, in... class TestNonContiguousBinaryOp (line 70) | class TestNonContiguousBinaryOp(flow.unittest.TestCase): method test_noncontiguous_binary_op (line 71) | def test_noncontiguous_binary_op(test_case): FILE: python/oneflow/test/modules/test_nonzero.py function np_nonzero (line 29) | def np_nonzero(input, as_tuple): function _test_nonzero (line 36) | def _test_nonzero(test_case, shape, as_tuple, device): class TestNonzero (line 50) | class TestNonzero(flow.unittest.TestCase): method test_nonzero (line 51) | def test_nonzero(test_case): method test_nonzero_with_random_data (line 64) | def test_nonzero_with_random_data(test_case): method test_nonzero_bool_with_random_data (line 74) | def test_nonzero_bool_with_random_data(test_case): method test_half_nonzero_with_random_data (line 83) | def test_half_nonzero_with_random_data(test_case): method test_nonzero_with_0dim_data (line 95) | def test_nonzero_with_0dim_data(test_case): method test_nonzero_tuple_with_random_data (line 105) | def test_nonzero_tuple_with_random_data(test_case): FILE: python/oneflow/test/modules/test_norm.py function _np_vector_norm_backward (line 27) | def _np_vector_norm_backward(x, ord=2, dim=None): function _np_matrix_norm_backward (line 78) | def _np_matrix_norm_backward(x, ord="fro"): function _test_norm_1d (line 111) | def _test_norm_1d(test_case, device): function _test_norm_2d (line 132) | def _test_norm_2d(test_case, device): function _test_norm_Nd (line 153) | def _test_norm_Nd(test_case, device): function _test_fro_order_norm_backward (line 171) | def _test_fro_order_norm_backward(test_case, device): function _test_1d_inf_order_norm_backward (line 184) | def _test_1d_inf_order_norm_backward(test_case, device): function _test_2d_inf_order_norm_backward (line 198) | def _test_2d_inf_order_norm_backward(test_case, device): function _test_1d_digits_order_norm_backward (line 212) | def _test_1d_digits_order_norm_backward(test_case, device): function _test_2d_digits_order_norm_backward (line 226) | def _test_2d_digits_order_norm_backward(test_case, device): function _test_linalg_norm_shape_not_match (line 240) | def _test_linalg_norm_shape_not_match(test_case, device): class TestNormModule (line 248) | class TestNormModule(flow.unittest.TestCase): method test_norm (line 249) | def test_norm(test_case): method test_no_dim_no_ord_norm_with_random_data (line 267) | def test_no_dim_no_ord_norm_with_random_data(test_case): method test_one_dim_norm_with_random_data (line 276) | def test_one_dim_norm_with_random_data(test_case): method test_no_dim_one_shape_norm_with_random_data (line 288) | def test_no_dim_one_shape_norm_with_random_data(test_case): method test_no_dim_two_shape_norm_with_random_data (line 299) | def test_no_dim_two_shape_norm_with_random_data(test_case): method test_tuple_dim_norm_with_random_data (line 308) | def test_tuple_dim_norm_with_random_data(test_case): method test_vector_norm_only_zero_with_random_data (line 318) | def test_vector_norm_only_zero_with_random_data(test_case): method test_ord_random_data (line 327) | def test_ord_random_data(test_case): FILE: python/oneflow/test/modules/test_normalize.py function _test_functional_normalize_double_dtype (line 28) | def _test_functional_normalize_double_dtype(test_case, device, dtype): class TestFunctionalNormalize (line 43) | class TestFunctionalNormalize(flow.unittest.TestCase): method test_functional_normalize_naive (line 44) | def test_functional_normalize_naive(test_case): method test_functional_normalize (line 54) | def test_functional_normalize(test_case): FILE: python/oneflow/test/modules/test_ofrecord_reader.py class OFRecordDataLoader (line 24) | class OFRecordDataLoader(flow.nn.Module): method __init__ (line 25) | def __init__(self, batch_size, device=None, placement=None, sbp=None): method forward (line 51) | def forward(self): class DataLoaderGraph (line 59) | class DataLoaderGraph(flow.nn.Graph): method __init__ (line 60) | def __init__(self, loader): method build (line 64) | def build(self): class DistributedOFRecordReaderTestCase (line 71) | class DistributedOFRecordReaderTestCase(oneflow.unittest.TestCase): method test (line 72) | def test(test_case): FILE: python/oneflow/test/modules/test_one_embedding_adagrad.py function compare_with_numpy_adagrad (line 33) | def compare_with_numpy_adagrad( class TestOptimizers (line 176) | class TestOptimizers(flow.unittest.TestCase): method test_one_embedding_adagrad (line 177) | def test_one_embedding_adagrad(test_case): FILE: python/oneflow/test/modules/test_one_embedding_adam.py function compare_with_numpy_adam (line 33) | def compare_with_numpy_adam( class TestOptimizers (line 248) | class TestOptimizers(flow.unittest.TestCase): method test_one_embedding_adam (line 250) | def test_one_embedding_adam(test_case): FILE: python/oneflow/test/modules/test_one_embedding_ftrl.py function compare_with_numpy_ftrl (line 32) | def compare_with_numpy_ftrl( class TestOptimizers (line 212) | class TestOptimizers(flow.unittest.TestCase): method test_ftrl (line 214) | def test_ftrl(test_case): FILE: python/oneflow/test/modules/test_one_embedding_sgd.py function compare_with_numpy_sgd (line 33) | def compare_with_numpy_sgd( class TestOptimizers (line 215) | class TestOptimizers(flow.unittest.TestCase): method test_one_embedding_sgd (line 216) | def test_one_embedding_sgd(test_case): FILE: python/oneflow/test/modules/test_one_hot.py function _test_one_hot (line 27) | def _test_one_hot(test_case, device, num_classes, size, on_value, off_va... class TestOnehot (line 40) | class TestOnehot(flow.unittest.TestCase): method test_onehot (line 41) | def test_onehot(test_case): method test_one_hot_scalar (line 55) | def test_one_hot_scalar(test_case): FILE: python/oneflow/test/modules/test_ones_like.py function _test_ones_like_float (line 27) | def _test_ones_like_float(test_case, shape, device): function _test_ones_like_int (line 39) | def _test_ones_like_int(test_case, shape, device): class TestModule (line 50) | class TestModule(flow.unittest.TestCase): method test_ones_like (line 51) | def test_ones_like(test_case): FILE: python/oneflow/test/modules/test_optim_adadelta.py function compare_with_numpy_adadelta (line 30) | def compare_with_numpy_adadelta( function compare_with_numpy_adadelta_clip_grad (line 107) | def compare_with_numpy_adadelta_clip_grad( class TestAdadelta (line 200) | class TestAdadelta(flow.unittest.TestCase): method test_adadelta (line 201) | def test_adadelta(test_case): method test_adadelta_clip_grad (line 218) | def test_adadelta_clip_grad(test_case): FILE: python/oneflow/test/modules/test_optim_adagrad.py function compare_with_numpy_adagrad (line 30) | def compare_with_numpy_adagrad( function compare_with_numpy_adagrad_clip_grad (line 109) | def compare_with_numpy_adagrad_clip_grad( class TestAdagrad (line 200) | class TestAdagrad(flow.unittest.TestCase): method test_adagrad (line 201) | def test_adagrad(test_case): method test_adagrad_clip_grad (line 218) | def test_adagrad_clip_grad(test_case): FILE: python/oneflow/test/modules/test_optim_adam.py function compare_with_numpy_adam (line 31) | def compare_with_numpy_adam( function compare_with_numpy_adam_clip_grad (line 166) | def compare_with_numpy_adam_clip_grad( class TestAdam (line 306) | class TestAdam(flow.unittest.TestCase): method test_adam (line 307) | def test_adam(test_case): method test_adam_clip_grad (line 327) | def test_adam_clip_grad(test_case): FILE: python/oneflow/test/modules/test_optim_adamw.py function compare_with_numpy_adamw (line 31) | def compare_with_numpy_adamw( function compare_with_numpy_adamw_clip_grad (line 162) | def compare_with_numpy_adamw_clip_grad( class TestAdamW (line 301) | class TestAdamW(flow.unittest.TestCase): method test_adamw (line 302) | def test_adamw(test_case): method test_adamw_clip_grad (line 321) | def test_adamw_clip_grad(test_case): FILE: python/oneflow/test/modules/test_optim_add_param_group.py function _test_sgd_add_param_group (line 22) | def _test_sgd_add_param_group(test_case): class TestAddParamGroup (line 43) | class TestAddParamGroup(flow.unittest.TestCase): method test_sgd_add_param_group (line 44) | def test_sgd_add_param_group(test_case): FILE: python/oneflow/test/modules/test_optim_ftrl.py function compare_with_numpy_ftrl (line 29) | def compare_with_numpy_ftrl( function compare_with_numpy_ftrl_clip_grad (line 122) | def compare_with_numpy_ftrl_clip_grad( class Testftrl (line 224) | class Testftrl(flow.unittest.TestCase): method test_ftrl (line 225) | def test_ftrl(test_case): method test_ftrl_clip_grad (line 243) | def test_ftrl_clip_grad(test_case): FILE: python/oneflow/test/modules/test_optim_lamb.py function compare_with_numpy_lamb (line 28) | def compare_with_numpy_lamb( class TestLamb (line 158) | class TestLamb(flow.unittest.TestCase): method test_lamb (line 159) | def test_lamb(test_case): FILE: python/oneflow/test/modules/test_optim_lbfgs.py function _quadratic_interpolate (line 29) | def _quadratic_interpolate(x1, f1, g1, x2, f2, g2, bounds=None): function _strong_wolfe (line 42) | def _strong_wolfe( function compare_with_numpy_lbfgs (line 143) | def compare_with_numpy_lbfgs( class TestLBFGS (line 370) | class TestLBFGS(flow.unittest.TestCase): method test_lbfgs_numpy (line 371) | def test_lbfgs_numpy(test_case): FILE: python/oneflow/test/modules/test_optim_rmsprop.py function compare_with_numpy_rmsprop (line 29) | def compare_with_numpy_rmsprop( function compare_with_numpy_rmsprop_clip_grad (line 122) | def compare_with_numpy_rmsprop_clip_grad( class TestRMSProp (line 223) | class TestRMSProp(flow.unittest.TestCase): method test_rmsprop (line 224) | def test_rmsprop(test_case): method test_rmsprop_clip_grad (line 241) | def test_rmsprop_clip_grad(test_case): FILE: python/oneflow/test/modules/test_optim_sgd.py function compare_with_numpy_sgd (line 32) | def compare_with_numpy_sgd( function compare_with_numpy_sgd_clip_grad (line 152) | def compare_with_numpy_sgd_clip_grad( class TestOptimizers (line 288) | class TestOptimizers(flow.unittest.TestCase): method test_sgd (line 289) | def test_sgd(test_case): method test_sgd_clip_grad (line 308) | def test_sgd_clip_grad(test_case): method test_eager_global_zero_grad_sbp (line 332) | def test_eager_global_zero_grad_sbp(test_case): FILE: python/oneflow/test/modules/test_pairwise_distance.py class TestPairwiseDistance (line 25) | class TestPairwiseDistance(flow.unittest.TestCase): method test_pairwise_distance_module_with_random_data (line 27) | def test_pairwise_distance_module_with_random_data(test_case): method test_pairwise_distance_module_with_nonequal_dim_random_data (line 37) | def test_pairwise_distance_module_with_nonequal_dim_random_data(test_c... method test_pairwise_distance_functional_with_random_data (line 47) | def test_pairwise_distance_functional_with_random_data(test_case): method test_pairwise_distance_functional_with_nonequal_dim_random_data (line 55) | def test_pairwise_distance_functional_with_nonequal_dim_random_data(te... FILE: python/oneflow/test/modules/test_param_group.py class TestParamGroup (line 29) | class TestParamGroup(flow.unittest.TestCase): method test_ParamGroup (line 30) | def test_ParamGroup(test_case): FILE: python/oneflow/test/modules/test_parameters_grouping.py function np_allclose_with_shape (line 27) | def np_allclose_with_shape(a, b, *args, **kwargs): function module_grouping (line 31) | def module_grouping(test_case, device): function direct_grouping (line 63) | def direct_grouping(test_case, device): function global_grouping (line 87) | def global_grouping(test_case, device): function multi_module_grad (line 103) | def multi_module_grad(test_case, device): function multi_module_lifecycle (line 144) | def multi_module_lifecycle(test_case, device): class TestCPG (line 173) | class TestCPG(flow.unittest.TestCase): method test_cpg (line 174) | def test_cpg(test_case): FILE: python/oneflow/test/modules/test_parital_fc.py class TestParitalFC (line 26) | class TestParitalFC(flow.unittest.TestCase): method test_parital_fc (line 28) | def test_parital_fc(test_case): FILE: python/oneflow/test/modules/test_pixel_shuffle.py function _np_pixel_shuffle (line 29) | def _np_pixel_shuffle(input, h_factor, w_factor): function _np_pixel_shuffle_grad (line 42) | def _np_pixel_shuffle_grad(input, h_factor, w_factor): function _test_pixel_shuffle_impl (line 51) | def _test_pixel_shuffle_impl( class TestPixelShuffleModule (line 72) | class TestPixelShuffleModule(flow.unittest.TestCase): method test_pixel_shuffle (line 73) | def test_pixel_shuffle(test_case): method test_pixel_shuffle_with_random_data (line 89) | def test_pixel_shuffle_with_random_data(test_case): FILE: python/oneflow/test/modules/test_prelu.py class TestPReLU (line 30) | class TestPReLU(flow.unittest.TestCase): method test_prelu_4dim_module_with_random_data (line 32) | def test_prelu_4dim_module_with_random_data(test_case): method test_prelu_4dim_default_alpha_module_with_random_data (line 44) | def test_prelu_4dim_default_alpha_module_with_random_data(test_case): method test_prelu_2dim_module_with_random_data (line 54) | def test_prelu_2dim_module_with_random_data(test_case): FILE: python/oneflow/test/modules/test_prod.py class TestReduceProd (line 24) | class TestReduceProd(flow.unittest.TestCase): method test_reduce_prod_without_dim (line 26) | def test_reduce_prod_without_dim(test_case): method test_reduce_prod_with_dim (line 35) | def test_reduce_prod_with_dim(test_case): method test_reduce_prod_bool_without_dim (line 46) | def test_reduce_prod_bool_without_dim(test_case): method test_reduce_prod_with_dtype (line 55) | def test_reduce_prod_with_dtype(test_case): FILE: python/oneflow/test/modules/test_pruning.py class TestPrune (line 35) | class TestPrune(flow.unittest.TestCase): method test_validate_pruning_amount_init (line 36) | def test_validate_pruning_amount_init(self): method test_validate_pruning_amount (line 69) | def test_validate_pruning_amount(self): method test_compute_nparams_to_prune (line 90) | def test_compute_nparams_to_prune(self): method test_random_pruning_sizes (line 102) | def test_random_pruning_sizes(self): method test_random_pruning_orig (line 130) | def test_random_pruning_orig(self): method test_random_pruning_new_weight (line 152) | def test_random_pruning_new_weight(self): method test_identity_pruning (line 176) | def test_identity_pruning(self): method test_random_pruning_0perc (line 210) | def test_random_pruning_0perc(self): method test_random_pruning (line 250) | def test_random_pruning(self): method test_random_pruning_forward (line 288) | def test_random_pruning_forward(self): method test_remove_pruning_forward (line 313) | def test_remove_pruning_forward(self): method test_pruning_id_consistency (line 339) | def test_pruning_id_consistency(self): method test_random_pruning_pickle (line 354) | def test_random_pruning_pickle(self): method test_multiple_pruning_calls (line 365) | def test_multiple_pruning_calls(self): method test_pruning_container (line 399) | def test_pruning_container(self): method test_pruning_container_compute_mask (line 424) | def test_pruning_container_compute_mask(self): method test_l1_unstructured_pruning (line 470) | def test_l1_unstructured_pruning(self): method test_l1_unstructured_pruning_with_importance_scores (line 494) | def test_l1_unstructured_pruning_with_importance_scores(self): method test_unstructured_pruning_same_magnitude (line 527) | def test_unstructured_pruning_same_magnitude(self): method test_random_structured_pruning_amount (line 543) | def test_random_structured_pruning_amount(self): method test_ln_structured_pruning (line 556) | def test_ln_structured_pruning(self): method test_ln_structured_pruning_importance_scores (line 583) | def test_ln_structured_pruning_importance_scores(self): method test_remove_pruning (line 623) | def test_remove_pruning(self): method test_remove_pruning_exception (line 650) | def test_remove_pruning_exception(self): method test_global_pruning (line 665) | def test_global_pruning(self): method test_global_pruning_importance_scores (line 695) | def test_global_pruning_importance_scores(self): method test_custom_from_mask_pruning (line 736) | def test_custom_from_mask_pruning(self): method test_pruning_rollback (line 759) | def test_pruning_rollback(self): method test_pruning_serialization_model (line 783) | def test_pruning_serialization_model(self): method test_prune (line 816) | def test_prune(self): method test_prune_importance_scores (line 829) | def test_prune_importance_scores(self): method test_prune_importance_scores_mimic_default (line 845) | def test_prune_importance_scores_mimic_default(self): method test_rnn_pruning (line 873) | def test_rnn_pruning(self): FILE: python/oneflow/test/modules/test_qat_conv_modules.py function _test_qat_conv1d (line 25) | def _test_qat_conv1d( function _test_qat_conv2d (line 70) | def _test_qat_conv2d( function _test_qat_conv3d (line 115) | def _test_qat_conv3d( class TestQatModules (line 162) | class TestQatModules(flow.unittest.TestCase): method test_qat_conv1d (line 164) | def test_qat_conv1d(test_case): method test_qat_conv2d (line 177) | def test_qat_conv2d(test_case): method test_qat_conv3d (line 190) | def test_qat_conv3d(test_case): FILE: python/oneflow/test/modules/test_quantile.py function _test_quantile (line 28) | def _test_quantile(test_cast, q): class TestQuantile (line 37) | class TestQuantile(flow.unittest.TestCase): method test_quantile (line 38) | def test_quantile(test_case): FILE: python/oneflow/test/modules/test_quantization.py function gen_quant_scale_for_min_max_symmetric (line 35) | def gen_quant_scale_for_min_max_symmetric(weight, quantization_bit): function gen_quant_scale_for_min_max_affine (line 41) | def gen_quant_scale_for_min_max_affine(weight, quantization_bit): function gen_quant_scale_for_min_max_cambricon (line 50) | def gen_quant_scale_for_min_max_cambricon(weight, quantization_bit): function product (line 56) | def product(tu): function quant_per_layer_symmetric (line 60) | def quant_per_layer_symmetric(input, quantization_bit, scale): function quant_per_layer_affine (line 66) | def quant_per_layer_affine(input, quantization_bit, scale, zero_point): function quant_per_layer_cambricon (line 72) | def quant_per_layer_cambricon(input, quantization_bit, shift): function _check_quantize (line 79) | def _check_quantize( function _run_test_quantize (line 132) | def _run_test_quantize( class TestQuantize (line 172) | class TestQuantize(flow.unittest.TestCase): method test_quantize (line 173) | def test_quantize(test_case): FILE: python/oneflow/test/modules/test_quick_gelu.py class QuickGELUActivation (line 28) | class QuickGELUActivation(torch.nn.Module): method forward (line 33) | def forward(self, input: torch.Tensor) -> torch.Tensor: function _test_quick_gelu (line 37) | def _test_quick_gelu(test_case, device): class TestModule (line 55) | class TestModule(flow.unittest.TestCase): method test_quick_gelu (line 56) | def test_quick_gelu(test_case): FILE: python/oneflow/test/modules/test_rand.py function _test_rand (line 28) | def _test_rand(test_case, device, shape): function _test_rand_tuple_shape (line 37) | def _test_rand_tuple_shape(test_case, device, shape): function _test_0d_rand (line 45) | def _test_0d_rand(test_case, device, shape): function _test_different_dtype (line 54) | def _test_different_dtype(test_case, device, shape): function _test_backward (line 64) | def _test_backward(test_case, device, shape): function _test_with_generator (line 71) | def _test_with_generator(test_case, device, shape): function _test_rand_with_flow_size (line 84) | def _test_rand_with_flow_size(test_case, device, shape): class TestRandModule (line 93) | class TestRandModule(flow.unittest.TestCase): method test_0d_randint (line 94) | def test_0d_randint(test_case): method test_cases (line 102) | def test_cases(test_case): method test_half_rand (line 118) | def test_half_rand(test_case): class TestRandOnNonDefaultDevice (line 127) | class TestRandOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 128) | def test_non_default_device(test_case): FILE: python/oneflow/test/modules/test_randint.py function _test_randint (line 27) | def _test_randint(test_case, device, shape, low, high): function _test_0d_randint (line 34) | def _test_0d_randint(test_case, device, shape, low, high): function _test_different_dtype (line 43) | def _test_different_dtype(test_case, device, shape, low, high): function _test_with_generator (line 57) | def _test_with_generator(test_case, device, shape, low, high): function _test_high (line 70) | def _test_high(test_case, device, shape, low, high): function _test_0rank (line 77) | def _test_0rank(test_case, device, shape, low, high): class TestRandint (line 83) | class TestRandint(flow.unittest.TestCase): method test_global_different_types (line 84) | def test_global_different_types(test_case): method test_randint (line 99) | def test_randint(test_case): method test_0d_randint (line 113) | def test_0d_randint(test_case): method test_high_randint (line 123) | def test_high_randint(test_case): method test_0rank_randint (line 133) | def test_0rank_randint(test_case): class TestRandintOnNonDefaultDevice (line 146) | class TestRandintOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 147) | def test_non_default_device(test_case): FILE: python/oneflow/test/modules/test_randint_like.py function _test_randint_like (line 27) | def _test_randint_like(test_case, device, shape, low, high): function _test_0d_randint_like (line 35) | def _test_0d_randint_like(test_case, device, shape, low, high): function _test_different_dtype (line 45) | def _test_different_dtype(test_case, device, shape, low, high): function _test_with_generator (line 60) | def _test_with_generator(test_case, device, shape, low, high): function _test_high (line 75) | def _test_high(test_case, device, shape, low, high): function _test_0rank (line 83) | def _test_0rank(test_case, device, shape, low, high): class TestRandIntLike (line 90) | class TestRandIntLike(flow.unittest.TestCase): method test_global_different_types (line 91) | def test_global_different_types(test_case): method test_randint_like (line 107) | def test_randint_like(test_case): method test_0d_randint_like (line 121) | def test_0d_randint_like(test_case): method test_high_randint_like (line 131) | def test_high_randint_like(test_case): method test_0rank_randint_like (line 141) | def test_0rank_randint_like(test_case): class TestRandIntLikeOnNonDefaultDevice (line 154) | class TestRandIntLikeOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 155) | def test_non_default_device(test_case): FILE: python/oneflow/test/modules/test_randn.py function _test_randn (line 29) | def _test_randn(test_case, device, shape): function _test_0d_rand (line 37) | def _test_0d_rand(test_case, device, shape): function _test_different_dtype (line 46) | def _test_different_dtype(test_case, device, shape): function _test_backward (line 56) | def _test_backward(test_case, device, shape): function _test_with_generator (line 65) | def _test_with_generator(test_case, device, shape): function _test_randn_tuple_shape (line 78) | def _test_randn_tuple_shape(test_case, device, shape): function _test_randn_with_flow_size (line 86) | def _test_randn_with_flow_size(test_case, device, shape): class TestRandnModule (line 95) | class TestRandnModule(flow.unittest.TestCase): method test_global_naive (line 96) | def test_global_naive(test_case): method test_randn (line 103) | def test_randn(test_case): method test_0d_randn (line 119) | def test_0d_randn(test_case): method test_half_randn (line 129) | def test_half_randn(test_case): method test_randn_layout_param (line 137) | def test_randn_layout_param(test_case): class TestRandnOnNonDefaultDevice (line 143) | class TestRandnOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 144) | def test_non_default_device(test_case): method test_with_generator (line 148) | def test_with_generator(test_case): FILE: python/oneflow/test/modules/test_randn_like.py function _test_randn_like (line 27) | def _test_randn_like(test_case, device, shape): function _test_0d_randn_like (line 33) | def _test_0d_randn_like(test_case, device, shape): function _test_different_dtype (line 39) | def _test_different_dtype(test_case, device, shape): function _test_with_generator (line 51) | def _test_with_generator(test_case, device, shape): function _test_0rank (line 66) | def _test_0rank(test_case, device, shape): class TestRandIntLike (line 73) | class TestRandIntLike(flow.unittest.TestCase): method test_global_different_types (line 74) | def test_global_different_types(test_case): method test_randn_like (line 89) | def test_randn_like(test_case): method test_0d_randn_like (line 101) | def test_0d_randn_like(test_case): method test_0rank_randn_like (line 109) | def test_0rank_randn_like(test_case): class TestRandIntLikeOnNonDefaultDevice (line 120) | class TestRandIntLikeOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 121) | def test_non_default_device(test_case): FILE: python/oneflow/test/modules/test_random_generator_and_seed.py function _inspect_rand_op_and_args (line 33) | def _inspect_rand_op_and_args(rand_op, **kwargs): function _test_rand_op_unidentical (line 54) | def _test_rand_op_unidentical(test_case, rand_op, input=None, **kwargs): function _test_global_rand_op_with_split (line 78) | def _test_global_rand_op_with_split(test_case, rand_op, input=None, **kw... function _test_global_rand_op_with_broadcast (line 106) | def _test_global_rand_op_with_broadcast(test_case, rand_op, input=None, ... class TestRandOpUnidentical (line 134) | class TestRandOpUnidentical(oneflow.unittest.TestCase): method test_usual_rand_op (line 136) | def test_usual_rand_op(self): method test_source_rand_op (line 150) | def test_source_rand_op(self): method test_bernoulli (line 162) | def test_bernoulli(self): method test_random_mask_like (line 169) | def test_random_mask_like(self): class TestGlobalRandOp (line 182) | class TestGlobalRandOp(oneflow.unittest.TestCase): method test_usual_rand_op_with_split (line 185) | def test_usual_rand_op_with_split(self): method test_usual_rand_op_with_broadcast (line 199) | def test_usual_rand_op_with_broadcast(self): method test_source_rand_op_with_split (line 213) | def test_source_rand_op_with_split(self): method test_source_rand_op_with_broadcast (line 226) | def test_source_rand_op_with_broadcast(self): method test_bernoulli_with_split (line 243) | def test_bernoulli_with_split(self): method test_bernoulli_with_broadcast (line 249) | def test_bernoulli_with_broadcast(self): method test_random_mask_like_with_split (line 256) | def test_random_mask_like_with_split(self): method test_random_mask_like_with_broadcast (line 268) | def test_random_mask_like_with_broadcast(self): FILE: python/oneflow/test/modules/test_randperm.py function _test_randperm_with_generator (line 25) | def _test_randperm_with_generator(test_case, N, device, dtype): function _test_randperm_backward (line 38) | def _test_randperm_backward(test_case, N, device, dtype): function _test_randperm_randomness (line 47) | def _test_randperm_randomness(test_case, N, device, dtype): function _test_randperm_large_seq_randomness (line 54) | def _test_randperm_large_seq_randomness(test_case, N, device, dtype): class Testrandperm (line 62) | class Testrandperm(flow.unittest.TestCase): method test_global_naive (line 63) | def test_global_naive(test_case): method test_global_different_types (line 70) | def test_global_different_types(test_case): method test_randperm (line 86) | def test_randperm(test_case): method test_randperm_backward (line 106) | def test_randperm_backward(test_case): method test_auto_1 (line 118) | def test_auto_1(test_case): method test_auto_0 (line 124) | def test_auto_0(test_case): class TestRandpermOnNonDefaultDevice (line 132) | class TestRandpermOnNonDefaultDevice(flow.unittest.TestCase): method test_non_default_device (line 133) | def test_non_default_device(test_case): FILE: python/oneflow/test/modules/test_reciprocal.py class TestReciprocalModule (line 30) | class TestReciprocalModule(flow.unittest.TestCase): method test_flow_reciprocal_list_with_random_data (line 32) | def test_flow_reciprocal_list_with_random_data(test_case): method test_flow_reciprocal_list_with_0dim_data (line 41) | def test_flow_reciprocal_list_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_reduce.py function _test_reduce (line 27) | def _test_reduce(test_case, dst, device): class TestReduce (line 66) | class TestReduce(flow.unittest.TestCase): method test_reduce (line 67) | def test_reduce(test_case): FILE: python/oneflow/test/modules/test_reduce_sum_like.py function _test_reduce_sum_like (line 27) | def _test_reduce_sum_like(test_case, device): function _test_reduce_sum_like_one (line 43) | def _test_reduce_sum_like_one(test_case, device): function _test_reduce_sum_like_different_dim (line 59) | def _test_reduce_sum_like_different_dim(test_case, device): function _test_reduce_sum_like_different_dim_with_input_axisvec (line 75) | def _test_reduce_sum_like_different_dim_with_input_axisvec(test_case, de... function _test_reduce_sum_like_3dim (line 91) | def _test_reduce_sum_like_3dim(test_case, device): function _test_reduce_sum_like_4dim (line 107) | def _test_reduce_sum_like_4dim(test_case, device): function _test_reduce_sum_like_backward (line 123) | def _test_reduce_sum_like_backward(test_case, device): class TestReduceSumLike (line 144) | class TestReduceSumLike(flow.unittest.TestCase): method test_reduce_sum_like (line 145) | def test_reduce_sum_like(test_case): FILE: python/oneflow/test/modules/test_reflection_pad.py function gen_numpy_test_sample (line 33) | def gen_numpy_test_sample(input, padding): function _test_reflection_pad2d (line 86) | def _test_reflection_pad2d(test_case, shape, padding, device): class TestReflectionPadModule (line 107) | class TestReflectionPadModule(flow.unittest.TestCase): method test_reflection_pad2d (line 108) | def test_reflection_pad2d(test_case): method test_reflection_pad_1d_with_3d_input (line 117) | def test_reflection_pad_1d_with_3d_input(test_case): method test_reflection_pad_1d_with_2d_input (line 129) | def test_reflection_pad_1d_with_2d_input(test_case): method test_reflection_pad_2d_with_random_data (line 140) | def test_reflection_pad_2d_with_random_data(test_case): method test_functional_reflection_pad_1d_with_random_data (line 153) | def test_functional_reflection_pad_1d_with_random_data(test_case): method test_functional_reflection_pad_2d_with_random_data (line 163) | def test_functional_reflection_pad_2d_with_random_data(test_case): FILE: python/oneflow/test/modules/test_repeat.py class TestRepeat (line 25) | class TestRepeat(flow.unittest.TestCase): method test_flow_tensor_repeat_with_random_data (line 27) | def test_flow_tensor_repeat_with_random_data(test_case): method test_flow_tensor_repeat_bool_with_random_data (line 34) | def test_flow_tensor_repeat_bool_with_random_data(test_case): method test_flow_tensor_repeat_with_0dim_data (line 41) | def test_flow_tensor_repeat_with_0dim_data(test_case): method test_complicated_repeat_case (line 48) | def test_complicated_repeat_case(test_case): method test_flow_tensor_0size_with_random_data (line 55) | def test_flow_tensor_0size_with_random_data(test_case): FILE: python/oneflow/test/modules/test_repeat_interleave.py class TestRepeatInterLeave (line 27) | class TestRepeatInterLeave(flow.unittest.TestCase): method test_flow_int_repeat_interleave_dim_none (line 29) | def test_flow_int_repeat_interleave_dim_none(test_case): method test_flow_int_repeat_interleave_with_dim (line 35) | def test_flow_int_repeat_interleave_with_dim(test_case): method test_flow_tensor_repeat_interleave_dim (line 42) | def test_flow_tensor_repeat_interleave_dim(test_case): method test_flow_tensor_repeat_interleave_dim_with_output_size (line 49) | def test_flow_tensor_repeat_interleave_dim_with_output_size(test_case): method test_flow_tensor_repeat_interleave_0size_tensor (line 55) | def test_flow_tensor_repeat_interleave_0size_tensor(test_case): FILE: python/oneflow/test/modules/test_replication_pad.py function _np_replication_pad2d_grad (line 34) | def _np_replication_pad2d_grad(src, dest, padding): function _test_ReplicationPad2d (line 74) | def _test_ReplicationPad2d(test_case, shape, padding, device): class TestReplicationPadModule (line 103) | class TestReplicationPadModule(flow.unittest.TestCase): method test_ReplicationPad2d (line 104) | def test_ReplicationPad2d(test_case): method test_replication_pad1d_with_3d_input (line 113) | def test_replication_pad1d_with_3d_input(test_case): method test_replication_pad1d_with_2d_input (line 126) | def test_replication_pad1d_with_2d_input(test_case): method test_replication_pad2d_with_random_data (line 137) | def test_replication_pad2d_with_random_data(test_case): method test_functional_replication_pad_1d_with_random_data (line 150) | def test_functional_replication_pad_1d_with_random_data(test_case): method test_functional_replication_pad_2d_with_random_data (line 160) | def test_functional_replication_pad_2d_with_random_data(test_case): FILE: python/oneflow/test/modules/test_reshape.py function _test_reshape (line 31) | def _test_reshape(test_case, device): function _test_reshape_tuple (line 41) | def _test_reshape_tuple(test_case, device): function _test_reshape_backward (line 51) | def _test_reshape_backward(test_case, device): function _test_reshape_scalar (line 71) | def _test_reshape_scalar(test_case, device): class TestModule (line 85) | class TestModule(flow.unittest.TestCase): method test_reshape (line 86) | def test_reshape(test_case): method test_reshape_flow_with_random_data (line 99) | def test_reshape_flow_with_random_data(test_case): method test_reshape_flow_with_0dim_data (line 106) | def test_reshape_flow_with_0dim_data(test_case): method test_reshape_with_0_size_data (line 113) | def test_reshape_with_0_size_data(test_case): method test_reshape_flow_bool_with_random_data (line 122) | def test_reshape_flow_bool_with_random_data(test_case): method test_reshape_like (line 129) | def test_reshape_like(test_case): method profile_reshape (line 161) | def profile_reshape(test_case): FILE: python/oneflow/test/modules/test_reshape_sbp.py class TestReshapeSbp (line 25) | class TestReshapeSbp(flow.unittest.TestCase): method test_reshape_sbp (line 26) | def test_reshape_sbp(test_case): class TestReshapeNdSbp (line 36) | class TestReshapeNdSbp(flow.unittest.TestCase): method test_reshape_nd_sbp (line 37) | def test_reshape_nd_sbp(test_case): FILE: python/oneflow/test/modules/test_resnet_load_torch_weight_compatibile.py class TestResNet18LoadWeightCompatibile (line 29) | class TestResNet18LoadWeightCompatibile(flow.unittest.TestCase): method test_resnet18_load_weight_compatibile (line 30) | def test_resnet18_load_weight_compatibile(test_case): FILE: python/oneflow/test/modules/test_rmsnorm.py function _get_norm_dims (line 27) | def _get_norm_dims(shape, normalized_shape): function _torch_rmsnorm (line 33) | def _torch_rmsnorm(x, weight, normalized_shape=None, eps=1e-6): function _test_rmsnorm (line 45) | def _test_rmsnorm( class TestRMSNorm (line 144) | class TestRMSNorm(flow.unittest.TestCase): method test_real_example (line 145) | def test_real_example(test_case): method test_no_affine (line 155) | def test_no_affine(test_case): method test_warp_impl (line 160) | def test_warp_impl(test_case): method test_block_smem_impl (line 169) | def test_block_smem_impl(test_case): method test_block_uncached_impl (line 180) | def test_block_uncached_impl(test_case): FILE: python/oneflow/test/modules/test_roc_auc_score.py function _test_roc_auc_score (line 25) | def _test_roc_auc_score(test_case, label_dtype, pred_dtype): function _compare_roc_auc_score (line 37) | def _compare_roc_auc_score(test_case, label_dtype, pred_dtype): class TestNMS (line 51) | class TestNMS(flow.unittest.TestCase): method test_roc_auc_score (line 52) | def test_roc_auc_score(test_case): FILE: python/oneflow/test/modules/test_roi_align.py function bilinear_interpolate (line 399) | def bilinear_interpolate(data, y, x, snap_border=False): function roi_align_np (line 431) | def roi_align_np( function _test_roi_align (line 483) | def _test_roi_align(test_case, device): function _test_roi_align_backward (line 516) | def _test_roi_align_backward(test_case, device): class TestRoIAlign (line 529) | class TestRoIAlign(flow.unittest.TestCase): method test_roi_align (line 530) | def test_roi_align(test_case): FILE: python/oneflow/test/modules/test_roll.py function _test_roll (line 27) | def _test_roll(test_case, device): function _test_roll_single_dims (line 58) | def _test_roll_single_dims(test_case, device): function _test_roll_none_dims (line 84) | def _test_roll_none_dims(test_case, device): class TestRoll (line 111) | class TestRoll(flow.unittest.TestCase): method test_expand_compare_with_torch (line 112) | def test_expand_compare_with_torch(test_case): FILE: python/oneflow/test/modules/test_round.py class TestRound (line 28) | class TestRound(flow.unittest.TestCase): method test_flow_round_with_random_data (line 30) | def test_flow_round_with_random_data(test_case): method test_flow_round_with_0dim_data (line 37) | def test_flow_round_with_0dim_data(test_case): method test_flow_round_half_to_even (line 44) | def test_flow_round_half_to_even(test_case): FILE: python/oneflow/test/modules/test_rrelu.py function do_test_rrelu_same_bound (line 30) | def do_test_rrelu_same_bound(test_case, shape, device, dtype): function do_test_rrelu_different_bound (line 64) | def do_test_rrelu_different_bound(test_case, shape, device, dtype): class TestModule (line 109) | class TestModule(flow.unittest.TestCase): method test_numpy_case (line 111) | def test_numpy_case(test_case): method test_functional_rrelu (line 135) | def test_functional_rrelu(test_case): method test_rrelu_train (line 151) | def test_rrelu_train(test_case): method test_rrelu_eval (line 159) | def test_rrelu_eval(test_case): method profile_rrelu (line 167) | def profile_rrelu(test_case): FILE: python/oneflow/test/modules/test_save_load.py class CustomModuleForSaveLoad (line 32) | class CustomModuleForSaveLoad(flow.nn.Module): method __init__ (line 33) | def __init__(self): method forward (line 37) | def forward(self, x): class TestSaveLoad (line 42) | class TestSaveLoad(flow.unittest.TestCase): method test_load_map_location (line 44) | def test_load_map_location(test_case): method test_save_dir (line 87) | def test_save_dir(test_case): method test_save_dir_fault_tolerance (line 97) | def test_save_dir_fault_tolerance(test_case): method test_save_state_dict (line 111) | def test_save_state_dict(test_case): method test_save_state_dict_bytes (line 133) | def test_save_state_dict_bytes(test_case): method _test_save_and_load_global_from_nested_dict (line 157) | def _test_save_and_load_global_from_nested_dict(test_case): method test_save_and_load_global_from_nested_dict_1n4d (line 212) | def test_save_and_load_global_from_nested_dict_1n4d(test_case): method test_save_and_load_global_from_nested_dict_2n2d (line 216) | def test_save_and_load_global_from_nested_dict_2n2d(test_case): method test_load_pytorch_weights (line 220) | def test_load_pytorch_weights(test_case): method test_load_pytorch_weights_global (line 252) | def test_load_pytorch_weights_global(test_case): method test_save_load_module_directly (line 295) | def test_save_load_module_directly(test_case): method test_save_load_module_directly_save_bytes (line 319) | def test_save_load_module_directly_save_bytes(test_case): method test_save_load_module_directly_load_filestream (line 349) | def test_save_load_module_directly_load_filestream(test_case): method test_load_old_dir_data (line 374) | def test_load_old_dir_data(test_case): method test_pytorch_non_tensor (line 387) | def test_pytorch_non_tensor(test_case): method test_pytorch_non_tensor_load_filestream (line 395) | def test_pytorch_non_tensor_load_filestream(test_case): method test_pytorch_non_tensor_save_bytes (line 404) | def test_pytorch_non_tensor_save_bytes(test_case): FILE: python/oneflow/test/modules/test_saved_tensor_hooks.py class TestSavedTensorHooks (line 27) | class TestSavedTensorHooks(flow.unittest.TestCase): method test_normal_saved_tensor_hooks (line 28) | def test_normal_saved_tensor_hooks(test_case): method test_saved_tensor_hooks_in_autograd_function (line 49) | def test_saved_tensor_hooks_in_autograd_function(test_case): FILE: python/oneflow/test/modules/test_sbp_symbol.py class TestSBPSymbol (line 22) | class TestSBPSymbol(flow.unittest.TestCase): method test_sbp_symbol (line 23) | def test_sbp_symbol(test_case): FILE: python/oneflow/test/modules/test_scatter_nd.py function _test_scatter_nd (line 27) | def _test_scatter_nd(test_case, device): function _test_scatter_nd_t (line 39) | def _test_scatter_nd_t(test_case, device): function _test_scatter_nd_backward (line 61) | def _test_scatter_nd_backward(test_case, device): class TestScatter_nd (line 81) | class TestScatter_nd(flow.unittest.TestCase): method test_scatter_nd (line 82) | def test_scatter_nd(test_case): FILE: python/oneflow/test/modules/test_scatter_ops.py function _get_indexes (line 25) | def _get_indexes(device): function _test_scatter (line 42) | def _test_scatter(test_case, test_scalar: bool, dim: int): function _test_scatter_add (line 50) | def _test_scatter_add(test_case, dim: int): function _test_scatter_reduce (line 58) | def _test_scatter_reduce(test_case, dim: int): class TestScatterOpsModule (line 73) | class TestScatterOpsModule(flow.unittest.TestCase): method test_scatter_with_random_data (line 75) | def test_scatter_with_random_data(test_case): method test_scatter_add_with_random_data (line 79) | def test_scatter_add_with_random_data(test_case): method test_scatter_reduce_with_random_data (line 85) | def test_scatter_reduce_with_random_data(test_case): FILE: python/oneflow/test/modules/test_searchsorted.py function _test_search_sorted (line 30) | def _test_search_sorted(test_case, input_dtype, device): function _test_search_sorted_1 (line 45) | def _test_search_sorted_1(test_case, input_dtype, device): function _test_search_sorted_2 (line 60) | def _test_search_sorted_2(test_case, input_dtype, device): function _test_search_sorted_3 (line 73) | def _test_search_sorted_3(test_case, input_dtype, device): function _test_search_sorted_4 (line 88) | def _test_search_sorted_4(test_case, input_dtype, device): function _test_search_sorted_5 (line 108) | def _test_search_sorted_5(test_case, input_dtype, device): function _test_search_sorted_6 (line 118) | def _test_search_sorted_6(test_case, input_dtype, device): function _test_search_sorted_7 (line 128) | def _test_search_sorted_7(test_case, input_dtype, device): class TestSearchSorted (line 139) | class TestSearchSorted(flow.unittest.TestCase): method test_search_sorted (line 140) | def test_search_sorted(test_case): method test_search_sorted (line 164) | def test_search_sorted(test_case): FILE: python/oneflow/test/modules/test_select.py class TestSelect (line 26) | class TestSelect(flow.unittest.TestCase): method test_flow_select (line 28) | def test_flow_select(test_case): method test_flow_select_with_stride (line 44) | def test_flow_select_with_stride(test_case): method test_flow_select_1dim (line 62) | def test_flow_select_1dim(test_case): FILE: python/oneflow/test/modules/test_shutting_down.py class _TestCallWhenShuttingDown (line 23) | class _TestCallWhenShuttingDown: method __init__ (line 24) | def __init__(self): method __del__ (line 29) | def __del__(self, of=oneflow): class _TestSyncWhenShuttingDown (line 41) | class _TestSyncWhenShuttingDown: method __init__ (line 42) | def __init__(self): method __del__ (line 45) | def __del__(self): FILE: python/oneflow/test/modules/test_sign.py function _test_sign_impl (line 29) | def _test_sign_impl(test_case, shape, device): class TestSign (line 44) | class TestSign(flow.unittest.TestCase): method test_sign (line 45) | def test_sign(test_case): method test_sign_with_random_data (line 53) | def test_sign_with_random_data(test_case): method test_sign_with_0_size_data (line 60) | def test_sign_with_0_size_data(test_case): method test_sign_with_random_data (line 67) | def test_sign_with_random_data(test_case): method test_sign_with_0dim_data (line 74) | def test_sign_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_single_threaded_vm.py class TestSingleThreadedVM (line 24) | class TestSingleThreadedVM(flow.unittest.TestCase): method test_ddp_in_single_threaded_vm (line 26) | def test_ddp_in_single_threaded_vm(test_case): FILE: python/oneflow/test/modules/test_skip_layer_norm.py function compare_result (line 29) | def compare_result(test_case, a, b, rtol=1e-5, atol=1e-8): class NaiveSkipLayerNorm (line 36) | class NaiveSkipLayerNorm(nn.Module): method __init__ (line 37) | def __init__(self): method forward (line 40) | def forward( class FusedSkipLayerNorm (line 67) | class FusedSkipLayerNorm(nn.Module): method __init__ (line 68) | def __init__(self): method forward (line 71) | def forward( function _test_skip_layer_norm (line 86) | def _test_skip_layer_norm( class TestSkipLayerNorm (line 176) | class TestSkipLayerNorm(flow.unittest.TestCase): method test_gather (line 177) | def test_gather(test_case): FILE: python/oneflow/test/modules/test_skip_rms_norm.py function compare_result (line 29) | def compare_result(test_case, a, b, rtol=1e-5, atol=1e-8): class NaiveSkipRMSNorm (line 36) | class NaiveSkipRMSNorm(nn.Module): method __init__ (line 37) | def __init__(self): method forward (line 40) | def forward( class FusedSkipRMSNorm (line 57) | class FusedSkipRMSNorm(nn.Module): method __init__ (line 58) | def __init__(self): method forward (line 61) | def forward( function _test_skip_rms_norm (line 75) | def _test_skip_rms_norm( class TestSkipRMSNorm (line 152) | class TestSkipRMSNorm(flow.unittest.TestCase): method test_gather (line 153) | def test_gather(test_case): FILE: python/oneflow/test/modules/test_slice.py function _test_slice (line 29) | def _test_slice(test_case, device): function _test_slice_empty (line 41) | def _test_slice_empty(test_case, device): function _test_slice_1_dim (line 50) | def _test_slice_1_dim(test_case, device): function _test_slice_3_dim (line 58) | def _test_slice_3_dim(test_case, device): function _test_slice_4_dim (line 64) | def _test_slice_4_dim(test_case, device): function _test_slice_with_int_index (line 74) | def _test_slice_with_int_index(test_case, device): function _test_slice_negative_index (line 97) | def _test_slice_negative_index(test_case, device): function _test_slice_ellipsis_type (line 106) | def _test_slice_ellipsis_type(test_case, device): function _test_slice_backward (line 123) | def _test_slice_backward(test_case, device): function _test_slice_scalar (line 135) | def _test_slice_scalar(test_case, device): class TestSlice (line 145) | class TestSlice(flow.unittest.TestCase): method test_slice (line 146) | def test_slice(test_case): class TestSliceUpdate (line 166) | class TestSliceUpdate(flow.unittest.TestCase): method test_slice_update (line 167) | def test_slice_update(test_case): method test_slice_update_negative_index (line 175) | def test_slice_update_negative_index(test_case): method test_slice_update_scalar_integer_tensor_index (line 182) | def test_slice_update_scalar_integer_tensor_index(test_case): method test_slice_update_scalar_boolean_tensor_index (line 200) | def test_slice_update_scalar_boolean_tensor_index(test_case): method test_slice_update_negative_index_graph (line 218) | def test_slice_update_negative_index_graph(test_case): method test_slice_update_different_dtype (line 231) | def test_slice_update_different_dtype(test_case): method test_slice_update_ellipsis_type (line 240) | def test_slice_update_ellipsis_type(test_case): method test_slice_update_ellipsis_type_graph (line 247) | def test_slice_update_ellipsis_type_graph(test_case): method test_slice_update_grad_graph (line 260) | def test_slice_update_grad_graph(test_case): method test_random_nd_slice_update_in_non_contiguous_tensor (line 304) | def test_random_nd_slice_update_in_non_contiguous_tensor(test_case): method test_slice_update_expand_value (line 368) | def test_slice_update_expand_value(test_case): FILE: python/oneflow/test/modules/test_softmax.py function _dtype_flow_to_np (line 28) | def _dtype_flow_to_np(dtype): function _np_softmax (line 32) | def _np_softmax(x, dtype=None): function _test_softmax_impl (line 40) | def _test_softmax_impl(test_case, shape, input_dtype, output_dtype): class Testsoftmax (line 53) | class Testsoftmax(flow.unittest.TestCase): method test_softmax (line 54) | def test_softmax(test_case): FILE: python/oneflow/test/modules/test_softplus.py function _test_softplus_impl (line 27) | def _test_softplus_impl(test_case, shape, device): class Testsoftplus (line 42) | class Testsoftplus(flow.unittest.TestCase): method test_softplus (line 43) | def test_softplus(test_case): FILE: python/oneflow/test/modules/test_sort.py function _test_sort (line 29) | def _test_sort(test_case, data_shape, axis, descending, data_type, device): function _test_tensor_sort (line 48) | def _test_tensor_sort(test_case, data_shape, axis, descending, data_type... class TestSort (line 68) | class TestSort(flow.unittest.TestCase): method test_sort (line 69) | def test_sort(test_case): method test_sort_with_random_data (line 81) | def test_sort_with_random_data(test_case): method test_sort_return_type_with_random_data_ (line 88) | def test_sort_return_type_with_random_data_(test_case): method test_sort_bool_with_random_data (line 97) | def test_sort_bool_with_random_data(test_case): method test_sort_return_type_bool_with_random_data (line 103) | def test_sort_return_type_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_sparse.py function _test_embedding_padding_idx (line 26) | def _test_embedding_padding_idx(test_case, device): function _test_embedding_scale_by_freq (line 76) | def _test_embedding_scale_by_freq(test_case, device): class TestEmbedding (line 139) | class TestEmbedding(flow.unittest.TestCase): method test_padding_idx (line 140) | def test_padding_idx(test_case): method test_embedding_impl (line 149) | def test_embedding_impl(test_case): method test_embedding_functional (line 168) | def test_embedding_functional(test_case): method test_embedding_renorm (line 190) | def test_embedding_renorm(test_case): FILE: python/oneflow/test/modules/test_sparse_softmax_cross_entropy.py function compare_with_torch (line 33) | def compare_with_torch( function compare_eager_global_with_torch (line 65) | def compare_eager_global_with_torch( function compare_eager_2d_global_with_torch (line 114) | def compare_eager_2d_global_with_torch( function compare_lazy_global_with_torch (line 178) | def compare_lazy_global_with_torch( class TestSparseSoftmaxCrossEntropyWithLogits (line 231) | class TestSparseSoftmaxCrossEntropyWithLogits(flow.unittest.TestCase): method test_sparse_softmax_cross_entropy (line 233) | def test_sparse_softmax_cross_entropy(test_case): class TestSparseSoftmaxCrossEntropyMsWithLogits (line 244) | class TestSparseSoftmaxCrossEntropyMsWithLogits(flow.unittest.TestCase): method test_distributed_sparse_softmax_cross_entropy (line 247) | def test_distributed_sparse_softmax_cross_entropy(test_case): FILE: python/oneflow/test/modules/test_special_ops.py class TestSpecialOps (line 33) | class TestSpecialOps(flow.unittest.TestCase): method test_flow_erf_with_random_data (line 35) | def test_flow_erf_with_random_data(test_case): method test_flow_erfc_with_random_data (line 43) | def test_flow_erfc_with_random_data(test_case): method test_flow_erfinv_with_random_data (line 51) | def test_flow_erfinv_with_random_data(test_case): method test_flow_exp2_with_random_data (line 59) | def test_flow_exp2_with_random_data(test_case): method test_flow_expm1_with_random_data (line 67) | def test_flow_expm1_with_random_data(test_case): method test_flow_round_with_random_data (line 75) | def test_flow_round_with_random_data(test_case): method test_flow_log1p_with_random_data (line 82) | def test_flow_log1p_with_random_data(test_case): method test_flow_log_softmax_with_random_data (line 90) | def test_flow_log_softmax_with_random_data(test_case): method test_flow_softmax_with_random_data (line 102) | def test_flow_softmax_with_random_data(test_case): method test_flow_logsumexp_with_random_data (line 110) | def test_flow_logsumexp_with_random_data(test_case): method test_flow_digamma_with_random_data (line 117) | def test_flow_digamma_with_random_data(test_case): method test_flow_psi_with_random_data (line 125) | def test_flow_psi_with_random_data(test_case): class TestZeta (line 134) | class TestZeta(flow.unittest.TestCase): method test_flow_zeta_with_random_data (line 137) | def test_flow_zeta_with_random_data(test_case): method test_flow_zeta_broadcast_input (line 154) | def test_flow_zeta_broadcast_input(test_case): method test_flow_zeta_broadcast_other (line 163) | def test_flow_zeta_broadcast_other(test_case): method test_flow_zeta_scalar_other (line 172) | def test_flow_zeta_scalar_other(test_case): method test_flow_zeta_scalar_other (line 180) | def test_flow_zeta_scalar_other(test_case): FILE: python/oneflow/test/modules/test_split.py class TestSplit (line 26) | class TestSplit(flow.unittest.TestCase): method test_flow_split_with_random_data (line 28) | def test_flow_split_with_random_data(test_case): method test_flow_split_with_stride (line 39) | def test_flow_split_with_stride(test_case): method test_flow_split_sizes_with_random_data (line 53) | def test_flow_split_sizes_with_random_data(test_case): method test_flow_split_sizes_neg_dim_with_random_data (line 63) | def test_flow_split_sizes_neg_dim_with_random_data(test_case): method test_flow_split_bool_with_random_data (line 73) | def test_flow_split_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_square_relu.py class SquareReLUActivation (line 28) | class SquareReLUActivation(torch.nn.Module): method forward (line 33) | def forward(self, input): function _test_square_relu (line 39) | def _test_square_relu(test_case, device): class TestModule (line 57) | class TestModule(flow.unittest.TestCase): method test_square_relu (line 58) | def test_square_relu(test_case): FILE: python/oneflow/test/modules/test_squeeze.py function _test_squeeze (line 29) | def _test_squeeze(test_case, device): function _test_squeeze_1d_input (line 45) | def _test_squeeze_1d_input(test_case, device): function _test_tensor_squeeze (line 52) | def _test_tensor_squeeze(test_case, device): function _test_squeeze_int (line 68) | def _test_squeeze_int(test_case, device): function _test_squeeze_backward (line 81) | def _test_squeeze_backward(test_case, device): class TestSqueeze (line 93) | class TestSqueeze(flow.unittest.TestCase): method test_squeeze (line 94) | def test_squeeze(test_case): method test_flow_squeeze_with_random_data (line 108) | def test_flow_squeeze_with_random_data(test_case): method test_inplace_squeeze_with_random_data (line 115) | def test_inplace_squeeze_with_random_data(test_case): method test_squeeze_with_0_size_data (line 122) | def test_squeeze_with_0_size_data(test_case): method test_flow_squeeze_bool_with_random_data (line 129) | def test_flow_squeeze_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_stack.py class TestStackModule (line 26) | class TestStackModule(flow.unittest.TestCase): method test_stack_with_random_data (line 28) | def test_stack_with_random_data(test_case): method test_stack_bool_with_random_data (line 36) | def test_stack_bool_with_random_data(test_case): method test_column_stack_with_random_data (line 48) | def test_column_stack_with_random_data(test_case): method test_column_stack_with_0dim_data (line 56) | def test_column_stack_with_0dim_data(test_case): method test_row_stack_with_random_data (line 64) | def test_row_stack_with_random_data(test_case): method test_row_stack_with_0dim_data (line 72) | def test_row_stack_with_0dim_data(test_case): method test_hstack_with_random_data (line 80) | def test_hstack_with_random_data(test_case): method test_hstack_with_0dim_data (line 88) | def test_hstack_with_0dim_data(test_case): method test_vstack_with_random_data (line 98) | def test_vstack_with_random_data(test_case): method test_vstack_with_0dim_data (line 107) | def test_vstack_with_0dim_data(test_case): method test_dstack_with_random_data (line 115) | def test_dstack_with_random_data(test_case): method test_dstack_with_0dim_data (line 124) | def test_dstack_with_0dim_data(test_case): method test_stack_kMaxInputCount_inputs (line 132) | def test_stack_kMaxInputCount_inputs(test_case): FILE: python/oneflow/test/modules/test_stateful_kernel_with_cache.py class TestStatefulKernelWithInpersistentState (line 27) | class TestStatefulKernelWithInpersistentState(flow.unittest.TestCase): method test_stateful_kernel_with_inpersistent_state (line 28) | def test_stateful_kernel_with_inpersistent_state(test_case): FILE: python/oneflow/test/modules/test_stateful_local_opkernel.py class TestStatefulLocalKernel (line 27) | class TestStatefulLocalKernel(flow.unittest.TestCase): method test_dynamic_attrs (line 29) | def test_dynamic_attrs(test_case): method test_stateful_local_kernel_in_global_mode (line 37) | def test_stateful_local_kernel_in_global_mode(test_case): FILE: python/oneflow/test/modules/test_std.py class TestStd (line 24) | class TestStd(flow.unittest.TestCase): method test_std_flow_with_random_data (line 26) | def test_std_flow_with_random_data(test_case): method test_std_tensor_with_random_data (line 37) | def test_std_tensor_with_random_data(test_case): FILE: python/oneflow/test/modules/test_stft.py function getRandBoolvalue (line 30) | def getRandBoolvalue(): function getRandFFtvalue (line 38) | def getRandFFtvalue(): function is_cufft_available (line 46) | def is_cufft_available(): class TestStft (line 54) | class TestStft(flow.unittest.TestCase): method test_stft_with_1D_random_data (line 58) | def test_stft_with_1D_random_data(test_case): method test_stft_with_2D_random_data (line 83) | def test_stft_with_2D_random_data(test_case): FILE: python/oneflow/test/modules/test_sub.py function _test_sub_impl (line 32) | def _test_sub_impl(test_case, shape, device): class TestSubModule (line 96) | class TestSubModule(flow.unittest.TestCase): method test_sub (line 97) | def test_sub(test_case): method test_random_dim_sub (line 105) | def test_random_dim_sub(test_case): method test_random_dim_scalar_sub (line 115) | def test_random_dim_scalar_sub(test_case): method test_sub_with_0_size_data (line 125) | def test_sub_with_0_size_data(test_case): method test_sub_with_0dim_data (line 136) | def test_sub_with_0dim_data(test_case): method test_sub_with_alpha (line 147) | def test_sub_with_alpha(test_case): method test_non_contiguous_inplace_sub (line 161) | def test_non_contiguous_inplace_sub(test_case): method test_scalar_sub_with_random_devices (line 171) | def test_scalar_sub_with_random_devices(test_case): FILE: python/oneflow/test/modules/test_sum.py function _test_sum_impl (line 29) | def _test_sum_impl(test_case, device, data_type): class TestSumModule (line 73) | class TestSumModule(flow.unittest.TestCase): method test_sum (line 74) | def test_sum(test_case): method test_sum_against_pytorch (line 82) | def test_sum_against_pytorch(test_case): method test_sum_dtype (line 89) | def test_sum_dtype(test_case): method test_sum_complex_dtype (line 108) | def test_sum_complex_dtype(test_case): method test_sum_complex_dtype (line 129) | def test_sum_complex_dtype(test_case): method test_sum_arithmetic_dtype (line 143) | def test_sum_arithmetic_dtype(test_case): method test_sum_with_0_size_tensor (line 150) | def test_sum_with_0_size_tensor(test_case): method test_sum_with_0dim_tensor (line 157) | def test_sum_with_0dim_tensor(test_case): FILE: python/oneflow/test/modules/test_swapaxes.py class TestSwapaxes (line 29) | class TestSwapaxes(flow.unittest.TestCase): method test_swapaxes_flow_with_random_data (line 31) | def test_swapaxes_flow_with_random_data(test_case): method test_swapaxes_flow_with_stride (line 38) | def test_swapaxes_flow_with_stride(test_case): FILE: python/oneflow/test/modules/test_swapdims.py class Testswapdims (line 30) | class Testswapdims(flow.unittest.TestCase): method test_swapdims_flow_with_random_data (line 32) | def test_swapdims_flow_with_random_data(test_case): method test_swapdims_flow_with_random_data2 (line 39) | def test_swapdims_flow_with_random_data2(test_case): FILE: python/oneflow/test/modules/test_swautils.py class TestLRScheduler (line 38) | class TestLRScheduler(flow.unittest.TestCase): method setUp (line 40) | def setUp(self): method test_multiplicative_lr (line 51) | def test_multiplicative_lr(self): method _test (line 65) | def _test(self, schedulers, targets, epochs=10): method test_swa_lr_state_dict (line 80) | def test_swa_lr_state_dict(self): method _check_scheduler_state_dict (line 88) | def _check_scheduler_state_dict(self, constr, constr2, epochs=10): method test_swalr_no_anneal (line 100) | def test_swalr_no_anneal(self): method test_swalr_cosine_anneal_after_multiplicative (line 110) | def test_swalr_cosine_anneal_after_multiplicative(self): method _test_swalr (line 138) | def _test_swalr(self, swa_scheduler, scheduler, targets, swa_start, ep... method test_swalr_hypers (line 156) | def test_swalr_hypers(self): class TestSWAUtils (line 170) | class TestSWAUtils(flow.unittest.TestCase): method _test_averaged_model (line 172) | def _test_averaged_model(self, net_device, swa_device): method test_averaged_model_all_devices (line 207) | def test_averaged_model_all_devices(self): method test_averaged_model_mixed_device (line 216) | def test_averaged_model_mixed_device(self): method test_averaged_model_state_dict (line 238) | def test_averaged_model_state_dict(self): method test_averaged_model_exponential (line 254) | def test_averaged_model_exponential(self): method test_averaged_model_exponential_buffers (line 293) | def test_averaged_model_exponential_buffers(self): method _test_update_bn (line 336) | def _test_update_bn(self, dnn, dl_x, dl_xy, momentum, cuda): method test_update_bn_dnn (line 395) | def test_update_bn_dnn(self): method test_update_bn_cnn (line 413) | def test_update_bn_cnn(self): method test_bn_update_eval_momentum (line 433) | def test_bn_update_eval_momentum(self): class SWATestDNN (line 450) | class SWATestDNN(flow.nn.Module): method __init__ (line 451) | def __init__(self, input_features): method compute_preactivation (line 457) | def compute_preactivation(self, x): method forward (line 460) | def forward(self, x): class SWATestCNN (line 466) | class SWATestCNN(flow.nn.Module): method __init__ (line 467) | def __init__(self, input_channels): method compute_preactivation (line 475) | def compute_preactivation(self, x): method forward (line 478) | def forward(self, x): class SchedulerTestNet (line 484) | class SchedulerTestNet(flow.nn.Module): method __init__ (line 485) | def __init__(self): method forward (line 490) | def forward(self, x): FILE: python/oneflow/test/modules/test_sync_and_async_allreduce.py function sync_allreduce (line 24) | def sync_allreduce(x): function async_allreduce (line 28) | def async_allreduce(x): class TestP2bOnGPU (line 34) | class TestP2bOnGPU(flow.unittest.TestCase): method test_p2b (line 35) | def test_p2b(test_case): FILE: python/oneflow/test/modules/test_sync_batchnorm.py class TestSyncBatchNorm (line 28) | class TestSyncBatchNorm(flow.unittest.TestCase): method test_sync_batchnorm3d (line 29) | def test_sync_batchnorm3d(test_case): method test_sync_batchnorm2d (line 52) | def test_sync_batchnorm2d(test_case): method test_sync_batchnorm1d (line 75) | def test_sync_batchnorm1d(test_case): FILE: python/oneflow/test/modules/test_t.py class TestTransposeAllDimFunction (line 30) | class TestTransposeAllDimFunction(flow.unittest.TestCase): method test_t_flow_with_random_data (line 32) | def test_t_flow_with_random_data(test_case): FILE: python/oneflow/test/modules/test_t5_layernorm.py class TorchT5LayerNorm (line 29) | class TorchT5LayerNorm(torch.nn.Module): method __init__ (line 30) | def __init__(self, hidden_size, eps=1e-6): method forward (line 38) | def forward(self, hidden_states): function _test_t5_layer_norm (line 55) | def _test_t5_layer_norm(test_case, device): class TestModule (line 82) | class TestModule(flow.unittest.TestCase): method test_t5_layernorm (line 83) | def test_t5_layernorm(test_case): FILE: python/oneflow/test/modules/test_tensor_buffer.py function _test_tensor_buffer_convert (line 27) | def _test_tensor_buffer_convert(test_case, device): class TestTensorBufferOps (line 39) | class TestTensorBufferOps(flow.unittest.TestCase): method test_tensor_buffer_convert (line 40) | def test_tensor_buffer_convert(test_case): FILE: python/oneflow/test/modules/test_tensor_ops.py function _test_is_floating_point (line 30) | def _test_is_floating_point(test_case, shape, device, dtype): function _test_type_dtype (line 40) | def _test_type_dtype(test_case, shape, device, src_dtype, tgt_dtype): function _test_type_str (line 49) | def _test_type_str( function _test_type_tensortype (line 61) | def _test_type_tensortype( function _test_type_noargs (line 73) | def _test_type_noargs(test_case, shape, device, dtype): class TestCuda (line 100) | class TestCuda(flow.unittest.TestCase): method test_cuda (line 102) | def test_cuda(test_case): method test_cuda_0dim (line 110) | def test_cuda_0dim(test_case): method test_cuda_int_device (line 118) | def test_cuda_int_device(test_case): class TestTensorOps (line 127) | class TestTensorOps(flow.unittest.TestCase): method test_cpu (line 129) | def test_cpu(test_case): method test_long (line 137) | def test_long(test_case): method test_long_0dim (line 144) | def test_long_0dim(test_case): method test_long_with_non_contiguous_input (line 151) | def test_long_with_non_contiguous_input(test_case): method test_int (line 161) | def test_int(test_case): method test_int_0dim (line 168) | def test_int_0dim(test_case): method test_half (line 175) | def test_half(test_case): method test_half_0dim (line 182) | def test_half_0dim(test_case): method test_float (line 189) | def test_float(test_case): method test_float_0dim (line 196) | def test_float_0dim(test_case): method test_double (line 203) | def test_double(test_case): method test_double_0dim (line 210) | def test_double_0dim(test_case): method test_bool (line 217) | def test_bool(test_case): method test_bool_0dim (line 224) | def test_bool_0dim(test_case): method test_bool_with_non_contiguous_input (line 231) | def test_bool_with_non_contiguous_input(test_case): method test_item (line 247) | def test_item(test_case): method test_item_0dim (line 260) | def test_item_0dim(test_case): method test_tolist (line 273) | def test_tolist(test_case): method test_tolist_0dim (line 286) | def test_tolist_0dim(test_case): method test_type_as (line 293) | def test_type_as(test_case): method test_is_floating_point (line 299) | def test_is_floating_point(test_case): method test_type_dtype (line 317) | def test_type_dtype(test_case): method test_type_tensortype_str_cpu (line 335) | def test_type_tensortype_str_cpu(test_case): method test_type_tensortype_str (line 363) | def test_type_tensortype_str(test_case): method test_type_tensortype_cpu (line 397) | def test_type_tensortype_cpu(test_case): method test_type_tensortype (line 425) | def test_type_tensortype(test_case): method test_type_noargs (line 460) | def test_type_noargs(test_case): method test_bincount (line 478) | def test_bincount(test_case): FILE: python/oneflow/test/modules/test_tensor_scatter_nd_update.py function _test_tensor_scatter_nd_update (line 27) | def _test_tensor_scatter_nd_update(test_case, device): function _test_tensor_scatter_nd_update_with_non_contiguous_input (line 40) | def _test_tensor_scatter_nd_update_with_non_contiguous_input(test_case, ... function _test_tensor_scatter_nd_update_t (line 62) | def _test_tensor_scatter_nd_update_t(test_case, device): function _test_tensor_scatter_nd_update_backward (line 87) | def _test_tensor_scatter_nd_update_backward(test_case, device): class TestTensorScatterNdUpdate (line 112) | class TestTensorScatterNdUpdate(flow.unittest.TestCase): method test_tensor_scatter_nd_update (line 113) | def test_tensor_scatter_nd_update(test_case): FILE: python/oneflow/test/modules/test_tensor_split.py class TestTensorSplitVec (line 25) | class TestTensorSplitVec(flow.unittest.TestCase): method test_flow_tensor_split_vec (line 27) | def test_flow_tensor_split_vec(test_case): method test_flow_tensor_split_vec_with_stride (line 41) | def test_flow_tensor_split_vec_with_stride(test_case): class TestTensorSplitInt (line 59) | class TestTensorSplitInt(flow.unittest.TestCase): method test_flow_tensor_split_int (line 61) | def test_flow_tensor_split_int(test_case): FILE: python/oneflow/test/modules/test_tensor_to.py class Test2DeviceGlobalTensorTo (line 29) | class Test2DeviceGlobalTensorTo(flow.unittest.TestCase): method test_asymmetric_global_tensor_clone (line 30) | def test_asymmetric_global_tensor_clone(test_case): method test_global_tensor_clone (line 42) | def test_global_tensor_clone(test_case): method test_global_tensor_to (line 53) | def test_global_tensor_to(test_case): method test_tensor_to_h2d1 (line 64) | def test_tensor_to_h2d1(test_case): class TestTo (line 76) | class TestTo(flow.unittest.TestCase): method test_global_tensor_clone (line 77) | def test_global_tensor_clone(test_case): method test_global_tensor_to (line 89) | def test_global_tensor_to(test_case): method test_empty_global_tensor_to (line 101) | def test_empty_global_tensor_to(test_case): method test_tensor_to_h2d (line 112) | def test_tensor_to_h2d(test_case): method test_tensor_to_d2h (line 125) | def test_tensor_to_d2h(test_case): method test_tensor_to_d2d (line 135) | def test_tensor_to_d2d(test_case): method test_tensor_to_h2h (line 145) | def test_tensor_to_h2h(test_case): method test_tensor_to_cast (line 153) | def test_tensor_to_cast(test_case): method test_tensor_to_cast_h2d (line 158) | def test_tensor_to_cast_h2d(test_case): method test_tensor_using_tensor (line 164) | def test_tensor_using_tensor(test_case): method test_int_to_args (line 172) | def test_int_to_args(test_case): method test_int_to_kwargs (line 178) | def test_int_to_kwargs(test_case): FILE: python/oneflow/test/modules/test_tensordot.py class TestTensordot (line 26) | class TestTensordot(flow.unittest.TestCase): method test_tensordot_intdim (line 28) | def test_tensordot_intdim(test_case): method test_tensordot_list_dim (line 46) | def test_tensordot_list_dim(test_case): method test_tensordot_tuple_dim (line 54) | def test_tensordot_tuple_dim(test_case): method test_tensordot_list_neg_dim (line 62) | def test_tensordot_list_neg_dim(test_case): method test_tensordot_backward (line 70) | def test_tensordot_backward(test_case): method test_tensordot_tensor_dim (line 78) | def test_tensordot_tensor_dim(test_case): method test_tensordot_single_item_tensor_dim (line 105) | def test_tensordot_single_item_tensor_dim(test_case): method test_tensordot_broadcast (line 114) | def test_tensordot_broadcast(test_case): FILE: python/oneflow/test/modules/test_tile.py class TestTile (line 25) | class TestTile(flow.unittest.TestCase): method test_flow_tile_with_random_data (line 27) | def test_flow_tile_with_random_data(test_case): method test_flow_tensor_tile_with_random_data (line 34) | def test_flow_tensor_tile_with_random_data(test_case): method test_flow_tile_bool_with_random_data (line 41) | def test_flow_tile_bool_with_random_data(test_case): method test_flow_tile_with_0dim_data (line 48) | def test_flow_tile_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_to_torch.py class TestToTroch (line 26) | class TestToTroch(flow.unittest.TestCase): method test_to_torch_cpu (line 28) | def test_to_torch_cpu(test_case): method test_to_torch_cpu_with_0_size_data (line 49) | def test_to_torch_cpu_with_0_size_data(test_case): method test_to_torch_cpu_with_0dim_data (line 59) | def test_to_torch_cpu_with_0dim_data(test_case): method test_to_torch_gpu (line 75) | def test_to_torch_gpu(test_case): method test_to_torch_global (line 89) | def test_to_torch_global(test_case): FILE: python/oneflow/test/modules/test_topk.py function _test_top_k (line 28) | def _test_top_k(test_case, shape, k, dim, device): class TestTopK (line 45) | class TestTopK(flow.unittest.TestCase): method test_in_top_k (line 46) | def test_in_top_k(test_case): FILE: python/oneflow/test/modules/test_transpose.py function _test_transpose (line 31) | def _test_transpose(test_case, device): function _test_tensor_transpose (line 40) | def _test_tensor_transpose(test_case, device): function _test_tranpose_negative_dim (line 49) | def _test_tranpose_negative_dim(test_case, device): function _test_transpose_backward (line 58) | def _test_transpose_backward(test_case, device): function _test_transpose_backward_v2 (line 72) | def _test_transpose_backward_v2(test_case, device): class TestTranspose (line 87) | class TestTranspose(flow.unittest.TestCase): method test_transpose (line 88) | def test_transpose(test_case): method test_transpose_flow_with_random_data (line 102) | def test_transpose_flow_with_random_data(test_case): method test_transpose_with_stride (line 109) | def test_transpose_with_stride(test_case): method test_transpose_with_0_size_data (line 119) | def test_transpose_with_0_size_data(test_case): method test_transpose_flow_bool_with_random_data (line 126) | def test_transpose_flow_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_tril.py class TestTril (line 24) | class TestTril(flow.unittest.TestCase): method test_tril_without_diag (line 26) | def test_tril_without_diag(test_case): method test_tril_with_diag (line 41) | def test_tril_with_diag(test_case): FILE: python/oneflow/test/modules/test_triu.py function _test_triu (line 30) | def _test_triu(test_case, diagonal, device, dtype): function _test_triu_ (line 46) | def _test_triu_(test_case, diagonal, device, dtype): class TestTriu (line 58) | class TestTriu(flow.unittest.TestCase): method test_triu (line 59) | def test_triu(test_case): method test_triu_with_0_size_data (line 69) | def test_triu_with_0_size_data(test_case): method test_triu_with_0_size_data_fp16 (line 76) | def test_triu_with_0_size_data_fp16(test_case): FILE: python/oneflow/test/modules/test_trunc.py class TestTrunc (line 24) | class TestTrunc(flow.unittest.TestCase): method test_trunc (line 26) | def test_trunc(test_case): FILE: python/oneflow/test/modules/test_trunc_divide.py class TestTruncDivide (line 26) | class TestTruncDivide(flow.unittest.TestCase): method test_elementwise_trunc_divide_random_data (line 28) | def test_elementwise_trunc_divide_random_data(test_case): method test_tensor_truncdiv_scalar_random_data (line 75) | def test_tensor_truncdiv_scalar_random_data(test_case): FILE: python/oneflow/test/modules/test_type_tensor.py class TestTypeTensor (line 74) | class TestTypeTensor(flow.unittest.TestCase): method test_type_tensor (line 75) | def test_type_tensor(test_case): method test_doubletensor_corner_cases (line 90) | def test_doubletensor_corner_cases(test_case): method test_type_tensor_ctor (line 101) | def test_type_tensor_ctor(test_case): FILE: python/oneflow/test/modules/test_unbind.py class TestUnbind (line 30) | class TestUnbind(flow.unittest.TestCase): method test_unbind_flow_with_random_data1 (line 32) | def test_unbind_flow_with_random_data1(test_case): method test_unbind_flow_with_random_data2 (line 39) | def test_unbind_flow_with_random_data2(test_case): method test_unbind_flow_with_random_data3 (line 46) | def test_unbind_flow_with_random_data3(test_case): method test_unbind_flow_with_random_data4 (line 53) | def test_unbind_flow_with_random_data4(test_case): method test_unbind_flow_with_random_data5 (line 60) | def test_unbind_flow_with_random_data5(test_case): FILE: python/oneflow/test/modules/test_unfold.py class TestUnfold (line 26) | class TestUnfold(flow.unittest.TestCase): method test_unfold_with_random_data (line 28) | def test_unfold_with_random_data(test_case): FILE: python/oneflow/test/modules/test_unfold_tensor.py class TestUnfoldTensor (line 28) | class TestUnfoldTensor(flow.unittest.TestCase): method test_unfold_tensor_with_random_data (line 30) | def test_unfold_tensor_with_random_data(test_case): method test_unfold_tensor_with_stride (line 40) | def test_unfold_tensor_with_stride(test_case): method test_unfold_tensor_with_0dim_data (line 53) | def test_unfold_tensor_with_0dim_data(test_case): FILE: python/oneflow/test/modules/test_unique.py function _test_unique_unsorted (line 31) | def _test_unique_unsorted(test_case, device, return_inverse, return_coun... function _test_unique_sorted (line 88) | def _test_unique_sorted(test_case, device, return_inverse, return_counts): class TestUnique (line 135) | class TestUnique(flow.unittest.TestCase): method test_unique (line 137) | def test_unique(test_case): method profile_unique (line 147) | def profile_unique(test_case): FILE: python/oneflow/test/modules/test_unsqueeze.py function _test_unsqueeze (line 29) | def _test_unsqueeze(test_case, device): function _test_unsqueeze_tensor_function (line 43) | def _test_unsqueeze_tensor_function(test_case, device): function _test_unsqueeze_different_dim (line 51) | def _test_unsqueeze_different_dim(test_case, device): function _test_unsqueeze_backward (line 60) | def _test_unsqueeze_backward(test_case, device): class TestUnsqueeze (line 73) | class TestUnsqueeze(flow.unittest.TestCase): method test_unsqueeze (line 74) | def test_unsqueeze(test_case): method test_flow_unsqueeze_with_random_data (line 87) | def test_flow_unsqueeze_with_random_data(test_case): method test_inplace_unsqueeze_with_random_data (line 94) | def test_inplace_unsqueeze_with_random_data(test_case): method test_unsqueeze_with_0_size_data (line 101) | def test_unsqueeze_with_0_size_data(test_case): method test_flow_unsqueeze_bool_with_random_data (line 108) | def test_flow_unsqueeze_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_upsample.py function _test_upsample2d_bilinear (line 28) | def _test_upsample2d_bilinear(test_case, device): function _test_upsample2d_bilinear_aligncorner (line 51) | def _test_upsample2d_bilinear_aligncorner(test_case, device): function _test_UpsamplingNearest2d (line 74) | def _test_UpsamplingNearest2d(test_case, device): function _test_UpsamplingBilinear2d (line 97) | def _test_UpsamplingBilinear2d(test_case, device): function _test_upsample2d_4dim (line 120) | def _test_upsample2d_4dim(test_case, device): function _test_upsample2d_bilinear_4dim (line 171) | def _test_upsample2d_bilinear_4dim(test_case, device): function _test_upsample2d_backward (line 222) | def _test_upsample2d_backward(test_case, device): function _test_upsample2d_bilinear_aligncorner_backward (line 237) | def _test_upsample2d_bilinear_aligncorner_backward(test_case, device): function _test_interpolate_nearest_float_scale (line 252) | def _test_interpolate_nearest_float_scale(test_case, device): function _test_interpolate_bilinear_float_scale (line 280) | def _test_interpolate_bilinear_float_scale(test_case, device): function _test_upsample_bilinear_align_corners (line 338) | def _test_upsample_bilinear_align_corners(test_case, device): class TestUpsample2d (line 356) | class TestUpsample2d(flow.unittest.TestCase): method test_upsample2d (line 357) | def test_upsample2d(test_case): method test_upsample2d_nearest (line 380) | def test_upsample2d_nearest(test_case): method test_upsample2d_nearest_half (line 391) | def test_upsample2d_nearest_half(test_case): method test_upsample2d_bilinear (line 403) | def test_upsample2d_bilinear(test_case): method test_upsample2d_bicubic (line 416) | def test_upsample2d_bicubic(test_case): method test_upsample1d_nearest_output_size (line 428) | def test_upsample1d_nearest_output_size(test_case): method test_upsample2d_nearest_output_size (line 436) | def test_upsample2d_nearest_output_size(test_case): method test_upsample3d_nearest_output_size (line 444) | def test_upsample3d_nearest_output_size(test_case): method test_upsample1d_linear_output_size (line 452) | def test_upsample1d_linear_output_size(test_case): method test_upsample2d_bilinear_output_size (line 461) | def test_upsample2d_bilinear_output_size(test_case): method test_upsample2d_bicubic_output_size (line 469) | def test_upsample2d_bicubic_output_size(test_case): method test_upsample3d_trilinear_output_size (line 477) | def test_upsample3d_trilinear_output_size(test_case): FILE: python/oneflow/test/modules/test_util_ops.py function _test_isnan (line 24) | def _test_isnan(test_case, shape, dtype, device): function _test_isinf (line 33) | def _test_isinf(test_case, shape, dtype, device): function _test_isfinite (line 42) | def _test_isfinite(test_case, shape, dtype, device): class TestUtilOps (line 54) | class TestUtilOps(flow.unittest.TestCase): method test_util_ops (line 55) | def test_util_ops(test_case): FILE: python/oneflow/test/modules/test_utils.py function _test_flatten_dense_tensors (line 27) | def _test_flatten_dense_tensors(test_case, device): function _test_unflatten_dense_tensors (line 43) | def _test_unflatten_dense_tensors(test_case, device): class TestUtilsFunction (line 72) | class TestUtilsFunction(flow.unittest.TestCase): method test_utils_function (line 73) | def test_utils_function(test_case): FILE: python/oneflow/test/modules/test_var.py class TestVar (line 26) | class TestVar(flow.unittest.TestCase): method test_flow_var_all_dim_with_random_data (line 28) | def test_flow_var_all_dim_with_random_data(test_case): method test_flow_var_one_dim_with_random_data (line 35) | def test_flow_var_one_dim_with_random_data(test_case): method test_flow_var_one_dim_with_random_half_data (line 50) | def test_flow_var_one_dim_with_random_half_data(test_case): method test_flow_var_0_size_data_with_random_data (line 62) | def test_flow_var_0_size_data_with_random_data(test_case): method test_flow_var_0_size_data_with_random_half_data (line 74) | def test_flow_var_0_size_data_with_random_half_data(test_case): method test_flow_var_all_dim_with_random_data_n5 (line 86) | def test_flow_var_all_dim_with_random_data_n5(test_case): FILE: python/oneflow/test/modules/test_view.py function _test_view (line 28) | def _test_view(test_case, device): function _test_view_flow_size (line 52) | def _test_view_flow_size(test_case, device): class TestView (line 77) | class TestView(flow.unittest.TestCase): method test_view (line 79) | def test_view(test_case): method test_view_with_0_dim_data (line 90) | def test_view_with_0_dim_data(test_case): FILE: python/oneflow/test/modules/test_vsplit.py class TestVsplitVec (line 25) | class TestVsplitVec(flow.unittest.TestCase): method test_flow_vsplit_vec (line 27) | def test_flow_vsplit_vec(test_case): method test_flow_vsplit_vec_with_stride (line 40) | def test_flow_vsplit_vec_with_stride(test_case): class TestVsplitInt (line 57) | class TestVsplitInt(flow.unittest.TestCase): method test_flow_vsplit_int (line 59) | def test_flow_vsplit_int(test_case): FILE: python/oneflow/test/modules/test_weight_norm.py function _test_weightnorm (line 40) | def _test_weightnorm(test_case, device, dim): function _test_weightnorm_backward (line 93) | def _test_weightnorm_backward(test_case, device, dim): class TestWeightNorm (line 134) | class TestWeightNorm(flow.unittest.TestCase): method test_weightnorm (line 135) | def test_weightnorm(test_case): method test_weight_norm_with_random_data (line 150) | def test_weight_norm_with_random_data(test_case): FILE: python/oneflow/test/modules/test_where.py function _test_where (line 29) | def _test_where(test_case, device): function _test_where_broadcast (line 46) | def _test_where_broadcast(test_case, device): function _test_where_scalar (line 71) | def _test_where_scalar(test_case, device): function _test_where_dim4 (line 90) | def _test_where_dim4(test_case, device): function _test_where_backward (line 109) | def _test_where_backward(test_case, device): function _test_where_broadcast_backward (line 136) | def _test_where_broadcast_backward(test_case, device): function _test_where_broadcast_x_backward (line 167) | def _test_where_broadcast_x_backward(test_case, device): function _test_where_x_y_none (line 189) | def _test_where_x_y_none(test_case, device): function _test_where_scalar (line 204) | def _test_where_scalar(test_case, device): class TestWhere (line 213) | class TestWhere(flow.unittest.TestCase): method test_where (line 214) | def test_where(test_case): method test_flow_where_tensor_with_random_data (line 232) | def test_flow_where_tensor_with_random_data(test_case): method test_flow_where_tensor_with_0dim_data (line 242) | def test_flow_where_tensor_with_0dim_data(test_case): method test_flow_where_tensor_broadcast_with_random_data (line 252) | def test_flow_where_tensor_broadcast_with_random_data(test_case): method test_flow_where_scalar_x_with_random_data (line 262) | def test_flow_where_scalar_x_with_random_data(test_case): method test_flow_where_scalar_x_broadcast_with_random_data (line 274) | def test_flow_where_scalar_x_broadcast_with_random_data(test_case): method test_flow_where_scalar_x_int_with_random_data (line 286) | def test_flow_where_scalar_x_int_with_random_data(test_case): method test_flow_where_scalar_y_with_random_data (line 296) | def test_flow_where_scalar_y_with_random_data(test_case): method test_flow_where_scalar_y_broadcast_with_random_data (line 308) | def test_flow_where_scalar_y_broadcast_with_random_data(test_case): method test_flow_where_scalar_y_int_with_random_data (line 320) | def test_flow_where_scalar_y_int_with_random_data(test_case): method test_flow_where_scalar_xy_with_random_data (line 330) | def test_flow_where_scalar_xy_with_random_data(test_case): method test_flow_where_scalar_xy_int_with_random_data (line 340) | def test_flow_where_scalar_xy_int_with_random_data(test_case): method test_flow_where_tensor_bool_with_random_data (line 350) | def test_flow_where_tensor_bool_with_random_data(test_case): method test_flow_where_tensor_broadcast_bool_with_random_data (line 360) | def test_flow_where_tensor_broadcast_bool_with_random_data(test_case): method test_flow_where_scalar_x_bool_with_random_data (line 370) | def test_flow_where_scalar_x_bool_with_random_data(test_case): method test_flow_where_scalar_x_broadcast_bool_with_random_data (line 382) | def test_flow_where_scalar_x_broadcast_bool_with_random_data(test_case): method test_flow_where_scalar_y_bool_with_random_data (line 394) | def test_flow_where_scalar_y_bool_with_random_data(test_case): method test_flow_where_scalar_y_broadcast_bool_with_random_data (line 406) | def test_flow_where_scalar_y_broadcast_bool_with_random_data(test_case): method test_flow_where_scalar_xy_bool_with_random_data (line 418) | def test_flow_where_scalar_xy_bool_with_random_data(test_case): FILE: python/oneflow/test/modules/test_zeropad2d.py function _np_zero_pad2d_grad (line 32) | def _np_zero_pad2d_grad(src, dest, padding): function _test_ZeroPad2d (line 68) | def _test_ZeroPad2d(test_case, shape, padding, value, device): class TestZeroPad2dModule (line 95) | class TestZeroPad2dModule(flow.unittest.TestCase): method test_ConstantPad2d (line 96) | def test_ConstantPad2d(test_case): FILE: python/oneflow/test/profiler/test_events.py class TestEventAndEvents (line 23) | class TestEventAndEvents(flow.unittest.TestCase): method test_event (line 24) | def test_event(test_case): method test_event_update (line 51) | def test_event_update(test_case): method test_events (line 59) | def test_events(test_case): FILE: python/oneflow/test/profiler/test_profile_lenet.py class LeNet (line 28) | class LeNet(nn.Module): method __init__ (line 29) | def __init__(self): method forward (line 37) | def forward(self, x): function get_event (line 49) | def get_event(events, name: str, input_shapes: str = "", attributes: str... function _test_lenet (line 64) | def _test_lenet( class TestProfileLenet (line 136) | class TestProfileLenet(flow.unittest.TestCase): method test_lenet_cpu (line 137) | def test_lenet_cpu(test_case): method test_lenet_cuda (line 145) | def test_lenet_cuda(test_case): FILE: python/oneflow/test/tensor/test_autocast.py class TestAutoCast (line 26) | class TestAutoCast(flow.unittest.TestCase): method test_autocast_half_mm (line 28) | def test_autocast_half_mm(test_case): method test_autocast_half_mm_add (line 36) | def test_autocast_half_mm_add(test_case): method test_autocast_graph (line 45) | def test_autocast_graph(test_case): FILE: python/oneflow/test/tensor/test_bfloat16_activation.py class TestBfloat16Activatian (line 24) | class TestBfloat16Activatian(flow.unittest.TestCase): method test_tan_with_random_data (line 25) | def test_tan_with_random_data(test_case): method test_tanh_with_random_data (line 40) | def test_tanh_with_random_data(test_case): method test_sin_with_random_data (line 55) | def test_sin_with_random_data(test_case): method test_sinh_with_random_data (line 70) | def test_sinh_with_random_data(test_case): method test_cos_with_random_data (line 85) | def test_cos_with_random_data(test_case): method test_cosh_with_random_data (line 100) | def test_cosh_with_random_data(test_case): method test_atan_with_random_data (line 115) | def test_atan_with_random_data(test_case): method test_atanh_with_random_data (line 130) | def test_atanh_with_random_data(test_case): method test_asin_with_random_data (line 145) | def test_asin_with_random_data(test_case): method test_asinh_with_random_data (line 160) | def test_asinh_with_random_data(test_case): method test_acos_with_random_data (line 175) | def test_acos_with_random_data(test_case): method test_acosh_with_random_data (line 190) | def test_acosh_with_random_data(test_case): method test_sqrt_with_random_data (line 205) | def test_sqrt_with_random_data(test_case): method test_square_with_random_data (line 220) | def test_square_with_random_data(test_case): method test_exp_with_random_data (line 235) | def test_exp_with_random_data(test_case): method test_exp2_with_random_data (line 250) | def test_exp2_with_random_data(test_case): method test_ceil_with_random_data (line 265) | def test_ceil_with_random_data(test_case): method test_erf_with_random_data (line 280) | def test_erf_with_random_data(test_case): method test_erfc_with_random_data (line 295) | def test_erfc_with_random_data(test_case): method test_floor_with_random_data (line 310) | def test_floor_with_random_data(test_case): method test_expm1_with_random_data (line 325) | def test_expm1_with_random_data(test_case): method test_lgamma_with_random_data (line 340) | def test_lgamma_with_random_data(test_case): method test_log_with_random_data (line 355) | def test_log_with_random_data(test_case): method test_log2_with_random_data (line 370) | def test_log2_with_random_data(test_case): method test_log1p_with_random_data (line 385) | def test_log1p_with_random_data(test_case): method test_sigmoid_with_random_data (line 400) | def test_sigmoid_with_random_data(test_case): method test_round_with_random_data (line 415) | def test_round_with_random_data(test_case): method test_rsqrt_with_random_data (line 430) | def test_rsqrt_with_random_data(test_case): method test_softplus_with_random_data (line 445) | def test_softplus_with_random_data(test_case): method test_softsign_with_random_data (line 460) | def test_softsign_with_random_data(test_case): method test_softshrink_with_random_data (line 475) | def test_softshrink_with_random_data(test_case): method test_silu_with_random_data (line 490) | def test_silu_with_random_data(test_case): method test_selu_with_random_data (line 505) | def test_selu_with_random_data(test_case): method test_mish_with_random_data (line 520) | def test_mish_with_random_data(test_case): method test_gelu_with_random_data (line 535) | def test_gelu_with_random_data(test_case): method test_elu_with_random_data (line 550) | def test_elu_with_random_data(test_case): method test_celu_with_random_data (line 566) | def test_celu_with_random_data(test_case): method test_hardswish_with_random_data (line 582) | def test_hardswish_with_random_data(test_case): method test_hardswish_with_random_data (line 598) | def test_hardswish_with_random_data(test_case): method test_hardshrink_with_random_data (line 614) | def test_hardshrink_with_random_data(test_case): method test_hardtanh_with_random_data (line 630) | def test_hardtanh_with_random_data(test_case): method test_leakyrelu_with_random_data (line 646) | def test_leakyrelu_with_random_data(test_case): method test_threshold_with_random_data (line 662) | def test_threshold_with_random_data(test_case): method test_logsinmoid_with_random_data (line 678) | def test_logsinmoid_with_random_data(test_case): method test_digamma_with_random_data (line 694) | def test_digamma_with_random_data(test_case): FILE: python/oneflow/test/tensor/test_complex.py function compare_result (line 69) | def compare_result(a, b, rtol=1e-5, atol=1e-8): function _np_zero_pad2d_grad (line 75) | def _np_zero_pad2d_grad(src, dest, padding): function _test_ZeroPad2d (line 111) | def _test_ZeroPad2d(test_case, shape, padding, value, device, rtol, atol): class TestTensorComplex64 (line 139) | class TestTensorComplex64(unittest.TestCase): method setUp (line 140) | def setUp(self): method test_from_numpy (line 162) | def test_from_numpy(self): method test_tensor (line 177) | def test_tensor(self): method test_tensor_cuda (line 194) | def test_tensor_cuda(self): method test_slice (line 210) | def test_slice(self): method test_slice_cuda (line 229) | def test_slice_cuda(self): method test_new_tensor (line 247) | def test_new_tensor(self): method test_new_empty (line 256) | def test_new_empty(self): method test_ones (line 264) | def test_ones(self): method test_new_ones (line 272) | def test_new_ones(self): method test_zeros (line 281) | def test_zeros(self): method test_new_zeros (line 289) | def test_new_zeros(self): method test_full (line 298) | def test_full(self): method test_new_full (line 306) | def test_new_full(self): method test_real (line 315) | def test_real(self): method test_imag (line 322) | def test_imag(self): method test_conj (line 329) | def test_conj(self): method test_conj_physical (line 337) | def test_conj_physical(self): method test_real_cuda (line 346) | def test_real_cuda(self): method test_imag_cuda (line 354) | def test_imag_cuda(self): method test_conj_cuda (line 362) | def test_conj_cuda(self): method test_conj_physical_cuda (line 371) | def test_conj_physical_cuda(self): method test_add_cpu (line 381) | def test_add_cpu(self): method test_add_cuda (line 410) | def test_add_cuda(self): method test_sub_cpu (line 444) | def test_sub_cpu(self): method test_sub_cuda (line 473) | def test_sub_cuda(self): method test_mul_cpu (line 507) | def test_mul_cpu(self): method test_mul_cuda (line 536) | def test_mul_cuda(self): method test_sum_cpu (line 570) | def test_sum_cpu(self): method test_sum_cuda (line 597) | def test_sum_cuda(self): method test_equal_cpu (line 626) | def test_equal_cpu(self): method test_equal_cuda (line 661) | def test_equal_cuda(self): method test_constant_pad (line 698) | def test_constant_pad(self): method test_cast (line 711) | def test_cast(self): method test_cast_cuda (line 745) | def test_cast_cuda(self): class TestTensorComplex128 (line 779) | class TestTensorComplex128(TestTensorComplex64): method setUp (line 780) | def setUp(self): class TestAutograd (line 803) | class TestAutograd(unittest.TestCase): method test_backward (line 804) | def test_backward(self): method test_backward_cuda (line 813) | def test_backward_cuda(self): method test_grad (line 821) | def test_grad(self): method test_grad_cuda (line 832) | def test_grad_cuda(self): FILE: python/oneflow/test/tensor/test_data_ptr.py class TestDataPtr (line 21) | class TestDataPtr(unittest.TestCase): method test_equality (line 23) | def test_equality(test_case): method test_global_tensor (line 35) | def test_global_tensor(test_case): FILE: python/oneflow/test/tensor/test_global_tensor.py class TestTensor (line 29) | class TestTensor(flow.unittest.TestCase): method test_creating_global_tensor (line 31) | def test_creating_global_tensor(test_case): method test_construct_local_from_global_tensor (line 59) | def test_construct_local_from_global_tensor(test_case): method test_global_set_data (line 72) | def test_global_set_data(test_case): method test_global_tensor_autograd_related_methods (line 87) | def test_global_tensor_autograd_related_methods(test_case): method test_global_tensor_unsupported_property (line 143) | def test_global_tensor_unsupported_property(test_case): method test_global_tensor_2d_sbp_init (line 159) | def test_global_tensor_2d_sbp_init(test_case): method test_copy (line 188) | def test_copy(test_case): FILE: python/oneflow/test/tensor/test_global_tensor_and_ndarray_compatibility.py function do_test_compute_op (line 48) | def do_test_compute_op(test_case, ndim, placement, sbp): class TestGlobalTensorAndNdarrayCompatibility (line 67) | class TestGlobalTensorAndNdarrayCompatibility(flow.unittest.TestCase): method test_tensor_and_ndarray_compatibility (line 69) | def test_tensor_and_ndarray_compatibility(test_case): FILE: python/oneflow/test/tensor/test_global_tensor_indexing.py function _randint (line 28) | def _randint(low, high): function _cpu_global_tensor (line 35) | def _cpu_global_tensor(tensor): function _assert_tensor_equal (line 39) | def _assert_tensor_equal(test_case, tensor1, tensor2, atol=0.0, rtol=0.0): function global_broadcast_consec (line 46) | def global_broadcast_consec(size, start=1): function _test_basic_slice (line 55) | def _test_basic_slice(test_case, placement): function _test_advanced_indexing (line 257) | def _test_advanced_indexing(test_case, placement, dtype): function _test_combined_indexing (line 434) | def _test_combined_indexing(test_case, placement, dtype): function _test_single_int (line 680) | def _test_single_int(test_case, placement): function _test_multiple_int (line 687) | def _test_multiple_int(test_case, placement): function _test_none (line 693) | def _test_none(test_case, placement): function _test_step (line 702) | def _test_step(test_case, placement): function _test_step_assignment (line 712) | def _test_step_assignment(test_case, placement): function _test_bool_indices (line 723) | def _test_bool_indices(test_case, placement): function _test_multiple_bool_indices (line 738) | def _test_multiple_bool_indices(test_case, placement): function _test_int_indices (line 752) | def _test_int_indices(test_case, placement): function _test_int_indices2d (line 760) | def _test_int_indices2d(test_case, placement): function _test_int_indices_broadcast (line 773) | def _test_int_indices_broadcast(test_case, placement): function _test_empty_index (line 787) | def _test_empty_index(test_case, placement): function _test_empty_ndim_index (line 809) | def _test_empty_ndim_index(test_case, placement): function _test_empty_ndim_index_bool (line 835) | def _test_empty_ndim_index_bool(test_case, placement): function _test_empty_slice (line 849) | def _test_empty_slice(test_case, placement): function _test_index_getitem_copy_bools_slices (line 857) | def _test_index_getitem_copy_bools_slices(test_case, placement): function _test_setitem_scalars (line 870) | def _test_setitem_scalars(test_case, placement): function _test_basic_advanced_combined (line 924) | def _test_basic_advanced_combined(test_case, placement): function _test_ellipsis_tensor (line 941) | def _test_ellipsis_tensor(test_case, placement): class TestGlobalIndexing (line 964) | class TestGlobalIndexing(flow.unittest.TestCase): method test_global_slice (line 966) | def test_global_slice(test_case): FILE: python/oneflow/test/tensor/test_lazy_tensor_indexing.py function get_graph_output (line 27) | def get_graph_output(*args, func): function setitem_and_return (line 42) | def setitem_and_return(ref, idx, value): function _randint (line 47) | def _randint(low, high): function _cpu_global_tensor (line 54) | def _cpu_global_tensor(tensor): function _assert_tensor_equal (line 58) | def _assert_tensor_equal(test_case, tensor1, tensor2, atol=0.0, rtol=0.0): function global_broadcast_consec (line 65) | def global_broadcast_consec(size, start=1): function _test_basic_slice (line 74) | def _test_basic_slice(test_case, placement): function _test_advanced_indexing (line 225) | def _test_advanced_indexing(test_case, placement, dtype): function _test_combined_indexing (line 419) | def _test_combined_indexing(test_case, placement, dtype): function _test_single_int (line 641) | def _test_single_int(test_case, placement): function _test_multiple_int (line 647) | def _test_multiple_int(test_case, placement): function _test_none (line 653) | def _test_none(test_case, placement): function _test_step (line 670) | def _test_step(test_case, placement): function _test_step_assignment (line 686) | def _test_step_assignment(test_case, placement): function _test_multiple_bool_indices (line 700) | def _test_multiple_bool_indices(test_case, placement): function _test_int_indices (line 714) | def _test_int_indices(test_case, placement): function _test_int_indices2d (line 728) | def _test_int_indices2d(test_case, placement): function _test_int_indices_broadcast (line 744) | def _test_int_indices_broadcast(test_case, placement): function _test_empty_index (line 758) | def _test_empty_index(test_case, placement): function _test_empty_ndim_index (line 780) | def _test_empty_ndim_index(test_case, placement): function _test_empty_slice (line 803) | def _test_empty_slice(test_case, placement): function _test_setitem_scalars (line 810) | def _test_setitem_scalars(test_case, placement): function _test_ellipsis_tensor (line 873) | def _test_ellipsis_tensor(test_case, placement): function _test_bool_indices (line 896) | def _test_bool_indices(test_case, placement): class TestGlobalIndexing (line 912) | class TestGlobalIndexing(flow.unittest.TestCase): method test_global_slice (line 918) | def test_global_slice(test_case): method test_bool_indices (line 941) | def test_bool_indices(test_case): FILE: python/oneflow/test/tensor/test_meta_tensor.py class CustomModule (line 24) | class CustomModule(nn.Module): method __init__ (line 25) | def __init__(self, foo, bar, device=None): class TestMetaTensor (line 43) | class TestMetaTensor(flow.unittest.TestCase): method test_meta_tensor_local_mode_without_data (line 45) | def test_meta_tensor_local_mode_without_data(test_case): method test_meta_tensor_local_mode_with_data (line 53) | def test_meta_tensor_local_mode_with_data(test_case): method test_meta_tensor_func_local_mode_without_data (line 61) | def test_meta_tensor_func_local_mode_without_data(test_case): method test_meta_tensor_func_local_mode_with_data (line 69) | def test_meta_tensor_func_local_mode_with_data(test_case): method test_meta_tensor_local_mode_ones (line 77) | def test_meta_tensor_local_mode_ones(test_case): method test_meta_tensor_local_mode_linear (line 85) | def test_meta_tensor_local_mode_linear(test_case): method test_skip_init_function (line 94) | def test_skip_init_function(test_case): method test_skip_init_function_custom_module (line 103) | def test_skip_init_function_custom_module(test_case): method test_meta_tensor_local_mode_clone (line 120) | def test_meta_tensor_local_mode_clone(test_case): method test_meta_tensor_global_mode_without_data (line 128) | def test_meta_tensor_global_mode_without_data(test_case): method test_meta_tensor_global_mode_with_data (line 143) | def test_meta_tensor_global_mode_with_data(test_case): method test_meta_tensor_func_global_mode_without_data (line 158) | def test_meta_tensor_func_global_mode_without_data(test_case): method test_meta_tensor_global_mode_clone (line 173) | def test_meta_tensor_global_mode_clone(test_case): method test_meta_tensor_calculate (line 184) | def test_meta_tensor_calculate(test_case): FILE: python/oneflow/test/tensor/test_new_tensor.py class TestNewTensor (line 23) | class TestNewTensor(flow.unittest.TestCase): method test_new_tensor_local_mode_with_default_args (line 25) | def test_new_tensor_local_mode_with_default_args(test_case): method test_new_tensor_local_mode_with_spec_args (line 34) | def test_new_tensor_local_mode_with_spec_args(test_case): method test_new_tensor_global_mode_with_default_args (line 42) | def test_new_tensor_global_mode_with_default_args(test_case): method test_new_tensor_global_mode_with_spec_args (line 54) | def test_new_tensor_global_mode_with_spec_args(test_case): method test_new_cuda_bfloat16_local_tensor_with_numpy (line 68) | def test_new_cuda_bfloat16_local_tensor_with_numpy(test_case): method test_new_cpu_bfloat16_local_tensor_with_numpy (line 79) | def test_new_cpu_bfloat16_local_tensor_with_numpy(test_case): FILE: python/oneflow/test/tensor/test_parameter.py class TestParameter (line 27) | class TestParameter(flow.unittest.TestCase): method test_parameter_grad_fn_none (line 29) | def test_parameter_grad_fn_none(test_case): method test_parameter_set_data_autograd_meta (line 36) | def test_parameter_set_data_autograd_meta(test_case): method test_parameter_inplace_modify_data (line 48) | def test_parameter_inplace_modify_data(test_case): method test_parameter_set_data (line 53) | def test_parameter_set_data(test_case): FILE: python/oneflow/test/tensor/test_safetensors.py function _test_save_safetensors (line 33) | def _test_save_safetensors(save_path): function _test_load_safetensors (line 40) | def _test_load_safetensors(load_path): class TestSafetensors (line 51) | class TestSafetensors(flow.unittest.TestCase): method test_safetensors (line 52) | def test_safetensors(test_case): FILE: python/oneflow/test/tensor/test_tensor_and_ndarray_compatibility.py function _test_compute_operator (line 49) | def _test_compute_operator(test_case, shape, dtype): function _test_logic_operator (line 79) | def _test_logic_operator(test_case, shape): function _test_compare_operator (line 91) | def _test_compare_operator(test_case, shape): class TestTensorAndNdarrayCompatibility (line 106) | class TestTensorAndNdarrayCompatibility(flow.unittest.TestCase): method test_op_compatibility (line 107) | def test_op_compatibility(test_case): FILE: python/oneflow/test/tensor/test_tensor_exponential.py function _test_exponential (line 28) | def _test_exponential(test_case, device, seed, lambd, dtype): class TestExponential (line 86) | class TestExponential(flow.unittest.TestCase): method test_exponential (line 87) | def test_exponential(test_case): FILE: python/oneflow/test/tensor/test_tensor_indexing.py function _test_numpy_scalar_indexing (line 27) | def _test_numpy_scalar_indexing(test_case, numpy_x, np_scalar): function _test_numpy_scalar_advance_indexing (line 52) | def _test_numpy_scalar_advance_indexing(test_case, numpy_x, np_scalar): function _test_basic_slice (line 84) | def _test_basic_slice(test_case, numpy_x): function _test_advanced_indexing (line 144) | def _test_advanced_indexing(test_case, numpy_x): function _test_advanced_indexing_array (line 202) | def _test_advanced_indexing_array(test_case, numpy_x, dtype): function _test_combining_indexing (line 233) | def _test_combining_indexing(test_case, numpy_x): function _test_mask_getitem (line 248) | def _test_mask_getitem(test_case, numpy_x): function _test_mask_setitem (line 265) | def _test_mask_setitem(test_case, numpy_x): function _test_list_indexing_using_scalar_tensor (line 290) | def _test_list_indexing_using_scalar_tensor(test_case, dtype): class TestTensorIndexing (line 298) | class TestTensorIndexing(flow.unittest.TestCase): method test_basic_slice (line 299) | def test_basic_slice(test_case): method test_advanced_indexing (line 309) | def test_advanced_indexing(test_case): method test_advanced_indexing_array (line 319) | def test_advanced_indexing_array(test_case): method test_combining_indexing (line 332) | def test_combining_indexing(test_case): method test_numpy_scalar_indexing (line 342) | def test_numpy_scalar_indexing(test_case): method test_mask_getitem (line 364) | def test_mask_getitem(test_case): method test_mask_setitem (line 388) | def test_mask_setitem(test_case): method test_combined_mask_setitem (line 398) | def test_combined_mask_setitem(test_case): method test_non_contiguous_combined_mask_setitem (line 412) | def test_non_contiguous_combined_mask_setitem(test_case): method test_combined_indexing_setitem (line 429) | def test_combined_indexing_setitem(test_case): method test_expand_dim_setitem (line 437) | def test_expand_dim_setitem(test_case): method test_advanced_indexing_with_scalar_index (line 446) | def test_advanced_indexing_with_scalar_index(test_case): method test_list_indexing_using_scalar_tensor (line 452) | def test_list_indexing_using_scalar_tensor(test_case): method test_advanced_indexing_with_0_size_tensor (line 462) | def test_advanced_indexing_with_0_size_tensor(test_case): method test_dataloader_indexing_with_1_dim_tensor (line 472) | def test_dataloader_indexing_with_1_dim_tensor(test_case): method test_indecies_on_different_devices (line 481) | def test_indecies_on_different_devices(test_case): class TestTensorIndexingMultiGpu (line 493) | class TestTensorIndexingMultiGpu(flow.unittest.TestCase): method test_indecies_on_different_devices (line 495) | def test_indecies_on_different_devices(test_case): FILE: python/oneflow/test/tensor/test_tensor_indexing2.py function _assert_tensor_equal (line 31) | def _assert_tensor_equal(test_case, tensor1, tensor2, atol=0.0, rtol=0.0): function consec (line 35) | def consec(size, start=1): function _test_basic_slice (line 44) | def _test_basic_slice(test_case, device, dtype): function _test_advanced_indexing (line 210) | def _test_advanced_indexing(test_case, device, dtype): function _test_combined_indexing (line 463) | def _test_combined_indexing(test_case, device, dtype): function _test_single_int (line 689) | def _test_single_int(test_case, device): function _test_multiple_int (line 694) | def _test_multiple_int(test_case, device): function _test_none (line 700) | def _test_none(test_case, device): function _test_step (line 708) | def _test_step(test_case, device): function _test_step_assignment (line 717) | def _test_step_assignment(test_case, device): function _test_bool_indices (line 724) | def _test_bool_indices(test_case, device): function _test_multiple_bool_indices (line 742) | def _test_multiple_bool_indices(test_case, device): function _test_int_indices (line 750) | def _test_int_indices(test_case, device): function _test_int_indices2d (line 757) | def _test_int_indices2d(test_case, device): function _test_int_indices_broadcast (line 764) | def _test_int_indices_broadcast(test_case, device): function _test_empty_index (line 772) | def _test_empty_index(test_case, device): function _test_empty_ndim_index (line 787) | def _test_empty_ndim_index(test_case, device): function _test_empty_ndim_index_bool (line 813) | def _test_empty_ndim_index_bool(test_case, device): function _test_empty_slice (line 820) | def _test_empty_slice(test_case, device): function _test_index_getitem_copy_bools_slices (line 830) | def _test_index_getitem_copy_bools_slices(test_case, device): function _test_setitem_scalars (line 846) | def _test_setitem_scalars(test_case, device): function _test_basic_advanced_combined (line 885) | def _test_basic_advanced_combined(test_case, device): function _test_ellipsis_tensor (line 901) | def _test_ellipsis_tensor(test_case, device): class TestIndexing (line 915) | class TestIndexing(flow.unittest.TestCase): method test_slice (line 916) | def test_slice(test_case): FILE: python/oneflow/test/tensor/test_tensor_is_view.py function _test_is_view (line 27) | def _test_is_view(test_case, device): class TestTensorIsView (line 37) | class TestTensorIsView(flow.unittest.TestCase): method test_is_view (line 38) | def test_is_view(test_case): FILE: python/oneflow/test/tensor/test_tensor_part_1.py class TestTensor (line 28) | class TestTensor(flow.unittest.TestCase): method test_numpy_and_default_dtype (line 30) | def test_numpy_and_default_dtype(test_case): method test_tensor_deepcopy (line 54) | def test_tensor_deepcopy(test_case): method test_tensor_property (line 64) | def test_tensor_property(test_case): method test_copy_to_and_from_numpy (line 73) | def test_copy_to_and_from_numpy(test_case): method test_inplace_copy_from_contiguous_numpy (line 88) | def test_inplace_copy_from_contiguous_numpy(test_case): method test_inplace_copy_from_non_contiguous_numpy (line 95) | def test_inplace_copy_from_non_contiguous_numpy(test_case): method test_construct_from_numpy_or_list (line 102) | def test_construct_from_numpy_or_list(test_case): method test_construct_from_another_tensor (line 119) | def test_construct_from_another_tensor(test_case): method test_construct_np_array_from_tensor (line 128) | def test_construct_np_array_from_tensor(test_case): method test_tensor_sign_with_random_data (line 138) | def test_tensor_sign_with_random_data(test_case): method test_flow_tensor_gather_with_random_data (line 146) | def test_flow_tensor_gather_with_random_data(test_case): method _test_tensor_init_methods (line 161) | def _test_tensor_init_methods(test_case, tensor_creator, get_numpy): method _test_non_contiguous_tensor_init_methods (line 209) | def _test_non_contiguous_tensor_init_methods(test_case, tensor_creator... method test_local_tensor_init_methods (line 255) | def test_local_tensor_init_methods(test_case): method test_global_tensor_init_methods (line 267) | def test_global_tensor_init_methods(test_case): method test_tensor_with_single_int (line 280) | def test_tensor_with_single_int(test_case): method test_tensor_device (line 287) | def test_tensor_device(test_case): method test_tensor_set_data_autograd_meta (line 298) | def test_tensor_set_data_autograd_meta(test_case): method test_tensor_set_data (line 306) | def test_tensor_set_data(test_case): method test_tensor_set_ref_tensor (line 318) | def test_tensor_set_ref_tensor(test_case): method test_tensor_unsupported_property (line 332) | def test_tensor_unsupported_property(test_case): method test_tensor_to_bool (line 352) | def test_tensor_to_bool(test_case): method test_tensor_autograd_fill_cpu (line 366) | def test_tensor_autograd_fill_cpu(test_case): method test_tensor_autograd_fill_cuda (line 412) | def test_tensor_autograd_fill_cuda(test_case): method test_tensor_register_post_grad_accumulation_hook (line 452) | def test_tensor_register_post_grad_accumulation_hook(test_case): method test_tensor_register_hook (line 478) | def test_tensor_register_hook(test_case): method test_non_leaf_tensor_register_hook (line 510) | def test_non_leaf_tensor_register_hook(test_case): method test_user_defined_data (line 523) | def test_user_defined_data(test_case): method test_local_tensor_and_op (line 535) | def test_local_tensor_and_op(test_case): method test_matmul_with_random_data (line 547) | def test_matmul_with_random_data(test_case): method test_mv_with_random_data (line 558) | def test_mv_with_random_data(test_case): method test_mm_with_random_data (line 568) | def test_mm_with_random_data(test_case): method test_tensor_to_list (line 578) | def test_tensor_to_list(test_case): method test_tensor_nelement (line 584) | def test_tensor_nelement(test_case): method test_tensor_numel (line 590) | def test_tensor_numel(test_case): method test_tensor_print (line 596) | def test_tensor_print(test_case): method test_indexing (line 613) | def test_indexing(test_case): method test_setitem_with_random_data (line 648) | def test_setitem_with_random_data(test_case): method test_div (line 669) | def test_div(test_case): method test_mul (line 689) | def test_mul(test_case): method test_mul_inplace_tensor (line 706) | def test_mul_inplace_tensor(test_case): method test_broadcast_mul_inplace_tensor (line 720) | def test_broadcast_mul_inplace_tensor(test_case): method test_div_inplace_tensor (line 730) | def test_div_inplace_tensor(test_case): method test_broadcast_div_inplace_tensor (line 744) | def test_broadcast_div_inplace_tensor(test_case): method test_add_inplace_tensor (line 754) | def test_add_inplace_tensor(test_case): method test_broadcast_add_inplace_tensor (line 768) | def test_broadcast_add_inplace_tensor(test_case): method test_sub_inplace_tensor (line 778) | def test_sub_inplace_tensor(test_case): method test_broadcast_sub_inplace_tensor (line 792) | def test_broadcast_sub_inplace_tensor(test_case): method test_add_tensor_method (line 801) | def test_add_tensor_method(test_case): method test_sub_tensor_method (line 817) | def test_sub_tensor_method(test_case): method test_sum (line 833) | def test_sum(test_case): method test_argwhere (line 840) | def test_argwhere(test_case): method test_tensor_argmax_with_random_data (line 852) | def test_tensor_argmax_with_random_data(test_case): method test_max_bool_input_with_random_data (line 860) | def test_max_bool_input_with_random_data(test_case): method test_min_bool_input_with_random_data (line 869) | def test_min_bool_input_with_random_data(test_case): method test_tensor_tanh_with_random_data (line 879) | def test_tensor_tanh_with_random_data(test_case): method test_flow_tensor_asin_with_random_data (line 887) | def test_flow_tensor_asin_with_random_data(test_case): method test_flow_tensor_arcsin_with_random_data (line 895) | def test_flow_tensor_arcsin_with_random_data(test_case): method test_flow_tensor_asinh_with_random_data (line 903) | def test_flow_tensor_asinh_with_random_data(test_case): method test_flow_tensor_arcsinh_with_random_data (line 911) | def test_flow_tensor_arcsinh_with_random_data(test_case): method test_flow_tensor_sinh_with_random_data (line 919) | def test_flow_tensor_sinh_with_random_data(test_case): method test_flow_tensor_atan2_with_random_data (line 927) | def test_flow_tensor_atan2_with_random_data(test_case): method test_dot (line 936) | def test_dot(test_case): method test_arccos_tensor_with_random_data (line 946) | def test_arccos_tensor_with_random_data(test_case): method test_arccosh_tensor_with_random_data (line 954) | def test_arccosh_tensor_with_random_data(test_case): method test_acosh_tensor_with_random_data (line 962) | def test_acosh_tensor_with_random_data(test_case): method test_sort_tensor_with_random_data (line 970) | def test_sort_tensor_with_random_data(test_case): method test_sort_tensor_return_type (line 978) | def test_sort_tensor_return_type(test_case): method test_argsort_tensor_with_random_data (line 986) | def test_argsort_tensor_with_random_data(test_case): method test_mean_with_random_data (line 993) | def test_mean_with_random_data(test_case): method test_log_tensor_with_random_data (line 1000) | def test_log_tensor_with_random_data(test_case): method test_log1p_tensor_with_random_data (line 1006) | def test_log1p_tensor_with_random_data(test_case): method test_log2_tensor_with_random_data (line 1012) | def test_log2_tensor_with_random_data(test_case): method test_log10_tensor_with_random_data (line 1018) | def test_log10_tensor_with_random_data(test_case): method test_neg_tensor_with_random_data (line 1024) | def test_neg_tensor_with_random_data(test_case): method test_negative_tensor_with_random_data (line 1030) | def test_negative_tensor_with_random_data(test_case): method test_neg_tensor_with_random_data (line 1036) | def test_neg_tensor_with_random_data(test_case): method test_greater_tensor_with_random_data (line 1042) | def test_greater_tensor_with_random_data(test_case): method test_less_tensor_with_random_data (line 1049) | def test_less_tensor_with_random_data(test_case): method test_tensor_topk_with_random_data (line 1056) | def test_tensor_topk_with_random_data(test_case): method test_tensor_topk_return_type (line 1068) | def test_tensor_topk_return_type(test_case): method test_flow_fmod_element_with_random_data (line 1080) | def test_flow_fmod_element_with_random_data(test_case): method test_flow_fmod_broadcast_with_random_data (line 1089) | def test_flow_fmod_broadcast_with_random_data(test_case): method test_flow_fmod_scalar_with_random_data (line 1098) | def test_flow_fmod_scalar_with_random_data(test_case): method test_fmod_with_0_size_data (line 1107) | def test_fmod_with_0_size_data(test_case): method test_tensor_flip_list_with_random_data (line 1114) | def test_tensor_flip_list_with_random_data(test_case): method test_tensor_flip_tuple_with_random_data (line 1123) | def test_tensor_flip_tuple_with_random_data(test_case): method test_tensor_chunk_list_with_random_data (line 1132) | def test_tensor_chunk_list_with_random_data(test_case): method test_tensor_reciprocal_list_with_random_data (line 1146) | def test_tensor_reciprocal_list_with_random_data(test_case): method test_tensor_slice (line 1155) | def test_tensor_slice(test_case): method test_zeros_ (line 1166) | def test_zeros_(test_case): method test_construct_small_tensor (line 1173) | def test_construct_small_tensor(test_case): method test_tensor_floor_with_random_data (line 1200) | def test_tensor_floor_with_random_data(test_case): method test_tensor_round_with_random_data (line 1207) | def test_tensor_round_with_random_data(test_case): method _test_tensor_reshape (line 1213) | def _test_tensor_reshape(test_case): method test_flatten_tensor_with_random_data (line 1223) | def test_flatten_tensor_with_random_data(test_case): method test_reshape_tensor_with_random_data (line 1233) | def test_reshape_tensor_with_random_data(test_case): method test_reshape_tensor_with_random_data_and_keyword (line 1240) | def test_reshape_tensor_with_random_data_and_keyword(test_case): method test_reshape_as_tensor_with_random_data (line 1247) | def test_reshape_as_tensor_with_random_data(test_case): method test_tensor_squeeze_with_random_data (line 1255) | def test_tensor_squeeze_with_random_data(test_case): method test_flow_unsqueeze_with_random_data (line 1262) | def test_flow_unsqueeze_with_random_data(test_case): method test_flow_invert_with_random_data (line 1269) | def test_flow_invert_with_random_data(test_case): method test_tensor_float (line 1275) | def test_tensor_float(test_case): method test_tensor_int (line 1280) | def test_tensor_int(test_case): method test_none_equal (line 1285) | def test_none_equal(test_case): method test_half (line 1294) | def test_half(test_case): method test_byte (line 1300) | def test_byte(test_case): method test_tensor_constructor (line 1306) | def test_tensor_constructor(test_case): method test_tensor_contains_magic_method (line 1334) | def test_tensor_contains_magic_method(test_case): method profile_fill_ (line 1340) | def profile_fill_(test_case): FILE: python/oneflow/test/tensor/test_tensor_part_2.py class TestTensor (line 30) | class TestTensor(flow.unittest.TestCase): method test_permute_flow_with_random_data (line 32) | def test_permute_flow_with_random_data(test_case): method test_permute_flow_with_random_data_and_keyword (line 41) | def test_permute_flow_with_random_data_and_keyword(test_case): method test_transpose_tensor_with_random_data (line 50) | def test_transpose_tensor_with_random_data(test_case): method test_t_tensor_with_random_data (line 58) | def test_t_tensor_with_random_data(test_case): method test_T_tensor_with_random_data (line 67) | def test_T_tensor_with_random_data(test_case): method test_tensor_where (line 73) | def test_tensor_where(test_case): method test_tensor_equal (line 84) | def test_tensor_equal(test_case): method test_tensor_equal_bool_dtype (line 93) | def test_tensor_equal_bool_dtype(test_case): method test_tensor_detach (line 100) | def test_tensor_detach(test_case): method _test_cast_tensor_function (line 110) | def _test_cast_tensor_function(test_case): method _test_sin_tensor_function (line 118) | def _test_sin_tensor_function(test_case, shape, device): method test_cos_tensor_function (line 124) | def test_cos_tensor_function(test_case): method test_std_tensor_function (line 131) | def test_std_tensor_function(test_case): method test_sqrt_tensor_function (line 138) | def test_sqrt_tensor_function(test_case): method test_rsqrt_tensor_function (line 147) | def test_rsqrt_tensor_function(test_case): method test_square_tensor_function (line 156) | def test_square_tensor_function(test_case): method test_addmm_tensor_with_random_data (line 168) | def test_addmm_tensor_with_random_data(test_case): method test_addmm_broadcast_tensor_with_random_data (line 184) | def test_addmm_broadcast_tensor_with_random_data(test_case): method test_clamp_tensor_with_random_data (line 198) | def test_clamp_tensor_with_random_data(test_case): method test_clamp_inplace_tensor_with_random_data (line 208) | def test_clamp_inplace_tensor_with_random_data(test_case): method test_clamp_inplace_tensor_no_grad_with_random_data (line 219) | def test_clamp_inplace_tensor_no_grad_with_random_data(test_case): method test_clamp_minnone_tensor_with_random_data (line 230) | def test_clamp_minnone_tensor_with_random_data(test_case): method test_clamp_minnone_tensor_no_grad_with_random_data (line 241) | def test_clamp_minnone_tensor_no_grad_with_random_data(test_case): method test_clamp_inplace_minnone_tensor_with_random_data (line 251) | def test_clamp_inplace_minnone_tensor_with_random_data(test_case): method test_clamp_inplace_minnone_tensor_no_grad_with_random_data (line 262) | def test_clamp_inplace_minnone_tensor_no_grad_with_random_data(test_ca... method test_clamp_maxnone_tensor_with_random_data (line 273) | def test_clamp_maxnone_tensor_with_random_data(test_case): method test_clamp_maxnone_tensor_no_grad_with_random_data (line 283) | def test_clamp_maxnone_tensor_no_grad_with_random_data(test_case): method test_clamp_inplace_maxnone_tensor_with_random_data (line 293) | def test_clamp_inplace_maxnone_tensor_with_random_data(test_case): method test_clamp_inplace_maxnone_tensor_no_grad_with_random_data (line 304) | def test_clamp_inplace_maxnone_tensor_no_grad_with_random_data(test_ca... method test_clamp_min_tensor_with_random_data (line 315) | def test_clamp_min_tensor_with_random_data(test_case): method test_clamp_min_inplace_tensor_with_random_data (line 322) | def test_clamp_min_inplace_tensor_with_random_data(test_case): method test_clamp_min_tensor_no_grad_with_random_data (line 330) | def test_clamp_min_tensor_no_grad_with_random_data(test_case): method test_clamp_min_inplace_tensor_no_grad_with_random_data (line 337) | def test_clamp_min_inplace_tensor_no_grad_with_random_data(test_case): method test_clamp_max_tensor_with_random_data (line 345) | def test_clamp_max_tensor_with_random_data(test_case): method test_clamp_max_inplace_tensor_with_random_data (line 352) | def test_clamp_max_inplace_tensor_with_random_data(test_case): method test_clamp_max_tensor_no_grad_with_random_data (line 360) | def test_clamp_max_tensor_no_grad_with_random_data(test_case): method test_clamp_max_inplace_tensor_no_grad_with_random_data (line 367) | def test_clamp_max_inplace_tensor_no_grad_with_random_data(test_case): method test_clip_tensor_with_random_data (line 375) | def test_clip_tensor_with_random_data(test_case): method test_clip_inplace_tensor_with_random_data (line 385) | def test_clip_inplace_tensor_with_random_data(test_case): method test_clip_minnone_tensor_with_random_data (line 396) | def test_clip_minnone_tensor_with_random_data(test_case): method test_clip_inplace_maxnone_tensor_with_random_data (line 406) | def test_clip_inplace_maxnone_tensor_with_random_data(test_case): method test_clip_maxnone_tensor_with_random_data (line 417) | def test_clip_maxnone_tensor_with_random_data(test_case): method test_clip_inplace_maxnone_tensor_with_random_data (line 427) | def test_clip_inplace_maxnone_tensor_with_random_data(test_case): method test_ceil_tensor_with_random_data (line 438) | def test_ceil_tensor_with_random_data(test_case): method test_ceil_tensor_with_random_data (line 445) | def test_ceil_tensor_with_random_data(test_case): method test_expm1_tensor_with_random_data (line 452) | def test_expm1_tensor_with_random_data(test_case): method test_floor_tensor_with_random_data (line 459) | def test_floor_tensor_with_random_data(test_case): method test_tensor_var_all_dim_with_random_data (line 466) | def test_tensor_var_all_dim_with_random_data(test_case): method test_tensor_var_one_dim_with_random_data (line 475) | def test_tensor_var_one_dim_with_random_data(test_case): method test_norm_tensor_function (line 485) | def test_norm_tensor_function(test_case): method test_pow_tensor_with_random_data (line 503) | def test_pow_tensor_with_random_data(test_case): method test_atanh_tensor_with_random_data (line 511) | def test_atanh_tensor_with_random_data(test_case): method test_acos_tensor_with_random_data (line 518) | def test_acos_tensor_with_random_data(test_case): method test_acosh_tensor_with_random_data (line 525) | def test_acosh_tensor_with_random_data(test_case): method test_atan_tensor_with_random_data (line 532) | def test_atan_tensor_with_random_data(test_case): method test_arctan_tensor_with_random_data (line 539) | def test_arctan_tensor_with_random_data(test_case): method test_tan_tensor_with_random_data (line 546) | def test_tan_tensor_with_random_data(test_case): method test_tan2_tensor_with_random_data (line 553) | def test_tan2_tensor_with_random_data(test_case): method test_arctanh_tensor_with_random_data (line 561) | def test_arctanh_tensor_with_random_data(test_case): method test_tensor_nonzero_with_random_data (line 571) | def test_tensor_nonzero_with_random_data(test_case): method test_tensor_fmod (line 582) | def test_tensor_fmod(test_case): method test_magic_fmod (line 599) | def test_magic_fmod(test_case): method test_tensor_mish (line 612) | def test_tensor_mish(test_case): method test_tensor_triu (line 630) | def test_tensor_triu(test_case): method test_tensor_grad_assignment (line 649) | def test_tensor_grad_assignment(test_case): method test_tensor_grad_assignment_sum (line 665) | def test_tensor_grad_assignment_sum(test_case): method test_tensor_mish (line 695) | def test_tensor_mish(test_case): method test_tensor_silu (line 715) | def test_tensor_silu(test_case): method test_tensor_selu (line 733) | def test_tensor_selu(test_case): method test_tensor_softsign (line 754) | def test_tensor_softsign(test_case): method test_eq_tensor_with_random_data (line 772) | def test_eq_tensor_with_random_data(test_case): method test_eq_tensor_with_same_random_data (line 780) | def test_eq_tensor_with_same_random_data(test_case): method test_erf_tensor_with_random_data (line 787) | def test_erf_tensor_with_random_data(test_case): method test_erfc_tensor_with_random_data (line 793) | def test_erfc_tensor_with_random_data(test_case): method test_erfinv_tensor_with_random_data (line 801) | def test_erfinv_tensor_with_random_data(test_case): method test_erfinv_inplace_tensor_with_random_data (line 809) | def test_erfinv_inplace_tensor_with_random_data(test_case): method test_exp_tensor_with_random_data (line 817) | def test_exp_tensor_with_random_data(test_case): method test_exp2_tensor_with_random_data (line 823) | def test_exp2_tensor_with_random_data(test_case): method test_round_tensor_with_random_data (line 829) | def test_round_tensor_with_random_data(test_case): method test_tensor_diag_one_dim (line 835) | def test_tensor_diag_one_dim(test_case): method test_flow_tensor_expand_with_random_data (line 841) | def test_flow_tensor_expand_with_random_data(test_case): method test_flow_tensor_expand_with_random_data (line 852) | def test_flow_tensor_expand_with_random_data(test_case): method test_flow_tensor_view_with_random_data (line 864) | def test_flow_tensor_view_with_random_data(test_case): method test_flow_tensor_view_as_with_random_data (line 877) | def test_flow_tensor_view_as_with_random_data(test_case): method test_tensor_diag_other_dim (line 892) | def test_tensor_diag_other_dim(test_case): method test_floordiv_elementwise_tensor_with_random_data (line 898) | def test_floordiv_elementwise_tensor_with_random_data(test_case): method test_scalar_floordiv_tensor_with_random_data (line 908) | def test_scalar_floordiv_tensor_with_random_data(test_case): method test_construct_global_tensor_by_numpy (line 918) | def test_construct_global_tensor_by_numpy(test_case): method test_digamma_tensor_with_random_data (line 940) | def test_digamma_tensor_with_random_data(test_case): class TestTensorNumpy (line 947) | class TestTensorNumpy(flow.unittest.TestCase): method test_1d_sbp_tensor_numpy_1n2d (line 949) | def test_1d_sbp_tensor_numpy_1n2d(test_case): method test_2d_sbp_tensor_numpy_1n2d (line 972) | def test_2d_sbp_tensor_numpy_1n2d(test_case): method test_2d_sbp_tensor_numpy_1n4d (line 993) | def test_2d_sbp_tensor_numpy_1n4d(test_case): method test_tensor_bmm (line 1016) | def test_tensor_bmm(test_case): method test_tensor_split (line 1026) | def test_tensor_split(test_case): method test_tensor_split_sizes (line 1038) | def test_tensor_split_sizes(test_case): method test_tensor_unbind (line 1049) | def test_tensor_unbind(test_case): method test_tensor_swapaxes (line 1057) | def test_tensor_swapaxes(test_case): method test_tensor_swapdimst (line 1065) | def test_tensor_swapdimst(test_case): method test_tensor_int_repeat_interleave_dim_none (line 1073) | def test_tensor_int_repeat_interleave_dim_none(test_case): method test_tensor_int_repeat_interleave_with_dim (line 1080) | def test_tensor_int_repeat_interleave_with_dim(test_case): method test_tensor_tensor_repeat_interleave_dim (line 1088) | def test_tensor_tensor_repeat_interleave_dim(test_case): method test_tensor_tensor_repeat_interleave_dim_with_output_size (line 1097) | def test_tensor_tensor_repeat_interleave_dim_with_output_size(test_case): method test_global_tensor_detach (line 1105) | def test_global_tensor_detach(test_case): method test_tensor_nansum (line 1115) | def test_tensor_nansum(test_case): FILE: python/oneflow/test/tensor/test_tensor_part_3.py function _get_indexes (line 25) | def _get_indexes(device): class TestTensor (line 44) | class TestTensor(flow.unittest.TestCase): method test_scatter_random_data (line 46) | def test_scatter_random_data(test_case): method test_scatter_add_or_multiply_random_data (line 61) | def test_scatter_add_or_multiply_random_data(test_case): method test_tensor_element_size_api (line 76) | def test_tensor_element_size_api(test_case): method test_tensor_new (line 80) | def test_tensor_new(test_case): method test_baddbmm (line 119) | def test_baddbmm(test_case): FILE: python/oneflow/test/tensor/test_tensor_pin_memory.py class TestTensor (line 29) | class TestTensor(flow.unittest.TestCase): method test_tensor_pin_memory (line 33) | def test_tensor_pin_memory(test_case): method test_0_dim_tensor_pin_memory (line 47) | def test_0_dim_tensor_pin_memory(test_case): method test_tensor_construct_with_pin_memory_param (line 62) | def test_tensor_construct_with_pin_memory_param(test_case): method test_tensor_is_pinned (line 76) | def test_tensor_is_pinned(test_case): FILE: python/oneflow/test/tensor/test_tensor_to_memory_format.py class TestTensor (line 28) | class TestTensor(flow.unittest.TestCase): method test_to_memory_format (line 30) | def test_to_memory_format(test_case): FILE: python/oneflow/test_utils/automated_test_util/generators.py function data_generator (line 45) | def data_generator(annotation): function torch_to_flow_converter (line 53) | def torch_to_flow_converter(annotation): function tensor_converter (line 62) | def tensor_converter(torch_tensor): function convert_torch_object_to_flow (line 66) | def convert_torch_object_to_flow(x): function pack (line 73) | def pack(x): class Nothing (line 79) | class Nothing: class generator (line 83) | class generator: method __init__ (line 84) | def __init__(self, children): method _init (line 89) | def _init(self): method eval (line 95) | def eval(self): method _calc_value (line 99) | def _calc_value(self): method value (line 102) | def value(self): method size (line 110) | def size(self): method __or__ (line 113) | def __or__(self, other): method __ror__ (line 119) | def __ror__(self, other): method __add__ (line 122) | def __add__(self, other): method __radd__ (line 125) | def __radd__(self, other): method __sub__ (line 128) | def __sub__(self, other): method __rsub__ (line 131) | def __rsub__(self, other): method __mul__ (line 134) | def __mul__(self, other): method __rmul__ (line 137) | def __rmul__(self, other): method to (line 140) | def to(self, annotation): method _to (line 146) | def _to(self, annotation): class add (line 150) | class add(generator): method __init__ (line 151) | def __init__(self, a, b): method _calc_value (line 156) | def _calc_value(self): class mul (line 160) | class mul(generator): method __init__ (line 161) | def __init__(self, a, b): method _calc_value (line 166) | def _calc_value(self): class neg (line 170) | class neg(generator): method __init__ (line 171) | def __init__(self, a): method _calc_value (line 175) | def _calc_value(self): class oneof (line 179) | class oneof(generator): method __init__ (line 180) | def __init__(self, *args, possibility=None): method _calc_value (line 190) | def _calc_value(self): method __call__ (line 199) | def __call__(self, *args: Any, **kwds: Any) -> Any: method size (line 202) | def size(self): class tuple (line 206) | class tuple(generator): method __init__ (line 207) | def __init__(self, *args): method _calc_value (line 211) | def _calc_value(self): class constant (line 215) | class constant(generator): method __init__ (line 216) | def __init__(self, x): method _calc_value (line 220) | def _calc_value(self): class nothing (line 224) | class nothing(generator): method __init__ (line 225) | def __init__(self): method _calc_value (line 228) | def _calc_value(self): class random (line 232) | class random(generator): method __init__ (line 233) | def __init__(self, low=1, high=6): method _to (line 239) | def _to(self, annotation): method _generate (line 246) | def _generate(self, annotation): method _calc_value (line 278) | def _calc_value(self): function random_or_nothing (line 282) | def random_or_nothing(low, high): class random_pytorch_tensor (line 287) | class random_pytorch_tensor(generator): method __init__ (line 288) | def __init__( method _calc_value (line 338) | def _calc_value(self): function random_bool (line 394) | def random_bool(): class random_device (line 398) | class random_device(generator): method __init__ (line 399) | def __init__(self): method _calc_value (line 402) | def _calc_value(self): class cpu_device (line 409) | class cpu_device(generator): method __init__ (line 410) | def __init__(self): method _calc_value (line 413) | def _calc_value(self): class gpu_device (line 417) | class gpu_device(generator): method __init__ (line 418) | def __init__(self): method _calc_value (line 421) | def _calc_value(self): class random_pytorch_dtype (line 426) | class random_pytorch_dtype(generator): method __init__ (line 459) | def __init__(self, seq_names): method _calc_value (line 466) | def _calc_value(self): class all_placement (line 470) | class all_placement(generator): method __init__ (line 471) | def __init__(self): method __len__ (line 477) | def __len__(self): method __getitem__ (line 480) | def __getitem__(self, key): method _calc_device (line 483) | def _calc_device(self): method _calc_all_placement (line 491) | def _calc_all_placement(self): method _calc_value (line 502) | def _calc_value(self): class all_cpu_placement (line 506) | class all_cpu_placement(all_placement): method __init__ (line 507) | def __init__(self): method _calc_device (line 510) | def _calc_device(self): class all_cuda_placement (line 514) | class all_cuda_placement(all_placement): method __init__ (line 515) | def __init__(self): method _calc_device (line 518) | def _calc_device(self): class random_placement (line 522) | class random_placement(all_placement): method __init__ (line 523) | def __init__(self): method _calc_value (line 526) | def _calc_value(self): class random_cpu_placement (line 530) | class random_cpu_placement(random_placement): method __init__ (line 531) | def __init__(self): method _calc_device (line 534) | def _calc_device(self): class random_gpu_placement (line 538) | class random_gpu_placement(random_placement): method __init__ (line 539) | def __init__(self): method _calc_device (line 542) | def _calc_device(self): class all_sbp (line 546) | class all_sbp(generator): method __init__ (line 547) | def __init__( method __len__ (line 583) | def __len__(self): method __getitem__ (line 586) | def __getitem__(self, key): method _calc_all_sbp (line 589) | def _calc_all_sbp(self): method _calc_value (line 606) | def _calc_value(self): class random_sbp (line 610) | class random_sbp(all_sbp): method __init__ (line 611) | def __init__( method _calc_value (line 631) | def _calc_value(self): class choice_pytorch_tensor (line 636) | class choice_pytorch_tensor(generator): method __init__ (line 637) | def __init__(self, a, size=None, replace=True, p=None, dtype=int): method _calc_value (line 647) | def _calc_value(self): FILE: python/oneflow/test_utils/automated_test_util/global_scope.py class GlobalScope (line 20) | class GlobalScope: method __init__ (line 21) | def __init__(self): method __enter__ (line 24) | def __enter__(self, *argc, **kwarg): method __exit__ (line 29) | def __exit__(self, *argc, **kwarg): function is_global (line 34) | def is_global(): FILE: python/oneflow/test_utils/automated_test_util/profiler.py function compose (line 30) | def compose(*fs): class ProfResult (line 37) | class ProfResult: method __init__ (line 38) | def __init__( method __getattr__ (line 58) | def __getattr__(self, attr): function run_torch (line 70) | def run_torch( function run_flow (line 121) | def run_flow( function profile_dual_object (line 175) | def profile_dual_object(op): function set_hardware_info_list (line 239) | def set_hardware_info_list(hardware_info_list: List[HardwareInfo]) -> None: function set_profiler_hook (line 244) | def set_profiler_hook(hook: Callable[[List[ProfResult]], Any]) -> None: function profile (line 249) | def profile(op): FILE: python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py function torch_tensor_to_flow (line 65) | def torch_tensor_to_flow(x): class PyTorchDoesNotSupportError (line 78) | class PyTorchDoesNotSupportError(Exception): method __init__ (line 79) | def __init__(self, exc): method __str__ (line 82) | def __str__(self): method __repr__ (line 85) | def __repr__(self): class OneFlowGraphBuildOrRunError (line 89) | class OneFlowGraphBuildOrRunError(Exception): method __init__ (line 90) | def __init__(self, exc): method __str__ (line 93) | def __str__(self): method __repr__ (line 96) | def __repr__(self): class BothDoNotSupportError (line 100) | class BothDoNotSupportError(Exception): method __init__ (line 101) | def __init__(self, th_exc, of_exc): method __str__ (line 105) | def __str__(self): method __repr__ (line 108) | def __repr__(self): function get_tensor_shape (line 115) | def get_tensor_shape(call_pytorch): function get_args (line 122) | def get_args(callable, *args, **kwargs): function to_string (line 226) | def to_string(*args, **kwargs) -> str: function check_eager_graph_tensor (line 244) | def check_eager_graph_tensor(eager_res, graph_res): function get_args_copy (line 263) | def get_args_copy(args, kwargs): function get_fake_program_more_detail (line 280) | def get_fake_program_more_detail(oneflow, mode, func, args=None, kwargs=... function get_global_test_device (line 300) | def get_global_test_device(oneflow_args, oneflow_kwargs=None): function get_module_graph_test (line 320) | def get_module_graph_test(graph_train_oneflow, oneflow, verbose, oneflow... function check_oneflow_args_first_element_is_int (line 364) | def check_oneflow_args_first_element_is_int(args): function get_functional_graph_res (line 375) | def get_functional_graph_res( function get_tensor_graph_res (line 470) | def get_tensor_graph_res( function get_oneflow_eager_res (line 511) | def get_oneflow_eager_res( function oneflow_eager_run_with_graph_check (line 526) | def oneflow_eager_run_with_graph_check( function oneflow_tensor_eager_run_with_graph_check (line 611) | def oneflow_tensor_eager_run_with_graph_check( function get_pytorch_oneflow_res (line 656) | def get_pytorch_oneflow_res( function get_pytorch_oneflow_tensor_res (line 745) | def get_pytorch_oneflow_tensor_res( function GetDualObject (line 783) | def GetDualObject(name, pytorch, oneflow): function note_print_args (line 878) | def note_print_args(x, end=True): function note_print_kwargs (line 891) | def note_print_kwargs(x, y, end=True): function print_note_fake_program (line 904) | def print_note_fake_program(detail=False): function clear_note_fake_program (line 981) | def clear_note_fake_program(): class DualObject (line 995) | class DualObject: method __init__ (line 996) | def __init__(self, name, pytorch, oneflow): method __repr__ (line 1047) | def __repr__(self): method __getattr__ (line 1050) | def __getattr__(self, key): method __setattr__ (line 1083) | def __setattr__(self, key, value): method __eq__ (line 1090) | def __eq__(self, other): function equality_checker (line 1102) | def equality_checker(torch_type, flow_type): function check_equality (line 1110) | def check_equality(dual_object: DualObject, rtol=0.0001, atol=1e-05, che... function check_tensor_equality (line 1132) | def check_tensor_equality( function check_basetype_equality (line 1184) | def check_basetype_equality(a, b, ignored1, ignored2, check_dtype=False): function check_basetype_equality (line 1192) | def check_basetype_equality(a, b, rtol=0.0001, atol=1e-05, check_dtype=F... function check_nonetype_equality (line 1218) | def check_nonetype_equality(a, b, ignored1, ignored2, check_dtype=False): function autotest (line 1222) | def autotest( function globaltest (line 1384) | def globaltest(f): function random_tensor (line 1393) | def random_tensor( function random_dtype (line 1437) | def random_dtype(seq_names): function choice_tensor (line 1446) | def choice_tensor( FILE: python/oneflow/test_utils/automated_test_util/util.py function broadcast (line 20) | def broadcast(obj, src: int = 0): FILE: python/oneflow/test_utils/oneflow_pytorch_compatibility/oneflow_pytorch_compatiblity_test.py function cos_sim (line 37) | def cos_sim(vector_a, vector_b): function import_file (line 47) | def import_file(source): function get_loss (line 57) | def get_loss( function do_test_train_loss_oneflow_pytorch (line 219) | def do_test_train_loss_oneflow_pytorch( FILE: python/oneflow/test_utils/test_util.py function GenCartesianProduct (line 28) | def GenCartesianProduct(sets): function GenArgList (line 38) | def GenArgList(arg_dict): function GenArgDict (line 45) | def GenArgDict(arg_dict): class Args (line 49) | class Args: method __init__ (line 50) | def __init__(self, flow_args, tf_args=None): method __str__ (line 57) | def __str__(self): method __repr__ (line 60) | def __repr__(self): function FlattenArray (line 92) | def FlattenArray(input_array): function Array2Numpy (line 99) | def Array2Numpy(input_array, target_shape): function Index2Coordinate (line 103) | def Index2Coordinate(idx, tensor_shape): function Coordinate2Index (line 114) | def Coordinate2Index(coordinate, tensor_shape): FILE: python/oneflow/test_utils/throttle.py function parse_args (line 23) | def parse_args(): function hash_cli2gpu (line 34) | def hash_cli2gpu(cmd: list): function main (line 44) | def main(): FILE: python/oneflow/unittest/dataset.py function dataset_dir (line 19) | def dataset_dir(sub_dir=None): FILE: python/oneflow/unittest/mlir.py class MLIRTestCase (line 20) | class MLIRTestCase(unittest.TestCase): method tearDown (line 21) | def tearDown(self): FILE: python/oneflow/utils/checkpoint.py function _checkpoint_without_reentrant (line 22) | def _checkpoint_without_reentrant(function, *args): function checkpoint (line 66) | def checkpoint(function, *args): FILE: python/oneflow/utils/data/_utils/__init__.py function _set_python_exit_flag (line 56) | def _set_python_exit_flag(): FILE: python/oneflow/utils/data/_utils/collate.py function default_convert (line 33) | def default_convert(data): function default_collate (line 74) | def default_collate(batch): FILE: python/oneflow/utils/data/_utils/fetch.py class _BaseDatasetFetcher (line 22) | class _BaseDatasetFetcher(object): method __init__ (line 23) | def __init__(self, dataset, auto_collation, collate_fn, drop_last): method fetch (line 29) | def fetch(self, possibly_batched_index): class _IterableDatasetFetcher (line 33) | class _IterableDatasetFetcher(_BaseDatasetFetcher): method __init__ (line 34) | def __init__(self, dataset, auto_collation, collate_fn, drop_last): method fetch (line 40) | def fetch(self, possibly_batched_index): class _MapDatasetFetcher (line 57) | class _MapDatasetFetcher(_BaseDatasetFetcher): method __init__ (line 58) | def __init__(self, dataset, auto_collation, collate_fn, drop_last): method fetch (line 63) | def fetch(self, possibly_batched_index): FILE: python/oneflow/utils/data/_utils/pin_memory.py function _pin_memory_loop (line 35) | def _pin_memory_loop(in_queue, out_queue, device_id, done_event): function pin_memory (line 66) | def pin_memory(data): FILE: python/oneflow/utils/data/_utils/signal_handling.py function _set_SIGCHLD_handler (line 62) | def _set_SIGCHLD_handler(): FILE: python/oneflow/utils/data/_utils/worker.py class ManagerWatchdog (line 43) | class ManagerWatchdog(object): method __init__ (line 44) | def __init__(self): method is_alive (line 65) | def is_alive(self): method __init__ (line 77) | def __init__(self): method is_alive (line 81) | def is_alive(self): class ManagerWatchdog (line 76) | class ManagerWatchdog(object): # type: ignore[no-redef] method __init__ (line 44) | def __init__(self): method is_alive (line 65) | def is_alive(self): method __init__ (line 77) | def __init__(self): method is_alive (line 81) | def is_alive(self): class WorkerInfo (line 90) | class WorkerInfo(object): method __init__ (line 93) | def __init__(self, **kwargs): method __setattr__ (line 99) | def __setattr__(self, key, val): method __repr__ (line 106) | def __repr__(self): function get_worker_info (line 113) | def get_worker_info(): class _IterableDatasetStopIteration (line 142) | class _IterableDatasetStopIteration(object): class _ResumeIteration (line 150) | class _ResumeIteration(object): function _generate_state (line 183) | def _generate_state(base_seed, worker_id): function _worker_loop (line 235) | def _worker_loop( FILE: python/oneflow/utils/data/dataloader.py class _DatasetKind (line 62) | class _DatasetKind(object): method create_fetcher (line 67) | def create_fetcher(kind, dataset, auto_collation, collate_fn, drop_last): class _InfiniteConstantSampler (line 78) | class _InfiniteConstantSampler(Sampler): method __init__ (line 86) | def __init__(self): method __iter__ (line 89) | def __iter__(self): class DataLoader (line 94) | class DataLoader(Generic[T_co]): method __init__ (line 178) | def __init__( method _get_iterator (line 338) | def _get_iterator(self) -> "_BaseDataLoaderIter": method __setattr__ (line 345) | def __setattr__(self, attr, val): method __iter__ (line 363) | def __iter__(self) -> "_BaseDataLoaderIter": method _auto_collation (line 379) | def _auto_collation(self): method _index_sampler (line 383) | def _index_sampler(self): method __len__ (line 394) | def __len__(self) -> int: method check_worker_number_rationality (line 426) | def check_worker_number_rationality(self): class _BaseDataLoaderIter (line 491) | class _BaseDataLoaderIter(object): method __init__ (line 492) | def __init__(self, loader: DataLoader) -> None: method __iter__ (line 516) | def __iter__(self) -> "_BaseDataLoaderIter": method _reset (line 519) | def _reset(self, loader, first_iter=False): method _next_index (line 525) | def _next_index(self): method _next_data (line 528) | def _next_data(self): method __next__ (line 531) | def __next__(self) -> Any: method __len__ (line 554) | def __len__(self) -> int: method __getstate__ (line 557) | def __getstate__(self): class _SingleProcessDataLoaderIter (line 561) | class _SingleProcessDataLoaderIter(_BaseDataLoaderIter): method __init__ (line 562) | def __init__(self, loader): method _next_data (line 575) | def _next_data(self): class _MultiProcessingDataLoaderIter (line 583) | class _MultiProcessingDataLoaderIter(_BaseDataLoaderIter): method __init__ (line 893) | def __init__(self, loader): method _reset (line 983) | def _reset(self, loader, first_iter=False): method _try_get_data (line 1016) | def _try_get_data(self, timeout=_utils.MP_STATUS_CHECK_INTERVAL): method _get_data (line 1069) | def _get_data(self): method _next_data (line 1104) | def _next_data(self): method _try_put_index (line 1155) | def _try_put_index(self): method _process_data (line 1175) | def _process_data(self, data): method _mark_worker_as_unavailable (line 1182) | def _mark_worker_as_unavailable(self, worker_id, shutdown=False): method _shutdown_workers (line 1210) | def _shutdown_workers(self): method __del__ (line 1285) | def __del__(self): FILE: python/oneflow/utils/data/dataset.py function _accumulate (line 38) | def _accumulate(iterable, fn=lambda x, y: x + y): class Dataset (line 57) | class Dataset(Generic[T_co]): method __getitem__ (line 73) | def __getitem__(self, index) -> T_co: method __add__ (line 76) | def __add__(self, other: "Dataset[T_co]") -> "ConcatDataset[T_co]": class IterableDataset (line 80) | class IterableDataset(Dataset[T_co]): method __iter__ (line 141) | def __iter__(self) -> Iterator[T_co]: method __add__ (line 144) | def __add__(self, other: Dataset[T_co]): method __getattr__ (line 147) | def __getattr__(self, attribute_name): method register_function (line 157) | def register_function(cls, function_name, function): method register_datapipe_as_function (line 161) | def register_datapipe_as_function(cls, function_name, cls_to_register): method __reduce_ex__ (line 175) | def __reduce_ex__(self, *args, **kwargs): method set_reduce_ex_hook (line 184) | def set_reduce_ex_hook(cls, hook_fn): class TensorDataset (line 190) | class TensorDataset(Dataset[Tuple[Tensor, ...]]): method __init__ (line 199) | def __init__(self, *tensors: Tensor) -> None: method __getitem__ (line 205) | def __getitem__(self, index): method __len__ (line 208) | def __len__(self): class ConcatDataset (line 212) | class ConcatDataset(Dataset[T_co]): method cumsum (line 224) | def cumsum(sequence): method __init__ (line 232) | def __init__(self, datasets: Iterable[Dataset]) -> None: method __len__ (line 243) | def __len__(self): method __getitem__ (line 246) | def __getitem__(self, idx): class ChainDataset (line 261) | class ChainDataset(IterableDataset): method __init__ (line 272) | def __init__(self, datasets: Iterable[Dataset]) -> None: method __iter__ (line 276) | def __iter__(self): method __len__ (line 284) | def __len__(self): class Subset (line 295) | class Subset(Dataset[T_co]): method __init__ (line 306) | def __init__(self, dataset: Dataset[T_co], indices: Sequence[int]) -> ... method __getitem__ (line 310) | def __getitem__(self, idx): method __len__ (line 313) | def __len__(self): function random_split (line 317) | def random_split( FILE: python/oneflow/utils/data/decorator.py class functional_datapipe (line 21) | class functional_datapipe(object): method __init__ (line 24) | def __init__(self, name: str) -> None: method __call__ (line 27) | def __call__(self, cls): class guaranteed_datapipes_determinism (line 44) | class guaranteed_datapipes_determinism(object): method __init__ (line 47) | def __init__(self) -> None: method __enter__ (line 52) | def __enter__(self) -> None: method __exit__ (line 55) | def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> N... class non_deterministic (line 60) | class non_deterministic(object): method __init__ (line 65) | def __init__(self, arg: Union[Type[IterDataPipe], Callable[[], bool]])... method __call__ (line 84) | def __call__(self, *args, **kwargs): method deterministic_wrapper_fn (line 111) | def deterministic_wrapper_fn(self, *args, **kwargs) -> IterDataPipe: FILE: python/oneflow/utils/data/distributed.py class DistributedSampler (line 27) | class DistributedSampler(Sampler[T_co]): method __init__ (line 75) | def __init__( method __iter__ (line 116) | def __iter__(self) -> Iterator[T_co]: method __len__ (line 145) | def __len__(self) -> int: method set_epoch (line 148) | def set_epoch(self, epoch: int) -> None: FILE: python/oneflow/utils/data/sampler.py class Sampler (line 25) | class Sampler(Generic[T_co]): method __init__ (line 37) | def __init__(self, data_source: Optional[Sized]) -> None: method __iter__ (line 40) | def __iter__(self) -> Iterator[T_co]: class SequentialSampler (line 71) | class SequentialSampler(Sampler[int]): method __init__ (line 79) | def __init__(self, data_source): method __iter__ (line 82) | def __iter__(self): method __len__ (line 85) | def __len__(self) -> int: class RandomSampler (line 89) | class RandomSampler(Sampler[int]): method __init__ (line 103) | def __init__( method num_samples (line 134) | def num_samples(self) -> int: method __iter__ (line 140) | def __iter__(self): method __len__ (line 165) | def __len__(self): class SubsetRandomSampler (line 169) | class SubsetRandomSampler(Sampler[int]): method __init__ (line 178) | def __init__(self, indices: Sequence[int], generator=None) -> None: method __iter__ (line 182) | def __iter__(self): method __len__ (line 188) | def __len__(self): class BatchSampler (line 192) | class BatchSampler(Sampler[List[int]]): method __init__ (line 208) | def __init__(self, sampler: Sampler[int], batch_size: int, drop_last: ... method __iter__ (line 230) | def __iter__(self): method __len__ (line 240) | def __len__(self): FILE: python/oneflow/utils/global_view/global_mode.py class global_mode (line 19) | class global_mode(internal_global_view.global_mode): method __init__ (line 69) | def __init__(self, enabled, placement=None, sbp=None) -> None: method __enter__ (line 75) | def __enter__(self): method __exit__ (line 78) | def __exit__(self, type, value, traceback): class current_global_mode (line 82) | class current_global_mode(internal_global_view.current_global_mode): method __init__ (line 102) | def __init__(self) -> None: method is_enabled (line 106) | def is_enabled(self): method sbp (line 110) | def sbp(self): method placement (line 114) | def placement(self): FILE: python/oneflow/utils/global_view/global_utils.py function to_global_tensor (line 24) | def to_global_tensor(input_tensor, placement=None, sbp=None, **kwargs): function to_local_tensor (line 43) | def to_local_tensor(input_tensor, copy): function check_input_global (line 50) | def check_input_global(input): function check_placement_on_all_ranks (line 71) | def check_placement_on_all_ranks(placement): function src_sbp_broadcast (line 84) | def src_sbp_broadcast(obj, src: int = 0): FILE: python/oneflow/utils/global_view/to_global.py function to_global (line 31) | def to_global(input, placement=None, sbp=None, warn_on_non_tensor_leaf=T... FILE: python/oneflow/utils/global_view/to_local.py function to_local (line 24) | def to_local(input, *, copy=False): FILE: python/oneflow/utils/hooks.py class RemovableHandle (line 28) | class RemovableHandle(object): method __init__ (line 34) | def __init__(self, hooks_dict: Any) -> None: method remove (line 39) | def remove(self) -> None: method __getstate__ (line 44) | def __getstate__(self): method __setstate__ (line 47) | def __setstate__(self, state) -> None: method __enter__ (line 56) | def __enter__(self) -> "RemovableHandle": method __exit__ (line 59) | def __exit__(self, type: Any, value: Any, tb: Any) -> None: class BackwardHook (line 63) | class BackwardHook(object): method __init__ (line 73) | def __init__(self, module, user_hooks, user_pre_hooks): method _pack_with_none (line 84) | def _pack_with_none(self, indices, values, size): method _unpack_none (line 91) | def _unpack_none(self, indices, values): method _set_user_hook (line 98) | def _set_user_hook(self, grad_fn): method _apply_on_tensors (line 143) | def _apply_on_tensors(self, fn, args): method setup_input_hook (line 189) | def setup_input_hook(self, args): method setup_output_hook (line 198) | def setup_output_hook(self, args): FILE: python/oneflow/utils/insight/sqlite_to_google_trace_event.py class DatabaseManager (line 22) | class DatabaseManager: method __init__ (line 23) | def __init__(self, db_file): method open_connection (line 28) | def open_connection(self): method close_connection (line 32) | def close_connection(self): method execute_sql (line 38) | def execute_sql(self, sql): function are_tables_exist (line 47) | def are_tables_exist(db_manager, table_names): function print_db_info (line 64) | def print_db_info(db_manager): function get_start_time (line 74) | def get_start_time(db_manager): function get_process_id (line 85) | def get_process_id(db_manager): function get_device_property (line 96) | def get_device_property(db_manager): function sqlite_to_google_trace_event (line 144) | def sqlite_to_google_trace_event(args, tables): FILE: python/oneflow/utils/tensor/from_or_to_torch_tensor.py function print_error_msg (line 21) | def print_error_msg(): function from_torch (line 31) | def from_torch(torch_tensor): function to_torch (line 68) | def to_torch(flow_tensor): FILE: python/setup.py class InstallPlatlib (line 31) | class InstallPlatlib(install): method finalize_options (line 32) | def finalize_options(self): function get_version (line 45) | def get_version(): class BinaryDistribution (line 86) | class BinaryDistribution(Distribution): method is_pure (line 87) | def is_pure(self): method has_ext_modules (line 90) | def has_ext_modules(self): function get_oneflow_internal_so_path (line 99) | def get_oneflow_internal_so_path(): FILE: tools/check_src.py function check_unwanted_test_scripts (line 10) | def check_unwanted_test_scripts(python_test_dir=None, allowed=None): function check_dir_empty (line 45) | def check_dir_empty(path): FILE: tools/clean_generated_api.py function main (line 11) | def main(): FILE: tools/create_pip_index.py function url4key (line 31) | def url4key(endpoint, bucket, key): function append_link (line 35) | def append_link(soup, link): function generate_index_file (line 44) | def generate_index_file(endpoint, bucket, dir_key, file_path, index_keys... FILE: tools/flags_from_git_diff.py function get_changed_files (line 4) | def get_changed_files(base=None, head=None): function should_run_single_client_tests (line 14) | def should_run_single_client_tests(changed=None): function print_github_action_output (line 33) | def print_github_action_output(name=None, value=None): FILE: tools/functional/generator.py function _escape_quote (line 186) | def _escape_quote(fmt): function _normalize (line 190) | def _normalize(fmt): function _remove_square_brackets_and_content_inside (line 195) | def _remove_square_brackets_and_content_inside(fmt): function _std_decay (line 200) | def _std_decay(fmt): function parse_function_params (line 206) | def parse_function_params(fmt): function render_file_if_different (line 239) | def render_file_if_different(target_file, content): function generate_return_types_named_tuple (line 252) | def generate_return_types_named_tuple(return_names, func_name, block_name): class Argument (line 275) | class Argument: method __init__ (line 276) | def __init__(self, fmt, keyword_only=False): method has_default_value (line 334) | def has_default_value(self): method to_string (line 337) | def to_string(self, to_cpp=False): class Return (line 344) | class Return: method __init__ (line 345) | def __init__(self, fmt): method type (line 355) | def type(self): method to_string (line 358) | def to_string(self, to_cpp=False): method check_named_tuple (line 361) | def check_named_tuple(self, fmt): class FunctionSignature (line 371) | class FunctionSignature: method __init__ (line 372) | def __init__(self, fmt): method num_of_args (line 395) | def num_of_args(self): method to_string (line 398) | def to_string(self, to_cpp=False, drop_name=False): method get_mangled_type (line 415) | def get_mangled_type(self): method get_schema_name (line 421) | def get_schema_name(self): class Block (line 425) | class Block: method __init__ (line 426) | def __init__(self, name, signature, bind_python): class Generator (line 432) | class Generator: method __init__ (line 433) | def __init__(self, input_file): method generate_cpp_header_file (line 456) | def generate_cpp_header_file(self, header_fmt, target_header_file): method generate_cpp_source_file (line 466) | def generate_cpp_source_file(self, source_fmt, target_source_file): method generate_pybind_for_python (line 486) | def generate_pybind_for_python( FILE: tools/generate_header_list.py function glob_by_pattern (line 11) | def glob_by_pattern(pattern): FILE: tools/oneflow-tblgen/backends.h function namespace (line 20) | namespace llvm { function namespace (line 25) | namespace oneflow { FILE: tools/oneflow-tblgen/op_schema_emitter.cpp type oneflow (line 35) | namespace oneflow { type tblgen (line 36) | namespace tblgen { type FileTarget (line 48) | enum class FileTarget { class OpSchemaEmitter (line 54) | class OpSchemaEmitter { method emitType (line 69) | static std::string emitType(const std::string& ods_type) { function EmitOpSchemaHeader (line 240) | void EmitOpSchemaHeader(RecordKeeper& RK, raw_ostream& os) { function EmitOpSchemaSource (line 244) | void EmitOpSchemaSource(RecordKeeper& RK, raw_ostream& os) { FILE: tools/oneflow-tblgen/tablegen.cpp type ActionType (line 28) | enum ActionType { type llvm (line 39) | namespace llvm { function LLVMTableGenMain (line 68) | bool LLVMTableGenMain(raw_ostream& OS, RecordKeeper& Records) { function main (line 99) | int main(int argc, char** argv) { FILE: tools/oss_file_exist.py function check_existence (line 5) | def check_existence(endpoint, bucket, path): FILE: tools/package_mirror.py function glob_by_pattern (line 17) | def glob_by_pattern(dir_path, pattern): function scan_urls (line 24) | def scan_urls(dir_path): function convert_url_to_oss_key (line 36) | def convert_url_to_oss_key(url): function convert_url_to_oss_key1 (line 50) | def convert_url_to_oss_key1(url): function convert_url_to_oss_https_url (line 55) | def convert_url_to_oss_https_url(url): function should_be_mirrored (line 63) | def should_be_mirrored(url: str): function calculate_data_md5 (line 77) | def calculate_data_md5(data): function upload_one_to_aliyun (line 84) | def upload_one_to_aliyun(url: str): function upload_to_aliyun (line 109) | def upload_to_aliyun(dir_path):