Home
last modified time | relevance | path

Searched refs:ProcessedFunction (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.h261 class TORCH_API ProcessedFunction {
263 ProcessedFunction(
292 Kind kind_{ProcessedFunction::Kind::kOutVariant};
550 std::vector<ProcessedFunction> functions_{};
822 ProcessedFunction* fn,
836 return fn_->kind() == ProcessedFunction::Kind::kOutVariant; in has_out_variant()
843 const ProcessedFunction* fn_;
958 return fn_->kind() == ProcessedFunction::Kind::kOutVariant; in has_out_variant()
962 return fn_->kind() == ProcessedFunction::Kind::kNativeFunction; in has_native()
1027 const ProcessedFunction* fn_;
H A DREADME.md173 ProcessedFunction ◄─────────────────────────────────────────────────────────────────────────────┘
217 ### `ProcessedNode` and `ProcessedFunction`
223 Each `ProcessedNode` stores a `ProcessedFunction`, which represents the actual op to execute. `Proc…
225 **Note that all `ProcessedFunction`s are shared amongst all runtime instances**, so all `ProcessedF…
H A Dimpl.cpp742 ProcessedFunction* fn = &functions_[node_idx]; in prepareStaticNodeInfos()
1992 ProcessedFunction::ProcessedFunction( in ProcessedFunction() function in torch::jit::ProcessedFunction
2001 kind_ = ProcessedFunction::Kind::kOutVariant; in ProcessedFunction()
2011 kind_ = ProcessedFunction::Kind::kNativeFunction; in ProcessedFunction()
2041 kind_ = ProcessedFunction::Kind::kInterpreterFallback; in ProcessedFunction()
2048 ProcessedFunction* fn, in StaticNodeInfo()
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc933 ProcessedFunction fn( in TEST()
953 ProcessedFunction fn( in TEST()
981 ProcessedFunction fn( in TEST()
996 ProcessedFunction fn( in TEST()
1084 TEST(ProcessedFunction, ProcessedFunction) { in TEST() argument
1096 ProcessedFunction sigmoid_fn( in TEST()
1100 EXPECT_EQ(sigmoid_fn.kind(), ProcessedFunction::Kind::kOutVariant); in TEST()
1104 ProcessedFunction transpose_fn( in TEST()
1108 EXPECT_EQ(transpose_fn.kind(), ProcessedFunction::Kind::kNativeFunction); in TEST()