Searched refs:total_cols (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_lib_gpu.cu.cc | 118 IntType total_rows, IntType total_cols, in split_v_kernel() argument
152 for (; gidx < total_cols; gidx += blockDim.x * gridDim.x) { in split_v_kernel()
166 input_ptr[gidy * total_cols + gidx]; in split_v_kernel()
214 int total_rows, int total_cols, in Run() argument
219 GetCudaLaunchConfig(total_rows * total_cols, gpu_device); in Run()
223 input_ptr, total_rows, total_cols, output_ptr_data); in Run()
225 auto config = GetCuda2DLaunchConfig(total_cols, total_rows, gpu_device); in Run()
236 total_cols, output_ptr_data); in Run()
241 total_cols, output_ptr_data); in Run()
|
| D | concat_lib_gpu_impl.cu.cc | 39 int total_rows, int total_cols, T* output) { in concat_fixed_kernel() argument
43 for (; gidx < total_cols; gidx += blockDim.x * gridDim.x) { in concat_fixed_kernel()
51 output[gidy * total_cols + gidx] = in concat_fixed_kernel()
64 IntType total_cols, T* output) { in concat_variable_kernel() argument
97 for (; gidx < total_cols; gidx += blockDim.x * gridDim.x) { in concat_variable_kernel()
110 output[gidy * total_cols + gidx] = in concat_variable_kernel()
|
| D | split_lib_gpu.h | 42 int total_cols, int total_rows,
|
| /external/u-boot/scripts/kconfig/ |
| D | nconf.gui.c | 559 int total_cols = 0; in show_scroll_win() local
574 total_cols = max(total_cols, len+2); in show_scroll_win()
578 pad = newpad(total_lines+10, total_cols+10); in show_scroll_win()
583 win_cols = min(total_cols+2, columns-2); in show_scroll_win()
657 if (start_x >= total_cols-text_cols) in show_scroll_win()
658 start_x = total_cols-text_cols; in show_scroll_win()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 852 total_cols = self._input_cols
862 total_cols = self._input_rows
1007 total_rows * total_cols, dtypes.float32)
|