MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / _validate_batch_size

Function _validate_batch_size

tensorflow/python/tpu/tpu_embedding.py:1112–1116  ·  view source on GitHub ↗
(batch_size, num_cores)

Source from the content-addressed store, hash-verified

1110
1111
1112def _validate_batch_size(batch_size, num_cores):
1113 if batch_size % num_cores:
1114 raise ValueError('`batch_size` is not a multiple of number of '
1115 'cores. `batch_size`={}, `_num_cores`={}.'.format(
1116 batch_size, num_cores))
1117
1118
1119def _validate_optimization_parameters(optimization_parameters,

Callers 1

__init__Method · 0.85

Calls 1

formatMethod · 0.45

Tested by

no test coverage detected