OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:REGISTER_CPU_KERNEL
(Results
1 - 15
of
15
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
quantized_reshape_op.cc
46
#define
REGISTER_CPU_KERNEL
(type) \
53
REGISTER_CPU_KERNEL
(::tensorflow::quint8);
54
REGISTER_CPU_KERNEL
(::tensorflow::qint32);
56
#undef
REGISTER_CPU_KERNEL
sequence_ops.cc
90
#define
REGISTER_CPU_KERNEL
(T) REGISTER_KERNEL(DEVICE_CPU, T)
101
TF_CALL_float(
REGISTER_CPU_KERNEL
);
102
TF_CALL_double(
REGISTER_CPU_KERNEL
);
103
TF_CALL_int32(
REGISTER_CPU_KERNEL
);
104
TF_CALL_int64(
REGISTER_CPU_KERNEL
);
116
#undef
REGISTER_CPU_KERNEL
170
#define
REGISTER_CPU_KERNEL
(T) REGISTER_KERNEL_ALL_NUMS(DEVICE_CPU, T)
171
TF_CALL_float(
REGISTER_CPU_KERNEL
);
172
TF_CALL_double(
REGISTER_CPU_KERNEL
);
188
#undef
REGISTER_CPU_KERNEL
[
all
...]
cross_op.cc
83
#define
REGISTER_CPU_KERNEL
(type) \
87
TF_CALL_REAL_NUMBER_TYPES(
REGISTER_CPU_KERNEL
);
88
#undef
REGISTER_CPU_KERNEL
check_numerics_op.cc
211
#define
REGISTER_CPU_KERNEL
(T) \
215
TF_CALL_half(
REGISTER_CPU_KERNEL
);
216
TF_CALL_float(
REGISTER_CPU_KERNEL
);
217
TF_CALL_double(
REGISTER_CPU_KERNEL
);
quantize_and_dequantize_op.cc
209
#define
REGISTER_CPU_KERNEL
(T) \
221
TF_CALL_float(
REGISTER_CPU_KERNEL
);
222
TF_CALL_double(
REGISTER_CPU_KERNEL
);
223
#undef
REGISTER_CPU_KERNEL
draw_bounding_box_op.cc
179
#define
REGISTER_CPU_KERNEL
(T) \
183
TF_CALL_half(
REGISTER_CPU_KERNEL
);
184
TF_CALL_float(
REGISTER_CPU_KERNEL
);
edit_distance_op.cc
233
#define
REGISTER_CPU_KERNEL
(T) \
238
TF_CALL_POD_STRING_TYPES(
REGISTER_CPU_KERNEL
);
240
#undef
REGISTER_CPU_KERNEL
depthwise_conv_op.cc
421
#define
REGISTER_CPU_KERNEL
(T) \
426
TF_CALL_half(
REGISTER_CPU_KERNEL
);
427
TF_CALL_float(
REGISTER_CPU_KERNEL
);
429
TF_CALL_double(
REGISTER_CPU_KERNEL
);
conv_ops_3d.cc
144
#define
REGISTER_CPU_KERNEL
(T) \
148
TF_CALL_half(
REGISTER_CPU_KERNEL
);
149
TF_CALL_float(
REGISTER_CPU_KERNEL
);
150
TF_CALL_double(
REGISTER_CPU_KERNEL
);
151
#undef
REGISTER_CPU_KERNEL
conv_grad_ops_3d.cc
232
#define
REGISTER_CPU_KERNEL
(T) \
239
TF_CALL_half(
REGISTER_CPU_KERNEL
);
240
TF_CALL_float(
REGISTER_CPU_KERNEL
);
241
TF_CALL_double(
REGISTER_CPU_KERNEL
);
242
#undef
REGISTER_CPU_KERNEL
379
#define
REGISTER_CPU_KERNEL
(T) \
387
TF_CALL_half(
REGISTER_CPU_KERNEL
);
388
TF_CALL_float(
REGISTER_CPU_KERNEL
);
389
TF_CALL_double(
REGISTER_CPU_KERNEL
);
390
#undef
REGISTER_CPU_KERNEL
[
all
...]
constant_op.cc
217
#define
REGISTER_CPU_KERNEL
(TYPE) REGISTER_KERNEL(CPU, TYPE)
218
TF_CALL_ALL_TYPES(
REGISTER_CPU_KERNEL
);
223
#undef
REGISTER_CPU_KERNEL
avgpooling_op.cc
353
#define
REGISTER_CPU_KERNEL
(T) \
360
TF_CALL_float(
REGISTER_CPU_KERNEL
);
361
TF_CALL_double(
REGISTER_CPU_KERNEL
);
362
TF_CALL_half(
REGISTER_CPU_KERNEL
);
depthwise_conv_grad_op.cc
595
#define
REGISTER_CPU_KERNEL
(T) \
600
TF_CALL_float(
REGISTER_CPU_KERNEL
);
601
TF_CALL_double(
REGISTER_CPU_KERNEL
);
602
#undef
REGISTER_CPU_KERNEL
[
all
...]
quantized_resize_bilinear_op.cc
720
#define
REGISTER_CPU_KERNEL
(type) \
727
REGISTER_CPU_KERNEL
(::tensorflow::quint8);
728
REGISTER_CPU_KERNEL
(::tensorflow::qint32);
729
REGISTER_CPU_KERNEL
(float);
/external/tensorflow/tensorflow/core/kernels/neon/
neon_depthwise_conv_op.cc
195
#define
REGISTER_CPU_KERNEL
(T) \
202
TF_CALL_float(
REGISTER_CPU_KERNEL
);
Completed in 153 milliseconds