Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dmodel_dataset_op.cc93 auto remove_node_hook = [](std::shared_ptr<model::Node> node) { in Iterator() local
96 model_ = std::make_shared<model::Model>(std::move(remove_node_hook)); in Iterator()
/external/tensorflow/tensorflow/core/framework/
Dmodel.h373 Model(NodeHook remove_node_hook) in Model() argument
375 remove_node_hook_(std::move(remove_node_hook)) { in Model()