Searched defs:num_tensors_ (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ | ||
| H A D | min_cost_flow_assignment.cc | 203 size_t num_tensors_; member in tflite::gpu::__anonde0fea1b0111::MinCostFlowSolver |
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ | ||
| H A D | argument_spec.h | 205 size_t num_tensors_ = 0; member |
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ | ||
| H A D | ProcessGroupWrapper.cpp | 27 std::size_t num_tensors_{}; member |
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ | ||
| H A D | iterator_ops.cc | 427 int64_t num_tensors_; member in tensorflow::data::__anonfca7ab5e0511::IteratorVariantSerializer |
| H A D | cache_dataset_ops.cc | 673 const size_t num_tensors_; member in tensorflow::data::CacheDatasetOp::FileDatasetBase |