Searched refs:split_size (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu_impl.cu.cc37 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 Dsparse_tensor.h252 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 121 milliseconds