Searched refs:llvm_codegen (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_external_calls.cpp | 73 LLVMCodeGen llvm_codegen(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
75 llvm_codegen.call({input_buf, weight_buf, bias_buf, result_buf}); in TEST()
127 LLVMCodeGen llvm_codegen(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
129 llvm_codegen.call({input_buf, weight_buf, bias_buf, result_buf}); in TEST()
168 LLVMCodeGen llvm_codegen(l.root_stmt(), {Input, Weight, Result}); in TEST() local
170 llvm_codegen.call({input_buf, weight_buf, result_buf}); in TEST()
227 LLVMCodeGen llvm_codegen(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
229 llvm_codegen.call({input_buf, weight_buf, bias_buf, result_buf}); in TEST()
288 LLVMCodeGen llvm_codegen(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
290 llvm_codegen.call({input_buf, weight_buf, bias_buf, result_buf}); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/tools/toolchains/clang6/ |
| D | CROSSTOOL.tpl | 193 name: "llvm_codegen"
304 requires { feature: "llvm_codegen" }
465 feature: "llvm_codegen"
485 requires { feature: "llvm_codegen" }
504 feature: "llvm_codegen"
|
| /external/pytorch/caffe2/ |
| D | CMakeLists.txt | 461 …set_source_files_properties(${TORCH_SRC_DIR}/csrc/jit/tensorexpr/llvm_codegen.cpp PROPERTIES COMPI…
|
| /external/pytorch/ |
| D | build_variables.bzl | 352 "torch/csrc/jit/tensorexpr/llvm_codegen.cpp",
|