Searched defs:GetBatchSize (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ | ||
| D | union_find.cc | 80 int ClusterBatchSize::GetBatchSize() const { in GetBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize |
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ | ||
| D | compute_batch_size_op.cc | 155 int64 GetBatchSize(const NodeDef& node, const grappler::GraphView& graph) { in GetBatchSize() function in tensorflow::data::experimental::__anon5e2a4ccc0111::ComputeBatchSizeOp |