Home
last modified time | relevance | path

Searched refs:ydim (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()
220 config.virtual_thread_count = dim3(xdim, ydim, 1); in GetCuda2DLaunchConfig()
226 grid_x, std::min(max_blocks / grid_x, std::max(ydim / block_rows, 1)), 1); 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()
265 std::min({ydim, std::max(thread_per_block / threadsx, 1), ythreadlimit}); in GetCuda3DLaunchConfig()
272 {DivUp(block_count, blocksx), DivUp(ydim, threadsy), ygridlimit}); in GetCuda3DLaunchConfig()
276 config.virtual_thread_count = dim3(xdim, ydim, zdim); in GetCuda3DLaunchConfig()
284 int xdim, int ydim, const Eigen::GpuDevice& d, DeviceFunc func, in GetCuda2DLaunchConfig() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder_test.cc105 constexpr int ydim = 2; in TEST_F() local
107 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
108 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
148 constexpr int ydim = 2; in TEST_F() local
150 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
151 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
Ddynamic_dimension_inference_test.cc245 constexpr int ydim = 2; in TEST_F() local
247 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
248 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
289 constexpr int ydim = 2; in TEST_F() local
291 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
292 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()