Home
last modified time | relevance | path

Searched refs:grid_dim (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/utils/
Dkernel_stats_utils_test.cc92 EXPECT_EQ(kernel.grid_dim()[0], 3); in TEST()
93 EXPECT_EQ(kernel.grid_dim()[1], 2); in TEST()
94 EXPECT_EQ(kernel.grid_dim()[2], 1); in TEST()
103 EXPECT_EQ(kernel.grid_dim()[0], 3); in TEST()
104 EXPECT_EQ(kernel.grid_dim()[1], 2); in TEST()
105 EXPECT_EQ(kernel.grid_dim()[2], 1); in TEST()
120 EXPECT_EQ(kernel.grid_dim()[0], 4); in TEST()
121 EXPECT_EQ(kernel.grid_dim()[1], 5); in TEST()
122 EXPECT_EQ(kernel.grid_dim()[2], 6); in TEST()
Dkernel_stats_utils.cc171 lhs.grid_dim(0), in operator ()()
172 lhs.grid_dim(1), in operator ()()
173 lhs.grid_dim(2), in operator ()()
186 rhs.grid_dim(0), in operator ()()
187 rhs.grid_dim(1), in operator ()()
188 rhs.grid_dim(2), in operator ()()
214 lhs.grid_dim(0) == rhs.grid_dim(0) && in operator ()()
215 lhs.grid_dim(1) == rhs.grid_dim(1) && in operator ()()
216 lhs.grid_dim(2) == rhs.grid_dim(2) && in operator ()()
Dkernel_stats_utils.h78 wrap.key->grid_dim(0), in AbslHashValue()
79 wrap.key->grid_dim(1), in AbslHashValue()
80 wrap.key->grid_dim(2), in AbslHashValue()
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_kernel_stats_db_test.cc88 EXPECT_EQ(kernel.grid_dim().at(0), 1); in TEST()
89 EXPECT_EQ(kernel.grid_dim().at(1), 1); in TEST()
90 EXPECT_EQ(kernel.grid_dim().at(2), 1); in TEST()
106 EXPECT_EQ(kernel.grid_dim().at(0), 2); in TEST()
107 EXPECT_EQ(kernel.grid_dim().at(1), 1); in TEST()
108 EXPECT_EQ(kernel.grid_dim().at(2), 1); in TEST()
124 EXPECT_EQ(kernel.grid_dim().at(0), 3); in TEST()
125 EXPECT_EQ(kernel.grid_dim().at(1), 1); in TEST()
126 EXPECT_EQ(kernel.grid_dim().at(2), 1); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dreduction_gpu_kernels.cu.h769 dim3 grid_dim(1,
774 grid_dim.y = std::min((int)grid_dim.y, TF_RED_WARPSIZE);
776 if (grid_dim.y > 2 && grid_dim.y < TF_RED_WARPSIZE) {
777 int log2 = Log2Floor(grid_dim.y);
778 grid_dim.y = 1 << log2;
781 if (grid_dim.y == 1) {
783 grid_dim, block_dim, 0, cu_stream, in, out,
790 sizeof(T) * extent_y * grid_dim.y)}),
793 grid_dim, block_dim, 0, cu_stream, in,
798 (grid_dim.y * extent_y + (TF_RED_WARPSIZE - 1)) / TF_RED_WARPSIZE, 1,
[all …]
/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper.h100 Status GpuLaunchKernel(void (*function)(Ts...), dim3 grid_dim, dim3 block_dim, in GpuLaunchKernel() argument
110 auto result = cudaLaunchKernel(func_ptr, grid_dim, block_dim, arg_ptrs.data(), in GpuLaunchKernel()
116 hipLaunchKernelGGL(function, grid_dim, block_dim, shared_memory_size_bytes, in GpuLaunchKernel()
/external/llvm-project/mlir/test/Conversion/GPUToSPIRV/
Dload-store.mlir52 %6 = "gpu.grid_dim"() {dimension = "x"} : () -> index
53 %7 = "gpu.grid_dim"() {dimension = "y"} : () -> index
54 %8 = "gpu.grid_dim"() {dimension = "z"} : () -> index
Dbuiltins.mlir184 %0 = "gpu.grid_dim"() {dimension = "x"} : () -> index
/external/llvm-project/mlir/test/Dialect/Linalg/
Dtile-and-distribute.mlir74 // CHECK: %[[NBLOCKSY:.*]] = "gpu.grid_dim"() {dimension = "y"}
76 // CHECK: %[[NBLOCKSX:.*]] = "gpu.grid_dim"() {dimension = "x"}
133 // CHECK: %[[NBLOCKSX:.*]] = "gpu.grid_dim"() {dimension = "x"}
163 // CHECK: %[[NBLOCKSY:.*]] = "gpu.grid_dim"() {dimension = "y"}
187 // CHECK: %[[NBLOCKSY:.*]] = "gpu.grid_dim"() {dimension = "y"}
189 // CHECK: %[[NBLOCKSX:.*]] = "gpu.grid_dim"() {dimension = "x"}
/external/llvm-project/mlir/test/Dialect/GPU/
Dops.mlir43 %gDimX = "gpu.grid_dim"() {dimension = "x"} : () -> (index)
44 %gDimY = "gpu.grid_dim"() {dimension = "y"} : () -> (index)
45 %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index)
Doutlining.mlir48 // CHECK-NEXT: = "gpu.grid_dim"() {dimension = "x"} : () -> index
49 // CHECK-NEXT: = "gpu.grid_dim"() {dimension = "y"} : () -> index
50 // CHECK-NEXT: = "gpu.grid_dim"() {dimension = "z"} : () -> index
Dinvalid.mlir226 %gDimX = "gpu.grid_dim"() {dimension = "o"} : () -> (index)
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dkernel_stats.proto18 repeated uint32 grid_dim = 6; field
/external/llvm-project/mlir/test/Conversion/GPUToROCDL/
Dgpu-to-rocdl.mlir44 %gDimX = "gpu.grid_dim"() {dimension = "x"} : () -> (index)
47 %gDimY = "gpu.grid_dim"() {dimension = "y"} : () -> (index)
50 %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index)
/external/llvm-project/polly/lib/External/ppcg/
Dgpu.h392 int grid_dim[2]; member
Dgpu.c653 kernel->grid_dim[0] = 32768; in read_grid_sizes()
656 kernel->grid_dim[0] = 256; in read_grid_sizes()
657 kernel->grid_dim[1] = 256; in read_grid_sizes()
662 read_sizes_from_set(size, kernel->grid_dim, &kernel->n_grid); in read_grid_sizes()
678 kernel->grid_dim, kernel->n_grid); in read_grid_and_block_sizes()
3900 kernel->grid_dim); in gpu_create_kernel()
3904 node = snap_band_to_sizes(node, kernel->grid_dim, in gpu_create_kernel()
/external/llvm-project/mlir/test/Conversion/GPUToNVVM/
Dgpu-to-nvvm.mlir44 %gDimX = "gpu.grid_dim"() {dimension = "x"} : () -> (index)
47 %gDimY = "gpu.grid_dim"() {dimension = "y"} : () -> (index)
50 %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index)
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_screen.c479 uint32_t *grid_dim = ret; in llvmpipe_get_compute_param() local
480 *grid_dim = 3; in llvmpipe_get_compute_param()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Dcustom_call.md78 const int64 grid_dim = 2048 / block_dim;
79 custom_call_kernel<<<grid_dim, block_dim,
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops_gpu.cu.cc248 const int grid_dim = in LSTMBlockCellFpropWithCUDA() local
250 TF_CHECK_OK(GpuLaunchKernel(concat_xh<T>, grid_dim, block_dim, 0, cu_stream, in LSTMBlockCellFpropWithCUDA()
/external/llvm-project/mlir/include/mlir/Dialect/GPU/
DGPUOps.td58 def GPU_GridDimOp : GPU_IndexOp<"grid_dim"> {
66 %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index)
355 %gDimX = "gpu.grid_dim"() {dimension = "x"} : () -> (index)
356 %gDimY = "gpu.grid_dim"() {dimension = "y"} : () -> (index)
357 %gDimZ = "gpu.grid_dim"() {dimension = "z"} : () -> (index)