Home
last modified time | relevance | path

Searched refs:REGISTER_KERNELS (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dsearchsorted_op.cc162 #define REGISTER_KERNELS(type) \ macro
169 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
170 #undef REGISTER_KERNELS
172 #define REGISTER_KERNELS(type) \ macro
179 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
180 #undef REGISTER_KERNELS
184 #define REGISTER_KERNELS(type) \ macro
191 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
192 #undef REGISTER_KERNELS
194 #define REGISTER_KERNELS(type) \ macro
[all …]
Dsparse_add_grad_op.cc131 #define REGISTER_KERNELS(type) \ macro
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
Dbincount_op.cc199 #define REGISTER_KERNELS(type) \ macro
204 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
205 #undef REGISTER_KERNELS
209 #define REGISTER_KERNELS(type) \ macro
216 TF_CALL_int32(REGISTER_KERNELS);
217 TF_CALL_float(REGISTER_KERNELS);
218 #undef REGISTER_KERNELS
285 #define REGISTER_KERNELS(Tidx, T) \ macro
292 REGISTER_KERNELS(int32, T); \
293 REGISTER_KERNELS(int64, T);
[all …]
Ddense_update_ops.cc91 #define REGISTER_KERNELS(type) \ macro
96 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
98 TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
100 TF_CALL_quint16(REGISTER_KERNELS);
101 #undef REGISTER_KERNELS
118 #define REGISTER_KERNELS(type) \ macro
126 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
127 #undef REGISTER_KERNELS
Dsparse_add_op.cc169 #define REGISTER_KERNELS(type, thresh_type) \ macro
176 REGISTER_KERNELS(float, float);
177 REGISTER_KERNELS(double, double);
178 REGISTER_KERNELS(int64, int64);
179 REGISTER_KERNELS(int32, int32);
180 REGISTER_KERNELS(int16, int16);
181 REGISTER_KERNELS(int8, int8);
182 REGISTER_KERNELS(complex64, float);
183 REGISTER_KERNELS(complex128, double);
184 #undef REGISTER_KERNELS
Dtraining_ops.cc973 #define REGISTER_KERNELS(D, T) \ macro
982 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T);
1007 REGISTER_KERNELS(GPU, Eigen::half);
1008 REGISTER_KERNELS(GPU, float);
1009 REGISTER_KERNELS(GPU, double);
1011 REGISTER_KERNELS(GPU, complex64);
1012 REGISTER_KERNELS(GPU, complex128);
1017 #undef REGISTER_KERNELS
1119 #define REGISTER_KERNELS(D, T) \ macro
1130 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T);
[all …]
Dsparse_reduce_op.cc235 #define REGISTER_KERNELS(T) \ macro
239 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
240 #undef REGISTER_KERNELS
242 #define REGISTER_KERNELS(T) \ macro
246 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
247 #undef REGISTER_KERNELS
331 #define REGISTER_KERNELS(T) \ macro
335 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
336 #undef REGISTER_KERNELS
338 #define REGISTER_KERNELS(T) \ macro
[all …]
Dhistogram_op.cc121 #define REGISTER_KERNELS(type) \ macro
133 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
134 #undef REGISTER_KERNELS
137 #define REGISTER_KERNELS(type) \ macro
146 TF_CALL_GPU_NUMBER_TYPES(REGISTER_KERNELS);
147 #undef REGISTER_KERNELS
Dragged_tensor_to_variant_op.cc256 #define REGISTER_KERNELS(value_type) \ macro
259 TF_CALL_POD_TYPES(REGISTER_KERNELS);
260 TF_CALL_tstring(REGISTER_KERNELS);
261 TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
262 TF_CALL_quint16(REGISTER_KERNELS);
263 TF_CALL_qint16(REGISTER_KERNELS);
264 #undef REGISTER_KERNELS
Dragged_tensor_from_variant_op.cc283 #define REGISTER_KERNELS(value_type) \ macro
286 TF_CALL_POD_TYPES(REGISTER_KERNELS);
287 TF_CALL_tstring(REGISTER_KERNELS);
288 TF_CALL_QUANTIZED_TYPES(REGISTER_KERNELS);
289 TF_CALL_quint16(REGISTER_KERNELS);
290 TF_CALL_qint16(REGISTER_KERNELS);
291 #undef REGISTER_KERNELS
Dtopk_op.cc239 #define REGISTER_KERNELS(type) \ macro
243 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
245 #undef REGISTER_KERNELS
266 #define REGISTER_KERNELS(type) \ macro
276 TF_CALL_GPU_NUMBER_TYPES(REGISTER_KERNELS);
277 TF_CALL_INTEGRAL_TYPES(REGISTER_KERNELS);
278 #undef REGISTER_KERNELS
Dsparse_reorder_op.cc86 #define REGISTER_KERNELS(type) \ macro
91 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
92 #undef REGISTER_KERNELS
Dsparse_conditional_accumulator_op.cc59 #define REGISTER_KERNELS(type, dev) \ macro
65 #define REGISTER_KERNELS_CPU(type) REGISTER_KERNELS(type, CPU)
72 #undef REGISTER_KERNELS
Dsparse_to_dense_op.cc150 #define REGISTER_KERNELS(type, index_type) \ macro
158 REGISTER_KERNELS(type, int32); \
159 REGISTER_KERNELS(type, int64);
166 #undef REGISTER_KERNELS
Dconditional_accumulator_op.cc57 #define REGISTER_KERNELS(type, dev) \ macro
107 #define REGISTER_KERNELS_CPU(type) REGISTER_KERNELS(type, CPU)
114 #undef REGISTER_KERNELS
123 #undef REGISTER_KERNELS
Dsparse_split_op.cc95 #define REGISTER_KERNELS(type) \ macro
100 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
101 #undef REGISTER_KERNELS
Dsparse_fill_empty_rows_op.cc226 #define REGISTER_KERNELS(D, T, Tindex) \ macro
233 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T, int64)
237 #undef REGISTER_KERNELS
331 #define REGISTER_KERNELS(D, T, Tindex) \ macro
337 #define REGISTER_CPU_KERNELS(T) REGISTER_KERNELS(CPU, T, int64)
341 #undef REGISTER_KERNELS
Dsparse_slice_op.cc99 #define REGISTER_KERNELS(type) \ macro
104 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
105 #undef REGISTER_KERNELS
Dbetainc_op.cc116 #define REGISTER_KERNELS(type) \ macro
121 REGISTER_KERNELS(float);
122 REGISTER_KERNELS(double);
123 #undef REGISTER_KERNELS
Dsoftplus_op.cc79 #define REGISTER_KERNELS(type) \ macro
87 TF_CALL_FLOAT_TYPES(REGISTER_KERNELS);
88 #undef REGISTER_KERNELS
Dsoftsign_op.cc80 #define REGISTER_KERNELS(type) \ macro
88 TF_CALL_FLOAT_TYPES(REGISTER_KERNELS);
89 #undef REGISTER_KERNELS
Dsparse_slice_grad_op.cc118 #define REGISTER_KERNELS(type) \ macro
123 TF_CALL_NUMBER_TYPES(REGISTER_KERNELS);
124 #undef REGISTER_KERNELS
Dserialize_sparse_op.cc396 #define REGISTER_KERNELS(type) \ macro
403 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
404 #undef REGISTER_KERNELS
407 #define REGISTER_KERNELS(type) \ macro
414 TF_CALL_ALL_TYPES(REGISTER_KERNELS);
415 #undef REGISTER_KERNELS
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_jpeg_shape_op.cc67 #define REGISTER_KERNELS(type) \ macro
73 TF_CALL_int32(REGISTER_KERNELS);
74 TF_CALL_int64(REGISTER_KERNELS);
75 #undef REGISTER_KERNELS
Drandom_crop_op.cc108 #define REGISTER_KERNELS(type) \ macro
113 TF_CALL_REAL_NUMBER_TYPES(REGISTER_KERNELS);
114 #undef REGISTER_KERNELS

12