Searched refs:is_graph_rewrite (Results 1 – 8 of 8) sorted by relevance
87 if (ctx->is_graph_rewrite()) { in AsGraphDefInternal()
102 if (!ctx->is_graph_rewrite()) { in AsGraphDefInternal()
120 if (!ctx->is_graph_rewrite()) { in AsGraphDefInternal()
111 if (!ctx->is_graph_rewrite()) { in AsGraphDefInternal()
654 bool is_graph_rewrite = false; member673 bool is_graph_rewrite() const { return params_.is_graph_rewrite; } in is_graph_rewrite() function
754 if (ctx->is_graph_rewrite()) { in AddInputDataset()795 if (t.dtype() == DT_RESOURCE && !ctx->is_graph_rewrite()) { in AddDatasetOrTensor()
427 params.is_graph_rewrite = true; in AsGraphDefForRewrite()
511 if (!ctx->is_graph_rewrite()) { in AddToGraph()