Searched refs:device_func (Results 1 – 8 of 8) sorted by relevance
| /external/llvm/test/CodeGen/NVPTX/ |
| D | simple-call.ll | 6 ; CHECK: .func ({{.*}}) device_func
7 define float @device_func(float %a) noinline {
16 ; CHECK: device_func,
17 %mul = call float @device_func(float %val)
|
| D | calling-conv.ll | 14 ; CHECK: .func device_func
15 define void @device_func(float* %a) {
|
| /external/llvm-project/llvm/test/CodeGen/NVPTX/ |
| D | simple-call.ll | 6 ; CHECK: .func ({{.*}}) device_func
7 define float @device_func(float %a) noinline {
16 ; CHECK: device_func,
17 %mul = call float @device_func(float %val)
|
| D | calling-conv.ll | 14 ; CHECK: .func device_func
15 define void @device_func(float* %a) {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 663 auto device_func = cluster_func->getFunc(); in runOnOperation() local
664 if (!device_func) return; in runOnOperation()
674 auto conv2d_result = device_func.walk([&](TF::Conv2DOp conv2d) { in runOnOperation()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_gpu.cu.cc | 791 auto* device_func = SparseApplyFtrlKernel<T, Tindex, has_l2_shrinkage>; in operator ()() local
793 GetGpuLaunchConfig(grad_size, d, device_func, 0, 0); in operator ()()
795 device_func, config.block_count, config.thread_per_block, 0, d.stream(), in operator ()()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_scope_test.py | 1169 def device_func(op): function
1175 with ops.device(device_func):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops_test.py | 3219 def device_func(unused_op): function
3227 with ops.device(device_func):
|