Searched defs:split_size (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | concat_lib_gpu_impl.cu.cc | 37 CudaDeviceArrayStruct<const T*> input_ptr_data, int split_size, 45 IntType split = gidx / split_size; 47 IntType col_offset = gidx % split_size; 51 input_ptr[gidy * split_size + col_offset]; 140 bool fixed_size, int split_size, 148 gpu_device.stream()>>>(input_ptrs, split_size, output->dimension(0), 192 int split_size, typename TTypes<T, 2>::Matrix* output); 199 int split_size, typename TTypes<T, 2>::Matrix* output); 36 concat_fixed_kernel( CudaDeviceArrayStruct<const T*> input_ptr_data, int split_size, int total_rows, int total_cols, T* output) argument 137 ConcatGPUImpl(const Eigen::GpuDevice& gpu_device, const CudaDeviceArrayStruct<const T*>& input_ptrs, const CudaDeviceArrayStruct<IntType>& output_scan, bool fixed_size, int split_size, typename TTypes<T, 2>::Matrix* output) argument
|
/external/tensorflow/tensorflow/core/util/sparse/ |
H A D | sparse_tensor.h | 252 static inline int GetSliceIndex(const int dim, const int split_size, argument 254 CHECK_GT(split_size, 0); 256 if (residual == 0) return dim / split_size; 257 const int offset = residual * (split_size + 1); 259 return dim / (split_size + 1); 261 return residual + ((dim - offset) / split_size); 266 static inline int GetDimensionInSlice(const int dim, const int split_size, argument 268 CHECK_GT(split_size, 0); 270 if (residual == 0) return dim % split_size; 271 const int offset = residual * (split_size 280 GetSliceShape(const int slice_index, const int split_size, const int residual) argument 515 const int split_size = split_dim_size / num_split; local [all...] |
Completed in 546 milliseconds