Home
last modified time | relevance | path

Searched defs:dimx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper_test.cu.cc239 #define TEST_LAUNCH_PARAMETER(dimx, dimy) \ in TEST_F() argument
284 #define TEST_LAUNCH_PARAMETER(dimx, dimy, dimz) \ in TEST_F() argument
/external/tensorflow/tensorflow/core/kernels/
Dscan_ops_gpu.h245 int dimx = in.dimension(0); in LaunchScan() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4465 int dimx = reduction_dimensions.dimensions[kDimX]; in CalculateVirtualThreadScalingFactorForReduction() local