Searched refs:shard_size (Results 1 – 7 of 7) sorted by relevance
| /external/autotest/server/site_tests/video_VDAStressSetup/ |
| D | video_VDAStressSetup.py | 39 shard_size = int(math.ceil(video_count / float(shard_count)))
40 begin = shard_size * shard_number
41 end = min(video_count, shard_size * (shard_number + 1))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 408 const size_t shard_size = in Compute() local
419 TensorShape col_buffer_shape = {static_cast<int64>(shard_size), in Compute()
520 image_id += shard_size) { in Compute()
522 std::min(static_cast<int>(shard_size), in Compute()
872 const size_t shard_size = in Compute() local
881 TensorShape col_buffer_shape = {static_cast<int64>(shard_size), in Compute()
942 for (int image_id = 0; image_id < dims.batch_size; image_id += shard_size) { in Compute()
944 std::min(static_cast<int>(shard_size), in Compute()
|
| D | conv_grad_filter_ops.cc | 322 const size_t shard_size = in Compute() local
329 TensorShape({static_cast<int64>(shard_size), in Compute()
363 for (int image_id = 0; image_id < dims.batch_size; image_id += shard_size) { in Compute()
365 std::min(static_cast<int>(shard_size), in Compute()
|
| D | topk_op_gpu.h | 390 auto shard_size = length / num_shards;
392 if (shard_size < min_shard_size) {
|
| D | conv_grad_input_ops.cc | 434 const size_t shard_size = in Compute() local
443 TensorShape({static_cast<int64>(shard_size), in Compute()
498 image_id += shard_size) { in Compute()
500 std::min(static_cast<int>(shard_size), in Compute()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 331 shard_size, residual = divmod(dims, num_shards)
332 return [shard_size + 1] * residual + [shard_size] * (num_shards - residual)
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 666 shard_size = batch_size // self._tpu_assignment.num_towers
670 x[index * shard_size:(index + 1) * shard_size] for x in inputs
|