Searched defs:buf_dtypes (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | external_functions_codegen.cpp | 22 int8_t* buf_dtypes, in nnc_aten_abs()
40 int8_t* buf_dtypes, in nnc_aten_absolute()
58 int8_t* buf_dtypes, in nnc_aten_angle()
76 int8_t* buf_dtypes, in nnc_aten_sgn()
94 int8_t* buf_dtypes, in nnc_aten_acos()
112 int8_t* buf_dtypes, in nnc_aten_arccos()
130 int8_t* buf_dtypes, in nnc_aten_acosh()
148 int8_t* buf_dtypes, in nnc_aten_arccosh()
166 int8_t* buf_dtypes, in nnc_aten_asinh()
184 int8_t* buf_dtypes, in nnc_aten_arcsinh()
[all …]
|
| D | external_functions.cpp | 82 int8_t* buf_dtypes, in constructTensors()
173 int8_t* buf_dtypes, in constructTensors()
186 int8_t* buf_dtypes, in constructTensors2()
283 int8_t* buf_dtypes, in constructTensors2()
363 int8_t* buf_dtypes, in nnc_aten_conv2d()
414 int8_t* buf_dtypes, in nnc_aten_quantized_conv1d()
445 int8_t* buf_dtypes, in nnc_aten_quantized_conv1d_out()
480 int8_t* buf_dtypes, in nnc_aten_quantized_conv2d()
509 int8_t* buf_dtypes, in nnc_aten_quantized_conv2d_out()
542 int8_t* buf_dtypes, in nnc_aten_quantized_conv2d_relu()
[all …]
|
| D | eval.cpp | 835 std::vector<int8_t> buf_dtypes; in visit() local
904 std::vector<int8_t> buf_dtypes; in visit() local
|
| D | llvm_codegen.cpp | 2238 llvm::Value* buf_dtypes = irb_.CreateAlloca( in visit() local
2399 llvm::Value* buf_dtypes = irb_.CreateAlloca( in visit() local
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_external_calls.cpp | 965 int64_t* extra_args) {}; in TEST()
|