Home
last modified time | relevance | path

Searched refs:output_scan (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu.cc48 GpuDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1); in ConcatGPUCall() local
49 OP_REQUIRES_OK(c, output_scan.Init()); in ConcatGPUCall()
51 output_scan.Set(0, scan); in ConcatGPUCall()
59 output_scan.Set(i + 1, scan); in ConcatGPUCall()
61 if (!one_size_input) OP_REQUIRES_OK(c, output_scan.Finalize()); in ConcatGPUCall()
64 output_scan.data(), one_size_input, in ConcatGPUCall()
Dconcat_lib_gpu_impl.cu.cc63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows, in concat_variable_kernel() argument
66 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in concat_variable_kernel()
80 for (IntType i = lidx; i < output_scan.size; i += blockSize) { in concat_variable_kernel()
140 const GpuDeviceArrayStruct<IntType>& output_scan, in ConcatGPUImpl() argument
153 IntType smem_usage = output_scan.size * sizeof(IntType); in ConcatGPUImpl()
162 gpu_device.stream()>>>(input_ptrs, output_scan, in ConcatGPUImpl()
168 gpu_device.stream()>>>(input_ptrs, output_scan, in ConcatGPUImpl()
Dsplit_lib_gpu.cu.cc117 GpuDeviceArrayStruct<IntType> output_scan, in split_v_kernel() argument
121 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in split_v_kernel()
135 for (IntType i = lidx; i < output_scan.size; i += blockSize) { in split_v_kernel()
215 const GpuDeviceArrayStruct<IntType>& output_scan, in Run() argument
227 IntType smem_usage = output_scan.size * sizeof(IntType); 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()
Dsplit_lib_gpu.h43 const GpuDeviceArrayStruct<IntType>& output_scan,