Home
last modified time | relevance | path

Searched refs:split_provider_index (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dsplit_utils.cc171 int64_t split_provider_index = 0; in CreateInputIteratorContexts() local
177 ctx->split_providers()[split_provider_index + j]); in CreateInputIteratorContexts()
179 split_provider_index += inputs[i]->num_sources(); in CreateInputIteratorContexts()
222 int64_t split_provider_index = 0; in CreateInputIteratorContexts() local
233 ctx->split_providers()[split_provider_index + j]); in CreateInputIteratorContexts()
235 split_provider_index += inputs[i]->num_sources(); in CreateInputIteratorContexts()
238 if (split_provider_index != ctx->split_providers().size()) { in CreateInputIteratorContexts()
242 split_provider_index, " sources"); in CreateInputIteratorContexts()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Dsplit_provider.h37 int64_t split_provider_index, int64_t timeout_ms) in DataServiceSplitProvider() argument
41 split_provider_index_(split_provider_index), in DataServiceSplitProvider()
H A Ddispatcher_client.cc94 int64_t split_provider_index, in GetSplit() argument
101 req.set_split_provider_index(split_provider_index); in GetSplit()
H A Ddispatcher_impl.cc414 int64_t provider_index = request->split_provider_index(); in GetSplit()
441 request->split_provider_index(), in GetSplit()
1092 int64_t iteration_id, int64_t repetition, int64_t split_provider_index, in RecordSplitProduced() argument
1098 produce_split->set_split_provider_index(split_provider_index); in RecordSplitProduced()
H A Ddispatcher_client.h65 int64_t split_provider_index, Tensor& split,
H A Djournal.proto76 int64 split_provider_index = 4; field
H A Ddispatcher_impl.h294 int64_t split_provider_index, bool finished)
H A Ddispatcher.proto56 int64 split_provider_index = 3; field
H A Ddispatcher_state.cc185 int64_t provider_index = produce_split.split_provider_index(); in ProduceSplit()