Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.h113 const std::map<string, int> host_compute_core; member
Dextract_outside_compilation_pass_test.cc243 const std::map<string, int> &host_compute_core, in ExtractOutsideCompilationTest()
292 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
410 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
485 std::map<string, int> host_compute_core; in TEST_F() local
635 std::map<string, int> host_compute_core; in TEST_F() local
771 std::map<string, int> host_compute_core; in TEST_F() local
877 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
959 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
Dextract_outside_compilation_pass.cc311 const Node* call_node, const std::map<string, int>& host_compute_core, in BuildXlaHostComputeNodeDef()
381 Graph* g, Node* call_node, const std::map<string, int>& host_compute_core, in ReplaceOutsideCompilationCallNode()
1808 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForFuncCallNode()
1894 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForIfNode()
2013 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForWhileNode()
2114 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2311 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction()
2515 auto const& host_compute_core = iter.second.host_compute_core; in ExtractOutsideCompilation() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc126 std::map<string, int>* host_compute_core) { in ParseHostComputeCoreList()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3509 const HostComputeCoreMap& host_compute_core, in ReplicateOutsideCompilationNodes()
3901 HostComputeCoreMap* host_compute_core) { in ParseHostComputeCores()
4358 HostComputeCoreMap host_compute_core; in RewriteTPUReplicateNode() local
Dencapsulate_tpu_computations_pass.cc2995 std::map<string, int> host_compute_core; in Run() local