Home
last modified time | relevance | path

Searched defs:NUM_BLOCK_DIMS (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_functor.cc154 #define INSTANTIATE(NUM_BLOCK_DIMS, T) \ argument
Dspacetobatch_functor_gpu.cu.cc153 #define INSTANTIATE(NUM_BLOCK_DIMS, T) \ argument
Dspacetobatch_op.cc187 #define TF_SPACETOBATCH_BLOCK_DIMS_CASE(NUM_BLOCK_DIMS) \ in SpaceToBatchOpCompute() argument
Dbatchtospace_op.cc185 #define TF_BATCHTOSPACE_BLOCK_DIMS_CASE(NUM_BLOCK_DIMS) \ in BatchToSpaceOpCompute() argument