Searched refs:REGISTER_SYCL_KERNEL (Results 1 - 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcwise_op_cosh.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(float);
28 REGISTER_SYCL_KERNEL(double);
29 #undef REGISTER_SYCL_KERNEL macro
H A Dcwise_op_sinh.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(float);
28 REGISTER_SYCL_KERNEL(double);
29 #undef REGISTER_SYCL_KERNEL macro
H A Dcwise_op_bitwise_and.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(int8);
28 REGISTER_SYCL_KERNEL(int16);
29 REGISTER_SYCL_KERNEL(int32);
30 REGISTER_SYCL_KERNEL(int64);
31 REGISTER_SYCL_KERNEL(uint8);
32 REGISTER_SYCL_KERNEL(uint16);
33 REGISTER_SYCL_KERNEL(uint32);
34 REGISTER_SYCL_KERNEL(uint64);
35 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dcwise_op_bitwise_or.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(int8);
28 REGISTER_SYCL_KERNEL(int16);
29 REGISTER_SYCL_KERNEL(int32);
30 REGISTER_SYCL_KERNEL(int64);
31 REGISTER_SYCL_KERNEL(uint8);
32 REGISTER_SYCL_KERNEL(uint16);
33 REGISTER_SYCL_KERNEL(uint32);
34 REGISTER_SYCL_KERNEL(uint64);
35 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dcwise_op_bitwise_xor.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(int8);
28 REGISTER_SYCL_KERNEL(int16);
29 REGISTER_SYCL_KERNEL(int32);
30 REGISTER_SYCL_KERNEL(int64);
31 REGISTER_SYCL_KERNEL(uint8);
32 REGISTER_SYCL_KERNEL(uint16);
33 REGISTER_SYCL_KERNEL(uint32);
34 REGISTER_SYCL_KERNEL(uint64);
35 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dcwise_op_left_shift.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(int8);
28 REGISTER_SYCL_KERNEL(int16);
29 REGISTER_SYCL_KERNEL(int32);
30 REGISTER_SYCL_KERNEL(int64);
31 REGISTER_SYCL_KERNEL(uint8);
32 REGISTER_SYCL_KERNEL(uint16);
33 REGISTER_SYCL_KERNEL(uint32);
34 REGISTER_SYCL_KERNEL(uint64);
35 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dcwise_op_right_shift.cc23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
27 REGISTER_SYCL_KERNEL(int8);
28 REGISTER_SYCL_KERNEL(int16);
29 REGISTER_SYCL_KERNEL(int32);
30 REGISTER_SYCL_KERNEL(int64);
31 REGISTER_SYCL_KERNEL(uint8);
32 REGISTER_SYCL_KERNEL(uint16);
33 REGISTER_SYCL_KERNEL(uint32);
34 REGISTER_SYCL_KERNEL(uint64);
35 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dsnapshot_op.cc36 #define REGISTER_SYCL_KERNEL(TYPE) \ macro
41 TF_CALL_POD_TYPES(REGISTER_SYCL_KERNEL);
43 #undef REGISTER_SYCL_KERNEL macro
H A Dreshape_op.cc50 #define REGISTER_SYCL_KERNEL(type) \ macro
63 REGISTER_SYCL_KERNEL(float)
64 REGISTER_SYCL_KERNEL(double)
65 REGISTER_SYCL_KERNEL(uint8)
66 REGISTER_SYCL_KERNEL(int8)
67 REGISTER_SYCL_KERNEL(int64)
68 REGISTER_SYCL_KERNEL(uint16)
86 #undef REGISTER_SYCL_KERNEL macro
H A Dshape_ops.cc36 #define REGISTER_SYCL_KERNEL(type) \ macro
50 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL);
51 TF_CALL_bool(REGISTER_SYCL_KERNEL);
52 #undef REGISTER_SYCL_KERNEL macro
161 #define REGISTER_SYCL_KERNEL(type) \ macro
175 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL);
176 TF_CALL_bool(REGISTER_SYCL_KERNEL);
177 #undef REGISTER_SYCL_KERNEL macro
200 #define REGISTER_SYCL_KERNEL(type) \ macro
206 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL);
207 #undef REGISTER_SYCL_KERNEL macro
306 #define REGISTER_SYCL_KERNEL macro
321 #undef REGISTER_SYCL_KERNEL macro
388 #define REGISTER_SYCL_KERNEL macro
403 #undef REGISTER_SYCL_KERNEL macro
447 #define REGISTER_SYCL_KERNEL macro
453 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dsession_ops.cc89 #define REGISTER_SYCL_KERNEL(type) \ macro
101 TF_CALL_NUMBER_TYPES(REGISTER_SYCL_KERNEL);
102 REGISTER_SYCL_KERNEL(bool);
103 #undef REGISTER_SYCL_KERNEL macro
137 #define REGISTER_SYCL_KERNEL(type) \ macro
144 TF_CALL_NUMBER_TYPES(REGISTER_SYCL_KERNEL);
145 REGISTER_SYCL_KERNEL(bool);
146 #undef REGISTER_SYCL_KERNEL macro
H A Dsequence_ops.cc93 #define REGISTER_SYCL_KERNEL(T) REGISTER_KERNEL(DEVICE_SYCL, T) macro
94 TF_CALL_float(REGISTER_SYCL_KERNEL);
95 TF_CALL_double(REGISTER_SYCL_KERNEL);
96 TF_CALL_int32(REGISTER_SYCL_KERNEL);
97 TF_CALL_int64(REGISTER_SYCL_KERNEL);
98 #undef REGISTER_SYCL_KERNEL macro
182 #define REGISTER_SYCL_KERNEL(T) REGISTER_KERNEL_ALL_NUMS(DEVICE_SYCL, T) macro
183 TF_CALL_float(REGISTER_SYCL_KERNEL);
184 TF_CALL_double(REGISTER_SYCL_KERNEL);
185 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Didentity_op.cc49 #define REGISTER_SYCL_KERNEL(type) \ macro
63 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL);
65 #undef REGISTER_SYCL_KERNEL macro
H A Dcontrol_flow_ops.cc265 #define REGISTER_SYCL_KERNEL(type) \ macro
271 REGISTER_SYCL_KERNEL(bool);
272 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL);
283 #undef REGISTER_SYCL_KERNEL macro
363 #define REGISTER_SYCL_KERNEL(type) \ macro
366 REGISTER_SYCL_KERNEL(bool);
367 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL);
375 #undef REGISTER_SYCL_KERNEL macro
458 #define REGISTER_SYCL_KERNEL(type) \ macro
463 REGISTER_SYCL_KERNEL(boo
466 #undef REGISTER_SYCL_KERNEL macro
559 #define REGISTER_SYCL_KERNEL macro
569 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dconstant_op.cc117 #define REGISTER_SYCL_KERNEL(D, TYPE) \ macro
121 REGISTER_SYCL_KERNEL(SYCL, float);
122 REGISTER_SYCL_KERNEL(SYCL, double);
123 REGISTER_SYCL_KERNEL(SYCL, uint8);
124 REGISTER_SYCL_KERNEL(SYCL, int8);
125 REGISTER_SYCL_KERNEL(SYCL, uint16);
126 REGISTER_SYCL_KERNEL(SYCL, int16);
127 REGISTER_SYCL_KERNEL(SYCL, int64);
128 REGISTER_SYCL_KERNEL(SYCL, bool);
129 #undef REGISTER_SYCL_KERNEL macro
[all...]
H A Dvariable_ops.cc190 #define REGISTER_SYCL_KERNEL(type) \ macro
211 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
212 #undef REGISTER_SYCL_KERNEL macro
H A Dstack_ops.cc372 #define REGISTER_SYCL_KERNEL(type) \ macro
379 TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL_KERNEL);
392 #undef REGISTER_SYCL_KERNEL macro
479 #define REGISTER_SYCL_KERNEL(type) \ macro
486 TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL_KERNEL);
499 #undef REGISTER_SYCL_KERNEL macro
H A Dpad_op.cc292 #define REGISTER_SYCL_KERNEL(T) \ macro
320 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
355 #undef REGISTER_SYCL_KERNEL macro

Completed in 234 milliseconds