Home
last modified time | relevance | path

Searched refs:operator_input_sizes_ (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/csrc/jit/mobile/
Dcode.h20 std::vector<int> operator_input_sizes_; member
Dfunction.cpp56 code_.operator_input_sizes_.emplace_back(num_specified_args.value_or(-1)); in append_operator()
76 int num_args = code_.operator_input_sizes_[i]; in initialize_operators()
Dmodule.cpp338 function.get_code().operator_input_sizes_[i]; in get_module_info()
/external/pytorch/torch/csrc/jit/serialization/
Dexport_bytecode.cpp174 mobile_code.operator_input_sizes_.emplace_back(num_args.value_or(-1)); in compileGraphToMobileCode()
274 const int size = code.operator_input_sizes_[i]; in convertMobileFunctionToCodeTable()
Dexport_module.cpp182 const int size = mobile_code.operator_input_sizes_[i]; in getFunctionTuple()
Dflatbuffer_serializer.cpp260 const int op_size = code.operator_input_sizes_[i]; in functionToFB()
/external/pytorch/test/cpp/jit/
Dtest_flatbuffer.cpp1230 func.get_code().operator_input_sizes_.size(), in TEST()
1237 func2.get_code().operator_input_sizes_.size(), in TEST()
1293 func.get_code().operator_input_sizes_.size(), in TEST()
1300 func2.get_code().operator_input_sizes_.size(), in TEST()
Dtest_lite_interpreter.cpp1630 func.get_code().operator_input_sizes_.size(), in TEST()
1665 func.get_code().operator_input_sizes_.size(), in TEST()