Searched defs:input_ptrs (Results 1 – 3 of 3) sorted by relevance
35 GpuDeviceArrayStruct<const T*> input_ptrs, in DynamicStitchKernel()55 const GpuDeviceArrayStruct<const T*>& input_ptrs, in DynamicStitchGPUImpl()
40 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_fixed_kernel() local65 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_variable_kernel() local139 const GpuDeviceArrayStruct<const T*>& input_ptrs, in ConcatGPUImpl()
41 GpuDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size()); in ConcatGPUCall() local