tensorflow/core/tpu/tpu_embedding_configuration_proto_rewrite.cc
// Computes the TensorCore batch size as the GCD of all input feature batch
// sizes.
Computes the TensorCore batch size as the GCD of all input feature batch
sizes.
Computes the TensorCore batch size as the GCD of all input feature batch sizes .
int ComputeBatchSizePerTensorCore(
    absl::Span<const int> input_feature_batch_sizes) {
  uint32_t batch_size = input_feature_batch_sizes[0];
  for (const uint32_t input_feature_batch_size : input_feature_batch_sizes) {
    batch_size =
        tensorflow::MathUtil::GCD(batch_size, input_feature_batch_size);
  }
  return batch_size;
}
int ComputeBatchSizePerTensorCore ( absl :: Span < const int > input_feature_batch_sizes ) { uint32_t batch_size = input_feature_batch_sizes [ 0 ] ; for ( const uint32_t input_feature_batch_size : input_feature_batch_sizes ) { batch_size = tensorflow :: MathUtil :: GCD ( batch_size , input_feature_batch_size ) ; } return batch_size ; }
Computes the TensorCore batch size as the GCD of all input feature batch
sizes.
Computes the TensorCore batch size as the GCD of all input feature batch sizes .