HomeSort by relevance Sort by last modified time
    Searched refs:TF_CALL_bfloat16 (Results 1 - 19 of 19) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
slice_op_gpu.cu.cc 42 TF_CALL_bfloat16(DEFINE_GPU_KERNELS);
tile_ops_cpu_impl.h 36 TF_CALL_bfloat16(DEFINE_TYPE);
59 TF_CALL_bfloat16(DEFINE_TYPE);
split_lib_gpu.h 56 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
tile_functor_cpu.cc 60 TF_CALL_bfloat16(DEFINE_TYPE);
82 TF_CALL_bfloat16(DEFINE_TYPE);
concat_lib.h 72 TF_CALL_bfloat16(REGISTER);
concat_lib_gpu.h 75 TF_CALL_bfloat16(REGISTER);
split_lib_gpu.cu.cc 59 TF_CALL_bfloat16(DEFINE_GPU_KERNELS);
69 TF_CALL_bfloat16(DEFINE_GPU_KERNELS);
250 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
259 TF_CALL_bfloat16(REGISTER_GPU_KERNEL);
concat_lib_gpu.cc 107 TF_CALL_bfloat16(REGISTER);
pack_op.cc 159 TF_CALL_bfloat16(REGISTER_GPU);
unpack_op.cc 146 TF_CALL_bfloat16(REGISTER_GPU);
tile_ops.cc 139 TF_CALL_bfloat16(HANDLE_TYPE_NAME);
218 TF_CALL_bfloat16(HANDLE_TYPE_NAME_CPU);
330 TF_CALL_bfloat16(HANDLE_TYPE_NAME);
slice_op.cc 294 TF_CALL_bfloat16(DECLARE_FOR_N);
315 TF_CALL_bfloat16(REGISTER_GPU);
check_numerics_op.cc 228 TF_CALL_bfloat16(REGISTER_CPU_KERNEL);
stateful_random_ops.cc 212 TF_CALL_bfloat16(REGISTER_CPU);
stateless_random_ops.cc 197 TF_CALL_bfloat16(REGISTER_CPU);
training_ops.cc 561 TF_CALL_bfloat16(REGISTER_CPU_KERNELS);
722 TF_CALL_bfloat16(REGISTER_CPU_KERNELS);
    [all...]
matmul_op.cc 590 TF_CALL_bfloat16(REGISTER_CPU);
618 TF_CALL_bfloat16(REGISTER_CPU);
random_op.cc 504 TF_CALL_bfloat16(REGISTER);
    [all...]
  /external/tensorflow/tensorflow/core/framework/
register_types.h 81 #define TF_CALL_bfloat16(m) m(::tensorflow::bfloat16)
112 #define TF_CALL_bfloat16(m)
142 #define TF_CALL_bfloat16(m)
165 TF_CALL_half(m) TF_CALL_bfloat16(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) \

Completed in 138 milliseconds