Searched defs:arg_fast_mem (Results 1 – 1 of 1) sorted by relevance
2132 std::vector<xla::OpSharding>* arg_sharding, std::vector<bool>* arg_fast_mem, in AssignArgsAndRetvalsToCores()2600 const std::vector<bool>& arg_fast_mem, in BuildCompileNode()4761 std::vector<bool> arg_fast_mem; in RewriteTPUReplicateNode() local