HomeSort by relevance Sort by last modified time
    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