Home
last modified time | relevance | path

Searched refs:track_allocations (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_init_op.cc82 if (ctx->track_allocations()) { in Compute()
86 if (ctx->track_allocations()) { in Compute()
145 if (ctx->track_allocations()) { in Compute()
153 if (ctx->track_allocations()) { in Compute()
Dvariable_ops.cc89 if (ctx->track_allocations() && var->tensor()->IsInitialized()) { in Compute()
121 if (context->track_allocations()) { in Compute()
167 if (context->track_allocations()) { in Compute()
Dlookup_table_op.h77 if (ctx->track_allocations()) { in Compute()
156 if (ctx->track_allocations()) { in Compute()
Dconstant_op_test.cc73 params.track_allocations = true; in PersistentMemoryTrackingTest()
Dlookup_table_op.cc957 if (ctx->track_allocations()) { in Compute()
961 if (ctx->track_allocations()) { in Compute()
990 if (ctx->track_allocations()) { in Compute()
994 if (ctx->track_allocations()) { in Compute()
1063 if (ctx->track_allocations()) { in Compute()
1067 if (ctx->track_allocations()) { in Compute()
Dqueue_op.cc40 if (resource_ && context->track_allocations()) { in Compute()
Dconstant_op.cc92 if (TF_PREDICT_FALSE(ctx->track_allocations())) { in Compute()
/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/
Dsimple_hash_table_kernel.cc266 if (ctx->track_allocations()) { in Compute()
270 if (ctx->track_allocations()) { in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc344 if (params_->track_allocations) { in OpKernelContext()
366 if (params_->track_allocations && in ~OpKernelContext()
384 if (TF_PREDICT_FALSE(track_allocations())) { in get_allocator()
393 new TrackingAllocator(allocator, params_->track_allocations); in get_allocator()
844 if (track_allocations() && s.ok() && out_temp->TotalBytes() > 0) { in allocate_temp()
960 if (TF_PREDICT_FALSE(track_allocations()) && tensor.TotalBytes() > 0) { in maybe_track_allocations_for_set_output()
Dop_kernel.h606 bool track_allocations = false; member
1177 bool track_allocations() const { return params_->track_allocations; } in track_allocations() function
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc804 params.track_allocations = false; in ProcessInline()
810 params.track_allocations = stats ? stats->TrackAllocations() : false; in ProcessInline()
832 } else if (item.const_tensor != nullptr && !params.track_allocations) { in ProcessInline()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc305 params->track_allocations = false; in PrepOpKernel()