Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc156 #define HANDLE_DIM(NDIM) \ in Compute() macro
164 HANDLE_DIM(1); in Compute()
165 HANDLE_DIM(2); in Compute()
166 HANDLE_DIM(3); in Compute()
167 HANDLE_DIM(4); in Compute()
168 HANDLE_DIM(5); in Compute()
169 HANDLE_DIM(6); in Compute()
170 HANDLE_DIM(7); in Compute()
172 #undef HANDLE_DIM in Compute()
254 #define HANDLE_DIM(NDIM) \ in Compute() macro
[all …]
Dwhere_op.cc160 #define HANDLE_DIM(NDIM) \ in Compute() macro
169 HANDLE_DIM(1); in Compute()
170 HANDLE_DIM(2); in Compute()
171 HANDLE_DIM(3); in Compute()
172 HANDLE_DIM(4); in Compute()
173 HANDLE_DIM(5); in Compute()
180 #undef HANDLE_DIM in Compute()
323 #define HANDLE_DIM(NDIM) \ in ComputeAsyncType() macro
332 HANDLE_DIM(1); in ComputeAsyncType()
333 HANDLE_DIM(2); in ComputeAsyncType()
[all …]
Dtranspose_functor_gpu.cu.cc171 #define HANDLE_DIM(DIM) \ macro
187 HANDLE_DIM(2); in run()
188 HANDLE_DIM(3); in run()
189 HANDLE_DIM(4); in run()
190 HANDLE_DIM(5); in run()
191 HANDLE_DIM(6); in run()
192 HANDLE_DIM(7); in run()
193 HANDLE_DIM(8); in run()
201 #undef HANDLE_DIM
Dargmax_op.cc83 #define HANDLE_DIM(NDIM) \ in Compute() macro
91 HANDLE_DIM(1); in Compute()
92 HANDLE_DIM(2); in Compute()
93 HANDLE_DIM(3); in Compute()
94 HANDLE_DIM(4); in Compute()
95 HANDLE_DIM(5); in Compute()
103 #undef HANDLE_DIM
Dslice_op.cc192 #define HANDLE_DIM(NDIM) \ in Compute() macro
198 HANDLE_DIM(1); in Compute()
199 HANDLE_DIM(2); in Compute()
200 HANDLE_DIM(3); in Compute()
201 HANDLE_DIM(4); in Compute()
202 HANDLE_DIM(5); in Compute()
203 HANDLE_DIM(6); in Compute()
204 HANDLE_DIM(7); in Compute()
206 #undef HANDLE_DIM in Compute()
Dtile_ops.cc306 #define HANDLE_DIM(DT, NDIM) \ in Compute() macro
314 HANDLE_DIM(T, 1) \ in Compute()
315 HANDLE_DIM(T, 2) \ in Compute()
316 HANDLE_DIM(T, 3) \ in Compute()
317 HANDLE_DIM(T, 4) \ in Compute()
318 HANDLE_DIM(T, 5) \ in Compute()
319 HANDLE_DIM(T, 6) \ in Compute()
320 HANDLE_DIM(T, 7) in Compute()
336 #undef HANDLE_DIM in Compute()
374 #define HANDLE_DIM(D) \ in HandleCaseImpl() macro
[all …]
Dreverse_sequence_op.cc138 #define HANDLE_DIM(NDIM) \ in Compute() macro
146 HANDLE_DIM(2); in Compute()
147 HANDLE_DIM(3); in Compute()
148 HANDLE_DIM(4); in Compute()
149 HANDLE_DIM(5); in Compute()