Searched defs:GraphFunction (Results 1 – 4 of 4) sorted by relevance
9 struct TORCH_API GraphFunction : public Function { struct21 bool isGraphFunction() const override { in isGraphFunction() argument27 std::function<void(GraphFunction&)> function_creator() const { in function_creator() argument35 std::shared_ptr<Graph> graph() const { in graph()41 const c10::QualifiedName& qualname() const override { in qualname()48 void _set_initial_executor_execution_mode(ExecutorExecutionMode mode) { in _set_initial_executor_execution_mode()54 void _set_ignore_amp(bool ignore_amp) { in _set_ignore_amp()78 "GraphFunction::is_optimized() is deprecated and always returns true. " in is_optimized() argument83 void check_single_output() { in check_single_output()89 GraphExecutor& get_executor() { in get_executor()[all …]
64 static void placeholderCreator(GraphFunction&) { in placeholderCreator() argument
20 GraphFunction::GraphFunction(FunctionDef fdef) in GraphFunction() function in tensorflow::tracing::graph::GraphFunction
3001 "name", graph, [](GraphFunction&) {}, ExecutorExecutionMode::PROFILING); in TEST() argument3014 "name", graph, [](GraphFunction&) {}, ExecutorExecutionMode::SIMPLE); in TEST() argument