| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.h | 101 const std::map<string, int>& host_compute_core) in XlaClusterInfo()
105 host_compute_core(host_compute_core) {} in XlaClusterInfo()
113 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
255 func_name_attrs, new_func_name, host_graph_func_name, host_compute_core, in ExtractOutsideCompilationTest()
292 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
300 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
410 std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}}; in TEST_F() local
418 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
485 std::map<string, int> host_compute_core; in TEST_F() local
493 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
635 std::map<string, int> host_compute_core; in TEST_F() local
643 host_compute_core, &fld, &shape_inference_graphs, in TEST_F()
[all …]
|
| D | extract_outside_compilation_pass.cc | 311 const Node* call_node, const std::map<string, int>& host_compute_core, in BuildXlaHostComputeNodeDef() argument
330 const auto iter = host_compute_core.find(original_oc_name); in BuildXlaHostComputeNodeDef()
331 if (iter != host_compute_core.end()) { in BuildXlaHostComputeNodeDef()
381 Graph* g, Node* call_node, const std::map<string, int>& host_compute_core, in ReplaceOutsideCompilationCallNode() argument
386 BuildXlaHostComputeNodeDef(call_node, host_compute_core, cluster_deps)); in ReplaceOutsideCompilationCallNode()
1808 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForFuncCallNode() argument
1839 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForFuncCallNode()
1894 const std::map<string, int>& host_compute_core, Graph* g, Node* n, in ExtractOutsideCompilationForIfNode() argument
1916 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
1921 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
[all …]
|
| D | extract_outside_compilation_pass.h | 94 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 126 std::map<string, int>* host_compute_core) { in ParseHostComputeCoreList() argument
140 if (host_compute_core->find(parts[0]) != host_compute_core->end()) { in ParseHostComputeCoreList()
144 (*host_compute_core)[parts[0]] = core; in ParseHostComputeCoreList()
|
| D | side_effect_util.h | 65 std::map<string, int>* host_compute_core);
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TPUReplicateMetadata.pbtxt | 46 name: "host_compute_core"
107 name: "host_compute_core"
175 name: "host_compute_core"
250 name: "host_compute_core"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TPUReplicateMetadata.pbtxt | 46 name: "host_compute_core"
107 name: "host_compute_core"
175 name: "host_compute_core"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TPUReplicateMetadata.pbtxt | 41 name: "host_compute_core"
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 479 const HostComputeCoreMap& host_compute_core,
517 HostComputeCoreMap* host_compute_core);
|
| D | distributed_tpu_rewrite_pass.cc | 3509 const HostComputeCoreMap& host_compute_core, in ReplicateOutsideCompilationNodes() argument
3520 int core = host_compute_core.at(oc_cluster_name); in ReplicateOutsideCompilationNodes()
3901 HostComputeCoreMap* host_compute_core) { in ParseHostComputeCores() argument
3906 ParseHostComputeCoreList(hc_core_string, host_compute_core)); in ParseHostComputeCores()
3909 if (host_compute_core->find(oc_cluster_name) == host_compute_core->end()) { in ParseHostComputeCores()
3911 (*host_compute_core)[oc_cluster_name] = 0; in ParseHostComputeCores()
4358 HostComputeCoreMap host_compute_core; in RewriteTPUReplicateNode() local
4360 *replicate_node, outside_compilation_nodes, &host_compute_core)); in RewriteTPUReplicateNode()
4362 tf_device_assignment, host_compute_core, outside_compilation_nodes, in RewriteTPUReplicateNode()
|
| D | encapsulate_tpu_computations_pass.cc | 2995 std::map<string, int> host_compute_core; in Run() local
2996 TF_RETURN_IF_ERROR(ParseHostComputeCoreList(core_list, &host_compute_core)); in Run()
2999 host_compute_core}); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | graph_pruning_preserve_ops.mlir | 12 …_placement = false, computation_shape = [], device_assignment = [], host_compute_core = [], num_co…
|
| D | tpu_space_to_depth_pass.mlir | 24 …= [], device = "", device_assignment = [0, 0, 0, 0], func = @_func, host_compute_core = [], input_…
106 …ion_shape = [], device = "", device_assignment = [], func = @_func, host_compute_core = [], input_…
180 …ion_shape = [], device = "", device_assignment = [], func = @_func, host_compute_core = [], input_…
|
| D | tpu_cluster_formation.mlir | 394 …_placement = false, computation_shape = [], device_assignment = [], host_compute_core = [], num_co…
419 …_placement = false, computation_shape = [], device_assignment = [], host_compute_core = [], num_co…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/ |
| D | end_to_end.mlir | 12 …false, computation_shape = [], device = "", device_assignment = [], host_compute_core = [], name =…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-function-defs.pbtxt | 108 key: "host_compute_core"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1513 host_compute_core = context.HostComputeCore()
1515 if host_compute_core:
1517 attr_value.list.s.extend(compat.as_bytes(x) for x in host_compute_core)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 838 DefaultValuedAttr<StrArrayAttr, "{}">:$host_compute_core,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4817 …ogy\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_compute_core\', \'padding…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4817 …ogy\', \'use_tpu\', \'device_assignment\', \'computation_shape\', \'host_compute_core\', \'padding…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 81483 name: "host_compute_core"
81544 name: "host_compute_core"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 54009 name: "host_compute_core"
|