Searched defs:REGISTER (Results 1 - 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
H A Dextract_image_patches_op_gpu.cu.cc29 #define REGISTER(T) template struct ExtractImagePatchesForward<GPUDevice, T>; macro
31 TF_CALL_GPU_NUMBER_TYPES(REGISTER);
33 #undef REGISTER macro
H A Dmkl_cwise_ops_common.cc60 // We will need to redefine "REGISTER" to include the mkl_op_registry flag
61 #pragma push_macro("REGISTER")
62 #undef REGISTER macro
63 #define REGISTER(OP, D, N, F, T) \ macro
82 #undef REGISTER macro
83 #pragma pop_macro("REGISTER")
H A Dstring_to_number_op.cc62 #define REGISTER(type) \ macro
67 REGISTER(float);
68 REGISTER(double);
69 REGISTER(int32);
70 REGISTER(int64);
71 #undef REGISTER macro
H A Dconcat_lib_cpu.cc65 #define REGISTER(T) \ macro
70 TF_CALL_ALL_TYPES(REGISTER)
71 REGISTER(quint8)
72 REGISTER(qint8)
73 REGISTER(quint16)
74 REGISTER(qint16)
75 REGISTER(qint32)
82 REGISTER(string);
H A Dconcat_lib_gpu.cc108 #define REGISTER(T) \ macro
115 TF_CALL_GPU_NUMBER_TYPES(REGISTER);
116 TF_CALL_complex64(REGISTER);
117 TF_CALL_complex128(REGISTER);
118 TF_CALL_int64(REGISTER);
119 TF_CALL_bfloat16(REGISTER);
120 TF_CALL_bool(REGISTER);
122 #undef REGISTER macro
H A Dcount_up_to_op.cc101 #define REGISTER(TYPE) \ macro
109 REGISTER(int32);
110 REGISTER(int64);
112 #undef REGISTER macro
H A Dparse_tensor_op.cc89 #define REGISTER(T) \ macro
93 TF_CALL_ALL_TYPES(REGISTER)
94 #undef REGISTER macro
H A Dsparse_xent_op_gpu.cu.cc45 #define REGISTER(Index) \ macro
50 REGISTER(int32)
51 REGISTER(int64)
52 #undef REGISTER macro
H A Dstateless_random_ops.cc110 #define REGISTER(TYPE) \ macro
135 TF_CALL_half(REGISTER);
136 TF_CALL_float(REGISTER);
137 TF_CALL_double(REGISTER);
139 #undef REGISTER macro
143 #define REGISTER(TYPE) \ macro
171 TF_CALL_half(REGISTER);
172 TF_CALL_float(REGISTER);
173 TF_CALL_double(REGISTER);
175 #undef REGISTER macro
[all...]
H A Dsummary_tensor_op.cc60 #define REGISTER(T) \ macro
65 TF_CALL_ALL_TYPES(REGISTER)
67 #undef REGISTER macro
99 #define REGISTER(T) \ macro
104 TF_CALL_ALL_TYPES(REGISTER)
106 #undef REGISTER macro
H A Ddecode_raw_op.cc99 #define REGISTER(type) \ macro
104 REGISTER(Eigen::half);
105 REGISTER(float);
106 REGISTER(double);
107 REGISTER(int32);
108 REGISTER(uint16);
109 REGISTER(uint8);
110 REGISTER(int16);
111 REGISTER(int8);
112 REGISTER(int6
114 #undef REGISTER macro
[all...]
H A Drandom_shuffle_op.cc108 #define REGISTER(T) \ macro
112 TF_CALL_ALL_TYPES(REGISTER)
H A Dfunction_ops.cc96 #define REGISTER(type) \ macro
99 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER)
100 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name("_Arg")
105 #undef REGISTER macro
106 #define REGISTER(type) \ macro
110 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER)
111 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name("_Retval")
116 #undef REGISTER macro
119 #define REGISTER(type) \ macro
122 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER)
128 #undef REGISTER macro
136 #define REGISTER macro
145 #undef REGISTER macro
[all...]
H A Dself_adjoint_eig_v2_op_gpu.cc166 #define REGISTER(Scalar) \ macro
171 REGISTER(float);
172 REGISTER(double);
173 REGISTER(complex64);
174 REGISTER(complex128);
176 #undef REGISTER macro
H A Dsparse_xent_op.cc115 #define REGISTER(Dev, T, Index) \ macro
122 REGISTER(CPU, float, int32)
123 REGISTER(CPU, float, int64)
124 REGISTER(CPU, double, int32)
125 REGISTER(CPU, double, int64)
126 REGISTER(CPU, Eigen::half, int32)
127 REGISTER(CPU, Eigen::half, int64)
130 REGISTER(GPU, float, int32)
131 REGISTER(GPU, float, int64)
132 REGISTER(GP
136 #undef REGISTER macro
[all...]
H A Dsummary_op.cc114 #define REGISTER(T) \ macro
121 TF_CALL_REAL_NUMBER_TYPES(REGISTER)
122 #undef REGISTER macro
H A Dtranspose_op.cc222 #define REGISTER(T) \ macro
233 TF_CALL_ALL_TYPES(REGISTER);
234 #undef REGISTER macro
238 #define REGISTER(T) \ macro
249 TF_CALL_ALL_TYPES(REGISTER)
250 #undef REGISTER macro
269 #define REGISTER(T) \ macro
280 TF_CALL_POD_TYPES(REGISTER);
281 #undef REGISTER macro
299 #define REGISTER( macro
311 #undef REGISTER macro
[all...]
H A Dextract_image_patches_op.cc124 #define REGISTER(T) \ macro
129 TF_CALL_REAL_NUMBER_TYPES(REGISTER);
131 #undef REGISTER macro
154 #define REGISTER(T) \ macro
159 TF_CALL_GPU_NUMBER_TYPES(REGISTER);
161 #undef REGISTER macro
H A Dinplace_ops.cc152 #define REGISTER(type) \ macro
157 TF_CALL_POD_STRING_TYPES(REGISTER)
158 #undef REGISTER macro
192 #define REGISTER(type) \ macro
197 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER)
198 #undef REGISTER macro
231 #define REGISTER(type) \ macro
236 TF_CALL_GPU_NUMBER_TYPES(REGISTER)
237 #undef REGISTER macro
H A Dmultinomial_op.cc212 #define REGISTER(TYPE) \ macro
224 TF_CALL_half(REGISTER);
225 TF_CALL_float(REGISTER);
226 TF_CALL_double(REGISTER);
227 #undef REGISTER macro
230 #define REGISTER(TYPE) \ macro
244 TF_CALL_half(REGISTER);
245 TF_CALL_float(REGISTER);
246 TF_CALL_double(REGISTER);
247 #undef REGISTER macro
[all...]
H A Drandom_poisson_op.cc322 #define REGISTER(TYPE) \ macro
327 TF_CALL_half(REGISTER);
328 TF_CALL_float(REGISTER);
329 TF_CALL_double(REGISTER);
353 #undef REGISTER macro
/external/tensorflow/tensorflow/contrib/image/kernels/
H A Dimage_ops.cc93 #define REGISTER(TYPE) \ macro
99 TF_CALL_uint8(REGISTER);
100 TF_CALL_int32(REGISTER);
101 TF_CALL_int64(REGISTER);
102 TF_CALL_float(REGISTER);
103 TF_CALL_double(REGISTER);
105 #undef REGISTER macro
130 #define REGISTER(TYPE) \ macro
136 TF_CALL_uint8(REGISTER);
137 TF_CALL_int32(REGISTER);
142 #undef REGISTER macro
[all...]
/external/nist-sip/java/javax/sip/message/
H A DRequest.java12 String REGISTER = "REGISTER"; field in interface:Request
/external/smali/smalidea/src/main/java/org/jf/smalidea/
H A DSmaliHighlightingColors.java74 public static final TextAttributesKey REGISTER = createTextAttributesKey( field in class:SmaliHighlightingColors
75 "REGISTER", DefaultLanguageHighlighterColors.LOCAL_VARIABLE);
/external/tensorflow/tensorflow/contrib/resampler/kernels/
H A Dresampler_ops.cc182 #define REGISTER(TYPE) \ macro
187 TF_CALL_half(REGISTER);
188 TF_CALL_float(REGISTER);
189 TF_CALL_double(REGISTER);
190 #undef REGISTER macro
193 #define REGISTER(TYPE) \ macro
197 TF_CALL_float(REGISTER);
198 TF_CALL_double(REGISTER);
199 #undef REGISTER macro
393 #define REGISTER(TYP macro
401 #undef REGISTER macro
404 #define REGISTER macro
413 #undef REGISTER macro
[all...]

Completed in 680 milliseconds

123