Searched refs:zdim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder_test.cc | 106 constexpr int zdim = 1; in TEST_F() local
108 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
109 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()
149 constexpr int zdim = 1; in TEST_F() local
151 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
152 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 246 constexpr int zdim = 1; in TEST_F() local
248 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
249 auto xz_shape = ShapeUtil::MakeShape(F32, {xdim, zdim}); in TEST_F()
290 constexpr int zdim = 1; in TEST_F() local
292 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
293 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_launch_config.h | 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()
267 std::min({zdim, std::max(thread_per_block / (threadsx * threadsy), 1), in GetCuda3DLaunchConfig()
274 DivUp(zdim, threadsz), zgridlimit}); in GetCuda3DLaunchConfig()
276 config.virtual_thread_count = dim3(xdim, ydim, zdim); in GetCuda3DLaunchConfig()
|