Searched refs:SetShouldStoreGraphs (Results 1 – 5 of 5) sorted by relevance
143 virtual void SetShouldStoreGraphs(bool value) = 0;
52 tensorflow::unwrap(ctx)->SetShouldStoreGraphs(true); in TFE_ContextEnableGraphCollection()56 tensorflow::unwrap(ctx)->SetShouldStoreGraphs(false); in TFE_ContextDisableGraphCollection()
954 tensorflow::unwrap(ctx)->SetShouldStoreGraphs(true); in TFE_ContextEnableRunMetadata()958 tensorflow::unwrap(ctx)->SetShouldStoreGraphs(false); in TFE_ContextDisableRunMetadata()
320 void SetShouldStoreGraphs(bool value) override;
868 void EagerContext::SetShouldStoreGraphs(bool value) { in SetShouldStoreGraphs() function in tensorflow::EagerContext