Home
last modified time | relevance | path

Searched defs:collective_graph_key (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session.h178 int64_t collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey; member
238 int64_t collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey; member
H A Dbuild_graph_options.h36 int64_t collective_graph_key = kNoCollectiveGraphKey; member
H A Dgraph_execution_state.h69 int64_t collective_graph_key; member
H A Dgraph_execution_state.cc907 int64_t collective_graph_key = options.collective_graph_key; in BuildGraph() local
H A Ddirect_session.cc1600 DataTypeVector* output_types, int64_t* collective_graph_key) { in CreateGraphs()
H A Ddirect_session_test.cc2747 int64_t* collective_graph_key) { in RunGraphWithCollectiveFunctions()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dgraph_mgr.cc129 int64_t collective_graph_key, WorkerSession* session, in InitItem()
301 int64_t collective_graph_key, WorkerSession* session, in Register()
H A Dgraph_mgr.h147 int64_t collective_graph_key; member
H A Dmaster_session.h178 int64_t collective_graph_key; member
H A Dmaster_session.cc122 int64_t collective_graph_key() { return collective_graph_key_; } in collective_graph_key() function in tensorflow::MasterSession::ReffedClientGraph
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto771 int64 collective_graph_key = 1; field
H A Dworker.proto147 int64 collective_graph_key = 7; field
/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/
Dconfig.pb.h7032 inline ::int64_t RunOptions_Experimental::collective_graph_key() const { in collective_graph_key() function
Dworker.pb.h7014 inline ::int64_t RegisterGraphRequest::collective_graph_key() const { in collective_graph_key() function