/external/tensorflow/tensorflow/core/kernels/ |
H A D | cwise_op_cosh.cc | 23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro 27 REGISTER_SYCL_KERNEL(float); 28 REGISTER_SYCL_KERNEL(double); 29 #undef REGISTER_SYCL_KERNEL macro
|
H A D | cwise_op_sinh.cc | 23 #define REGISTER_SYCL_KERNEL(TYPE) \ macro 27 REGISTER_SYCL_KERNEL(float); 28 REGISTER_SYCL_KERNEL(double); 29 #undef REGISTER_SYCL_KERNEL macro
|
H A D | cwise_op_bitwise_and.cc | 23 #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 D | cwise_op_bitwise_or.cc | 23 #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 D | cwise_op_bitwise_xor.cc | 23 #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 D | cwise_op_left_shift.cc | 23 #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 D | cwise_op_right_shift.cc | 23 #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 D | snapshot_op.cc | 36 #define REGISTER_SYCL_KERNEL(TYPE) \ macro 41 TF_CALL_POD_TYPES(REGISTER_SYCL_KERNEL); 43 #undef REGISTER_SYCL_KERNEL macro
|
H A D | reshape_op.cc | 50 #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 D | shape_ops.cc | 36 #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 D | session_ops.cc | 89 #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 D | sequence_ops.cc | 93 #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 D | identity_op.cc | 49 #define REGISTER_SYCL_KERNEL(type) \ macro 63 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL); 65 #undef REGISTER_SYCL_KERNEL macro
|
H A D | control_flow_ops.cc | 265 #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 D | constant_op.cc | 117 #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 D | variable_ops.cc | 190 #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 D | stack_ops.cc | 372 #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 D | pad_op.cc | 292 #define REGISTER_SYCL_KERNEL(T) \ macro 320 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL); 355 #undef REGISTER_SYCL_KERNEL macro
|