Searched refs:ProcessedFunction (Results 1 – 4 of 4) sorted by relevance
261 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_;
173 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…
742 ProcessedFunction* fn = &functions_[node_idx]; in prepareStaticNodeInfos()1992 ProcessedFunction::ProcessedFunction( in ProcessedFunction() function in torch::jit::ProcessedFunction2001 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()
933 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() argument1096 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()