Searched refs:track_allocations (Results 1 – 15 of 15) sorted by relevance
80 if (ctx->track_allocations()) { in Compute()84 if (ctx->track_allocations()) { in Compute()140 if (ctx->track_allocations()) { in Compute()146 if (ctx->track_allocations()) { in Compute()
75 if (ctx->track_allocations() && var->tensor()->IsInitialized()) { in Compute()109 if (context->track_allocations()) { in Compute()153 if (context->track_allocations()) { in Compute()
46 if (resource_ && context->track_allocations()) { in Compute()
68 params.track_allocations = true; in PersistentMemoryTrackingTest()
707 if (ctx->track_allocations()) { in Compute()711 if (ctx->track_allocations()) { in Compute()784 if (ctx->track_allocations()) { in Compute()788 if (ctx->track_allocations()) { in Compute()
66 if (ctx->track_allocations()) { in Compute()
85 if (TF_PREDICT_FALSE(ctx->track_allocations())) { in Compute()
126 void HeapProfiler::StartHeapObjectsTracking(bool track_allocations) { in StartHeapObjectsTracking() argument130 if (track_allocations) { in StartHeapObjectsTracking()
40 void StartHeapObjectsTracking(bool track_allocations);
287 if (track_allocations()) { in get_allocator()295 new TrackingAllocator(allocator, params_->track_allocations); in get_allocator()672 if (track_allocations() && s.ok() && out_temp->TotalBytes() > 0) { in allocate_temp()694 if (track_allocations()) { in allocate_persistent()727 if (track_allocations() && tensor.TotalBytes() > 0) { in set_output()
517 bool track_allocations = false; member1048 bool track_allocations() const { return params_->track_allocations; } in track_allocations() function
723 void StartTrackingHeapObjects(bool track_allocations = false);
288 params.track_allocations = true; in Run()
1548 params.track_allocations = false; in Process()1552 params.track_allocations = true; in Process()
9971 void HeapProfiler::StartTrackingHeapObjects(bool track_allocations) { in StartTrackingHeapObjects() argument9973 track_allocations); in StartTrackingHeapObjects()