Searched refs:RoundUpToNearest (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 38 static T RoundUpToNearest(T value, T divisor) { in RoundUpToNearest() function
56 redzone_size_(RoundUpToNearest( in RedzoneAllocator()
74 int64 rhs_slop = RoundUpToNearest(byte_size, kRhsRedzoneAlign) - byte_size; in AllocateBytes()
257 RoundUpToNearest<int64>(user_allocation_size, kRhsRedzoneAlign) - in CheckRedzonesForBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | launch_dimensions.cc | 85 RoundUpToNearest(threads_per_block / unroll_factor, int64{32}); in CalculateLaunchDimensions()
|
| D | cudnn_pad_for_convolutions.cc | 235 s->set_dimensions(dim, RoundUpToNearest<int64>(s->dimensions(dim), 8)); in TryResolvePaddedShapesForTensorCore()
311 s->set_dimensions(dim, RoundUpToNearest<int64>(s->dimensions(dim), 4)); in TryResolvePaddedShapesForIntegerConvolution()
|
| D | cublas_gemm_pad_for_tensor_cores.cc | 43 s.set_dimensions(dim, RoundUpToNearest<int64>(s.dimensions(dim), 8)); in PadForTensorCores()
|
| D | ir_emitter_unnested.cc | 2938 int64 rounded_bound = RoundUpToNearest(dimension_to_sort_bound, kTileSize); in EmitSortFromMlir()
5336 RoundUpToNearest(CeilOfRatio(reduction_dimensions.dimensions[2], in ComputeReductionCodegenInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.h | 417 T RoundUpToNearest(T value, T divisor) { in RoundUpToNearest() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 231 RoundUpToNearest<int64>(pre_split_size, 2); in CombineShapePair()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization_test.cc | 575 dim = RoundUpToNearest<int64>(dim, 64); in ShapeSizePadMinorTo64()
|
| D | heap_simulator.cc | 931 offset = std::max(offset, RoundUpToNearest(chunk.chunk_end(), alignment_)); in FindChunkCandidate()
|
| D | buffer_assignment.cc | 661 RoundUpToNearest(combined_allocation->size(), alignment); in CombineTempAllocations()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1088 RoundUpToNearest(original.base_shape().dimensions(i), in HaloExchangeToPadOnLeft()
|
| D | spmd_partitioner.cc | 1145 RoundUpToNearest(padded_shape.dimensions(target_dim), in ReshardWithAllToAll()
|