Searched refs:IsParallelMap (Results 1 – 2 of 2) sorted by relevance
38 bool IsParallelMap(const NodeDef& node) { in IsParallelMap() function53 if (IsParallelMap(map_node)) { in MakeMapAndBatchNode()124 if (node2->op() != "MapDataset" && !IsParallelMap(*node2)) { in OptimizeAndCollectStats()
120 bool IsParallelMap(const std::string& op) { in IsParallelMap() function139 return IsParallelInterleave(op) || IsParallelMap(op) || IsMapAndBatch(op); in IntroducesFunctionParallelism()185 DCHECK(IsParallelMap(node->op())); in ConvertMapOrInterleave()742 DCHECK(IsParallelInterleave(node.op()) || IsParallelMap(node.op())); in OptimizeAndCollectStats()