Searched refs:GetBatchSize (Results 1 – 3 of 3) sorted by relevance
108 int64 batch_size = GetBatchSize(*node, graph_view); in Compute()153 int64 GetBatchSize(const NodeDef& node, const grappler::GraphView& graph) { in GetBatchSize() function in tensorflow::data::experimental::__anonb93db4c70111::ComputeBatchSizeOp159 int64 batch_size_0 = GetBatchSize(*input_0, graph); in GetBatchSize()162 auto batch_size_i = GetBatchSize(*input, graph); in GetBatchSize()176 return GetBatchSize(*input, graph); in GetBatchSize()
80 int ClusterBatchSize::GetBatchSize() const { in GetBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize
84 int GetBatchSize() const;