Searched defs:symbolic_shape_inputs (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_dynamic_shapes.cpp | 62 std::vector<int64_t> symbolic_shape_inputs = c10::fmap( in TEST() local
136 std::vector<int64_t> symbolic_shape_inputs = c10::fmap( in TEST() local
228 std::vector<int64_t> symbolic_shape_inputs( in TEST() local
311 std::vector<int64_t> symbolic_shape_inputs({x_dim1_sym.value()}); in TEST() local
381 std::vector<int64_t> symbolic_shape_inputs = {-3, -2}; in TEST() local
491 std::vector<int64_t> symbolic_shape_inputs( in TEST() local
572 std::vector<int64_t> symbolic_shape_inputs = { in TEST() local
654 std::vector<int64_t> symbolic_shape_inputs = {-2, -3}; in TEST() local
|
| H A D | test_kernel.cpp | 1926 std::vector<int64_t> symbolic_shape_inputs = {-2, -3}; in TEST_F() local
2001 std::vector<int64_t> symbolic_shape_inputs = {-2, -3, -4, -5}; in TEST_F() local
2078 std::vector<int64_t> symbolic_shape_inputs = {-2, -3, -4, -5, -6}; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | kernel.h | 145 std::move(symbolic_shape_inputs), in TensorExprKernel()
|
| H A D | tensorexpr_init.cpp | 780 bool pre_alloc = false) { in initTensorExprBindings()
|
| H A D | kernel.cpp | 1863 std::vector<int64_t> symbolic_shape_inputs, in TensorExprKernel()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | symbolic_shape_runtime_fusion.cpp | 424 std::vector<int64_t> symbolic_shape_inputs; in insertDynamicShapesGuard() local
|