Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
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
Dencapsulate_util.h113 const std::map<string, int> host_compute_core; member
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()
1807 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForFuncCallNode()
1893 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForIfNode()
2012 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForWhileNode()
2113 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2310 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction()
2514 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.cc4016 const HostComputeCoreMap& host_compute_core, in ReplicateOutsideCompilationNodes()
4408 HostComputeCoreMap* host_compute_core) { in ParseHostComputeCores()
4920 HostComputeCoreMap host_compute_core; in RewriteTPUReplicateNode() local
Dencapsulate_tpu_computations_pass.cc2995 std::map<string, int> host_compute_core; in Run() local