Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc39 int total_rows, int total_cols, T* __restrict__ output) { in concat_fixed_kernel()
63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows, in concat_variable_kernel()
Dsplit_lib_gpu.cu.cc114 IntType total_rows, IntType total_cols, in split_v_kernel()
211 int total_rows, int total_cols, in Run()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_band_part_op.cc147 const int64 total_rows = b * m; in operator ()() local