OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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