Searched refs:dataset_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | slack.cc | 81 NodeDef* dataset_node) { in RecursivelyHandleOp() argument
82 if (dataset_node->op() == kPrefetchDatasetOp) { in RecursivelyHandleOp()
83 if (HasNodeAttr(*dataset_node, "slack_period")) { in RecursivelyHandleOp()
84 (*dataset_node->mutable_attr())["slack_period"].set_i(slack_period_); in RecursivelyHandleOp()
86 AddNodeAttr("slack_period", slack_period_, dataset_node); in RecursivelyHandleOp()
90 if (IsDatasetNodeOfType(*dataset_node, kPassThroughOps)) { in RecursivelyHandleOp()
91 NodeDef* input_node = graph_utils::GetInputNode(*dataset_node, graph, 0); in RecursivelyHandleOp()
94 if (IsDatasetNodeOfType(*dataset_node, kMultipleInputsDatasetOps)) { in RecursivelyHandleOp()
96 for (int i = 0; i < dataset_node->input_size(); ++i) { in RecursivelyHandleOp()
97 NodeDef* input_node = graph_utils::GetInputNode(*dataset_node, graph, i); in RecursivelyHandleOp()
[all …]
|
| D | slack.h | 62 NodeDef* dataset_node);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | serialization_utils.cc | 58 GraphDef* result, string* dataset_node) { in AsGraphDefMinimal() argument
73 *dataset_node = node.input(0); in AsGraphDefMinimal()
|
| D | serialization_utils.h | 37 GraphDef* result, string* dataset_node);
|