Searched refs:input_split_metadata (Results 1 – 4 of 4) sorted by relevance
50 InputSplitMetadata input_split_metadata( in TEST() local52 EXPECT_EQ(input_split_metadata.task_sizes().size(), expected_num_batches); in TEST()65 EXPECT_THAT(input_split_metadata.task_sizes(), in TEST()67 EXPECT_EQ(input_split_metadata.DebugString(), in TEST()
22 name = "input_split_metadata",23 srcs = ["input_split_metadata.cc"],24 hdrs = ["input_split_metadata.h"],37 ":input_split_metadata",53 ":input_split_metadata",387 ":input_split_metadata",
160 const internal::InputSplitMetadata input_split_metadata( in get_split_func() local164 input_split_metadata.task_sizes(); in get_split_func()
530 const internal::InputSplitMetadata input_split_metadata( in SplitInputTask() local533 const absl::FixedArray<int>& task_sizes = input_split_metadata.task_sizes(); in SplitInputTask()