Searched refs:SetShouldStoreStepStats (Results 1 – 3 of 3) sorted by relevance
186 void SetShouldStoreStepStats(bool value);
379 void EagerContext::SetShouldStoreStepStats(bool value) { in SetShouldStoreStepStats() function in tensorflow::EagerContext
861 ctx->context.SetShouldStoreStepStats(true); in TFE_ContextEnableRunMetadata()866 ctx->context.SetShouldStoreStepStats(false); in TFE_ContextDisableRunMetadata()