Home
last modified time | relevance | path

Searched refs:op_input_t (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/profiler/
Dcollection.h101 using op_input_t = std::variant< variable
143 std::vector<op_input_t>&& inputs,
144 std::vector<op_input_t>&& concrete_inputs,
168 std::vector<op_input_t> inputs_;
169 std::vector<op_input_t> concrete_inputs_;
Dcollection.cpp219 std::vector<op_input_t> out; in getIValueGenerator()
220 auto push_value = [&out, io_type](const Tag& tag, op_input_t input) { in getIValueGenerator()
/external/pytorch/torch/csrc/autograd/
Dprofiler_kineto.cpp68 using torch::profiler::impl::op_input_t;
89 const std::vector<op_input_t>& input_shapes, in parseArgData()
90 const std::vector<op_input_t>& concreteInputs) { in parseArgData()