Home
last modified time | relevance | path

Searched refs:CeilOfRatio (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc123 tensorflow::MathUtil::CeilOfRatio(input_dimension, window_stride); in MakePadding()
130 tensorflow::MathUtil::CeilOfRatio(padding_size, 2ll)); in MakePadding()
Dsharding_builder.cc58 tile_dimension = CeilOfRatio(static_cast<int64>(tile_dimension), num_tiles); in Tile1D()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h353 CeilOfRatio(limits[0] - starts[0], strides[0]), in Slice2D()
354 CeilOfRatio(limits[1] - starts[1], strides[1])); in Slice2D()
379 CeilOfRatio(limits[0] - starts[0], strides[0]), in Slice3D()
380 CeilOfRatio(limits[1] - starts[1], strides[1]), in Slice3D()
381 CeilOfRatio(limits[2] - starts[2], strides[2])); in Slice3D()
413 CeilOfRatio(limits[0] - starts[0], strides[0]), in Slice4D()
414 CeilOfRatio(limits[1] - starts[1], strides[1]), in Slice4D()
415 CeilOfRatio(limits[2] - starts[2], strides[2]), in Slice4D()
416 CeilOfRatio(limits[3] - starts[3], strides[3])); in Slice4D()
Dutil.h402 T CeilOfRatio(T dividend, T divisor) { in CeilOfRatio() function
403 return tensorflow::MathUtil::CeilOfRatio<T>(dividend, divisor); in CeilOfRatio()
414 return CeilOfRatio<T>(dividend, divisor); in ElementWiseCeilOfRatio()
423 return CeilOfRatio(value, divisor) * divisor; in RoundUpToNearest()
Dreference_util.cc155 return tensorflow::MathUtil::CeilOfRatio(unpadded_width, stride); in WindowCount()
/external/tensorflow/tensorflow/core/lib/math/
Dmath_util_test.cc50 IntegralType ceil_1 = MathUtil::CeilOfRatio(numerator, denominator); in TestCeilOfRatio()
172 const int64 correct_result = MathUtil::CeilOfRatio(numerator, denominator); in TestThatCeilOfRatioDenomMinusOneIsIncorrect()
189 TEST(MathUtil, CeilOfRatio) { in TEST() argument
Dmath_util.h51 static IntegralType CeilOfRatio(IntegralType numerator, in CeilOfRatio() function
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc147 return port::MathUtil::CeilOfRatio(x, y); in DivideCeil()
154 *block_count = port::MathUtil::CeilOfRatio(element_count, *threads_per_block); in CalculateDimensionality()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dquantize_test.cc58 int64 padded_output_width = CeilOfRatio(width, size_per_pack); in PackLargeInput()
85 int64 padded_output_width = CeilOfRatio(width, size_per_pack) * size_per_pack; in GenerateLargeSizeMinCombinedOutput()
152 CeilOfRatio(output_size, pack_size) * pack_size - output_size; in GenerateMinCombinedOutput()
Dquantize.h53 const int64 output_size = CeilOfRatio(input_size, kElementsPerPack); in PackToUint32()
Dprng.cc118 const int64 half_size = CeilOfRatio<int64>(size, 2); in StatelessRngUniformU32()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpartition_assignment.cc91 int64 block_count = CeilOfRatio(num_elements, threads_per_block); in CalculateLaunchDimensions()
Dir_emission_utils.cc224 int num_segments = CeilOfRatio(bit_width, 32); in EmitFullWarpShuffleDown()
Dir_emitter_unnested.cc1315 uint64 num_blocks = CeilOfRatio(num_iterations, kThreadsPerBlock); in HandleSort()
/external/tensorflow/tensorflow/stream_executor/lib/
Dmathutil.h34 static IntegralType CeilOfRatio(IntegralType numerator, in CeilOfRatio() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc176 index[i] * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), shape_dim); in TileOffsetForDevice()
195 (index[i] + 1) * CeilOfRatio(shape_dim, tile_assignment_.dim(i)), in TileLimitForDevice()
441 i, CeilOfRatio<int64>(shape.dimensions(i), tile_assignment_.dim(i))); in TileShape()
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d_gpu.h875 MathUtil::CeilOfRatio<int>(input_dims[1], requested_tile_size_i),
876 MathUtil::CeilOfRatio<int>(input_dims[2], requested_tile_size_j),
915 MathUtil::CeilOfRatio<int>(input_dims[1], kTileSize),
916 MathUtil::CeilOfRatio<int>(input_dims[2], kTileSize),
Dclustering_ops.cc67 int64 CeilOfRatio(int64 a, int64 b) { return (a + b - 1) / b; } in CeilOfRatio() function
390 NextMultiple(num_threads, CeilOfRatio(num_points, rows_per_block)); in Compute()
/external/tensorflow/tensorflow/contrib/cloud/kernels/
Dbigquery_reader_ops.cc149 const int64 partition_size = tensorflow::MathUtil::CeilOfRatio<int64>( in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateful_random_ops.cc55 const int64 half_size = xla::CeilOfRatio<int64>(size, 2); in StatefulRngUniformU32()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2614 int64 total_size = port::MathUtil::CeilOfRatio(batch, int64{16}) * in ShouldIncludeWinogradNonfusedAlgo()