Searched defs:buf_dims (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions_codegen.cpp | 20 int64_t* buf_dims, in nnc_aten_abs()
38 int64_t* buf_dims, in nnc_aten_absolute()
56 int64_t* buf_dims, in nnc_aten_angle()
74 int64_t* buf_dims, in nnc_aten_sgn()
92 int64_t* buf_dims, in nnc_aten_acos()
110 int64_t* buf_dims, in nnc_aten_arccos()
128 int64_t* buf_dims, in nnc_aten_acosh()
146 int64_t* buf_dims, in nnc_aten_arccosh()
164 int64_t* buf_dims, in nnc_aten_asinh()
182 int64_t* buf_dims, in nnc_aten_arcsinh()
[all …]
|
| H A D | external_functions.cpp | 80 int64_t* buf_dims, in constructTensors()
171 int64_t* buf_dims, in constructTensors()
184 int64_t* buf_dims, in constructTensors2()
281 int64_t* buf_dims, in constructTensors2()
361 int64_t* buf_dims, in nnc_aten_conv2d()
412 int64_t* buf_dims, in nnc_aten_quantized_conv1d()
443 int64_t* buf_dims, in nnc_aten_quantized_conv1d_out()
478 int64_t* buf_dims, in nnc_aten_quantized_conv2d()
507 int64_t* buf_dims, in nnc_aten_quantized_conv2d_out()
540 int64_t* buf_dims, in nnc_aten_quantized_conv2d_relu()
[all …]
|
| H A D | eval.cpp | 833 std::vector<int64_t> buf_dims; in visit() local
902 std::vector<int64_t> buf_dims; in visit() local
|
| H A D | llvm_codegen.cpp | 2234 llvm::Value* buf_dims = irb_.CreateAlloca( in visit() local
2395 llvm::Value* buf_dims = irb_.CreateAlloca( in visit() local
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_external_calls.cpp | 965 int64_t* extra_args) {}; in TEST()
|