| /external/tensorflow/tensorflow/compiler/xla/client/ |
| padding.cc | 124 tensorflow::MathUtil::CeilOfRatio(input_dimension, window_stride);
131 tensorflow::MathUtil::CeilOfRatio(padding_size, 2ll));
|
| sharding_builder.cc | 58 tile_dimension = CeilOfRatio(static_cast<int64>(tile_dimension), num_tiles);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| partition_assignment.cc | 91 int64 block_count = CeilOfRatio(num_elements, threads_per_block);
|
| ir_emission_utils.cc | 247 int num_segments = CeilOfRatio(bit_width, 32);
|
| cudnn_convolution_algorithm_picker.cc | 111 int64 total_size = CeilOfRatio(batch, int64{16}) *
|
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/lib/ |
| mathutil.h | 35 static IntegralType CeilOfRatio(IntegralType numerator,
|
| /external/tensorflow/tensorflow/core/lib/math/ |
| math_util.h | 29 // CeilOfRatio<IntegralType>
51 static IntegralType CeilOfRatio(IntegralType numerator,
|
| math_util_test.cc | 50 IntegralType ceil_1 = MathUtil::CeilOfRatio(numerator, denominator);
55 << "CeilOfRatio fails with numerator = " << numerator
142 // An implementation of CeilOfRatio that is correct for small enough values,
172 const int64 correct_result = MathUtil::CeilOfRatio(numerator, denominator);
189 TEST(MathUtil, CeilOfRatio) {
|
| /external/tensorflow/tensorflow/stream_executor/ |
| device_description.cc | 139 return port::MathUtil::CeilOfRatio(x, y);
155 return port::MathUtil::CeilOfRatio(value, n) * n;
186 port::MathUtil::CeilOfRatio(thread_dims.x * thread_dims.y * thread_dims.z,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util.h | 344 MakeUnique<Array2D<T>>(CeilOfRatio(limits[0] - starts[0], strides[0]),
345 CeilOfRatio(limits[1] - starts[1], strides[1]));
370 MakeUnique<Array3D<T>>(CeilOfRatio(limits[0] - starts[0], strides[0]),
371 CeilOfRatio(limits[1] - starts[1], strides[1]),
372 CeilOfRatio(limits[2] - starts[2], strides[2]));
404 MakeUnique<Array4D<T>>(CeilOfRatio(limits[0] - starts[0], strides[0]),
405 CeilOfRatio(limits[1] - starts[1], strides[1]),
406 CeilOfRatio(limits[2] - starts[2], strides[2]),
407 CeilOfRatio(limits[3] - starts[3], strides[3]));
|
| util.h | 355 // Imports the templated CeilOfRatio math function from the TensorFlow
358 T CeilOfRatio(T dividend, T divisor) {
359 return tensorflow::MathUtil::CeilOfRatio<T>(dividend, divisor);
362 // Rounds the value up to a multiple of the divisor by first calling CeilOfRatio
366 return CeilOfRatio(value, divisor) * divisor;
|
| reference_util.cc | 188 return tensorflow::MathUtil::CeilOfRatio(unpadded_width, stride);
|
| /external/tensorflow/tensorflow/contrib/cloud/kernels/ |
| bigquery_reader_ops.cc | 149 const int64 partition_size = tensorflow::MathUtil::CeilOfRatio<int64>(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_sharding.cc | 35 tile_dimension = CeilOfRatio(static_cast<int64>(tile_dimension), num_tiles);
278 CeilOfRatio(shape.dimensions(i), tile_shape_.dimensions(i));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops_gpu_3.cu.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| clustering_ops.cc | 66 int64 CeilOfRatio(int64 a, int64 b) { return (a + b - 1) / b; }
389 NextMultiple(num_threads, CeilOfRatio(num_points, rows_per_block));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| stateless_random_ops.cc | 134 const int64 half_size = MathUtil::CeilOfRatio<int64>(size, 2);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_gpu_executor.cc | 426 port::MathUtil::CeilOfRatio(block_count, multiprocessor_count);
[all...] |
