Searched refs:SetShouldStoreGraphs (Results 1 – 4 of 4) sorted by relevance
74 ctx->context.SetShouldStoreGraphs(true); in TFE_ContextEnableGraphCollection()78 ctx->context.SetShouldStoreGraphs(false); in TFE_ContextDisableGraphCollection()
860 ctx->context.SetShouldStoreGraphs(true); in TFE_ContextEnableRunMetadata()865 ctx->context.SetShouldStoreGraphs(false); in TFE_ContextDisableRunMetadata()
188 void SetShouldStoreGraphs(bool value);
371 void EagerContext::SetShouldStoreGraphs(bool value) { in SetShouldStoreGraphs() function in tensorflow::EagerContext