Searched defs:GetBatchSize (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ | ||
| H A D | union_find.cc | 80 int ClusterBatchSize::GetBatchSize() const { in GetBatchSize() function in tensorflow::tensorrt::segment::ClusterBatchSize |
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ | ||
| H A D | compute_batch_size_op.cc | 155 int64_t GetBatchSize(const NodeDef& node, const grappler::GraphView& graph) { in GetBatchSize() function in tensorflow::data::experimental::__anon36dfd0560111::ComputeBatchSizeOp |
| /aosp_15_r20/external/cronet/base/test/launcher/ | ||
| H A D | unit_test_launcher.cc | 463 size_t UnitTestLauncherDelegate::GetBatchSize() { in GetBatchSize() function in base::UnitTestLauncherDelegate |