Searched refs:NewScopedAllocatorId (Results 1 – 2 of 2) sorted by relevance
64 int NewScopedAllocatorId(int num_fields);
598 int sa_id = sa_opti->NewScopedAllocatorId(input_shapes.size()); in Rewrite()705 int ScopedAllocatorOptimizer::NewScopedAllocatorId(int num_fields) { in NewScopedAllocatorId() function in tensorflow::grappler::ScopedAllocatorOptimizer