Searched refs:entry_args_same_across_replicas (Results 1 – 1 of 1) sorted by relevance
485 const std::vector<bool>& entry_args_same_across_replicas,512 Block* block, const std::vector<bool>& entry_args_same_across_replicas,1403 std::vector<bool> entry_args_same_across_replicas; in RunOnFunction() local1408 entry_args_same_across_replicas.reserve(f.getNumArguments()); in RunOnFunction()1411 entry_args_same_across_replicas.push_back(attr != nullptr); in RunOnFunction()1412 any_arg_replicated |= entry_args_same_across_replicas.back(); in RunOnFunction()1429 if (!any_arg_replicated) entry_args_same_across_replicas.clear(); in RunOnFunction()1434 &f.front(), &builder, entry_function, entry_args_same_across_replicas, in RunOnFunction()1443 Block* block, const std::vector<bool>& entry_args_same_across_replicas, in SetEntryTupleShapesAndLeafReplication() argument1470 if (entry_args_same_across_replicas.empty()) continue; in SetEntryTupleShapesAndLeafReplication()[all …]