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