Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
193 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
194 const GpuDeviceArrayStruct<int32>& ptr_offsets, bool fixed_size, \
200 const GpuDeviceArrayStruct<const T*>& input_ptrs, \
201 const GpuDeviceArrayStruct<int64_t>& ptr_offsets, bool fixed_size, \
H A 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);
H A Dsparse_concat_op_gpu.cu.cc42 GpuDeviceArrayStruct<const int64*> ind_ptrs_data, in SparseConcatKernel()
43 GpuDeviceArrayStruct<const T*> val_ptrs_data, in SparseConcatKernel()
44 GpuDeviceArrayStruct<int64_t> nnz_scan_data, in SparseConcatKernel()
45 GpuDeviceArrayStruct<int64_t> concat_size_scan_data, in SparseConcatKernel()
46 GpuDeviceArrayStruct<int64_t> output_shape_data, in SparseConcatKernel()
85 int64 output_nnz, int rank, GpuDeviceArrayStruct<const T*> val_ptrs_data, in SparseConcatPermuteKernel()
86 GpuDeviceArrayStruct<int64_t> nnz_scan_data, in SparseConcatPermuteKernel()
87 GpuDeviceArrayStruct<int64_t> output_shape_data, in SparseConcatPermuteKernel()
205 : GpuDeviceArrayStruct<int64_t>()), in operator ()()
H A 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_t>& ptr_offsets, bool fixed_size, \
H A Dsplit_lib_gpu.cu.cc76 GpuDeviceArrayStruct<T*> output_ptr_data) { in SplitOpKernel()
113 GpuDeviceArrayStruct<IntType> output_scan, in split_v_kernel()
115 GpuDeviceArrayStruct<T*> output_ptr_data) { in split_v_kernel()
172 GpuDeviceArrayStruct<T*> output_ptr_data) { in SplitVOpKernel_fixed()
198 const GpuDeviceArrayStruct<T*>& output_ptr_data) { in Run()
212 const GpuDeviceArrayStruct<IntType>& output_scan, in Run()
213 const GpuDeviceArrayStruct<T*>& output_ptr_data) { in Run()
H A Dsparse_slice_op_gpu.cu.cc51 GpuDeviceArrayStruct<int64_t> input_start_data, in SparseSliceSelectFunctor()
52 GpuDeviceArrayStruct<int64_t> input_size_data, in SparseSliceSelectFunctor()
81 GpuDeviceArrayStruct<int64_t> input_start_data_;
82 GpuDeviceArrayStruct<int64_t> input_size_data_;
90 GpuDeviceArrayStruct<int64_t> input_start_data, in SparseSliceGatherKernel()
H A Dsplit_lib_gpu.h36 const GpuDeviceArrayStruct<T*>& output_ptr_data);
43 const GpuDeviceArrayStruct<IntType>& output_scan,
44 const GpuDeviceArrayStruct<T*>& output_ptr_data);
H A Ddynamic_stitch_op.cc142 const GpuDeviceArrayStruct<int>& input_indices,
143 const GpuDeviceArrayStruct<const T*>& input_ptrs,
149 const GpuDeviceArrayStruct<int32>& input_indices, \
150 const GpuDeviceArrayStruct<const T*>& input_ptrs, T* output);
H A Dsparse_split_op_gpu.cu.cc166 GpuDeviceArrayStruct<Index*> output_indices_data, in SparseSplitScatterKernel()
167 GpuDeviceArrayStruct<T*> output_values_data) { in SparseSplitScatterKernel()
195 GpuDeviceArrayStruct<Index*> output_indices_data, in LaunchSparseSplitScatterKernel()
196 GpuDeviceArrayStruct<T*> output_values_data) { in LaunchSparseSplitScatterKernel()
H A Dgpu_device_array_gpu.h29 struct GpuDeviceArrayStruct { struct
38 GpuDeviceArrayStruct<ValueType, MaxInlineValues>* data) { in GetGpuDeviceArrayOnDevice() argument
H A Dgpu_device_array.h100 const GpuDeviceArrayStruct<ValueType, MaxInlineValues>& data() const { in data()
112 GpuDeviceArrayStruct<ValueType, MaxInlineValues> data_;
H A Dbucketize_op_gpu.cu.cc38 GpuDeviceArrayStruct<float> boundaries_array, int32* __restrict__ out) { in BucketizeCustomKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dkernels_gpu.cu.cc195 GpuDeviceArrayStruct<int> batch_ptr_s, const int batch_size, in COOMatrixToSparseTensorKernel3D()
261 GpuDeviceArrayStruct<int> batch_ptr_s, const int batch_size, in CSRSparseMatrixBatchMulVecKernel3D()
384 GpuDeviceArrayStruct<int> cuda_ptr_s, int* local_ptr, int length) { in CopyFromGpuDeviceArrayToLocal()
396 const int size, const int rows, GpuDeviceArrayStruct<int> batch_ptr_s, in CSRSparseMatrixSoftmaxKernel3D()
566 GpuDeviceArrayStruct<int> softmax_and_grad_batch_ptr_s, in CSRSparseMatrixSoftmaxGradKernel3D()