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

1 2

  /external/tensorflow/tensorflow/core/kernels/
batch_matmul_op_real.cc 33 TF_CALL_half(REGISTER_BATCH_MATMUL_CPU);
40 TF_CALL_half(REGISTER_BATCH_MATMUL_GPU);
reverse_op_gpu.cu.cc 44 TF_CALL_half(DEFINE_REVERSE_ALL_DIMS);
batch_norm_op.cc 173 TF_CALL_half(REGISTER_KERNEL);
193 TF_CALL_half(DECLARE_GPU_SPECS);
205 TF_CALL_half(REGISTER_GPU_KERNEL);
229 TF_CALL_half(REGISTER_KERNEL);
252 TF_CALL_half(DECLARE_GPU_SPECS);
264 TF_CALL_half(REGISTER_GPU_KERNEL);
softmax_op.cc 85 TF_CALL_half(REGISTER_CPU);
94 TF_CALL_half(REGISTER_CPU);
tile_ops_cpu_impl.h 41 TF_CALL_half(DEFINE_TYPE);
conditional_accumulator_op.cc 55 TF_CALL_half(REGISTER_KERNELS_CPU);
sparse_conditional_accumulator_op.cc 55 TF_CALL_half(REGISTER_KERNELS_CPU);
tile_functor_cpu.cc 66 TF_CALL_half(DEFINE_TYPE);
tile_ops.cc 146 TF_CALL_half(HANDLE_TYPE_NAME);
224 TF_CALL_half(HANDLE_TYPE_NAME_CPU);
236 TF_CALL_half(HANDLE_TYPE_NAME_GPU);
329 TF_CALL_half(HANDLE_TYPE_NAME);
491 TF_CALL_half(HANDLE_TYPE_NAME_CPU);
501 TF_CALL_half(HANDLE_TYPE_NAME_GPU);
580 TF_CALL_half(REGISTER_GPU);
multinomial_op.cc 251 TF_CALL_half(REGISTER);
271 TF_CALL_half(REGISTER);
319 TF_CALL_half(REGISTER);
341 TF_CALL_half(REGISTER);
stateful_random_ops.cc 211 TF_CALL_half(REGISTER_CPU);
218 TF_CALL_half(REGISTER_GPU);
stateless_random_ops.cc 196 TF_CALL_half(REGISTER_CPU);
205 TF_CALL_half(REGISTER_GPU);
depthwise_conv_op.cc 458 TF_CALL_half(REGISTER_CPU_KERNEL);
471 TF_CALL_half(REGISTER_GPU_KERNEL);
493 TF_CALL_half(REGISTER_GROUPED_CONV_KERNEL);
xent_op.cc 133 TF_CALL_half(REGISTER_CPU);
depthwise_conv_grad_op.cc     [all...]
parameterized_truncated_normal_op.cc 447 TF_CALL_half(REGISTER);
462 TF_CALL_half(REGISTER);
check_numerics_op.cc 227 TF_CALL_half(REGISTER_CPU_KERNEL);
draw_bounding_box_op.cc 183 TF_CALL_half(REGISTER_CPU_KERNEL);
random_poisson_op.cc 327 TF_CALL_half(REGISTER);
cudnn_rnn_ops.cc     [all...]
lrn_op.cc 291 TF_CALL_half(REGISTER_CPU);
514 TF_CALL_half(REGISTER_CPU);
  /external/tensorflow/tensorflow/core/framework/
register_types.h 87 #define TF_CALL_half(m) m(Eigen::half)
118 #define TF_CALL_half(m) m(Eigen::half)
148 #define TF_CALL_half(m)
165 TF_CALL_half(m) TF_CALL_bfloat16(m) TF_CALL_float(m) TF_CALL_double(m)
172 TF_CALL_INTEGRAL_TYPES(m) TF_CALL_half(m) TF_CALL_float(m) TF_CALL_double(m)
175 TF_CALL_half(m) TF_CALL_bfloat16(m) TF_CALL_float(m) TF_CALL_double(m) \
198 TF_CALL_half(m) TF_CALL_float(m) TF_CALL_double(m)
215 TF_CALL_half(m) TF_CALL_float(m) TF_CALL_double(m) TF_CALL_complex64(m) \
register_types_traits.h 102 TF_CALL_double(m) TF_CALL_float(m) TF_CALL_half(m) TF_CALL_int32(m) \
  /external/tensorflow/tensorflow/contrib/image/kernels/
image_ops.cc 135 TF_CALL_half(REGISTER);
159 TF_CALL_half(DECLARE_FUNCTOR);
179 TF_CALL_half(REGISTER);
  /external/tensorflow/tensorflow/contrib/resampler/kernels/
resampler_ops.cc 187 TF_CALL_half(REGISTER);
398 TF_CALL_half(REGISTER);
409 // TF_CALL_half(REGISTER);

Completed in 943 milliseconds

1 2