Home
last modified time | relevance | path

Searched refs:HANDLE_DIM (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc158 #define HANDLE_DIM(NDIM) \ in Compute() macro
166 HANDLE_DIM(1); in Compute()
167 HANDLE_DIM(2); in Compute()
168 HANDLE_DIM(3); in Compute()
169 HANDLE_DIM(4); in Compute()
170 HANDLE_DIM(5); in Compute()
171 HANDLE_DIM(6); in Compute()
172 HANDLE_DIM(7); in Compute()
173 HANDLE_DIM(8); in Compute()
175 #undef HANDLE_DIM in Compute()
[all …]
Dwhere_op.cc159 #define HANDLE_DIM(NDIM) \ in Compute() macro
168 HANDLE_DIM(1); in Compute()
169 HANDLE_DIM(2); in Compute()
170 HANDLE_DIM(3); in Compute()
171 HANDLE_DIM(4); in Compute()
172 HANDLE_DIM(5); in Compute()
173 HANDLE_DIM(6); in Compute()
174 HANDLE_DIM(7); in Compute()
175 HANDLE_DIM(8); in Compute()
182 #undef HANDLE_DIM in Compute()
[all …]
Dargmax_op.cc84 #define HANDLE_DIM(NDIM) \ in Compute() macro
92 HANDLE_DIM(1); in Compute()
93 HANDLE_DIM(2); in Compute()
94 HANDLE_DIM(3); in Compute()
95 HANDLE_DIM(4); in Compute()
96 HANDLE_DIM(5); in Compute()
97 HANDLE_DIM(6); in Compute()
98 HANDLE_DIM(7); in Compute()
108 #undef HANDLE_DIM
Dtranspose_functor_gpu.cu.cc172 #define HANDLE_DIM(DIM) \ macro
188 HANDLE_DIM(2); in run()
189 HANDLE_DIM(3); in run()
190 HANDLE_DIM(4); in run()
191 HANDLE_DIM(5); in run()
192 HANDLE_DIM(6); in run()
193 HANDLE_DIM(7); in run()
194 HANDLE_DIM(8); in run()
202 #undef HANDLE_DIM
Dslice_op.cc195 #define HANDLE_DIM(NDIM) \ in Compute() macro
201 HANDLE_DIM(1); in Compute()
202 HANDLE_DIM(2); in Compute()
203 HANDLE_DIM(3); in Compute()
204 HANDLE_DIM(4); in Compute()
205 HANDLE_DIM(5); in Compute()
206 HANDLE_DIM(6); in Compute()
207 HANDLE_DIM(7); in Compute()
208 HANDLE_DIM(8); in Compute()
210 #undef HANDLE_DIM in Compute()
Dcwise_op_select.cc194 #define HANDLE_DIM(NDIMS) \ in Compute() macro
210 HANDLE_DIM(1); in Compute()
213 HANDLE_DIM(2); in Compute()
216 HANDLE_DIM(3); in Compute()
219 HANDLE_DIM(4); in Compute()
222 HANDLE_DIM(5); in Compute()
225 HANDLE_DIM(6); in Compute()
228 HANDLE_DIM(7); in Compute()
231 HANDLE_DIM(8); in Compute()
Dreverse_sequence_op.cc140 #define HANDLE_DIM(NDIM) \ in Compute() macro
148 HANDLE_DIM(2); in Compute()
149 HANDLE_DIM(3); in Compute()
150 HANDLE_DIM(4); in Compute()
151 HANDLE_DIM(5); in Compute()
Dtile_ops.cc373 #define HANDLE_DIM(DT, NDIM) \ in Compute() macro
381 HANDLE_DIM(T, 1) \ in Compute()
382 HANDLE_DIM(T, 2) \ in Compute()
383 HANDLE_DIM(T, 3) \ in Compute()
384 HANDLE_DIM(T, 4) \ in Compute()
385 HANDLE_DIM(T, 5) \ in Compute()
386 HANDLE_DIM(T, 6) \ in Compute()
387 HANDLE_DIM(T, 7) in Compute()
403 #undef HANDLE_DIM in Compute()
441 #define HANDLE_DIM(D) \ in HandleCaseImpl() macro
[all …]