Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
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 …]
Dargmax_op.cc158 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ macro
169 DECLARE_GPU_SPEC(T, int64, 1); \
170 DECLARE_GPU_SPEC(T, int64, 2); \
171 DECLARE_GPU_SPEC(T, int64, 3); \
172 DECLARE_GPU_SPEC(T, int64, 4); \
173 DECLARE_GPU_SPEC(T, int64, 5); \
174 DECLARE_GPU_SPEC(T, int32, 1); \
175 DECLARE_GPU_SPEC(T, int32, 2); \
176 DECLARE_GPU_SPEC(T, int32, 3); \
177 DECLARE_GPU_SPEC(T, int32, 4); \
[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
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
390 #define DECLARE_GPU_SPEC(T) \ macro
398 DECLARE_GPU_SPEC(float);
[all …]
Done_hot_op.cc156 #define DECLARE_GPU_SPEC(T) \ macro
161 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
162 TF_CALL_bool(DECLARE_GPU_SPEC);
163 TF_CALL_int32(DECLARE_GPU_SPEC);
164 TF_CALL_int64(DECLARE_GPU_SPEC);
167 #undef DECLARE_GPU_SPEC
Dslice_op.cc273 #define DECLARE_GPU_SPEC(T, NDIM) \ macro
283 DECLARE_GPU_SPEC(T, 1); \
284 DECLARE_GPU_SPEC(T, 2); \
285 DECLARE_GPU_SPEC(T, 3); \
286 DECLARE_GPU_SPEC(T, 4); \
287 DECLARE_GPU_SPEC(T, 5); \
288 DECLARE_GPU_SPEC(T, 6); \
289 DECLARE_GPU_SPEC(T, 7);
301 #undef DECLARE_GPU_SPEC
Dmatrix_set_diag_op.cc145 #define DECLARE_GPU_SPEC(T) \ macro
154 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
155 TF_CALL_bool(DECLARE_GPU_SPEC);
156 TF_CALL_complex64(DECLARE_GPU_SPEC);
157 TF_CALL_complex128(DECLARE_GPU_SPEC);
Dreverse_op.cc332 #define DECLARE_GPU_SPEC(T) \ macro
343 TF_CALL_uint8(DECLARE_GPU_SPEC);
344 TF_CALL_int8(DECLARE_GPU_SPEC);
345 TF_CALL_bool(DECLARE_GPU_SPEC);
346 TF_CALL_half(DECLARE_GPU_SPEC);
347 TF_CALL_float(DECLARE_GPU_SPEC);
348 TF_CALL_double(DECLARE_GPU_SPEC);
349 TF_CALL_complex64(DECLARE_GPU_SPEC);
350 TF_CALL_complex128(DECLARE_GPU_SPEC);
351 #undef DECLARE_GPU_SPEC
Ddata_format_ops.cc172 #define DECLARE_GPU_SPEC(T) \ macro
178 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
181 #undef DECLARE_GPU_SPEC
183 #define DECLARE_GPU_SPEC(T) \ macro
190 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
193 #undef DECLARE_GPU_SPEC
Dmatrix_band_part_op.cc202 #define DECLARE_GPU_SPEC(T) \ macro
212 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
213 TF_CALL_bool(DECLARE_GPU_SPEC);
214 TF_CALL_complex64(DECLARE_GPU_SPEC);
215 TF_CALL_complex128(DECLARE_GPU_SPEC);
216 #undef DECLARE_GPU_SPEC
Dmatrix_diag_op.cc177 #define DECLARE_GPU_SPEC(T) \ macro
189 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
190 TF_CALL_bool(DECLARE_GPU_SPEC);
191 TF_CALL_complex64(DECLARE_GPU_SPEC);
192 TF_CALL_complex128(DECLARE_GPU_SPEC);
Dcholesky_op.cc79 #define DECLARE_GPU_SPEC(T) \ macro
89 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
90 TF_CALL_complex64(DECLARE_GPU_SPEC);
91 TF_CALL_complex128(DECLARE_GPU_SPEC);
Dresize_nearest_neighbor_op_gpu.cu.cc195 #define DECLARE_GPU_SPEC(T) \ macro
201 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
203 #undef DECLARE_GPU_SPEC
251 #define DECLARE_GPU_SPEC(T) \ macro
257 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
259 #undef DECLARE_GPU_SPEC
Dbatch_norm_op.cc181 #define DECLARE_GPU_SPEC(T) \ macro
191 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
195 #undef DECLARE_GPU_SPEC
237 #define DECLARE_GPU_SPEC(T) \ macro
250 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T);
254 #undef DECLARE_GPU_SPEC
Dpad_op.cc300 #define DECLARE_GPU_SPEC(T, Dims) \ macro
315 DECLARE_GPU_SPEC(T, 0); \
316 DECLARE_GPU_SPEC(T, 1); \
317 DECLARE_GPU_SPEC(T, 2); \
318 DECLARE_GPU_SPEC(T, 3); \
319 DECLARE_GPU_SPEC(T, 4); \
320 DECLARE_GPU_SPEC(T, 5); \
321 DECLARE_GPU_SPEC(T, 6);
Dbetainc_op.cc146 #define DECLARE_GPU_SPEC(T) \ macro
150 DECLARE_GPU_SPEC(float);
151 DECLARE_GPU_SPEC(double);
153 #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
Dextract_image_patches_op.cc138 #define DECLARE_GPU_SPEC(T) \ macro
147 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
149 #undef DECLARE_GPU_SPEC
Dextract_volume_patches_op.cc167 #define DECLARE_GPU_SPEC(T) \ macro
179 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
181 #undef DECLARE_GPU_SPEC
Dtraining_ops.cc568 #define DECLARE_GPU_SPEC(T) \ macro
575 DECLARE_GPU_SPEC(Eigen::half);
576 DECLARE_GPU_SPEC(float);
577 DECLARE_GPU_SPEC(double);
578 #undef DECLARE_GPU_SPEC
729 #define DECLARE_GPU_SPEC(T) \ macro
738 DECLARE_GPU_SPEC(Eigen::half);
739 DECLARE_GPU_SPEC(float);
740 DECLARE_GPU_SPEC(double);
741 #undef DECLARE_GPU_SPEC
[all …]
Dtopk_op.cc248 #define DECLARE_GPU_SPEC(T) \ macro
257 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
258 TF_CALL_INTEGRAL_TYPES(DECLARE_GPU_SPEC);
260 #undef DECLARE_GPU_SPEC
Dsparse_tensor_dense_matmul_op.cc180 #define DECLARE_GPU_SPEC(T, Tindices, ADJ_A, ADJ_B) \ macro
192 DECLARE_GPU_SPEC(T, int32, ADJ_A, ADJ_B); \
193 DECLARE_GPU_SPEC(T, int64, ADJ_A, ADJ_B)
203 #undef DECLARE_GPU_SPEC
/external/tensorflow/tensorflow/contrib/framework/kernels/
Dzero_initializer_op.cc72 #define DECLARE_GPU_SPEC(T) \ macro
78 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPEC);
79 #undef DECLARE_GPU_SPEC
/external/tensorflow/tensorflow/contrib/rnn/kernels/
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
/external/tensorflow/tensorflow/contrib/seq2seq/kernels/
Dbeam_search_ops.cc170 #define DECLARE_GPU_SPEC(T) \ macro
180 DECLARE_GPU_SPEC(int32);
181 #undef DECLARE_GPU_SPEC

12