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

  /external/tensorflow/tensorflow/core/kernels/
tile_ops_gpu_impl.h 41 #define DEFINE_DIM(T, NDIM) \
48 DEFINE_DIM(int16, NDIM) \
49 DEFINE_DIM(int32, NDIM) \
50 DEFINE_DIM(int64, NDIM) \
51 DEFINE_DIM(Eigen::half, NDIM) \
52 DEFINE_DIM(float, NDIM) \
53 DEFINE_DIM(double, NDIM) \
54 DEFINE_DIM(complex64, NDIM) \
55 DEFINE_DIM(complex128, NDIM) \
tile_ops_cpu_impl.h 30 #define DEFINE_DIM(T, NDIM) \
33 #define DEFINE_TYPE(T) DEFINE_DIM(T, CPU_PROVIDED_IXDIM)
44 #undef DEFINE_DIM
51 #define DEFINE_DIM(T, NDIM) \
54 #define DEFINE_TYPE(T) DEFINE_DIM(T, CPU_PROVIDED_IXDIM)
64 #undef DEFINE_DIM

Completed in 122 milliseconds