Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc304 NodeDef node_in_func = func->node_def(i); in ReaderOpInFunction() local
305 if (IsDatasetNodeOfType(node_in_func, kReaderDatasetOps) && in ReaderOpInFunction()
306 node_in_func.input_size() > 0 && in ReaderOpInFunction()
307 absl::StartsWith(node_in_func.input(0), "args_0")) { in ReaderOpInFunction()