Home
last modified time | relevance | path

Searched refs:DECLARE_GPU_SPEC (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dargmax_op.cc165 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ macro
176 DECLARE_GPU_SPEC(T, int64, 1); \
177 DECLARE_GPU_SPEC(T, int64, 2); \
178 DECLARE_GPU_SPEC(T, int64, 3); \
179 DECLARE_GPU_SPEC(T, int64, 4); \
180 DECLARE_GPU_SPEC(T, int64, 5); \
181 DECLARE_GPU_SPEC(T, int64, 6); \
182 DECLARE_GPU_SPEC(T, int64, 7); \
183 DECLARE_GPU_SPEC(T, int32, 1); \
184 DECLARE_GPU_SPEC(T, int32, 2); \
[all …]
Dpopulation_count_op.cc143 #define DECLARE_GPU_SPEC(T) \ macro
150 TF_CALL_uint8(DECLARE_GPU_SPEC);
151 TF_CALL_int8(DECLARE_GPU_SPEC);
152 TF_CALL_uint16(DECLARE_GPU_SPEC);
153 TF_CALL_int16(DECLARE_GPU_SPEC);
154 TF_CALL_int32(DECLARE_GPU_SPEC);
155 TF_CALL_int64(DECLARE_GPU_SPEC);
157 #undef DECLARE_GPU_SPEC
Dslice_op.cc281 #define DECLARE_GPU_SPEC(T, NDIM) \ macro
291 DECLARE_GPU_SPEC(T, 1); \
292 DECLARE_GPU_SPEC(T, 2); \
293 DECLARE_GPU_SPEC(T, 3); \
294 DECLARE_GPU_SPEC(T, 4); \
295 DECLARE_GPU_SPEC(T, 5); \
296 DECLARE_GPU_SPEC(T, 6); \
297 DECLARE_GPU_SPEC(T, 7); \
298 DECLARE_GPU_SPEC(T, 8);
307 #undef DECLARE_GPU_SPEC
Dtraining_ops.cc990 #define DECLARE_GPU_SPEC(T) \ macro
997 DECLARE_GPU_SPEC(Eigen::half);
998 DECLARE_GPU_SPEC(float);
999 DECLARE_GPU_SPEC(double);
1001 DECLARE_GPU_SPEC(complex64);
1002 DECLARE_GPU_SPEC(complex128);
1004 #undef DECLARE_GPU_SPEC
1138 #define DECLARE_GPU_SPEC(T) \ macro
1147 DECLARE_GPU_SPEC(Eigen::half);
1148 DECLARE_GPU_SPEC(float);
[all …]
Done_hot_op.cc158 #define DECLARE_GPU_SPEC(T) \ macro
163 TF_CALL_int32(DECLARE_GPU_SPEC);
164 TF_CALL_int64(DECLARE_GPU_SPEC);
165 TF_CALL_GPU_ALL_TYPES(DECLARE_GPU_SPEC);
168 #undef DECLARE_GPU_SPEC
Din_topk_op.cc123 #define DECLARE_GPU_SPEC(T, TARGET_T) \ macro
132 DECLARE_GPU_SPEC(float, int32);
133 DECLARE_GPU_SPEC(float, int64);
135 #undef DECLARE_GPU_SPEC
Dpad_op.cc292 #define DECLARE_GPU_SPEC(T, Dims) \ macro
307 DECLARE_GPU_SPEC(T, 0); \
308 DECLARE_GPU_SPEC(T, 1); \
309 DECLARE_GPU_SPEC(T, 2); \
310 DECLARE_GPU_SPEC(T, 3); \
311 DECLARE_GPU_SPEC(T, 4); \
312 DECLARE_GPU_SPEC(T, 5); \
313 DECLARE_GPU_SPEC(T, 6);
Dbatch_norm_op.cc178 #define DECLARE_GPU_SPEC(T) \ macro
188 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
192 #undef DECLARE_GPU_SPEC
223 #define DECLARE_GPU_SPEC(T) \ macro
236 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
240 #undef DECLARE_GPU_SPEC
Ddata_format_ops.cc258 #define DECLARE_GPU_SPEC(T) \ macro
264 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
267 #undef DECLARE_GPU_SPEC
269 #define DECLARE_GPU_SPEC(T) \ macro
276 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
279 #undef DECLARE_GPU_SPEC
Dcompare_and_bitpack_op.cc181 #define DECLARE_GPU_SPEC(T) \ macro
189 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC)
190 TF_CALL_bool(DECLARE_GPU_SPEC)
192 #undef DECLARE_GPU_SPEC
Dbetainc_op.cc147 #define DECLARE_GPU_SPEC(T) \ macro
151 DECLARE_GPU_SPEC(float);
152 DECLARE_GPU_SPEC(double);
154 #undef DECLARE_GPU_SPEC
Dtopk_op.cc250 #define DECLARE_GPU_SPEC(T) \ macro
259 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
260 TF_CALL_INTEGRAL_TYPES(DECLARE_GPU_SPEC);
262 #undef DECLARE_GPU_SPEC
Dreverse_op.cc329 #define DECLARE_GPU_SPEC(T) \ macro
340 TF_CALL_uint8(DECLARE_GPU_SPEC);
341 TF_CALL_int8(DECLARE_GPU_SPEC);
342 TF_CALL_GPU_ALL_TYPES(DECLARE_GPU_SPEC);
343 #undef DECLARE_GPU_SPEC
Dsparse_tensor_dense_matmul_op.cc182 #define DECLARE_GPU_SPEC(T, Tindices, ADJ_A, ADJ_B) \ macro
194 DECLARE_GPU_SPEC(T, int32, ADJ_A, ADJ_B); \
195 DECLARE_GPU_SPEC(T, int64, ADJ_A, ADJ_B)
205 #undef DECLARE_GPU_SPEC
Dreverse_sequence_op.cc186 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ macro
196 DECLARE_GPU_SPEC(T, int32, Dims); \
197 DECLARE_GPU_SPEC(T, int64, Dims);
/external/tensorflow/tensorflow/core/kernels/image/
Dmirror_pad_op.cc204 #define DECLARE_GPU_SPEC(T, Tpaddings, i) \ macro
213 DECLARE_GPU_SPEC(T, int32, 1); \
214 DECLARE_GPU_SPEC(T, int32, 2); \
215 DECLARE_GPU_SPEC(T, int32, 3); \
216 DECLARE_GPU_SPEC(T, int32, 4); \
217 DECLARE_GPU_SPEC(T, int32, 5); \
218 DECLARE_GPU_SPEC(T, int64, 1); \
219 DECLARE_GPU_SPEC(T, int64, 2); \
220 DECLARE_GPU_SPEC(T, int64, 3); \
221 DECLARE_GPU_SPEC(T, int64, 4); \
[all …]
Dadjust_contrast_op.cc106 #define DECLARE_GPU_SPEC(T) \ macro
117 DECLARE_GPU_SPEC(uint8);
118 DECLARE_GPU_SPEC(int8);
119 DECLARE_GPU_SPEC(int16);
120 DECLARE_GPU_SPEC(int32);
121 DECLARE_GPU_SPEC(float);
122 DECLARE_GPU_SPEC(double);
123 #undef DECLARE_GPU_SPEC
391 #define DECLARE_GPU_SPEC(T) \ macro
399 DECLARE_GPU_SPEC(float);
[all …]
Dresize_nearest_neighbor_op_gpu.cu.cc192 #define DECLARE_GPU_SPEC(T) \ macro
198 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
200 #undef DECLARE_GPU_SPEC
246 #define DECLARE_GPU_SPEC(T) \ macro
252 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
254 #undef DECLARE_GPU_SPEC
Dextract_volume_patches_op.cc170 #define DECLARE_GPU_SPEC(T) \ macro
182 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
184 #undef DECLARE_GPU_SPEC
Dextract_image_patches_op.cc142 #define DECLARE_GPU_SPEC(T) \ macro
151 TF_CALL_GPU_ALL_TYPES(DECLARE_GPU_SPEC);
153 #undef DECLARE_GPU_SPEC
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_matrix_components_op.cc131 #define DECLARE_GPU_SPEC(T) \ macro
140 DECLARE_GPU_SPEC(int32);
141 DECLARE_GPU_SPEC(float);
142 DECLARE_GPU_SPEC(double);
143 DECLARE_GPU_SPEC(complex64);
144 DECLARE_GPU_SPEC(complex128);
146 #undef DECLARE_GPU_SPEC
Dsoftmax_op.cc100 #define DECLARE_GPU_SPEC(T) \ macro
107 DECLARE_GPU_SPEC(float);
108 DECLARE_GPU_SPEC(double);
110 #undef DECLARE_GPU_SPEC
209 #define DECLARE_GPU_SPEC(T) \ macro
217 DECLARE_GPU_SPEC(float);
218 DECLARE_GPU_SPEC(double);
220 #undef DECLARE_GPU_SPEC
Dmul_op.cc153 #define DECLARE_GPU_SPEC(T) \ macro
160 DECLARE_GPU_SPEC(float);
161 DECLARE_GPU_SPEC(double);
163 DECLARE_GPU_SPEC(std::complex<float>);
164 DECLARE_GPU_SPEC(std::complex<double>);
167 #undef DECLARE_GPU_SPEC
/external/tensorflow/tensorflow/core/kernels/linalg/
Dcholesky_op.cc79 #define DECLARE_GPU_SPEC(T) \ macro
89 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
90 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPEC);
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.cc388 #define DECLARE_GPU_SPEC(T) \ macro
403 DECLARE_GPU_SPEC(float);
404 #undef DECLARE_GPU_SPEC
418 #define DECLARE_GPU_SPEC(T) \ macro
436 DECLARE_GPU_SPEC(float);
437 #undef DECLARE_GPU_SPEC

12