Searched refs:TPU_FAST_MEM_ATTR (Results 1 – 3 of 3) sorted by relevance
21 const char* const TPU_FAST_MEM_ATTR = "_TPU_FAST_MEM"; variable
48 extern const char* const TPU_FAST_MEM_ATTR; // "_TPU_FAST_MEM"
1302 if (neighbor_node.attrs().Find(TPU_FAST_MEM_ATTR) != nullptr) { in ParseAndValidateShardingFromNeighbors()1305 << arg_node_name << " has " << TPU_FAST_MEM_ATTR << " attribute"; in ParseAndValidateShardingFromNeighbors()1326 if (e->dst()->attrs().Find(TPU_FAST_MEM_ATTR) != nullptr) { in ParseAndValidateShardingFromNeighbors()1329 << e->dst()->name() << TPU_FAST_MEM_ATTR << " attribute"; in ParseAndValidateShardingFromNeighbors()2106 VLOG(3) << "Add " << TPU_FAST_MEM_ATTR << " attribute to " in AssignArgsAndRetvalsToCores()