Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Drewrite_utils.h72 StatusOr<NodeDef> GetDatasetNodeDef(const GraphDef& graph_def);
H A Drewrite_utils_test.cc91 TF_ASSERT_OK_AND_ASSIGN(NodeDef dataset_nodedef, GetDatasetNodeDef(graph)); in TEST()
H A Drewrite_utils.cc344 StatusOr<NodeDef> GetDatasetNodeDef(const GraphDef& graph_def) { in GetDatasetNodeDef() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dassert_prev_dataset_op.cc88 TF_ASSIGN_OR_RETURN(NodeDef node, GetDatasetNodeDef(graph_def)); in GetDatasetNode()