Home
last modified time | relevance | path

Searched refs:graph_executor (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/graph_executor/
Dgraph_executor_test.cc64 GraphExecutor* graph_executor = status_or_graph_executor.ValueOrDie().get(); in TEST_F() local
73 TF_ASSERT_OK(graph_executor->Run(/*run_options=*/{}, inputs, in TEST_F()
104 auto graph_executor, in TEST_F()
118 TF_ASSERT_OK(graph_executor->Extend(extension)); in TEST_F()
127 TF_ASSERT_OK(graph_executor->Run(/*run_options=*/{}, inputs, in TEST_F()
DBUILD32 name = "graph_executor",
33 srcs = ["graph_executor.cc"],
34 hdrs = ["graph_executor.h"],
80 ":graph_executor",
/external/tensorflow/tensorflow/core/tfrt/saved_model/
DBUILD19 "//tensorflow/core/tfrt/graph_executor/...",
65 "//tensorflow/core/tfrt/graph_executor",
66 "//tensorflow/core/tfrt/graph_executor:graph_execution_options",
Dsaved_model.cc579 "graph_executor creation", auto graph_executor, in LoadSavedModel()
590 std::move(resource_context), std::move(graph_executor))}; in LoadSavedModel()
607 std::unique_ptr<GraphExecutor> graph_executor) in SavedModelImpl() argument
620 graph_executor_(std::move(graph_executor)), in SavedModelImpl()
Dsaved_model.h217 std::unique_ptr<GraphExecutor> graph_executor);
/external/tensorflow/tensorflow/core/tfrt/tpu/
DBUILD11 "//tensorflow/core/tfrt/graph_executor:__pkg__",
/external/pytorch/torch/contrib/
D_tensorboard_vis.py22 def dump_tensorboard_summary(graph_executor, logdir): argument
24 pb_graph = visualize(graph_executor)
/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_backend_impl.cpp214 torch::jit::GraphExecutor& graph_executor = ts_computation->graph_executor(); in ExecuteComputation() local
230 graph_executor.run(stack); in ExecuteComputation()
Dts_lowering_context.h55 torch::jit::GraphExecutor& graph_executor() { in graph_executor() function
/external/pytorch/torch/_dynamo/backends/
Dtvm.py33 from tvm.contrib import graph_executor # type: ignore[import]
130 m = graph_executor.GraphModule(lib["default"](dev))
/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
DBUILD96 "//tensorflow/core/tfrt/graph_executor:__pkg__",
/external/pytorch/torch/csrc/lazy/python/
Dinit.cpp299 computationPtr->graph_executor().run(stack); in initLazyBindings()
/external/tensorflow/tensorflow/core/tfrt/utils/
DBUILD288 "//tensorflow/core/tfrt/graph_executor:__pkg__",
/external/tensorflow/tensorflow/compiler/mlir/tfrt/
DBUILD28 "//tensorflow/core/tfrt/graph_executor/...",
552 "//tensorflow/core/tfrt/graph_executor:__pkg__",
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
DBUILD349 "//tensorflow/core/tfrt/graph_executor:__pkg__",
/external/tensorflow/
D.bazelrc671 …e/tfrt/runtime,tensorflow/core/tfrt/saved_model,tensorflow/core/tfrt/graph_executor,tensorflow/cor…
674 …e/tfrt/runtime,tensorflow/core/tfrt/saved_model,tensorflow/core/tfrt/graph_executor,tensorflow/cor…
/external/pytorch/torch/csrc/jit/
DOVERVIEW.md877 [runtime/graph_executor.cpp](runtime/graph_executor.cpp)
1306 [runtime/graph_executor.cpp](runtime/graph_executor.cpp)
/external/pytorch/torch/_C/
D__init__.pyi.in732 # Defined in torch/csrc/jit/runtime/graph_executor.h
/external/pytorch/
Dbuild_variables.bzl309 "torch/csrc/jit/runtime/graph_executor.cpp",