Searched defs:buf_ranks (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions_codegen.cpp | 19 int64_t* buf_ranks, in nnc_aten_abs()
37 int64_t* buf_ranks, in nnc_aten_absolute()
55 int64_t* buf_ranks, in nnc_aten_angle()
73 int64_t* buf_ranks, in nnc_aten_sgn()
91 int64_t* buf_ranks, in nnc_aten_acos()
109 int64_t* buf_ranks, in nnc_aten_arccos()
127 int64_t* buf_ranks, in nnc_aten_acosh()
145 int64_t* buf_ranks, in nnc_aten_arccosh()
163 int64_t* buf_ranks, in nnc_aten_asinh()
181 int64_t* buf_ranks, in nnc_aten_arcsinh()
[all …]
|
| H A D | external_functions.cpp | 79 int64_t* buf_ranks, in constructTensors()
170 int64_t* buf_ranks, in constructTensors()
183 int64_t* buf_ranks, in constructTensors2()
280 int64_t* buf_ranks, in constructTensors2()
360 int64_t* buf_ranks, in nnc_aten_conv2d()
411 int64_t* buf_ranks, in nnc_aten_quantized_conv1d()
442 int64_t* buf_ranks, in nnc_aten_quantized_conv1d_out()
477 int64_t* buf_ranks, in nnc_aten_quantized_conv2d()
506 int64_t* buf_ranks, in nnc_aten_quantized_conv2d_out()
539 int64_t* buf_ranks, in nnc_aten_quantized_conv2d_relu()
[all …]
|
| H A D | eval.cpp | 832 std::vector<int64_t> buf_ranks; in visit() local
901 std::vector<int64_t> buf_ranks; in visit() local
|
| H A D | llvm_codegen.cpp | 2232 llvm::Value* buf_ranks = irb_.CreateAlloca( in visit() local
2393 llvm::Value* buf_ranks = 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()
|