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

  /external/tensorflow/tensorflow/core/kernels/
slice_op_cpu_impl.h 29 #define DEFINE_CPU_KERNELS(T) \
32 TF_CALL_ALL_TYPES(DEFINE_CPU_KERNELS);
34 #undef DEFINE_CPU_KERNELS
split_lib_cpu.cc 40 #define DEFINE_CPU_KERNELS(T) template struct Split<Eigen::ThreadPoolDevice, T>;
42 TF_CALL_ALL_TYPES(DEFINE_CPU_KERNELS)
43 DEFINE_CPU_KERNELS(quint8)

Completed in 59 milliseconds