Searched refs:HostMemory (Results 1 – 25 of 150) sorted by relevance
123456
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.cc | 31 .HostMemory("allocation")
32 .HostMemory("handle"),
36 .HostMemory("allocation")
37 .HostMemory("handle"),
42 .HostMemory("inputs")
43 .HostMemory("handle"),
47 .HostMemory("inputs")
48 .HostMemory("handle"),
53 .HostMemory("base_handle")
54 .HostMemory("shape_index")
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.cc | 27 .HostMemory("output")
32 .HostMemory("output")
40 .HostMemory("output") \
46 .HostMemory("output") \
57 .HostMemory("input")
58 .HostMemory("output")
64 .HostMemory("input")
65 .HostMemory("output")
75 .HostMemory("output") \
81 .HostMemory("output") \
[all …]
|
| D | strided_slice_op.cc | 380 .HostMemory("begin") \
381 .HostMemory("end") \
382 .HostMemory("strides"), \
387 .HostMemory("shape") \
388 .HostMemory("begin") \
389 .HostMemory("end") \
390 .HostMemory("strides"), \
395 .HostMemory("begin") \
396 .HostMemory("end") \
397 .HostMemory("strides"), \
[all …]
|
| D | bcast_ops.cc | 122 .HostMemory("s0")
123 .HostMemory("s1")
124 .HostMemory("r0"),
129 .HostMemory("s0")
130 .HostMemory("s1")
131 .HostMemory("r0"),
136 .HostMemory("s0")
137 .HostMemory("s1")
138 .HostMemory("r0"),
143 .HostMemory("s0")
[all …]
|
| D | stack_ops.cc | 44 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_GPU).HostMemory("handle"),
49 .HostMemory("max_size")
50 .HostMemory("handle"),
53 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_SYCL).HostMemory("handle"),
57 .HostMemory("max_size")
58 .HostMemory("handle"),
70 .HostMemory("handle") \
75 .HostMemory("handle") \
88 .HostMemory("handle") \
89 .HostMemory("elem") \
[all …]
|
| D | control_flow_ops.cc | 45 .HostMemory("pred") \
52 .HostMemory("pred") \
59 .HostMemory("pred") \
66 .HostMemory("pred") \
94 .HostMemory("data") \
95 .HostMemory("pred") \
96 .HostMemory("output_false") \
97 .HostMemory("output_true") \
104 .HostMemory("data") \
105 .HostMemory("pred") \
[all …]
|
| D | list_kernels.cu.cc | 43 .HostMemory("element_shape"), \
48 .HostMemory("indices") \
49 .HostMemory("element_shape"), \
54 .HostMemory("index") \
55 .HostMemory("element_shape"), \
60 .HostMemory("element_shape"), \
65 .HostMemory("lengths"), \
70 .HostMemory("leading_dims") \
71 .HostMemory("element_shape") \
72 .HostMemory("lengths"), \
[all …]
|
| D | reshape_op.cc | 23 .HostMemory("shape")
28 .HostMemory("shape")
35 .HostMemory("shape") \
41 .HostMemory("shape") \
53 .HostMemory("shape") \
59 .HostMemory("shape") \
72 .HostMemory("tensor")
73 .HostMemory("shape")
74 .HostMemory("output")
80 .HostMemory("tensor")
[all …]
|
| D | pad_op.cc | 270 .HostMemory("paddings"), \
276 .HostMemory("paddings"), \
282 .HostMemory("paddings") \
283 .HostMemory("constant_values"), \
289 .HostMemory("paddings") \
290 .HostMemory("constant_values"), \
334 .HostMemory("paddings"), \
340 .HostMemory("paddings"), \
346 .HostMemory("paddings") \
347 .HostMemory("constant_values"), \
[all …]
|
| D | debug_ops.cc | 36 .HostMemory("input")
37 .HostMemory("output"),
46 .HostMemory("input")
47 .HostMemory("output"),
58 .HostMemory("input")
59 .HostMemory("output"),
66 .HostMemory("input")
67 .HostMemory("output"),
82 .HostMemory("input") \
83 .HostMemory("output") \
[all …]
|
| D | reduction_ops_sum.cc | 44 .HostMemory("reduction_indices"), \
51 .HostMemory("reduction_indices"), \
67 .HostMemory("input")
68 .HostMemory("output")
69 .HostMemory("reduction_indices"),
76 .HostMemory("input")
77 .HostMemory("output")
78 .HostMemory("reduction_indices"),
89 .HostMemory("reduction_indices"), \
96 .HostMemory("reduction_indices"), \
[all …]
|
| D | reduction_ops_max.cc | 44 .HostMemory("reduction_indices"), \
51 .HostMemory("reduction_indices"), \
65 .HostMemory("reduction_indices")
66 .HostMemory("input")
67 .HostMemory("output")
74 .HostMemory("reduction_indices")
75 .HostMemory("input")
76 .HostMemory("output")
91 .HostMemory("reduction_indices"), \
98 .HostMemory("reduction_indices"), \
[all …]
|
| D | reduction_ops_min.cc | 44 .HostMemory("reduction_indices"), \
51 .HostMemory("reduction_indices"), \
63 .HostMemory("reduction_indices")
64 .HostMemory("input")
65 .HostMemory("output")
72 .HostMemory("reduction_indices")
73 .HostMemory("input")
74 .HostMemory("output")
89 .HostMemory("reduction_indices"), \
96 .HostMemory("reduction_indices"), \
[all …]
|
| D | cwise_op_add_1.cc | 33 .HostMemory("x")
34 .HostMemory("y")
35 .HostMemory("z")
40 .HostMemory("x")
41 .HostMemory("y")
42 .HostMemory("z")
56 .HostMemory("x")
57 .HostMemory("y")
58 .HostMemory("z")
63 .HostMemory("x")
[all …]
|
| D | in_topk_op.cc | 103 .HostMemory("predictions")
104 .HostMemory("targets")
105 .HostMemory("precision")
110 .HostMemory("predictions")
111 .HostMemory("targets")
112 .HostMemory("precision")
118 .HostMemory("predictions")
119 .HostMemory("targets")
120 .HostMemory("k")
121 .HostMemory("precision")
[all …]
|
| D | tensor_array_ops.cc | 243 .HostMemory("size") \
244 .HostMemory("handle"), \
249 .HostMemory("size") \
250 .HostMemory("handle"), \
255 .HostMemory("size") \
256 .HostMemory("handle"), \
393 .HostMemory("handle")
394 .HostMemory("grad_handle"),
398 .HostMemory("handle")
399 .HostMemory("grad_handle"),
[all …]
|
| D | unique_op.cc | 250 .HostMemory("x")
251 .HostMemory("y")
252 .HostMemory("idx"),
258 .HostMemory("x")
259 .HostMemory("y")
260 .HostMemory("idx"),
266 .HostMemory("x")
267 .HostMemory("y")
268 .HostMemory("idx"),
274 .HostMemory("x")
[all …]
|
| D | concat_op.cc | 183 .HostMemory("concat_dim"), \
188 .HostMemory("axis"), \
206 .HostMemory("concat_dim"), \
211 .HostMemory("axis"), \
229 .HostMemory("concat_dim")
230 .HostMemory("values")
231 .HostMemory("output"),
236 .HostMemory("values")
237 .HostMemory("axis")
238 .HostMemory("output"),
[all …]
|
| D | identity_op.cc | 70 .HostMemory("input") \
71 .HostMemory("output") \
76 .HostMemory("input") \
77 .HostMemory("output") \
122 .HostMemory("input") \
123 .HostMemory("output") \
128 .HostMemory("input") \
129 .HostMemory("output") \
134 .HostMemory("input") \
135 .HostMemory("output") \
[all …]
|
| D | reverse_op.cc | 302 .HostMemory("dims"), \
308 .HostMemory("axis"), \
314 .HostMemory("axis"), \
360 .HostMemory("dims"), \
366 .HostMemory("axis"), \
372 .HostMemory("axis"), \
390 .HostMemory("tensor")
391 .HostMemory("dims")
392 .HostMemory("output"),
398 .HostMemory("tensor")
[all …]
|
| D | map_stage_op.cc | 548 Name("MapStage").HostMemory("key").HostMemory("indices").Device(DEVICE_GPU),
551 .HostMemory("key")
552 .HostMemory("indices")
559 .HostMemory("key")
560 .HostMemory("indices")
564 .HostMemory("key")
565 .HostMemory("indices")
608 .HostMemory("key")
609 .HostMemory("indices")
613 .HostMemory("key")
[all …]
|
| D | transpose_op.cc | 84 .HostMemory("x")
85 .HostMemory("y"),
90 .HostMemory("x")
91 .HostMemory("y"),
98 .HostMemory("x")
99 .HostMemory("y"),
104 .HostMemory("x")
105 .HostMemory("y"),
226 .HostMemory("perm"), \
231 .HostMemory("perm"), \
[all …]
|
| D | cwise_op_floor_mod.cc | 28 .HostMemory("x")
29 .HostMemory("y")
30 .HostMemory("z")
38 .HostMemory("x")
39 .HostMemory("y")
40 .HostMemory("z")
|
| D | cwise_op_maximum.cc | 30 .HostMemory("x")
31 .HostMemory("y")
32 .HostMemory("z")
41 .HostMemory("x")
42 .HostMemory("y")
43 .HostMemory("z")
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 67 .HostMemory("constants") \
68 .HostMemory("resources"), \
74 .HostMemory("constants") \
75 .HostMemory("key") \
76 .HostMemory("compilation_successful") \
77 .HostMemory("resources"), \
87 Name("_HostSend").Device(DEVICE).HostMemory("tensor"), SendOp); \
89 Name("_HostRecv").Device(DEVICE).HostMemory("tensor"), RecvOp); \
91 Name("_HostCast").Device(DEVICE).HostMemory("x").HostMemory("y"), \
98 Name("HostConst").Device(DEVICE).HostMemory("output"), _HostConstantOp); \
[all …]
|
123456