Home
last modified time | relevance | path

Searched refs:TF_SPACETOBATCH_FOR_EACH_NUM_BLOCK_DIMS (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_functor.h42 #define TF_SPACETOBATCH_FOR_EACH_NUM_BLOCK_DIMS(MACRO, ...) \ macro
Dspacetobatch_functor_gpu.cu.cc159 TF_SPACETOBATCH_FOR_EACH_NUM_BLOCK_DIMS(INSTANTIATE, T)
Dspacetobatch_functor.cc160 TF_SPACETOBATCH_FOR_EACH_NUM_BLOCK_DIMS(INSTANTIATE, T)
Dbatchtospace_op.cc198 TF_SPACETOBATCH_FOR_EACH_NUM_BLOCK_DIMS(TF_BATCHTOSPACE_BLOCK_DIMS_CASE) in BatchToSpaceOpCompute()
Dspacetobatch_op.cc199 TF_SPACETOBATCH_FOR_EACH_NUM_BLOCK_DIMS(TF_SPACETOBATCH_BLOCK_DIMS_CASE) in SpaceToBatchOpCompute()