Home
last modified time | relevance | path

Searched refs:GpuDeviceArrayStruct (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op_gpu.cu.cc34 GpuDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel()
35 GpuDeviceArrayStruct<const T*> input_ptrs, in DynamicStitchKernel()
54 const GpuDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl()
55 const GpuDeviceArrayStruct<const T*>& input_ptrs, in DynamicStitchGPUImpl()
70 const GpuDeviceArrayStruct<int32>& input_indices, \
71 const GpuDeviceArrayStruct<const T*>& input_ptrs, T* output);
Dconcat_lib_gpu.h41 const GpuDeviceArrayStruct<const T*>& input_ptrs,
42 const GpuDeviceArrayStruct<IntType>& ptr_offsets,
60 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
61 const GpuDeviceArrayStruct<int32>& ptr_offsets, bool fixed_size, \
65 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
66 const GpuDeviceArrayStruct<int64>& ptr_offsets, bool fixed_size, \
Dconcat_lib_gpu_impl.cu.cc38 GpuDeviceArrayStruct<const T*> input_ptr_data, int split_size, in concat_fixed_kernel()
62 GpuDeviceArrayStruct<const T*> input_ptr_data, in concat_variable_kernel()
63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows, in concat_variable_kernel()
139 const GpuDeviceArrayStruct<const T*>& input_ptrs, in ConcatGPUImpl()
140 const GpuDeviceArrayStruct<IntType>& output_scan, in ConcatGPUImpl()
191 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
192 const GpuDeviceArrayStruct<int32>& ptr_offsets, bool fixed_size, \
198 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
199 const GpuDeviceArrayStruct<int64>& ptr_offsets, bool fixed_size, \
Dsplit_lib_gpu.cu.cc80 GpuDeviceArrayStruct<T*> output_ptr_data) { in SplitOpKernel()
117 GpuDeviceArrayStruct<IntType> output_scan, in split_v_kernel()
119 GpuDeviceArrayStruct<T*> output_ptr_data) { in split_v_kernel()
175 GpuDeviceArrayStruct<T*> output_ptr_data) { in SplitVOpKernel_fixed()
201 const GpuDeviceArrayStruct<T*>& output_ptr_data) { in Run()
215 const GpuDeviceArrayStruct<IntType>& output_scan, in Run()
216 const GpuDeviceArrayStruct<T*>& output_ptr_data) { in Run()
Dsplit_lib_gpu.h36 const GpuDeviceArrayStruct<T*>& output_ptr_data);
43 const GpuDeviceArrayStruct<IntType>& output_scan,
44 const GpuDeviceArrayStruct<T*>& output_ptr_data);
Dgpu_device_array_gpu.h29 struct GpuDeviceArrayStruct { struct
38 GpuDeviceArrayStruct<ValueType, MaxInlineValues>* data) { in GetGpuDeviceArrayOnDevice() argument
Dgpu_device_array.h96 const GpuDeviceArrayStruct<ValueType, MaxInlineValues>& data() const { in data()
108 GpuDeviceArrayStruct<ValueType, MaxInlineValues> data_;
Ddynamic_stitch_op.cc141 const GpuDeviceArrayStruct<int>& input_indices,
142 const GpuDeviceArrayStruct<const T*>& input_ptrs,
148 const GpuDeviceArrayStruct<int32>& input_indices, \
149 const GpuDeviceArrayStruct<const T*>& input_ptrs, T* output);
Dbucketize_op_gpu.cu.cc39 GpuDeviceArrayStruct<float> boundaries_array, int32* out) { in BucketizeCustomKernel()