HomeSort by relevance Sort by last modified time
    Searched refs:REGISTER_GPU_KERNEL (Results 1 - 25 of 26) sorted by null

1 2

  /external/tensorflow/tensorflow/core/kernels/
split_lib_gpu.h 48 #define REGISTER_GPU_KERNEL(T) \
53 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
54 TF_CALL_complex64(REGISTER_GPU_KERNEL);
55 TF_CALL_complex128(REGISTER_GPU_KERNEL);
56 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
57 TF_CALL_uint8(REGISTER_GPU_KERNEL);
58 TF_CALL_bool(REGISTER_GPU_KERNEL);
59 #undef REGISTER_GPU_KERNEL
l2loss_op_gpu.cu.cc 64 #define REGISTER_GPU_KERNEL(T) \
69 REGISTER_GPU_KERNEL(float);
70 REGISTER_GPU_KERNEL(double);
71 REGISTER_GPU_KERNEL(Eigen::half);
72 #undef REGISTER_GPU_KERNEL
shape_ops.cc 72 #define REGISTER_GPU_KERNEL(type) \
86 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
87 TF_CALL_bool(REGISTER_GPU_KERNEL);
88 TF_CALL_variant(REGISTER_GPU_KERNEL);
89 #undef REGISTER_GPU_KERNEL
124 #define REGISTER_GPU_KERNEL(type) \
138 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
139 TF_CALL_bool(REGISTER_GPU_KERNEL);
140 #undef REGISTER_GPU_KERNEL
226 #define REGISTER_GPU_KERNEL(type)
    [all...]
session_ops.cc 72 #define REGISTER_GPU_KERNEL(type) \
84 TF_CALL_NUMBER_TYPES(REGISTER_GPU_KERNEL);
85 REGISTER_GPU_KERNEL(bool);
86 #undef REGISTER_GPU_KERNEL
125 #define REGISTER_GPU_KERNEL(type) \
132 TF_CALL_NUMBER_TYPES(REGISTER_GPU_KERNEL);
133 REGISTER_GPU_KERNEL(bool);
134 #undef REGISTER_GPU_KERNEL
reshape_op.cc 32 #define REGISTER_GPU_KERNEL(type) \
45 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
46 TF_CALL_bool(REGISTER_GPU_KERNEL);
47 #undef REGISTER_GPU_KERNEL
cwise_op_clip.cc 186 #define REGISTER_GPU_KERNEL(type) \
190 REGISTER_GPU_KERNEL(Eigen::half);
191 REGISTER_GPU_KERNEL(float);
192 REGISTER_GPU_KERNEL(double);
193 REGISTER_GPU_KERNEL(int8);
194 REGISTER_GPU_KERNEL(int16);
195 REGISTER_GPU_KERNEL(int64);
196 REGISTER_GPU_KERNEL(uint8);
197 REGISTER_GPU_KERNEL(uint16);
211 #undef REGISTER_GPU_KERNEL
    [all...]
sequence_ops.cc 91 #define REGISTER_GPU_KERNEL(T) REGISTER_KERNEL(DEVICE_GPU, T)
108 TF_CALL_float(REGISTER_GPU_KERNEL);
109 TF_CALL_double(REGISTER_GPU_KERNEL);
110 TF_CALL_int32(REGISTER_GPU_KERNEL);
111 TF_CALL_int64(REGISTER_GPU_KERNEL);
117 #undef REGISTER_GPU_KERNEL
177 #define REGISTER_GPU_KERNEL(T) REGISTER_KERNEL_ALL_NUMS(DEVICE_GPU, T)
178 TF_CALL_float(REGISTER_GPU_KERNEL);
179 TF_CALL_double(REGISTER_GPU_KERNEL);
180 #undef REGISTER_GPU_KERNEL
    [all...]
stack_ops.cc 67 #define REGISTER_GPU_KERNEL(type) \
79 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
80 #undef REGISTER_GPU_KERNEL
134 #define REGISTER_GPU_KERNEL(type) \
146 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
147 #undef REGISTER_GPU_KERNEL
split_lib_gpu.cu.cc 245 #define REGISTER_GPU_KERNEL(T) template struct SplitOpGPULaunch<T>;
247 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
248 TF_CALL_complex64(REGISTER_GPU_KERNEL);
249 TF_CALL_complex128(REGISTER_GPU_KERNEL);
250 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
251 #undef REGISTER_GPU_KERNEL
252 #define REGISTER_GPU_KERNEL(T) \
256 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
257 TF_CALL_complex64(REGISTER_GPU_KERNEL);
258 TF_CALL_complex128(REGISTER_GPU_KERNEL);
    [all...]
identity_op.cc 88 #define REGISTER_GPU_KERNEL(type) \
110 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
111 REGISTER_GPU_KERNEL(Variant);
113 #undef REGISTER_GPU_KERNEL
control_flow_ops.cc 249 #define REGISTER_GPU_KERNEL(type) \
263 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
265 TF_CALL_QUANTIZED_TYPES(REGISTER_GPU_KERNEL);
267 REGISTER_GPU_KERNEL(bool);
269 REGISTER_GPU_KERNEL(uint64);
270 TF_CALL_variant(REGISTER_GPU_KERNEL);
272 #undef REGISTER_GPU_KERNEL
358 #define REGISTER_GPU_KERNEL(type) \
365 TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
367 REGISTER_GPU_KERNEL(bool)
    [all...]
data_format_ops.cc 197 #define REGISTER_GPU_KERNEL(T) \
201 TF_CALL_int32(REGISTER_GPU_KERNEL);
202 TF_CALL_int64(REGISTER_GPU_KERNEL);
203 #undef REGISTER_GPU_KERNEL
205 #define REGISTER_GPU_KERNEL(T) \
216 TF_CALL_int32(REGISTER_GPU_KERNEL);
217 TF_CALL_int64(REGISTER_GPU_KERNEL);
218 #undef REGISTER_GPU_KERNEL
cross_op.cc 104 #define REGISTER_GPU_KERNEL(type) \
109 TF_CALL_REAL_NUMBER_TYPES(REGISTER_GPU_KERNEL);
110 #undef REGISTER_GPU_KERNEL
batch_norm_op.cc 199 #define REGISTER_GPU_KERNEL(T) \
205 TF_CALL_half(REGISTER_GPU_KERNEL);
206 TF_CALL_float(REGISTER_GPU_KERNEL);
207 #undef REGISTER_GPU_KERNEL
258 #define REGISTER_GPU_KERNEL(T) \
264 TF_CALL_half(REGISTER_GPU_KERNEL);
265 TF_CALL_float(REGISTER_GPU_KERNEL);
266 #undef REGISTER_GPU_KERNEL
adjust_contrast_op.cc 127 #define REGISTER_GPU_KERNEL(T) \
131 REGISTER_GPU_KERNEL(uint8);
132 REGISTER_GPU_KERNEL(int8);
133 REGISTER_GPU_KERNEL(int16);
134 REGISTER_GPU_KERNEL(int32);
135 REGISTER_GPU_KERNEL(float);
136 REGISTER_GPU_KERNEL(double);
137 #undef REGISTER_GPU_KERNEL
mirror_pad_op.cc 230 #define REGISTER_GPU_KERNEL(T) \
244 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
245 #undef REGISTER_GPU_KERNEL
437 #define REGISTER_GPU_KERNEL(T) \
451 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
452 #undef REGISTER_GPU_KERNEL
quantize_and_dequantize_op.cc 245 #define REGISTER_GPU_KERNEL(T) \
262 TF_CALL_float(REGISTER_GPU_KERNEL);
263 TF_CALL_double(REGISTER_GPU_KERNEL);
264 #undef REGISTER_GPU_KERNEL
bias_op.cc 378 #define REGISTER_GPU_KERNEL(type) \
386 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
387 #undef REGISTER_GPU_KERNEL
619 #define REGISTER_GPU_KERNEL(type) \
624 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNEL);
625 #undef REGISTER_GPU_KERNEL
depthwise_conv_op.cc 466 #define REGISTER_GPU_KERNEL(T) \
471 TF_CALL_half(REGISTER_GPU_KERNEL);
472 TF_CALL_float(REGISTER_GPU_KERNEL);
473 TF_CALL_double(REGISTER_GPU_KERNEL);
depthwise_conv_grad_op.cc     [all...]
pad_op.cc 329 #define REGISTER_GPU_KERNEL(T) \
357 TF_CALL_GPU_ALL_TYPES(REGISTER_GPU_KERNEL);
358 TF_CALL_int8(REGISTER_GPU_KERNEL);
359 TF_CALL_uint8(REGISTER_GPU_KERNEL);
conv_grad_ops_3d.cc     [all...]
  /external/tensorflow/tensorflow/contrib/seq2seq/kernels/
beam_search_ops.cc 184 #define REGISTER_GPU_KERNEL(T) \
191 REGISTER_GPU_KERNEL(int32);
192 #undef REGISTER_GPU_KERNEL
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
gru_ops.cc 408 #define REGISTER_GPU_KERNEL(T) \
413 REGISTER_GPU_KERNEL(float);
414 #undef REGISTER_GPU_KERNEL
441 #define REGISTER_GPU_KERNEL(T) \
446 REGISTER_GPU_KERNEL(float);
447 #undef REGISTER_GPU_KERNEL
lstm_ops.cc 406 #define REGISTER_GPU_KERNEL(T) \
411 REGISTER_GPU_KERNEL(float);
412 REGISTER_GPU_KERNEL(Eigen::half);
413 // REGISTER_GPU_KERNEL(double);
414 #undef REGISTER_GPU_KERNEL
701 #define REGISTER_GPU_KERNEL(T) \
    [all...]

Completed in 2017 milliseconds

1 2