Searched refs:TensorExprKernel (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
H A D | test_kernel.cpp | 47 TensorExprKernel k(graph); in TEST_F() 68 TensorExprKernel k(graph); in TEST_F() 94 TensorExprKernel k(graph); in TEST_F() 123 TensorExprKernel k(graph, {}, {}, true); in TEST_F() 160 TensorExprKernel k(graph); in TEST_F() 198 TensorExprKernel k(graph); in TEST_F() 236 TensorExprKernel k(graph); in TEST_F() 268 TensorExprKernel k(graph); in TEST_F() 297 TensorExprKernel k(graph); in TEST_F() 327 TensorExprKernel k(graph); in TEST_F() [all …]
|
H A D | test_graph_opt.cpp | 49 TensorExprKernel kernel(g); in TEST_F() 91 TensorExprKernel kernel(g); in TEST_F() 139 TensorExprKernel kernel(g); in TEST_F() 184 TensorExprKernel kernel(g); in TEST_F() 227 TensorExprKernel kernel(g); in TEST_F() 256 TensorExprKernel kernel(g); in TEST_F() 288 TensorExprKernel kernel(g); in TEST_F()
|
H A D | test_quantization.cpp | 47 TensorExprKernel k(graph); in TEST_F() 77 TensorExprKernel k(graph); in TEST_F() 109 TensorExprKernel k(graph); in TEST_F() 161 TensorExprKernel k(graph); in TEST_F() 205 TensorExprKernel k(graph); in TEST_F() 242 TensorExprKernel k(graph); in TEST_F() 297 TensorExprKernel k(graph); in TEST_F() 336 TensorExprKernel k(graph); in TEST_F() 367 TensorExprKernel k(graph); in TEST_F() 429 TensorExprKernel k(graph); in TEST_F()
|
H A D | test_dynamic_shapes.cpp | 66 TensorExprKernel kernel( in TEST() 150 TensorExprKernel kernel( in TEST() 241 TensorExprKernel kernel( in TEST() 323 TensorExprKernel kernel( in TEST() 382 TensorExprKernel k(graph, {}, symbolic_shape_inputs, false, symbolic_strides); in TEST() 508 TensorExprKernel kernel( in TEST() 574 TensorExprKernel k(graph, {}, symbolic_shape_inputs, false, symbolic_strides); in TEST() 666 TensorExprKernel kernel( in TEST()
|
H A D | tutorial.cpp | 464 TensorExprKernel kernel(graph); in main()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
H A D | kernel.cpp | 398 ExprHandle TensorExprKernel::constant(const torch::jit::Value* v) { in constant() 424 ArgValue TensorExprKernel::toArg(const torch::jit::Value* v) const { in toArg() 477 ExprHandle TensorExprKernel::getVarForShape(const c10::ShapeSymbol& ss) { in getVarForShape() 491 std::vector<ExprHandle> TensorExprKernel::sizesFromSymbolicShape( in sizesFromSymbolicShape() 503 std::vector<ExprHandle> TensorExprKernel::sizesForValue( in sizesForValue() 569 Tensor TensorExprKernel::computeValue(const torch::jit::Value* v) { in computeValue() 789 StmtPtr TensorExprKernel::transformLoops(BackendType backendType, StmtPtr st) { in transformLoops() 940 std::string TensorExprKernel::getCodeGenName(BackendType backendType) { in getCodeGenName() 962 TensorExprKernel::BackendType TensorExprKernel::inferBackendTypeFromDevice( in inferBackendTypeFromDevice() 986 void TensorExprKernel::genInputDebugNames() { in genInputDebugNames() [all …]
|
H A D | kernel.h | 98 class TORCH_API TensorExprKernel { 121 explicit TensorExprKernel( 132 explicit TensorExprKernel( 141 : TensorExprKernel( in TensorExprKernel() function
|
H A D | tensorexpr_init.cpp | 772 py::class_<TensorExprKernel>(te, "TensorExprKernel") in initTensorExprBindings() 788 return std::make_unique<TensorExprKernel>( in initTensorExprBindings() 800 [](TensorExprKernel& self, const py::tuple& inputs) { in initTensorExprBindings() 817 [](TensorExprKernel& self, const py::tuple& inputs) { in initTensorExprBindings() 834 [](TensorExprKernel& self) { return self.getCodeGenStmt(); }, in initTensorExprBindings() 838 [](TensorExprKernel& self, const std::string& attr = "") { in initTensorExprBindings() 842 .def("recompile", [](TensorExprKernel& self) { self.recompile(); }); in initTensorExprBindings()
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_tensorexpr_pybind.py | 146 kernel = te.TensorExprKernel(graph) 173 kernel = te.TensorExprKernel(graph) 195 kernel = te.TensorExprKernel(graph) 208 kernel = te.TensorExprKernel(graph) 226 kernel = te.TensorExprKernel(graph) 250 kernel = te.TensorExprKernel(graph) 275 kernel = te.TensorExprKernel(graph) 299 kernel = te.TensorExprKernel(graph) 325 kernel = te.TensorExprKernel(graph) 357 kernel = te.TensorExprKernel(graph, {"aten::nan_to_num": my_custom_lowering}) [all …]
|
H A D | test_jit_fuser_te.py | 1695 k = torch._C._te.TensorExprKernel(graph) 2376 kernel = torch._C._te.TensorExprKernel(f.graph) 2890 kernel = torch._C._te.TensorExprKernel(ts_g.graph)
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | tensorexpr_fuser.cpp | 1360 std::make_shared<tensorexpr::TensorExprKernel>(node->g(attr::Subgraph)); in createTensorExprOp() 1413 std::shared_ptr<tensorexpr::TensorExprKernel> kernel = in createTensorExprOp() 1414 std::make_shared<tensorexpr::TensorExprKernel>( in createTensorExprOp()
|