Searched refs:is_fast_mem (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1300 const Node& neighbor_node, int64* inferred_core_id, bool* is_fast_mem, in ParseAndValidateShardingFromNeighbors() argument
1303 *is_fast_mem = true; in ParseAndValidateShardingFromNeighbors()
1327 *is_fast_mem = true; in ParseAndValidateShardingFromNeighbors()
1827 bool* is_fast_mem) { in InferXlaShardingFromNeighbors() argument
1833 *is_fast_mem = false; in InferXlaShardingFromNeighbors()
1838 num_cores_per_replica, n.name(), *edge->dst(), &core, is_fast_mem, in InferXlaShardingFromNeighbors()
1871 is_fast_mem, &result)); in InferXlaShardingFromNeighbors()
1991 bool is_fast_mem; in AssignArgsAndRetvalsToCores() local
1994 &node_and_sharding, &is_fast_mem)); in AssignArgsAndRetvalsToCores()
2103 (*arg_fast_mem)[i] = is_fast_mem; in AssignArgsAndRetvalsToCores()
[all …]
|