Searched defs:input_ptrs (Results 1 – 4 of 4) 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
140 std::vector<void*> input_ptrs; in Run() local