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