Home
last modified time | relevance | path

Searched defs:partition_graph (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmessage_wrappers.cc669 const GraphDef& partition_graph) { in AddPartitionGraph()
739 const GraphDef& partition_graph) { in AddPartitionGraph()
816 const GraphDef& partition_graph) { in AddPartitionGraph()
/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dwhile_gradients_test.cc227 for (const GraphDef& partition_graph : run_metadata.partition_graphs()) { in TEST_F() local
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Daccelerator_partitioner.py303 def partition_graph( member in Partitioner
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto298 repeated GraphDef partition_graph = 4; field
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session.cc1375 std::unique_ptr<Graph>& partition_graph = iter->second; in CreateExecutors() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dworker.pb.h8072 inline const ::tensorflow::GraphDef& RunGraphResponse::partition_graph(int index) const { in partition_graph() function
8085 RunGraphResponse::partition_graph() const { in partition_graph() function