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

  /external/tensorflow/tensorflow/core/kernels/
gpu_device_array.h 31 // GpuDeviceArrayOnHost ptrs(context, size);
41 class GpuDeviceArrayOnHost {
43 GpuDeviceArrayOnHost(OpKernelContext* context, int32 size)
113 TF_DISALLOW_COPY_AND_ASSIGN(GpuDeviceArrayOnHost);
concat_lib_gpu.cc 41 GpuDeviceArrayOnHost<const T*> input_ptrs(c, inputs_flat.size());
48 GpuDeviceArrayOnHost<IntType> output_scan(c, inputs_flat.size() + 1);
bucketize_op_gpu.cu.cc 88 GpuDeviceArrayOnHost<float> boundaries_array(context,
dynamic_stitch_op.cc 188 GpuDeviceArrayOnHost<int32> indices_flat(c, first_dim_size);
189 GpuDeviceArrayOnHost<const T*> data_flat(c, data_elements_size);
split_v_op.cc 369 GpuDeviceArrayOnHost<T*> ptrs(context, num_split);
372 GpuDeviceArrayOnHost<Tlen> offsets(context, num_split + 1);
split_op.cc 305 GpuDeviceArrayOnHost<T*> ptrs(context, num_split);

Completed in 145 milliseconds