HomeSort by relevance Sort by last modified time
    Searched defs:num_kernels (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
pooling_ops_3d_gpu.cu.cc 143 int num_kernels = local
145 CudaLaunchConfig config = GetCudaLaunchConfig(num_kernels, d);
149 num_kernels, bottom_data, output_data, pooled_plane, pooled_height,
156 num_kernels, bottom_data, output_data, pooled_plane, pooled_height,
maxpooling_op_gpu.cu.cc 468 const int num_kernels = batch * channels * pooled_height * pooled_width; local
469 CudaLaunchConfig config = GetCudaLaunchConfig(num_kernels, d);
474 num_kernels, bottom_data, output_data, pooled_height, pooled_width,
480 num_kernels, bottom_data, output_data, pooled_height, pooled_width,
  /external/vboot_reference/cgpt/
cgpt_prioritize.c 102 int num_kernels; local
136 num_kernels = 0;
139 num_kernels++;
142 if (num_kernels) {
144 groups = NewGroupList(num_kernels);

Completed in 149 milliseconds