| /external/pytorch/torch/csrc/jit/passes/ |
| D | refine_tuple_types.cpp | 8 namespace torch::jit { namespace
23 std::vector<c10::TypePtr> types; in VisitTupleNode() local
|
| D | fixup_trace_scope_blocks.cpp | 12 namespace torch::jit { namespace
337 std::vector<TypePtr> types; in convertReturnsToTuples() local
|
| D | symbolic_shape_runtime_fusion.cpp | 18 namespace torch::jit { namespace
536 const auto& types = node->tys(attr::types); in __anon74f36cf50602() local
|
| D | shape_analysis.cpp | 32 namespace torch::jit { namespace
334 std::vector<TensorTypePtr>& types, in broadcastBinary()
1568 auto types = entry.second(node); in PropagateTensorShapeOnNode() local
|
| /external/pytorch/torch/csrc/jit/codegen/onednn/ |
| D | interface.cpp | 19 namespace torch { namespace
126 std::vector<TypePtr> types = node->tys(attr::types); in createLlgaGuardKernel() local
|
| /external/pytorch/torch/csrc/utils/ |
| D | disable_torch_function.cpp | 9 namespace torch { namespace
209 PyObject *func = nullptr, *types = nullptr, *args = nullptr, in THPModule_disable_torch_function() local
243 PyObject *func = nullptr, *types = nullptr, *args = nullptr, in THPModule_disable_torch_dispatch() local
315 namespace torch { namespace
|
| D | invalid_arguments.cpp | 11 namespace torch { namespace
48 std::vector<std::string> types; member
78 std::vector<std::unique_ptr<Type>> types; member
148 std::vector<std::unique_ptr<Type>> types; in _buildType() local
|
| /external/pytorch/torch/csrc/jit/ir/ |
| D | graph_utils.cpp | 3 namespace torch::jit { namespace
19 std::vector<TypePtr> types; in inferShapeAndTypeForInput() local
|
| /external/pytorch/torch/csrc/jit/mobile/ |
| D | interpreter.cpp | 18 namespace torch::jit { namespace
35 void isinstance(Stack& stack, at::ArrayRef<at::TypePtr> types) { in isinstance()
349 at::ArrayRef<TypePtr> types(&code.types_.at(inst.X), inst.N); in run() local
|
| D | function.cpp | 11 namespace torch::jit { namespace
247 const std::vector<c10::TypePtr>& types, in registerFunc()
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | symbolic_script.cpp | 6 namespace torch::jit { namespace
1517 std::vector<TypePtr> types = tup->elements().vec(); in originalReturnType() local
|
| D | vararg_functions.cpp | 8 namespace torch::jit { namespace
383 void isinstance(Stack& stack, at::ArrayRef<at::TypePtr> types) { in isinstance()
|
| D | interpreter.cpp | 59 namespace torch::jit { namespace
680 at::ArrayRef<TypePtr> types( in runTemplate() local
|
| /external/pytorch/torch/csrc/jit/frontend/ |
| D | schema_type_parser.cpp | 42 namespace torch::jit { namespace
369 std::vector<TypePtr> types; in parseFakeAndRealType() local
420 std::vector<TypePtr> types; in parseFakeAndRealType() local
|
| D | schema_matching.cpp | 17 namespace torch::jit { namespace
633 auto types = fmap(values, [](Value* v) { return v->type(); }); in packOutputs() local
|
| /external/pytorch/torch/csrc/jit/backends/ |
| D | backend_init.cpp | 10 namespace torch { namespace
16 std::unordered_set<TypePtr> types; in getSharedModuleTypes() local
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | scalar_type_analysis.cpp | 7 namespace torch::jit { namespace
100 const std::vector<c10::ScalarType>& types) { in PromoteScalarTypes()
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | export_bytecode.cpp | 33 namespace torch::jit { namespace
283 std::vector<IValue> types; in convertMobileFunctionToCodeTable() local
|
| D | export_module.cpp | 42 namespace torch::jit { namespace
192 std::vector<IValue> types; in getFunctionTuple() local
|
| D | python_print.cpp | 25 namespace torch::jit { namespace
1194 const auto& types = node->tys(attr::types); in printRHS() local
|
| /external/pytorch/torch/csrc/profiler/ |
| D | util.cpp | 17 namespace torch::profiler::impl { namespace
281 std::string strListToStr(const std::vector<std::string>& types) { in strListToStr()
334 std::vector<std::string> types; in inputTypes() local
|
| /external/pytorch/torch/csrc/profiler/standalone/ |
| D | execution_trace_observer.cpp | 53 namespace torch::profiler::impl { namespace
466 std::vector<std::string>& types, in appendValueInfo()
|
| /external/pytorch/torch/csrc/ |
| D | Module.cpp | 141 THPObjectPtr types(PySequence_Fast(arg, "expected a sequence")); in THPModule_initNames() local
1587 namespace torch::cuda { namespace
1596 namespace torch::xpu { namespace
1602 namespace torch::profiler { namespace
|
| /external/pytorch/test/cpp/jit/ |
| D | test_lite_interpreter.cpp | 27 namespace torch { namespace
763 std::unordered_set<std::string> types = {"List", "int", "NamedTuple"}; in TEST() local
828 std::unordered_set<std::string> types = {"List", "int", "Sequence"}; in TEST() local
1133 std::vector<IValue> types{"List[int]", "List[int]"}; in TEST() local
2199 std::vector<TypePtr> types; member in torch::jit::LiteInterpreterDynamicTypeTestFixture
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | python_ir.cpp | 26 namespace torch::jit { namespace
775 [](Node& n, const char* name, const std::vector<TypePtr>& types) { in initPythonIRBindings()
1029 std::vector<TypePtr> types; in initPythonIRBindings() local
|