| /external/tensorflow/tensorflow/core/kernels/ |
| extract_jpeg_shape_op.cc | 67 #define REGISTER_KERNELS(type) \
73 TF_CALL_int32(REGISTER_KERNELS);
74 TF_CALL_int64(REGISTER_KERNELS);
75 #undef REGISTER_KERNELS
|
| sparse_add_grad_op.cc | 131 #define REGISTER_KERNELS(type) \
137 REGISTER_KERNELS(float);
138 REGISTER_KERNELS(double);
139 REGISTER_KERNELS(int64);
140 REGISTER_KERNELS(int32);
141 REGISTER_KERNELS(int16);
142 REGISTER_KERNELS(int8);
143 REGISTER_KERNELS(complex64);
144 REGISTER_KERNELS(complex128);
145 #undef REGISTER_KERNELS
[all...] |
| bincount_op.cc | 115 #define REGISTER_KERNELS(type) \
120 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
121 #undef REGISTER_KERNELS
125 #define REGISTER_KERNELS(type) \
132 TF_CALL_int32(REGISTER_KERNELS);
133 TF_CALL_float(REGISTER_KERNELS);
134 #undef REGISTER_KERNELS
|
| dense_update_ops.cc | 94 #define REGISTER_KERNELS(type) \
99 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
100 TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
101 #undef REGISTER_KERNELS
126 #define REGISTER_KERNELS(type) \
134 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
135 #undef REGISTER_KERNELS
|
| sparse_add_op.cc | 167 #define REGISTER_KERNELS(type, thresh_type) \
174 REGISTER_KERNELS(float, float);
175 REGISTER_KERNELS(double, double);
176 REGISTER_KERNELS(int64, int64);
177 REGISTER_KERNELS(int32, int32);
178 REGISTER_KERNELS(int16, int16);
179 REGISTER_KERNELS(int8, int8);
180 REGISTER_KERNELS(complex64, float);
181 REGISTER_KERNELS(complex128, double);
182 #undef REGISTER_KERNELS
[all...] |
| histogram_op.cc | 117 #define REGISTER_KERNELS(type) \
129 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
130 #undef REGISTER_KERNELS
133 #define REGISTER_KERNELS(type) \
142 TF_CALL_GPU_NUMBER_TYPES(REGISTER_KERNELS);
143 #undef REGISTER_KERNELS
|
| conditional_accumulator_op.cc | 46 #define REGISTER_KERNELS(type, dev) \
52 #define REGISTER_KERNELS_CPU(type) REGISTER_KERNELS(type, CPU)
59 #undef REGISTER_KERNELS
|
| sparse_conditional_accumulator_op.cc | 47 #define REGISTER_KERNELS(type, dev) \
53 #define REGISTER_KERNELS_CPU(type) REGISTER_KERNELS(type, CPU)
60 #undef REGISTER_KERNELS
|
| sparse_reorder_op.cc | 80 #define REGISTER_KERNELS(type) \
85 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
86 #undef REGISTER_KERNELS
|
| random_crop_op.cc | 112 #define REGISTER_KERNELS(type) \
117 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
118 #undef REGISTER_KERNELS
|
| sparse_reduce_op.cc | 233 #define REGISTER_KERNELS(T) \
237 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
238 #undef REGISTER_KERNELS
240 #define REGISTER_KERNELS(T) \
244 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
245 #undef REGISTER_KERNELS
327 #define REGISTER_KERNELS(T) \
331 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
332 #undef REGISTER_KERNELS
334 #define REGISTER_KERNELS(T)
[all...] |
| sparse_slice_op.cc | 96 #define REGISTER_KERNELS(type) \
101 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
102 #undef REGISTER_KERNELS
|
| sparse_split_op.cc | 90 #define REGISTER_KERNELS(type) \
95 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
96 #undef REGISTER_KERNELS
|
| sparse_to_dense_op.cc | 140 #define REGISTER_KERNELS(type, index_type) \
148 REGISTER_KERNELS(type, int32); \
149 REGISTER_KERNELS(type, int64);
156 #undef REGISTER_KERNELS
|
| training_ops.cc | 485 #define REGISTER_KERNELS(D, T) \
494 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T);
516 REGISTER_KERNELS(GPU, Eigen::half);
517 REGISTER_KERNELS(GPU, float);
518 REGISTER_KERNELS(GPU, double);
522 #define REGISTER_SYCL_KERNELS(T) REGISTER_KERNELS(SYCL, T);
529 #undef REGISTER_KERNELS
636 #define REGISTER_KERNELS(D, T) \
647 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T);
671 REGISTER_KERNELS(GPU, Eigen::half)
[all...] |
| topk_op.cc | 237 #define REGISTER_KERNELS(type) \
241 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
243 #undef REGISTER_KERNELS
264 #define REGISTER_KERNELS(type) \
274 TF_CALL_GPU_NUMBER_TYPES(REGISTER_KERNELS);
275 TF_CALL_INTEGRAL_TYPES(REGISTER_KERNELS);
277 #undef REGISTER_KERNELS
|
| softplus_op.cc | 84 #define REGISTER_KERNELS(type) \
92 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
93 #undef REGISTER_KERNELS
|
| softsign_op.cc | 85 #define REGISTER_KERNELS(type) \
93 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
94 #undef REGISTER_KERNELS
|
| sparse_concat_op.cc | 154 #define REGISTER_KERNELS(type) \
159 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
160 #undef REGISTER_KERNELS
|
| sparse_dense_binary_op_shared.cc | 173 #define REGISTER_KERNELS(T) \
185 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
186 #undef REGISTER_KERNELS
|
| sparse_tensor_dense_add_op.cc | 153 #define REGISTER_KERNELS(T) \
157 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
158 #undef REGISTER_KERNELS
|
| sparse_fill_empty_rows_op.cc | 188 #define REGISTER_KERNELS(type) \
194 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
195 #undef REGISTER_KERNELS
258 #define REGISTER_KERNELS(type) \
264 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
265 #undef REGISTER_KERNELS
|
| betainc_op.cc | 116 #define REGISTER_KERNELS(type) \
121 REGISTER_KERNELS(float);
122 REGISTER_KERNELS(double);
123 #undef REGISTER_KERNELS
|
| sparse_sparse_binary_op_shared.cc | 229 #define REGISTER_KERNELS(T) \
238 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
239 #undef REGISTER_KERNELS
|
| /external/tensorflow/tensorflow/contrib/framework/kernels/ |
| zero_initializer_op.cc | 61 #define REGISTER_KERNELS(D, T) \
65 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T);
81 #define REGISTER_GPU_KERNELS(T) REGISTER_KERNELS(GPU, T);
86 #undef REGISTER_KERNELS
|
