Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dset_stats_aggregator_dataset_op.cc157 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
158 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
164 this, {input_graph_node, resource_handle_node, tag_node, prefix_node}, in AsGraphDefInternal()
Dthreadpool_dataset_op.cc192 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
193 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
195 this, {input_graph_node, resource_handle_node}, output)); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc531 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
534 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
537 {input_graph_node, buffer_size_node, resource_handle_node}, // Inputs in AsGraphDefInternal()
590 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
593 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
600 seed2_node, resource_handle_node}, // Inputs in AsGraphDefInternal()
841 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
844 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
851 seed2_node, count_node, resource_handle_node}, // Inputs in AsGraphDefInternal()
Dcache_dataset_ops.cc650 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
651 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
653 this, {input_node, filename_node, resource_handle_node}, output)); in AsGraphDefInternal()
1003 Node* resource_handle_node = nullptr; in AsGraphDefInternal() local
1006 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
1008 this, {input_node, filename_node, resource_handle_node}, output)); in AsGraphDefInternal()