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

  /external/tensorflow/tensorflow/core/kernels/
cuda_device_array.h 31 // CudaDeviceArrayOnHost ptrs(context, size);
41 class CudaDeviceArrayOnHost {
43 CudaDeviceArrayOnHost(OpKernelContext* context, int32 size)
113 TF_DISALLOW_COPY_AND_ASSIGN(CudaDeviceArrayOnHost);
concat_lib_gpu.cc 55 CudaDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size());
62 CudaDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1);
bucketize_op_gpu.cu.cc 88 CudaDeviceArrayOnHost<float> boundaries_array(context,
dynamic_stitch_op.cc 176 CudaDeviceArrayOnHost<int32> indices_flat(c, first_dim_size);
177 CudaDeviceArrayOnHost<const T*> data_flat(c, data_elements_size);
split_v_op.cc 325 CudaDeviceArrayOnHost<T*> ptrs(context, num_split);
328 CudaDeviceArrayOnHost<Tlen> offsets(context, num_split + 1);
split_op.cc 264 CudaDeviceArrayOnHost<T*> ptrs(context, num_split);

Completed in 309 milliseconds