Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc637 ClusterBatchSize cluster_batch_size; in GetClusterBatchSizeForNode() local
639 return cluster_batch_size; in GetClusterBatchSizeForNode()
644 cluster_batch_size.SetMaxBatchSize( in GetClusterBatchSizeForNode()
653 return cluster_batch_size; in GetClusterBatchSizeForNode()
664 return cluster_batch_size.SetBatchSize(leading_shape->dim(0).size()); in GetClusterBatchSizeForNode()