Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
H A Dsplit_lib_gpu.cu.cc113 GpuDeviceArrayStruct<IntType> output_scan, in split_v_kernel()
212 const GpuDeviceArrayStruct<IntType>& output_scan, in Run()
H A Dconcat_lib_gpu.cc48 GpuDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1); in ConcatGPUCall() local