| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 141 XlaJitCompiledCpuFunction* jit = jit_unique_ptr.get(); in Compile() local
142 jit->executable_ = std::move(executable); in Compile()
143 jit->buffer_infos_ = std::move(buffer_infos); in Compile()
144 jit->arg_index_table_ = std::move(arg_index_table); in Compile()
145 jit->program_shape_ = in Compile()
147 XlaCompiledCpuFunction::set_static_data_raw_function(&jit->static_data_, in Compile()
150 &jit->static_data_, jit->buffer_infos_.data()); in Compile()
152 &jit->static_data_, jit->buffer_infos_.size()); in Compile()
154 &jit->static_data_, jit->arg_index_table_.data()); in Compile()
156 &jit->static_data_, jit->arg_index_table_.size()); in Compile()
[all …]
|
| D | xla_jit_compiled_cpu_function_test.cc | 175 std::unique_ptr<XlaJitCompiledCpuFunction> jit, in TEST()
178 XlaCompiledCpuFunction function(jit->StaticData()); in TEST()
237 std::unique_ptr<XlaJitCompiledCpuFunction> jit, in TEST()
240 XlaCompiledCpuFunction function(jit->StaticData()); in TEST()
355 std::unique_ptr<XlaJitCompiledCpuFunction> jit, in TEST()
|
| /external/swiftshader/src/Reactor/ |
| D | LLVMReactor.cpp | 72 thread_local rr::JITBuilder *jit = nullptr; variable
93 x = jit->builder->CreateZExt(x, extTy); in lowerPAVG()
94 y = jit->builder->CreateZExt(y, extTy); in lowerPAVG()
98 llvm::Value *res = jit->builder->CreateAdd(x, y); in lowerPAVG()
99 res = jit->builder->CreateAdd(res, one); in lowerPAVG()
100 res = jit->builder->CreateLShr(res, one); in lowerPAVG()
101 return jit->builder->CreateTrunc(res, ty); in lowerPAVG()
107 return jit->builder->CreateSelect(jit->builder->CreateICmp(pred, x, y), x, y); in lowerPMINMAX()
113 return jit->builder->CreateSExt(jit->builder->CreateICmp(pred, x, y), dstTy, ""); in lowerPCMP()
125 llvm::Value *v = jit->builder->CreateShuffleVector(op, undef, mask); in lowerPMOV()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | BUILD | 14 "//tensorflow/compiler/jit:common",
15 "//tensorflow/compiler/jit:compilation_passes",
16 "//tensorflow/compiler/jit:flags",
17 "//tensorflow/compiler/jit:xla_activity_listener",
18 "//tensorflow/compiler/jit:xla_activity_proto_cc",
19 "//tensorflow/compiler/jit:xla_compilation_cache",
20 "//tensorflow/compiler/jit:xla_device_no_jit_rewrite_registration",
21 "//tensorflow/compiler/jit:xla_cluster_util",
22 "//tensorflow/compiler/jit:xla_launch_util",
42 # Linked by tensorflow core, without registration of jit compilation passes.
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.cc | 23 namespace jit { namespace
100 xla::StatusOr<absl::optional<jit::DeviceId>> PickDeviceForXlaImpl( in PickDeviceForXlaImpl()
101 const jit::DeviceInfoCache& device_info_cache, in PickDeviceForXlaImpl()
102 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu, in PickDeviceForXlaImpl()
113 absl::optional<jit::DeviceId> maybe_gpu_device; in PickDeviceForXlaImpl()
114 absl::optional<jit::DeviceId> maybe_cpu_device; in PickDeviceForXlaImpl()
115 absl::optional<jit::DeviceId> maybe_unknown_device; in PickDeviceForXlaImpl()
125 [&](const jit::DeviceId& d0, absl::optional<jit::DeviceId>* d1) -> bool { in PickDeviceForXlaImpl()
148 devices.ForEach([&](jit::DeviceId device) { in PickDeviceForXlaImpl()
218 xla::StatusOr<jit::DeviceId> PickDeviceForXla( in PickDeviceForXla()
[all …]
|
| D | device_util_test.cc | 27 jit::DeviceInfoCache cache; in PickDeviceHelper()
28 jit::DeviceSet device_set; in PickDeviceHelper()
30 TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, cache.GetIdFor(name)); in PickDeviceHelper()
35 jit::DeviceId result_id, in PickDeviceHelper()
110 jit::DeviceSet device_set; in SimpleRoundTripTestForDeviceSet()
111 jit::DeviceInfoCache device_info_cache; in SimpleRoundTripTestForDeviceSet()
118 TF_ASSERT_OK_AND_ASSIGN(jit::DeviceId device_id, in SimpleRoundTripTestForDeviceSet()
124 device_set.ForEach([&](jit::DeviceId device_id) { in SimpleRoundTripTestForDeviceSet()
|
| D | device_util.h | 31 namespace jit {
201 xla::StatusOr<jit::DeviceId> PickDeviceForXla(
202 const jit::DeviceInfoCache& device_info_cache,
203 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu);
210 xla::StatusOr<absl::optional<jit::DeviceId>> MaybePickDeviceForXla(
211 const jit::DeviceInfoCache& device_info_cache,
212 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu);
|
| D | build_xla_ops_pass.cc | 286 Status DeviceRequiresCompilation(const jit::DeviceInfoCache& device_info_cache, in DeviceRequiresCompilation()
287 jit::DeviceId device, bool* result) { in DeviceRequiresCompilation()
346 xla::StatusOr<jit::DeviceId> InferDeviceForCluster( in InferDeviceForCluster()
347 jit::DeviceInfoCache* device_info_cache, Node* n, in InferDeviceForCluster()
352 jit::DeviceSet device_set; in InferDeviceForCluster()
357 TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, in InferDeviceForCluster()
367 TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, in InferDeviceForCluster()
372 TF_ASSIGN_OR_RETURN(jit::DeviceId result, in InferDeviceForCluster()
468 jit::DeviceInfoCache* device_info_cache, in ReplaceNodeWithXlaCompileAndXlaRun()
476 jit::DeviceId device, in ReplaceNodeWithXlaCompileAndXlaRun()
[all …]
|
| D | BUILD | 58 name = "jit",
83 "//tensorflow/compiler/jit/kernels:xla_ops",
99 "//tensorflow/compiler/jit/kernels:xla_ops",
127 "//tensorflow/compiler/jit/kernels:xla_ops",
153 "//tensorflow/compiler/jit/kernels:xla_ops",
178 "//tensorflow/compiler/jit/kernels:xla_ops",
231 "//tensorflow/compiler/jit/ops:xla_ops",
556 "//tensorflow/compiler/jit/kernels:xla_ops_no_jit_rewrite_registration",
623 "//tensorflow/compiler/jit/kernels:xla_ops",
769 "//tensorflow/compiler/jit/ops:xla_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | jit_test.py | 23 from tensorflow.python.compiler.xla import jit
52 with jit.experimental_jit_scope(use_jit):
63 with jit.experimental_jit_scope(True):
103 with jit.experimental_jit_scope(True):
106 with jit.experimental_jit_scope(True):
109 with jit.experimental_jit_scope(True):
112 with jit.experimental_jit_scope(True):
117 with jit.experimental_jit_scope(True):
172 with jit.experimental_jit_scope(True):
194 with jit.experimental_jit_scope():
[all …]
|
| D | BUILD | 12 "jit.py",
55 "//tensorflow/compiler/jit:xla_ops_py",
56 "//tensorflow/compiler/jit/ops:xla_ops_grad",
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | BUILD | 15 "//tensorflow/compiler/jit:compilation_passes",
16 "//tensorflow/compiler/jit:jit_compilation_passes",
17 "//tensorflow/compiler/jit:xla_cluster_util",
18 "//tensorflow/compiler/jit:xla_cpu_jit",
19 "//tensorflow/compiler/jit:xla_gpu_jit",
|
| /external/pcre/dist2/ |
| D | RunTest.bat | 18 @rem 16 requires absence of jit support
19 @rem 17 requires presence of jit support
20 @rem Sheri P also added override tests for study and jit testing
68 %pcre2test% -C jit >NUL
69 set jit=%ERRORLEVEL% variable
305 if %jit% EQU 1 call :runsub 1 testoutjit "Test with JIT Override" -q -jit
312 if %jit% EQU 1 call :runsub 2 testoutjit "Test with JIT Override" -q -jit
317 if %jit% EQU 1 call :runsub 3 testoutjit "Test with JIT Override" -q -jit
326 if %jit% EQU 1 call :runsub 4 testoutjit "Test with JIT Override" -q -jit
335 if %jit% EQU 1 call :runsub 5 testoutjit "Test with JIT Override" -q -jit
[all …]
|
| D | RunTest | 152 -jit) with=" with JIT";;
401 $sim ./pcre2test -C jit >/dev/null
402 jit=$?
403 if [ $jit -ne 0 -a "$nojit" != "yes" ] ; then
404 jitopt=-jit
562 -jit) with=" with JIT";;
747 if [ $jit -ne 0 ] ; then
759 if [ $jit -eq 0 -o "$nojit" = "yes" ] ; then
|
| /external/bcc/src/lua/bcc/ |
| D | run.lua | 30 local jit = require("jit")
33 jit.version, jit.os, jit.arch))
|
| /external/llvm-project/lldb/test/Shell/Breakpoint/ |
| D | jitbp_elf.test | 5 # RUN: %lldb -b -o 'settings set plugin.jit-loader.gdb.enable on' -o 'b jitbp' -o 'run -jit-kind=mc…
8 # CHECK: (lldb) run -jit-kind=mcjit {{.*}}/jitbp_elf.test.tmp.ll
|
| /external/llvm-project/mlir/lib/ExecutionEngine/ |
| D | ExecutionEngine.cpp | 106 auto &mainJitDylib = jit->getMainJITDylib(); in registerSymbols()
109 mainJitDylib.getExecutionSession(), jit->getDataLayout()))))); in registerSymbols()
295 auto jit = in create() local
306 cantFail(jit->addIRModule(std::move(tsm))); in create()
307 engine->jit = std::move(jit); in create()
310 llvm::orc::JITDylib &mainJD = engine->jit->getMainJITDylib(); in create()
319 auto expectedSymbol = jit->lookup(makePackedFunctionName(name)); in lookup()
|
| /external/llvm/test/ExecutionEngine/OrcMCJIT/ |
| D | load-object-a.ll | 4 ; RUN: %lli -jit-kind=orc-mcjit -extra-module=%p/Inputs/multi-module-b.ll -extra-module=%p/Inputs/m…
10 ; RUN: %lli -jit-kind=orc-mcjit -extra-object=%t.cachedir2/multi-module-b.o -extra-object=%t.cached…
17 ; RUN: %lli -jit-kind=orc-mcjit -extra-archive=%t.cachedir3/load-object.a %s
|
| /external/llvm-project/llvm/test/ExecutionEngine/OrcLazy/ |
| D | hidden-visibility.ll | 1 ; RUN: lli -jit-kind=orc-lazy -extra-module %p/Inputs/hidden-definitions.ll %s
2 ; RUN: not lli -jit-kind=orc-lazy -jd libFoo -extra-module %p/Inputs/hidden-definitions.ll %s
|
| D | basic-whole-module-partitions.ll | 1 ; RUN: lli -jit-kind=orc-lazy -orc-lazy-debug=funcs-to-stdout \
3 ; RUN: lli -jit-kind=orc-lazy -per-module-lazy -orc-lazy-debug=funcs-to-stdout \
|
| D | static-initializers-in-objectfiles.ll | 3 ; RUN: lli -jit-kind=orc-lazy -enable-cache-manager -object-cache-dir=%t %s
4 ; RUN: lli -jit-kind=orc-lazy -enable-cache-manager -object-cache-dir=%t %s
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | BUILD | 38 "//tensorflow/compiler/jit:xla_cpu_jit",
39 "//tensorflow/compiler/jit:xla_gpu_jit",
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | BUILD | 46 "//tensorflow/compiler/jit:xla_device",
77 "//tensorflow/compiler/jit:xla_device",
112 "//tensorflow/compiler/jit:flags",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/cc/ |
| D | BUILD | 35 deps = ["//tensorflow/compiler/jit/ops:xla_ops"],
46 "//tensorflow/compiler/jit/ops:xla_ops",
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | BUILD | 87 "//tensorflow/compiler/jit:compilation_passes",
88 "//tensorflow/compiler/jit:encapsulate_util",
133 "//tensorflow/compiler/jit:encapsulate_util",
134 "//tensorflow/compiler/jit:shape_inference",
|