/external/tensorflow/tensorflow/compiler/xla/client/ |
D | padding.cc | 123 tensorflow::MathUtil::CeilOfRatio(input_dimension, window_stride); in MakePadding() 130 tensorflow::MathUtil::CeilOfRatio(padding_size, 2ll)); in MakePadding()
|
D | sharding_builder.cc | 58 tile_dimension = CeilOfRatio(static_cast<int64>(tile_dimension), num_tiles); in Tile1D()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util.h | 353 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()
|
D | util.h | 402 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()
|
D | reference_util.cc | 155 return tensorflow::MathUtil::CeilOfRatio(unpadded_width, stride); in WindowCount()
|
/external/tensorflow/tensorflow/core/lib/math/ |
D | math_util_test.cc | 50 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
|
D | math_util.h | 51 static IntegralType CeilOfRatio(IntegralType numerator, in CeilOfRatio() function
|
/external/tensorflow/tensorflow/stream_executor/ |
D | device_description.cc | 147 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/ |
D | quantize_test.cc | 58 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()
|
D | quantize.h | 53 const int64 output_size = CeilOfRatio(input_size, kElementsPerPack); in PackToUint32()
|
D | prng.cc | 118 const int64 half_size = CeilOfRatio<int64>(size, 2); in StatelessRngUniformU32()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | partition_assignment.cc | 91 int64 block_count = CeilOfRatio(num_elements, threads_per_block); in CalculateLaunchDimensions()
|
D | ir_emission_utils.cc | 224 int num_segments = CeilOfRatio(bit_width, 32); in EmitFullWarpShuffleDown()
|
D | ir_emitter_unnested.cc | 1315 uint64 num_blocks = CeilOfRatio(num_iterations, kThreadsPerBlock); in HandleSort()
|
/external/tensorflow/tensorflow/stream_executor/lib/ |
D | mathutil.h | 34 static IntegralType CeilOfRatio(IntegralType numerator, in CeilOfRatio() function
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_sharding.cc | 176 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/ |
D | conv_2d_gpu.h | 875 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),
|
D | clustering_ops.cc | 67 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/ |
D | bigquery_reader_ops.cc | 149 const int64 partition_size = tensorflow::MathUtil::CeilOfRatio<int64>( in Compute()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | stateful_random_ops.cc | 55 const int64 half_size = xla::CeilOfRatio<int64>(size, 2); in StatefulRngUniformU32()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 2614 int64 total_size = port::MathUtil::CeilOfRatio(batch, int64{16}) * in ShouldIncludeWinogradNonfusedAlgo()
|