Searched defs:host_compute_core (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.h | 89 const std::map<string, int> host_compute_core; member
|
| D | extract_outside_compilation_pass_test.cc | 243 const std::map<string, int> &host_compute_core, in ExtractOutsideCompilationTest()
291 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
419 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
467 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
590 std::map<string, int> host_compute_core; in TEST_F() local
749 std::map<string, int> host_compute_core; in TEST_F() local
889 std::map<string, int> host_compute_core; in TEST_F() local
|
| D | extract_outside_compilation_pass.cc | 292 const Node* call_node, const std::map<string, int>& host_compute_core) { in BuildXlaHostComputeNodeDef()
374 Graph* g, Node* call_node, const std::map<string, int>& host_compute_core) { in ReplaceOrRemoveOutsideCompilationCallNode()
1279 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1589 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction()
1729 auto const& host_compute_core = iter.second.host_compute_core; in ExtractOutsideCompilation() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 114 std::map<string, int>* host_compute_core) { in ParseHostComputeCoreList()
|