Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows, in concat_variable_kernel()
140 const GpuDeviceArrayStruct<IntType>& output_scan, in ConcatGPUImpl()
Dconcat_lib_gpu.cc48 GpuDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1); in ConcatGPUCall() local
Dsplit_lib_gpu.cu.cc117 GpuDeviceArrayStruct<IntType> output_scan, in split_v_kernel()
215 const GpuDeviceArrayStruct<IntType>& output_scan, in Run()