Searched refs:DEVICE_SYCL (Results 1 – 25 of 93) sorted by relevance
1234
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.cc | 39 .Device(DEVICE_SYCL) \
45 .Device(DEVICE_SYCL) \
56 .Device(DEVICE_SYCL)
63 .Device(DEVICE_SYCL)
164 .Device(DEVICE_SYCL) \
170 .Device(DEVICE_SYCL) \
181 .Device(DEVICE_SYCL)
188 .Device(DEVICE_SYCL)
203 .Device(DEVICE_SYCL) \
211 .Device(DEVICE_SYCL)
[all …]
|
| D | control_flow_ops.cc | 125 .Device(DEVICE_SYCL) \
133 .Device(DEVICE_SYCL) \
144 .Device(DEVICE_SYCL) \
158 .Device(DEVICE_SYCL) \
278 .Device(DEVICE_SYCL) \
287 .Device(DEVICE_SYCL) \
326 .Device(DEVICE_SYCL) \
333 .Device(DEVICE_SYCL) \
377 Name("Enter").Device(DEVICE_SYCL).TypeConstraint<type>("T"), EnterOp)
383 Name("RefEnter").Device(DEVICE_SYCL).TypeConstraint<type>("T"), EnterOp)
[all …]
|
| D | stack_ops.cc | 53 REGISTER_KERNEL_BUILDER(Name("Stack").Device(DEVICE_SYCL).HostMemory("handle"),
56 .Device(DEVICE_SYCL)
109 .Device(DEVICE_SYCL) \
118 .Device(DEVICE_SYCL) \
174 .Device(DEVICE_SYCL) \
183 .Device(DEVICE_SYCL) \
204 Name("StackClose").Device(DEVICE_SYCL).HostMemory("handle"), StackCloseOp);
206 Name("StackCloseV2").Device(DEVICE_SYCL).HostMemory("handle"),
|
| D | identity_op.cc | 51 Name("Identity").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
54 Name("PreventGradient").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
57 Name("RefIdentity").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
60 Name("StopGradient").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
69 .Device(DEVICE_SYCL) \
75 .Device(DEVICE_SYCL) \
|
| D | relu_op.cc | 218 Name("Relu").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
221 Name("ReluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
224 Name("Relu6").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
227 Name("Relu6Grad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
230 Name("LeakyRelu").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
233 Name("LeakyReluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
236 Name("Elu").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
239 Name("EluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
242 Name("Selu").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
245 Name("SeluGrad").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
|
| D | debug_ops.cc | 42 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_SYCL), CopyOp);
45 .Device(DEVICE_SYCL)
65 .Device(DEVICE_SYCL)
94 .Device(DEVICE_SYCL) \
131 .Device(DEVICE_SYCL) \
|
| D | reshape_op.cc | 52 .Device(DEVICE_SYCL) \
58 .Device(DEVICE_SYCL) \
71 .Device(DEVICE_SYCL)
79 .Device(DEVICE_SYCL)
|
| D | function_ops.cc | 79 Name(kArgOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), ArgOp);
82 .Device(DEVICE_SYCL)
89 Name(kRetOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), RetvalOp);
92 .Device(DEVICE_SYCL)
210 Name("_ListToArray").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
213 Name("_ArrayToList").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
222 .Device(DEVICE_SYCL)
228 .Device(DEVICE_SYCL)
291 REGISTER_KERNEL_BUILDER(Name(kGradientOp).Device(DEVICE_SYCL),
402 Name("RemoteCall").Device(DEVICE_SYCL).HostMemory("target"), RemoteCallOp);
|
| D | reduction_ops_sum.cc | 86 .Device(DEVICE_SYCL) \
93 .Device(DEVICE_SYCL) \
104 .Device(DEVICE_SYCL)
113 .Device(DEVICE_SYCL)
|
| D | reduction_ops_max.cc | 88 .Device(DEVICE_SYCL) \
95 .Device(DEVICE_SYCL) \
106 .Device(DEVICE_SYCL)
115 .Device(DEVICE_SYCL)
|
| D | reduction_ops_min.cc | 86 .Device(DEVICE_SYCL) \
93 .Device(DEVICE_SYCL) \
104 .Device(DEVICE_SYCL)
113 .Device(DEVICE_SYCL)
|
| D | map_stage_op.cc | 561 .Device(DEVICE_SYCL),
566 .Device(DEVICE_SYCL),
622 .Device(DEVICE_SYCL),
627 .Device(DEVICE_SYCL),
679 Name("MapPeek").HostMemory("key").HostMemory("indices").Device(DEVICE_SYCL),
684 .Device(DEVICE_SYCL),
746 .Device(DEVICE_SYCL),
751 .Device(DEVICE_SYCL),
786 REGISTER_KERNEL_BUILDER(Name("MapSize").Device(DEVICE_SYCL).HostMemory("size"),
789 Name("OrderedMapSize").Device(DEVICE_SYCL).HostMemory("size"),
[all …]
|
| D | session_ops.cc | 91 .Device(DEVICE_SYCL) \
96 .Device(DEVICE_SYCL) \
139 .Device(DEVICE_SYCL) \
171 Name("DeleteSessionTensor").Device(DEVICE_SYCL).HostMemory("handle"),
|
| D | variable_ops.cc | 189 Name("Variable").Device(DEVICE_SYCL).TypeConstraint<type>("dtype"), \
192 Name("VariableV2").Device(DEVICE_SYCL).TypeConstraint<type>("dtype"), \
195 .Device(DEVICE_SYCL) \
199 .Device(DEVICE_SYCL) \
203 .Device(DEVICE_SYCL) \
|
| D | pad_op.cc | 404 .Device(DEVICE_SYCL) \
410 .Device(DEVICE_SYCL) \
416 .Device(DEVICE_SYCL) \
423 .Device(DEVICE_SYCL) \
432 .Device(DEVICE_SYCL)
440 .Device(DEVICE_SYCL)
448 .Device(DEVICE_SYCL)
457 .Device(DEVICE_SYCL)
|
| D | strided_slice_op.cc | 498 .Device(DEVICE_SYCL) \
505 .Device(DEVICE_SYCL) \
513 .Device(DEVICE_SYCL) \
520 .Device(DEVICE_SYCL) \
531 .Device(DEVICE_SYCL)
540 .Device(DEVICE_SYCL)
550 .Device(DEVICE_SYCL)
558 .Device(DEVICE_SYCL)
|
| D | stage_op.cc | 223 REGISTER_KERNEL_BUILDER(Name("Stage").Device(DEVICE_SYCL), StageOp);
256 REGISTER_KERNEL_BUILDER(Name("Unstage").Device(DEVICE_SYCL), UnstageOp);
293 Name("StagePeek").HostMemory("index").Device(DEVICE_SYCL), StagePeekOp);
323 Name("StageSize").HostMemory("size").Device(DEVICE_SYCL), StageSizeOp);
346 REGISTER_KERNEL_BUILDER(Name("StageClear").Device(DEVICE_SYCL), StageClearOp);
|
| D | bcast_ops.cc | 150 .Device(DEVICE_SYCL)
157 .Device(DEVICE_SYCL)
200 .Device(DEVICE_SYCL)
208 .Device(DEVICE_SYCL)
|
| D | sendrecv_ops.cc | 114 REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_SYCL), SendOp);
116 Name("_HostSend").Device(DEVICE_SYCL).HostMemory("tensor"), SendOp);
200 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_SYCL), RecvOp);
209 Name("_HostRecv").Device(DEVICE_SYCL).HostMemory("tensor"), RecvOp);
|
| D | concat_op.cc | 246 .Device(DEVICE_SYCL) \
251 .Device(DEVICE_SYCL) \
259 .Device(DEVICE_SYCL)
266 .Device(DEVICE_SYCL)
364 .Device(DEVICE_SYCL)
|
| D | dense_update_ops.cc | 121 Name("Assign").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
155 Name("AssignAdd").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
158 Name("AssignSub").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
|
| D | host_constant_op.cc | 59 .Device(DEVICE_SYCL)
71 Name("HostConst").Device(DEVICE_SYCL).HostMemory("output"),
|
| D | unpack_op.cc | 172 Name("Unpack").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
178 .Device(DEVICE_SYCL)
185 .Device(DEVICE_SYCL)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | memory_types_test.cc | 38 TF_EXPECT_OK(ValidateMemoryTypes(DEVICE_SYCL, g)); in TEST()
62 EXPECT_TRUE(errors::IsInternal(ValidateMemoryTypes(DEVICE_SYCL, g))); in TEST()
65 TF_EXPECT_OK(EnsureMemoryTypes(DEVICE_SYCL, "/device:SYCL:0", g)); in TEST()
66 TF_EXPECT_OK(ValidateMemoryTypes(DEVICE_SYCL, g)); in TEST()
91 TF_EXPECT_OK(MemoryTypeForOutput(DEVICE_SYCL, g, si, 0, &memory_type)); in TEST()
|
| /external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
| D | memory_stats_ops.cc | 64 Name("BytesInUse").Device(DEVICE_SYCL).HostMemory("out"), BytesInUseOp);
86 Name("BytesLimit").Device(DEVICE_SYCL).HostMemory("out"), BytesLimitOp);
110 Name("MaxBytesInUse").Device(DEVICE_SYCL).HostMemory("out"),
|
1234