Home
last modified time | relevance | path

Searched refs:OperatorGenerator (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_special_ops.cpp244 OperatorGenerator(
260 OperatorGenerator( \
277 OperatorGenerator( \
321 OperatorGenerator(
329 OperatorGenerator(
347 OperatorGenerator(
352 OperatorGenerator(
370 OperatorGenerator(
375 OperatorGenerator(
381 OperatorGenerator(
[all …]
H A Doperator.h301 std::optional<Operator> OperatorGenerator( in OperatorGenerator() function
310 std::optional<Operator> OperatorGenerator( in OperatorGenerator() function
314 return OperatorGenerator( in OperatorGenerator()
321 std::optional<Operator> OperatorGenerator( in OperatorGenerator() function
329 std::optional<Operator> OperatorGenerator( in OperatorGenerator() function
H A Dregister_prim_ops_fulljit.cpp178 OperatorGenerator(
H A Dregister_prim_ops.cpp1394 result.push_back(OperatorGenerator( in createOperators()
1397 result.push_back(OperatorGenerator( in createOperators()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_custom_operators.cpp204 torch::jit::RegisterOperators reg({OperatorGenerator( in TEST()
223 torch::jit::RegisterOperators reg({OperatorGenerator( in TEST()