HomeSort by relevance Sort by last modified time
    Searched refs:total_rows (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
split_lib_gpu.h 42 int total_cols, int total_rows,
split_lib_gpu.cu.cc 118 IntType total_rows, IntType total_cols,
164 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y)
214 int total_rows, int total_cols,
219 GetCudaLaunchConfig(total_rows * total_cols, gpu_device);
223 input_ptr, total_rows, total_cols, output_ptr_data);
225 auto config = GetCuda2DLaunchConfig(total_cols, total_rows, gpu_device);
235 gpu_device.stream()>>>(input_ptr, output_scan, total_rows,
240 gpu_device.stream()>>>(input_ptr, output_scan, total_rows,
concat_lib_gpu_impl.cu.cc 39 int total_rows, int total_cols, T* output) {
50 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) {
63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows,
109 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y)
matrix_band_part_op.cc 146 const int64 total_rows = b * m; local
188 thread_pool->ParallelFor(total_rows, row_cost, std::move(compute_shard));
  /external/ltp/android/tools/
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/contrib/factorization/python/ops/
factorization_ops.py     [all...]

Completed in 680 milliseconds