| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.h | 77 const std::map<string, int>& host_compute_core) in XlaClusterInfo()
81 host_compute_core(host_compute_core) {} in XlaClusterInfo()
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() argument
254 func_name_attrs, new_func_name, host_graph_func_name, host_compute_core, in ExtractOutsideCompilationTest()
291 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
299 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
419 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
427 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
467 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
475 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
590 std::map<string, int> host_compute_core; in TEST_F() local
598 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
[all …]
|
| D | extract_outside_compilation_pass.cc | 292 const Node* call_node, const std::map<string, int>& host_compute_core) { in BuildXlaHostComputeNodeDef() argument
306 const auto iter = host_compute_core.find(original_oc_name); in BuildXlaHostComputeNodeDef()
307 if (iter != host_compute_core.end()) { in BuildXlaHostComputeNodeDef()
374 Graph* g, Node* call_node, const std::map<string, int>& host_compute_core) { in ReplaceOrRemoveOutsideCompilationCallNode() argument
397 BuildXlaHostComputeNodeDef(call_node, host_compute_core)); in ReplaceOrRemoveOutsideCompilationCallNode()
1279 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions() argument
1314 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1371 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1376 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1436 cond, cond_xla_func_name, cond_host_func_name, host_compute_core, flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
[all …]
|
| D | extract_outside_compilation_pass.h | 92 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 114 std::map<string, int>* host_compute_core) { in ParseHostComputeCoreList() argument
128 if (host_compute_core->find(parts[0]) != host_compute_core->end()) { in ParseHostComputeCoreList()
132 (*host_compute_core)[parts[0]] = core; in ParseHostComputeCoreList()
|
| D | side_effect_util.h | 56 std::map<string, int>* host_compute_core);
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TPUReplicateMetadata.pbtxt | 41 name: "host_compute_core"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 927 host_compute_core = context.HostComputeCore()
929 if host_compute_core:
931 attr_value.list.s.extend([compat.as_bytes(x) for x in host_compute_core])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3849 …_per_replica\', \'topology\', \'use_tpu\', \'device_assignment\', \'host_compute_core\', \'padding…
3853 …ogy\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_compute_core\', \'padding…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3849 …_per_replica\', \'topology\', \'use_tpu\', \'device_assignment\', \'host_compute_core\', \'padding…
3853 …ogy\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_compute_core\', \'padding…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 82164 name: "host_compute_core"
82269 name: "host_compute_core"
82363 name: "host_compute_core"
82424 name: "host_compute_core"
|
| D | ops_history.v1.pbtxt | 82791 name: "host_compute_core"
82896 name: "host_compute_core"
82990 name: "host_compute_core"
83051 name: "host_compute_core"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 39946 name: "host_compute_core"
40040 name: "host_compute_core"
|