Home
last modified time | relevance | path

Searched refs:xdim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h202 inline Cuda2DLaunchConfig GetCuda2DLaunchConfig(int xdim, int ydim, in GetCuda2DLaunchConfig() argument
206 if (xdim <= 0 || ydim <= 0) { in GetCuda2DLaunchConfig()
211 int block_cols = std::min(xdim, kThreadsPerBlock); in GetCuda2DLaunchConfig()
220 config.virtual_thread_count = dim3(xdim, ydim, 1); in GetCuda2DLaunchConfig()
223 int grid_x = std::min(DivUp(xdim, block_cols), max_blocks); in GetCuda2DLaunchConfig()
237 int xdim, int ydim, int zdim, const Eigen::GpuDevice& d, DeviceFunc func, in GetCuda3DLaunchConfig() argument
241 if (xdim <= 0 || ydim <= 0 || zdim <= 0) { in GetCuda3DLaunchConfig()
263 int threadsx = std::min({xdim, thread_per_block, xthreadlimit}); in GetCuda3DLaunchConfig()
270 int blocksx = std::min({block_count, DivUp(xdim, threadsx), xgridlimit}); in GetCuda3DLaunchConfig()
276 config.virtual_thread_count = dim3(xdim, ydim, zdim); in GetCuda3DLaunchConfig()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder_test.cc104 constexpr int xdim = 3; in TEST_F() local
107 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
109 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()
147 constexpr int xdim = 3; in TEST_F() local
150 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
152 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()
Ddynamic_dimension_inference_test.cc244 constexpr int xdim = 3; in TEST_F() local
247 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
249 auto xz_shape = ShapeUtil::MakeShape(F32, {xdim, zdim}); in TEST_F()
288 constexpr int xdim = 3; in TEST_F() local
291 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
293 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()