Searched refs:track_allocations (Results 1 – 11 of 11) sorted by relevance
79 if (ctx->track_allocations()) { in Compute()83 if (ctx->track_allocations()) { in Compute()140 if (ctx->track_allocations()) { in Compute()146 if (ctx->track_allocations()) { in Compute()
89 if (ctx->track_allocations() && var->tensor()->IsInitialized()) { in Compute()121 if (context->track_allocations()) { in Compute()167 if (context->track_allocations()) { in Compute()
877 if (ctx->track_allocations()) { in Compute()881 if (ctx->track_allocations()) { in Compute()910 if (ctx->track_allocations()) { in Compute()914 if (ctx->track_allocations()) { in Compute()983 if (ctx->track_allocations()) { in Compute()987 if (ctx->track_allocations()) { in Compute()
73 params.track_allocations = true; in PersistentMemoryTrackingTest()
75 if (ctx->track_allocations()) { in Compute()
40 if (resource_ && context->track_allocations()) { in Compute()
89 if (TF_PREDICT_FALSE(ctx->track_allocations())) { in Compute()
331 if (params_->track_allocations) { in OpKernelContext()353 if (params_->track_allocations && in ~OpKernelContext()371 if (TF_PREDICT_FALSE(track_allocations())) { in get_allocator()380 new TrackingAllocator(allocator, params_->track_allocations); in get_allocator()799 if (track_allocations() && s.ok() && out_temp->TotalBytes() > 0) { in allocate_temp()837 if (track_allocations()) { in allocate_persistent()949 if (TF_PREDICT_FALSE(track_allocations()) && tensor.TotalBytes() > 0) { in maybe_track_allocations_for_set_output()
633 bool track_allocations = false; member1216 bool track_allocations() const { return params_->track_allocations; } in track_allocations() function
752 params.track_allocations = false; in Process()758 params.track_allocations = stats ? stats->TrackAllocations() : false; in Process()780 } else if (item.const_tensor != nullptr && !params.track_allocations) { in Process()
305 params->track_allocations = false; in PrepOpKernel()