HomeSort by relevance Sort by last modified time
    Searched refs:GpuDeviceArrayStruct (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
gpu_device_array_gpu.h 29 struct GpuDeviceArrayStruct {
38 GpuDeviceArrayStruct<ValueType, MaxInlineValues>* data) {
dynamic_stitch_op_gpu.cu.cc 34 GpuDeviceArrayStruct<int32> input_indices,
35 GpuDeviceArrayStruct<const T*> input_ptrs,
54 const GpuDeviceArrayStruct<int>& input_indices,
55 const GpuDeviceArrayStruct<const T*>& input_ptrs,
70 const GpuDeviceArrayStruct<int32>& input_indices, \
71 const GpuDeviceArrayStruct<const T*>& input_ptrs, T* output);
split_lib_gpu.h 36 const GpuDeviceArrayStruct<T*>& output_ptr_data);
43 const GpuDeviceArrayStruct<IntType>& output_scan,
44 const GpuDeviceArrayStruct<T*>& output_ptr_data);
concat_lib_gpu.h 41 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, \
concat_lib_gpu_impl.cu.cc 38 GpuDeviceArrayStruct<const T*> input_ptr_data, int split_size,
62 GpuDeviceArrayStruct<const T*> input_ptr_data,
63 GpuDeviceArrayStruct<IntType> output_scan, IntType total_rows,
139 const GpuDeviceArrayStruct<const T*>& input_ptrs,
140 const GpuDeviceArrayStruct<IntType>& output_scan,
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, \
gpu_device_array.h 27 // GpuDeviceArrayStruct.
96 const GpuDeviceArrayStruct<ValueType, MaxInlineValues>& data() const {
108 GpuDeviceArrayStruct<ValueType, MaxInlineValues> data_;
split_lib_gpu.cu.cc 80 GpuDeviceArrayStruct<T*> output_ptr_data) {
117 GpuDeviceArrayStruct<IntType> output_scan,
119 GpuDeviceArrayStruct<T*> output_ptr_data) {
175 GpuDeviceArrayStruct<T*> output_ptr_data) {
201 const GpuDeviceArrayStruct<T*>& output_ptr_data) {
215 const GpuDeviceArrayStruct<IntType>& output_scan,
216 const GpuDeviceArrayStruct<T*>& output_ptr_data) {
bucketize_op_gpu.cu.cc 39 GpuDeviceArrayStruct<float> boundaries_array, int32* out) {
dynamic_stitch_op.cc 141 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);

Completed in 294 milliseconds