Home
last modified time | relevance | path

Searched defs:torch_op (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Ddata_flow.cpp76 [&](ExtraFields<EventType::TorchOp>& torch_op) { in calculateUniqueTensorIDs()
/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/
H A Dmicrobenchmarks.py66 def gen_unary_torch_fun(torch_op): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_binary_ufuncs.py1857 def _scalar_helper(python_op, torch_op): argument