Searched defs:block_dim (Results 1 – 13 of 13) sorted by relevance
73 def _compare_expand_tensor_with_kronecker_product(self, tensor, block_dim): argument103 def testExpandTensor(self, input_shape, block_dim): argument
136 def _generate_indices(num_rows, block_dim): argument
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
605 dim3 block_dim; in LaunchDepthwiseConv2dGPUSmall() local
73 const BlockDim &block_dim = block_dim_limit(); in ToMap() local
271 int block_dim = kNumCacheLevels; in BlockDimForCacheLevel() local
331 Status CudaLaunchKernel(void (*function)(Ts...), dim3 grid_dim, dim3 block_dim, in CudaLaunchKernel()
242 const int block_dim = 128; in LSTMBlockCellFpropWithCUDA() local
4837 auto block_dim = FormatTexelBlockExtent(image_state->createInfo.format); in ValidateBufferBounds() local