Home
last modified time | relevance | path

Searched refs:mutable_partition_graph (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmessage_wrappers.cc664 GraphDef* InMemoryRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::InMemoryRunGraphResponse
734 GraphDef* OwnedProtoRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::OwnedProtoRunGraphResponse
735 return response_.mutable_partition_graph(i); in mutable_partition_graph()
740 GraphDef* graph_def = response_.mutable_partition_graph()->Add(); in AddPartitionGraph()
811 GraphDef* NonOwnedProtoRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::NonOwnedProtoRunGraphResponse
812 return response_->mutable_partition_graph(i); in mutable_partition_graph()
H A Dmessage_wrappers_test.cc128 EXPECT_EQ(1234, response->mutable_partition_graph(0)->versions().producer()); in CheckRunGraphResponse()
130 response->mutable_partition_graph(0)->versions().min_consumer()); in CheckRunGraphResponse()
145 run_graph_response->mutable_partition_graph(i)); in BuildRunStepResponse()
H A Dmessage_wrappers.h494 virtual GraphDef* mutable_partition_graph(size_t i) = 0;
530 GraphDef* mutable_partition_graph(size_t i) override;
564 GraphDef* mutable_partition_graph(size_t i) override;
592 GraphDef* mutable_partition_graph(size_t i) override;
H A Dmaster_session.cc804 run_graph_resp->mutable_partition_graph(i)); in RunPartitionsHelper()
/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.h2612 ::tensorflow::GraphDef* mutable_partition_graph(int index); in Swap()
2614 mutable_partition_graph(); in Swap()
8060 inline ::tensorflow::GraphDef* RunGraphResponse::mutable_partition_graph(int index) { in mutable_partition_graph() function
8065 RunGraphResponse::mutable_partition_graph() { in mutable_partition_graph() function