Home
last modified time | relevance | path

Searched refs:ComputeBatchSizePerTensorCore (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_configuration_proto_rewrite.cc147 int ComputeBatchSizePerTensorCore( in ComputeBatchSizePerTensorCore() function
183 ComputeBatchSizePerTensorCore(input_feature_batch_sizes); in PopulateBatchSizeAndFeatureCounts()