Searched defs:buf_strides (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions_codegen.cpp | 21 int64_t* buf_strides, in nnc_aten_abs()
39 int64_t* buf_strides, in nnc_aten_absolute()
57 int64_t* buf_strides, in nnc_aten_angle()
75 int64_t* buf_strides, in nnc_aten_sgn()
93 int64_t* buf_strides, in nnc_aten_acos()
111 int64_t* buf_strides, in nnc_aten_arccos()
129 int64_t* buf_strides, in nnc_aten_acosh()
147 int64_t* buf_strides, in nnc_aten_arccosh()
165 int64_t* buf_strides, in nnc_aten_asinh()
183 int64_t* buf_strides, in nnc_aten_arcsinh()
[all …]
|
| H A D | external_functions.cpp | 81 int64_t* buf_strides, in constructTensors()
172 int64_t* buf_strides, in constructTensors()
185 int64_t* buf_strides, in constructTensors2()
282 int64_t* buf_strides, in constructTensors2()
362 int64_t* buf_strides, in nnc_aten_conv2d()
413 int64_t* buf_strides, in nnc_aten_quantized_conv1d()
444 int64_t* buf_strides, in nnc_aten_quantized_conv1d_out()
479 int64_t* buf_strides, in nnc_aten_quantized_conv2d()
508 int64_t* buf_strides, in nnc_aten_quantized_conv2d_out()
541 int64_t* buf_strides, in nnc_aten_quantized_conv2d_relu()
[all …]
|
| H A D | eval.cpp | 834 std::vector<int64_t> buf_strides; in visit() local
903 std::vector<int64_t> buf_strides; in visit() local
|
| H A D | llvm_codegen.cpp | 2236 llvm::Value* buf_strides = irb_.CreateAlloca( in visit() local
2397 llvm::Value* buf_strides = irb_.CreateAlloca( in visit() local
|
| /aosp_15_r20/external/mesa3d/src/panfrost/vulkan/ |
| H A D | panvk_shader.h | 174 unsigned buf_strides[PANVK_VARY_BUF_MAX]; member
|
| H A D | panvk_vX_shader.c | 1245 unsigned buf_strides[PANVK_VARY_BUF_MAX] = {0}; in panvk_per_arch() local
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_external_calls.cpp | 965 int64_t* extra_args) {}; in TEST()
|