Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dutil.cpp79 const std::list<std::pair<at::RecordFunctionHandle, int>>& input_op_ids) { in getNvtxStr()
265 const std::list<std::pair<at::RecordFunctionHandle, int>>& input_op_ids) { in inputOpIdsToStr()
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dnvtx_observer.cpp129 auto input_op_ids = getInputTensorOpIds(fn); in enterNVTX() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Dpod_tpu_driver.cc666 std::vector<int64_t> input_op_ids; in ExecuteProgram() local