Searched defs:host_compute_core (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass_test.cc | 243 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
|
| H A D | extract_outside_compilation_pass.cc | 310 const Node* call_node, const std::map<string, int>& host_compute_core, in BuildXlaHostComputeNodeDef()
380 Graph* g, Node* call_node, const std::map<string, int>& host_compute_core, in ReplaceOutsideCompilationCallNode()
1811 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForFuncCallNode()
1897 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForIfNode()
2016 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForWhileNode()
2117 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2312 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction()
2516 auto const& host_compute_core = iter.second.host_compute_core; in ExtractOutsideCompilation() local
|
| H A D | encapsulate_util.h | 113 const std::map<string, int> host_compute_core; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | side_effect_util.cc | 126 std::map<string, int>* host_compute_core) { in ParseHostComputeCoreList()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 3963 const HostComputeCoreMap& host_compute_core, in ReplicateOutsideCompilationNodes()
4355 HostComputeCoreMap* host_compute_core) { in ParseHostComputeCores()
4863 HostComputeCoreMap host_compute_core; in RewriteTPUReplicateNode() local
|
| H A D | encapsulate_tpu_computations_pass.cc | 2943 std::map<string, int> host_compute_core; in Run() local
|