Home
last modified time | relevance | path

Searched defs:task_sizes (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dinput_split_metadata.cc69 const absl::FixedArray<int>& InputSplitMetadata::task_sizes() const { in task_sizes() function in tensorflow::serving::internal::InputSplitMetadata
86 absl::FixedArray<int> task_sizes(num_batches, batch_size_limit); in generate_task_sizes() local
H A Dbatch_input_task.h231 const absl::FixedArray<int>& task_sizes = input_split_metadata_.task_sizes(); in ToTaskHandles() local
H A Dshared_batch_scheduler_test.cc163 const absl::FixedArray<int> task_sizes = in get_split_func() local
H A Dbatch_resource_base.cc533 const absl::FixedArray<int>& task_sizes = input_split_metadata.task_sizes(); in SplitInputTask() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dcumsum.cc29 std::map<Axis, std::string> task_sizes = { in GetCumsumCode() local