Searched refs:total_rows (Results 1 – 6 of 6) sorted by relevance
| /external/ltp/android/tools/ |
| D | compare_ltp_projects.py | 163 total_rows = max(len(deleted_test_suites), len(added_test_suites))
164 if total_rows > 0:
167 for i in range(total_rows):
173 total_rows = max(len(deleted_tests), len(added_tests))
174 if total_rows:
177 for i in range(total_rows):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_lib_gpu.cu.cc | 118 IntType total_rows, IntType total_cols, in split_v_kernel() argument
164 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) 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()
235 gpu_device.stream()>>>(input_ptr, output_scan, total_rows, in Run()
240 gpu_device.stream()>>>(input_ptr, output_scan, total_rows, in Run()
|
| D | concat_lib_gpu_impl.cu.cc | 39 int total_rows, int total_cols, T* output) { in concat_fixed_kernel() argument
50 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) { in concat_fixed_kernel()
63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows, in concat_variable_kernel() argument
109 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) in concat_variable_kernel()
|
| D | split_lib_gpu.h | 42 int total_cols, int total_rows,
|
| D | matrix_band_part_op.cc | 146 const int64 total_rows = b * m; in operator ()() local
188 thread_pool->ParallelFor(total_rows, row_cost, std::move(compute_shard)); in operator ()()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 851 total_rows = self._input_rows
861 total_rows = self._input_cols
989 normalization_factor = total_rows / math_ops.cast(num_rows, dtypes.float32)
1007 total_rows * total_cols, dtypes.float32)
|