Home
last modified time | relevance | path

Searched refs:grid_x (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc218 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
221 return int3(grid_x, grid_y, grid_z); in GetGridSize()
224 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
227 return int3(grid_x, grid_y, grid_z); in GetGridSize()
230 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
233 return int3(grid_x, grid_y, grid_z); in GetGridSize()
236 const int grid_x = dst_[0]->Batch(); in GetGridSize() local
239 return int3(grid_x, grid_y, grid_z); in GetGridSize()
/external/llvm-project/mlir/test/Dialect/GPU/
Doutlining.mlir26 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %gDimX, %grid_y = %gDimY,
68 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst,
99 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst,
123 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst,
151 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst,
178 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1,
204 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1,
223 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst,
Dmultiple-all-reduce.mlir10 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1, %grid_z = %c1)
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dresize.cc147 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
150 return int3(grid_x, grid_y, grid_z); in GetGridSize()
304 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
307 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dmean_stddev_normalization.cc211 const int grid_x = work_group_size_.x; in GetGridSize() local
214 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dconv_metal.cc624 int grid_x = DivideRoundUp(dst_shape.w, block_size.x); in GetGroupsCount() local
628 return DivideRoundUp(grid_x, wg_size.x) * DivideRoundUp(grid_y, wg_size.y) * in GetGroupsCount()
636 int grid_x = DivideRoundUp(dst_shape.w, block_size.x); in GetGroupsCountForLinearWH() local
640 return DivideRoundUp(grid_x * grid_y, wg_size.x) * in GetGroupsCountForLinearWH()
648 int grid_x = DivideRoundUp(dst_shape.w, block_size.x); in GetGroupsCountForLinearWHS() local
652 return DivideRoundUp(grid_x * grid_y * grid_z, wg_size.x); in GetGroupsCountForLinearWHS()
960 const int grid_x = in BindArguments() local
963 RETURN_IF_ERROR(args->SetInt("task_size_x", grid_x)); in BindArguments()
964 RETURN_IF_ERROR(args->SetInt("task_size_y", grid_x * grid_y)); in BindArguments()
969 int grid_x = in GetGridSize() local
[all …]
Dsplit.cc145 const int grid_x = width * batch; in GetGridSize() local
148 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dstrided_slice.cc170 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
173 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dconv_weights_converter.cc180 const int grid_x = in GetGridSize() local
184 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Ddepthwise_conv_3x3_stride_h2.cc191 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
194 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dconvolution_transposed_thin.cc151 const int grid_x = src_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
154 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dwinograd.cc461 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
464 return int3(grid_x, grid_y, grid_z); in GetGridSize()
675 const int grid_x = tiles_x * tiles_y * dst_[0]->Batch(); in GetGridSize() local
678 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dconvolution_transposed_3x3_thin.cc193 const int grid_x = src_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
196 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Ddepthwise_conv_3x3.cc285 const int grid_x = DivideRoundUp(dst_[0]->Width(), 2) * dst_[0]->Batch(); in GetGridSize() local
288 return int3(grid_x, grid_y, grid_z); in GetGridSize()
Dreduce.cc452 int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
456 grid_x *= work_group_size_.x; in GetGridSize()
460 return int3(grid_x, grid_y, grid_z); in GetGridSize()
/external/llvm-project/mlir/test/mlir-cuda-runner/
Dtwo-modules.mlir12 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %one, %grid_y = %one, %grid_z = %one)
18 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %one, %grid_y = %one, %grid_z = %one)
Dgpu-to-cubin.mlir7 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst, %grid_z = %cst)
Dall-reduce-region.mlir12 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %one, %grid_y = %one, %grid_z = %one)
Dshuffle.mlir12 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %one, %grid_y = %one, %grid_z = %one)
Dall-reduce-op.mlir15 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1, %grid_z = %c1)
/external/llvm-project/mlir/test/mlir-rocm-runner/
Dtwo-modules.mlir13 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1, %grid_z = %c1)
19 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1, %grid_z = %c1)
Dvector-transferops.mlir5 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst, %grid_z = %cst)
27 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst, %grid_z = %cst)
Dgpu-to-hsaco.mlir7 gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1, %grid_z = %c1)
/external/tensorflow/tensorflow/core/util/
Dgpu_launch_config.h247 int grid_x = std::min(DivUp(xdim, block_cols), max_blocks); in GetGpu2DLaunchConfig() local
250 grid_x, std::min(max_blocks / grid_x, std::max(ydim / block_rows, 1)), 1); in GetGpu2DLaunchConfig()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.h91 uint32 grid_x; member
104 " grid:", kernel_info.grid_x, ",", kernel_info.grid_y, ",", in ToXStat()

12