Searched defs:block_dim (Results 1 – 14 of 14) sorted by relevance
98 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in BatchToSpaceOpCompute() local140 for (int block_dim = 0; block_dim < removed_prefix_block_dims; ++block_dim) { in BatchToSpaceOpCompute() local148 for (int block_dim = removed_prefix_block_dims; in BatchToSpaceOpCompute() local
101 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in SpaceToBatchOpCompute() local138 for (int block_dim = 0; block_dim < removed_prefix_block_dims; ++block_dim) { in SpaceToBatchOpCompute() local146 for (int block_dim = removed_prefix_block_dims; in SpaceToBatchOpCompute() local
107 for (int block_dim = 0; block_dim < NUM_BLOCK_DIMS; ++block_dim) { in operator ()() local135 for (int block_dim = NUM_BLOCK_DIMS - 1; block_dim >= 0; --block_dim) { in operator ()() local
67 for (int block_dim = NUM_BLOCK_DIMS - 1; block_dim >= 0; --block_dim) { in S2B() local117 for (int block_dim = 0; block_dim < NUM_BLOCK_DIMS; ++block_dim) { in operator ()() local
32 for (int block_dim = 0; block_dim < num_block_dims; ++block_dim) { in ConstructSpaceToBatchGraph() local
16 repeated uint32 block_dim = 5; field
62 int block_dim = 0; in FormatConverter() local242 int block_dim = 0; in FormatConverter() local
72 const BlockDim &block_dim = block_dim_limit(); in ToMap() local
270 int block_dim = kNumCacheLevels; in BlockDimForCacheLevel() local
102 Status GpuLaunchKernel(void (*function)(Ts...), dim3 grid_dim, dim3 block_dim, in GpuLaunchKernel()
153 int64_t block_dim = (left_side) ? bdims - 2 : bdims - 1; in SolveWithInvertedDiagonalBlocks() local
247 const int block_dim = 128; in LSTMBlockCellFpropWithCUDA() local
686 dim3 block_dim, thread_block; in NmsGpu() local
4934 auto block_dim = FormatTexelBlockExtent(image_state->createInfo.format); in ValidateBufferBounds() local