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

  /external/tensorflow/tensorflow/core/util/
cuda_launch_config.h 108 inline int DivUp(int a, int b) { return (a + b - 1) / b; }
135 std::min(DivUp(physical_thread_count, thread_per_block),
164 std::min(block_count, DivUp(work_element_count, thread_per_block));
188 DivUp(work_element_count, fixed_block_size));
223 int grid_x = std::min(DivUp(xdim, block_cols), max_blocks);
270 int blocksx = std::min({block_count, DivUp(xdim, threadsx), xgridlimit});
272 {DivUp(block_count, blocksx), DivUp(ydim, threadsy), ygridlimit});
273 int blocksz = std::min({DivUp(block_count, (blocksx * blocksy)),
274 DivUp(zdim, threadsz), zgridlimit})
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
depthwise_conv_op_gpu.cu.cc 601 args.batch * DivUp(args.out_depth, kBlockDepth) * kBlockDepth;
610 DivUp(args.batch * args.out_depth, kBlockDepth) * kBlockDepth;
    [all...]

Completed in 391 milliseconds