Searched refs:sa_builder (Results 1 – 1 of 1) sorted by relevance
320 NodeDefBuilder sa_builder(sa_name, "_ScopedAllocator"); in ConstructScopedAllocatorNode() local321 sa_builder.Device(device_name); in ConstructScopedAllocatorNode()322 sa_builder.Attr("sa_name", sa_name); in ConstructScopedAllocatorNode()323 sa_builder.Attr("T", dtype); in ConstructScopedAllocatorNode()324 sa_builder.Attr("id", sa_id); in ConstructScopedAllocatorNode()325 sa_builder.Attr("shapes", input_shapes); in ConstructScopedAllocatorNode()326 sa_builder.Attr("shape", sa_shape); in ConstructScopedAllocatorNode()327 sa_builder.Attr("expected_call_count", static_cast<int64>(ops.size())); in ConstructScopedAllocatorNode()329 LOG_WARNING_AND_RETURN_IF_ERROR(sa_builder.Finalize(sa_node)); in ConstructScopedAllocatorNode()