Home
last modified time | relevance | path

Searched refs:function_table_ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_code.cc106 auto idx = function_table_.find(key); in GetIndex()
107 if (idx != function_table_.end()) { in GetIndex()
110 pprof::Function* func_pb = &function_table_[key]; in GetIndex()
112 func_pb->set_id(function_table_.size()); in GetIndex()
125 return function_table_; in functions()
130 std::map<std::tuple<string, string, uint64>, pprof::Function> function_table_; member in tensorflow::tfprof::__anon370f23e00111::FunctionTable
137 : function_table_(function_table) {} in LocationTable()
155 line_pb->set_function_id(function_table_->GetIndex( in GetIndex()
167 FunctionTable* function_table_; member in tensorflow::tfprof::__anon370f23e00111::LocationTable
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/interpreter/
H A Dcode_impl.h95 std::vector<Function*> function_table_; member
493 insertInstruction(CALL, function_table_.size()); in emitCall()
494 function_table_.emplace_back(func); in emitCall()
554 insertInstruction(TAIL_CALL, function_table_.size()); in emitBailOut()
567 function_table_.emplace_back(func.get()); in emitBailOut()
671 function_table_.emplace_back(forked_functions_.back().get()); in emitFork()
672 insertInstruction(FORK, function_table_.size() - 1, node->inputs().size()); in emitFork()
680 function_table_.emplace_back(awaited_functions_.back().get()); in emitAwaitable()
682 AWAITABLE, function_table_.size() - 1, node->inputs().size()); in emitAwaitable()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dgraphdef_export.cc83 function_table_(function_table) {} in GraphDefExporter()
128 function_table_; member in mlir::tfg::__anon5cc7450f0111::GraphDefExporter
599 if (auto *table = function_table_.dyn_cast<SymbolTable *>()) { in GetOutputSegment()
607 *function_table_.get<const FunctionLibraryDefinition *>())); in GetOutputSegment()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dinterpreter.cpp446 Function* fn = frame.function->function_table_[inst.X]; in runTemplate()
614 frame.function->function_table_[inst.X]->ensure_defined(); in runTemplate()
619 auto& f = *frame.function->function_table_[inst.X]; in runTemplate()
767 toGraphFunction(*frame.function->function_table_[inst.X]); in runTemplate()
781 auto fn_ptr = frame.function->function_table_[inst.X]; in runTemplate()