| /external/pytorch/torch/csrc/autograd/ |
| D | function.cpp | 20 namespace torch::autograd { namespace
57 Node* func, in gatherFunctions()
101 auto func = std::move(stack.back()); in deleteNode() local
|
| D | VariableTypeManual.cpp | 22 namespace torch { namespace
477 std::unique_ptr<torch::autograd::ViewFunc> func(nullptr); in _fw_primal() local
510 std::unique_ptr<torch::autograd::ViewFunc> func(nullptr); in _make_dual() local
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | multi_wait.cpp | 7 namespace torch { namespace
49 std::function<void()> MultiWait::Completer(std::function<void()> func) { in Completer()
56 std::function<void()> func) { in Completer()
63 void MultiWait::Complete(const std::function<void()>& func) { in Complete()
|
| /external/pytorch/torch/csrc/jit/mobile/model_tracer/ |
| D | TensorUtils.cpp | 4 namespace torch::jit::mobile { namespace
5 void for_each_tensor_in_ivalue( in for_each_tensor_in_ivalue()
|
| D | MobileModelRunner.cpp | 4 namespace torch::jit::mobile { namespace
185 auto func = module_->get_method(function_name); in run_with_inputs() local
211 void MobileModelRunner::for_each_tensor_in_bundled_inputs( in for_each_tensor_in_bundled_inputs()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | external_functions_core.cpp | 3 namespace torch::jit::tensorexpr { namespace
11 int8_t* func, in DispatchParallel()
|
| /external/pytorch/torch/csrc/api/src/serialize/ |
| D | output-archive.cpp | 15 namespace torch { namespace
47 const std::function<size_t(const void*, size_t)>& func) { in save_to()
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | python_interpreter.cpp | 24 namespace torch::jit { namespace
32 const py::function func = py::reinterpret_borrow<const py::function>( in createPythonOperation() local
|
| D | python_tracer.cpp | 21 namespace torch::jit::tracer { namespace
75 const py::function& func, in createGraphByTracingWithDict()
133 const py::function& func, in createGraphByTracing()
268 m.def("_tracer_set_get_unique_name_fn", [](const py::function& func) { in initPythonTracerBindings()
|
| /external/pytorch/binaries/ |
| D | dump_operator_names.cc | 25 namespace torch { namespace
30 const auto& func = method.function(); in dump_opnames() local
|
| /external/pytorch/torch/csrc/utils/ |
| D | disable_torch_function.cpp | 9 namespace torch { namespace
209 PyObject *func = nullptr, *types = nullptr, *args = nullptr, in THPModule_disable_torch_function() local
243 PyObject *func = nullptr, *types = nullptr, *args = nullptr, in THPModule_disable_torch_dispatch() local
315 namespace torch { namespace
|
| D | python_dispatch.cpp | 33 namespace torch::impl::dispatch { namespace
116 py::object func, in PythonKernelHolder()
194 auto func = in operator ()() local
420 bool with_keyset) { in initDispatchBindings()
483 bool with_keyset) { in initDispatchBindings()
1002 const auto& func = python_registrations_[op.operator_name()][key]; in python_op_registration_trampoline_impl() local
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | profiling_record.cpp | 13 namespace torch::jit { namespace
24 void registerProfileNode(const std::function<bool(const Node*)>& func) { in registerProfileNode()
36 for (const auto& func : registry_funcs_) { in shouldProfileNode() local
51 void RegisterProfilingNode(const std::function<bool(const Node*)>& func) { in RegisterProfilingNode()
|
| D | decomposition_registry.cpp | 20 namespace torch::jit { namespace
132 auto& func = toGraphFunction(*cache_it->second); in GetDecompositionFunction() local
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | export_bytecode.cpp | 33 namespace torch::jit { namespace
176 auto func = mobile::makeOperatorFunction(opname, num_args); in compileGraphToMobileCode() local
261 const mobile::Function& func, in convertMobileFunctionToCodeTable()
348 for (const auto& func : module.compilation_unit().methods()) { in get_min_operator_version_from_version_map() local
378 for (const auto& func : in jitModuleToMobile() local
|
| /external/pytorch/torch/csrc/distributed/autograd/context/ |
| D | context.cpp | 9 namespace torch { namespace
36 const std::shared_ptr<SendRpcBackward>& func, in addSendFunction()
48 std::shared_ptr<RecvRpcBackward>& func, in addRecvFunction()
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| D | finalize.cpp | 19 namespace torch { namespace
197 for (auto func : module.type()->methods()) { in Finalize() local
219 for (auto func : module.type()->methods()) { in FinalizeOnDevicePTQ() local
|
| /external/pytorch/torch/csrc/jit/ |
| D | jit_log.cpp | 18 namespace torch::jit { namespace
132 torch::jit::GraphFunction func("source_dump", graph, nullptr); in log_function() local
|
| /external/pytorch/aten/src/ATen/core/ |
| D | library.cpp | 5 namespace torch { namespace
47 CppFunction::CppFunction(c10::KernelFunction func, std::optional<c10::impl::CppSignature> cpp_signa… in CppFunction()
|
| /external/pytorch/torch/csrc/jit/mobile/nnc/ |
| D | context.cpp | 10 namespace torch { namespace
312 c10::fmap(functions_, [](decltype(functions_)::const_reference func) { in serialize()
321 Function* func = find_function(name); in run() local
|
| /external/pytorch/torch/csrc/jit/mobile/ |
| D | function.cpp | 11 namespace torch::jit { namespace
79 auto func = makeOperatorFunction(opname, num_specified_args); in initialize_operators() local
257 auto& func = name_function_pair.first->second; in registerFunc() local
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | create_functional_graphs.cpp | 11 namespace torch::jit { namespace
215 FunctionalGraphSlicer func(graph); in CreateFunctionalGraphs() local
|
| D | graph_rewrite_helper.cpp | 7 namespace torch::jit::graph_rewrite_helper { namespace
10 auto func = func_value->type()->expectRef<FunctionType>().function(); in getFuncName() local
|
| /external/pytorch/torch/csrc/ |
| D | utils.cpp | 205 bool maybeThrowBackCompatKeepdimWarn(char* func) { in maybeThrowBackCompatKeepdimWarn()
247 namespace torch::gdb { namespace
|
| /external/pytorch/torch/csrc/profiler/ |
| D | util.cpp | 17 namespace torch::profiler::impl { namespace
32 const char* func, in logSoftAssert()
52 const char* func, in logSoftAssert()
|