Method Run

tensorflow/core/kernels/split_lib_gpu.cu.cc:197–208

template <typename T>
void SplitOpGPULaunch<T>::Run(const Eigen::GpuDevice& d, const T* input,
                              int32 prefix_dim_size, int32 split_dim_size,
                              int32 suffix_dim_size,
                              const GpuDeviceArrayStruct<T*>& output_ptr_data) {
  GpuLaunchConfig config =
      GetGpuLaunchConfig(prefix_dim_size * split_dim_size * suffix_dim_size, d);

  TF_CHECK_OK(GpuLaunchKernel(SplitOpKernel<T>, config.block_count,
                              config.thread_per_block, 0, d.stream(), input,
                              prefix_dim_size, split_dim_size, suffix_dim_size,
                              output_ptr_data));
}

template <typename T, typename IntType>
void SplitVOpGPULaunch<T, IntType>::Run(