Home
last modified time | relevance | path

Searched refs:input_ptrs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc40 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_fixed_kernel() local
47 const T* input_ptr = input_ptrs[split]; in concat_fixed_kernel()
65 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_variable_kernel() local
106 const T* input_ptr = input_ptrs[curr_segment]; in concat_variable_kernel()
139 const GpuDeviceArrayStruct<const T*>& input_ptrs, in ConcatGPUImpl() argument
149 gpu_device.stream()>>>(input_ptrs, split_size, output->dimension(0), in ConcatGPUImpl()
162 gpu_device.stream()>>>(input_ptrs, output_scan, in ConcatGPUImpl()
168 gpu_device.stream()>>>(input_ptrs, output_scan, in ConcatGPUImpl()
191 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
198 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
Ddynamic_stitch_op_gpu.cu.cc35 GpuDeviceArrayStruct<const T*> input_ptrs, in DynamicStitchKernel() argument
38 const T** data_ptrs = GetGpuDeviceArrayOnDevice(&input_ptrs); in DynamicStitchKernel()
55 const GpuDeviceArrayStruct<const T*>& input_ptrs, in DynamicStitchGPUImpl() argument
63 input_ptrs, output)); in DynamicStitchGPUImpl()
71 const GpuDeviceArrayStruct<const T*>& input_ptrs, T* output);
Dconcat_lib_gpu.cc41 GpuDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size()); in ConcatGPUCall() local
42 OP_REQUIRES_OK(c, input_ptrs.Init()); in ConcatGPUCall()
44 input_ptrs.Set(i, inputs_flat[i]->data()); in ConcatGPUCall()
46 OP_REQUIRES_OK(c, input_ptrs.Finalize()); in ConcatGPUCall()
63 ConcatGPUImpl<T, IntType>(c->eigen_gpu_device(), input_ptrs.data(), in ConcatGPUCall()
Dconcat_lib_gpu.h41 const GpuDeviceArrayStruct<const T*>& input_ptrs,
60 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
65 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
Ddynamic_stitch_op.cc142 const GpuDeviceArrayStruct<const T*>& input_ptrs,
149 const GpuDeviceArrayStruct<const T*>& input_ptrs, T* output);