Home
last modified time | relevance | path

Searched refs:collect_resource_usage (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddataset.h1040 if (collect_resource_usage(ctx)) { in RecordBufferDequeue()
1049 if (collect_resource_usage(ctx)) { in RecordBufferEnqueue()
1070 if (collect_resource_usage(ctx)) { in RecordStart()
1079 if (collect_resource_usage(ctx)) { in RecordStop()
1088 return collect_resource_usage(ctx) && node_->is_recording(); in IsRecording()
1092 bool collect_resource_usage(IteratorContext* ctx) { in collect_resource_usage() function
1094 return model && model->collect_resource_usage() && node_; in collect_resource_usage()
Ddataset.cc558 if (model && model->collect_resource_usage() && node_) { in GetNext()
571 if (model && model->collect_resource_usage() && node_) { in GetNext()
597 if (model && model->collect_resource_usage() && node_) { in Skip()
606 if (model && model->collect_resource_usage() && node_) { in Skip()
Dmodel.proto111 bool collect_resource_usage = 3; field
Dmodel.h660 bool collect_resource_usage() const { return collect_resource_usage_; } in collect_resource_usage() function
Dmodel_test.cc969 EXPECT_EQ(model.collect_resource_usage(), in TEST()
970 restored_model->collect_resource_usage()); in TEST()
Dmodel.cc1950 model_proto.collect_resource_usage()); in FromProto()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc857 node && ctx->model() && ctx->model()->collect_resource_usage(); in Run()
920 node && ctx->model() && ctx->model()->collect_resource_usage(); in RunWithBorrowedArgs()
1022 node && ctx->model() && ctx->model()->collect_resource_usage(); in RunAsync()