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

  /external/tensorflow/tensorflow/core/kernels/
cuda_device_array_gpu.h 29 struct CudaDeviceArrayStruct {
38 CudaDeviceArrayStruct<ValueType, MaxInlineValues>* data) {
dynamic_stitch_op_gpu.cu.cc 34 CudaDeviceArrayStruct<int32> input_indices,
35 CudaDeviceArrayStruct<const T*> input_ptrs,
54 const CudaDeviceArrayStruct<int>& input_indices,
55 const CudaDeviceArrayStruct<const T*>& input_ptrs,
69 const CudaDeviceArrayStruct<int32>& input_indices, \
70 const CudaDeviceArrayStruct<const T*>& input_ptrs, T* output);
concat_lib_gpu_impl.cu.cc 37 CudaDeviceArrayStruct<const T*> input_ptr_data, int split_size,
61 CudaDeviceArrayStruct<const T*> input_ptr_data,
62 CudaDeviceArrayStruct<IntType> output_scan, IntType total_rows,
138 const CudaDeviceArrayStruct<const T*>& input_ptrs,
139 const CudaDeviceArrayStruct<IntType>& output_scan,
190 const CudaDeviceArrayStruct<const T*>& input_ptrs, \
191 const CudaDeviceArrayStruct<int32>& ptr_offsets, bool fixed_size, \
197 const CudaDeviceArrayStruct<const T*>& input_ptrs, \
198 const CudaDeviceArrayStruct<int64>& ptr_offsets, bool fixed_size, \
cuda_device_array.h 27 // CudaDeviceArrayStruct.
96 const CudaDeviceArrayStruct<ValueType, MaxInlineValues>& data() const {
108 CudaDeviceArrayStruct<ValueType, MaxInlineValues> data_;
split_lib_gpu.cu.cc 74 CudaDeviceArrayStruct<T*> output_ptr_data) {
111 CudaDeviceArrayStruct<IntType> output_scan,
113 CudaDeviceArrayStruct<T*> output_ptr_data) {
169 CudaDeviceArrayStruct<T*> output_ptr_data) {
195 const CudaDeviceArrayStruct<T*>& output_ptr_data) {
210 const CudaDeviceArrayStruct<IntType>& output_scan,
211 const CudaDeviceArrayStruct<T*>& output_ptr_data) {
concat_lib_gpu.cc 42 const CudaDeviceArrayStruct<const T*>& input_ptrs,
43 const CudaDeviceArrayStruct<IntType>& ptr_offsets,
bucketize_op_gpu.cu.cc 39 CudaDeviceArrayStruct<float> boundaries_array, int32* out) {
dynamic_stitch_op.cc 141 const CudaDeviceArrayStruct<int>& input_indices,
142 const CudaDeviceArrayStruct<const T*>& input_ptrs,
split_v_op.cc 285 const CudaDeviceArrayStruct<IntType>& output_scan,
286 const CudaDeviceArrayStruct<T*>& output_ptr_data);
split_op.cc 227 const CudaDeviceArrayStruct<T*>& output_ptr_data);

Completed in 689 milliseconds