Home
last modified time | relevance | path

Searched refs:TensorExprKernel (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_kernel.cpp47 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 Dtest_graph_opt.cpp49 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 Dtest_quantization.cpp47 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 Dtest_dynamic_shapes.cpp66 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 Dtutorial.cpp464 TensorExprKernel kernel(graph); in main()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp398 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 Dkernel.h98 class TORCH_API TensorExprKernel {
121 explicit TensorExprKernel(
132 explicit TensorExprKernel(
141 : TensorExprKernel( in TensorExprKernel() function
H A Dtensorexpr_init.cpp772 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 Dtest_tensorexpr_pybind.py146 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 Dtest_jit_fuser_te.py1695 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 Dtensorexpr_fuser.cpp1360 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()